diff --git a/ai.py b/ai.py index 8d18032..f69c4af 100644 --- a/ai.py +++ b/ai.py @@ -3,8 +3,6 @@ import json from zhipuai import ZhipuAI import flask - - # client = ZhipuAI(api_key="73bdeed728677bc80efc6956478a2315.VerNWJMCwN9L5gTi") # 请填写您自己的APIKey # response = client.chat.completions.create( # model="glm-4", # 请填写您要调用的模型名称 @@ -21,9 +19,9 @@ def app_chat(): data = json.loads(flask.globals.request.get_data()) # print(data) uid = data["user_id"] - if not data["text"][-1] in ['?', '?', '.', '。', ',', ',', '!', '!']: data["text"] += "。" + data["text"] += "" # 使用ZhipuAI库调用模型生成回复 client = ZhipuAI(api_key="73bdeed728677bc80efc6956478a2315.VerNWJMCwN9L5gTi") # 请填写您自己的APIKey diff --git a/examples/SchoolCalendar.png b/examples/SchoolCalendar.png new file mode 100644 index 0000000..c91a22a Binary files /dev/null and b/examples/SchoolCalendar.png differ diff --git a/examples/classExample.jpg b/examples/classExample.jpg new file mode 100644 index 0000000..99cc9fe Binary files /dev/null and b/examples/classExample.jpg differ diff --git a/wechat_client.go b/wechat_client.go index 635460c..ce16805 100644 --- a/wechat_client.go +++ b/wechat_client.go @@ -18,7 +18,10 @@ func Use(vals ...interface{}) { _ = val } } - +var keyWordMap = map[string]string{ + "生成课程表": "./examples/classExample.jpg", + "生成校历": "./examples/SchoolCalendar.png", +} type SendTextRequest struct { InGroup bool `json:"in_group"` //本来想用于区分在群聊和非群聊时的上下文记忆规则,但是最终没有实现... UserID string `json:"user_id"` @@ -94,7 +97,7 @@ func main() { func sendMessage(msg *openwechat.Message, self *openwechat.Self) { if msg.IsTickledMe() { - msg.ReplyText("别拍了,机器人是会被拍坏掉的。") + msg.ReplyText("你好。我是大数据分析与知识处理课程的AI助手,专门为学习相关课程的学生提供帮助,包括解答疑问、提供学习资料、辅助分析和理解复杂概念等。如果您有任何与大数据分析、知识处理相关的问题,欢迎随时向我提问。") return } @@ -147,7 +150,16 @@ func sendMessage(msg *openwechat.Message, self *openwechat.Self) { } } - } else { + }else { + // 遍历 map + for keyword, value := range keyWordMap { + if strings.Contains(msg.Content, keyword) { + img, _ := os.Open(value) + defer img.Close() + msg.ReplyImage(img) + return + } + } // 调用GPT sender, _ := msg.Sender()