lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: enhance headmasterHandler to include group nickname in JSON responses for better user context and improve message formatting consistency

Browse Source
This commit is contained in:
lixiangwuxian 2025-01-09 03:04:51 +08:00
parent d3e9c9b0d6
commit ffc73605e8
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
handler/headmaster/headmaster.go
View File

@ -33,13 +33,13 @@ func headmasterHandler(msg model.Message) (reply model.Reply) {
}
if len(msg.RawMsg) > 7 && msg.RawMsg[0:7] == "校长 " {
return model.Reply{
ReplyMsg: ask(from, fmt.Sprintf("{\"qqid\":%d,\"nickname\":\"%s\",\"referid\":%d,\"msg\":\"%s\"}", msg.UserId, msg.UserNickName, msg.OriginMsgId, msg.RawMsg[7:])),
ReplyMsg: ask(from, fmt.Sprintf("{\"qqid\":%d,\"nickname\":\"%s\",\"group_nickname\":\"%s\",\"referid\":%d,\"msg\":\"%s\"}", msg.UserId, msg.UserNickName, msg.GroupInfo.UserCard, msg.OriginMsgId, msg.RawMsg[7:])),
ReferOriginMsg: false,
FromMsg: msg,
}
}
return model.Reply{
ReplyMsg: ask(from, fmt.Sprintf("{\"qqid\":%d,\"nickname\":\"%s\",\"referid\":%d,\"msg\":\"%s\"}", msg.UserId, msg.UserNickName, msg.OriginMsgId, msg.RawMsg)),
ReplyMsg: ask(from, fmt.Sprintf("{\"qqid\":%d,\"nickname\":\"%s\",\"group_nickname\":\"%s\",\"referid\":%d,\"msg\":\"%s\"}", msg.UserId, msg.UserNickName, msg.GroupInfo.UserCard, msg.OriginMsgId, msg.RawMsg)),
ReferOriginMsg: false,
FromMsg: msg,
}
@ -55,6 +55,22 @@ func ask(from string, question string) (reply string) {
openai.ChatCompletionRequest{
Model: config.ConfigManager.GetProperty("openai_model_name"),
Messages: GenRequestFromUsr(from, question),
// Tools: []openai.Tool{
// {
// Type: openai.ToolTypeFunction,
// Function: &openai.FunctionDefinition{
// Name: "try_get_qqid_by_nickname",
// Description: "尝试在群内获取昵称对应的qq号",
// Parameters: json.RawMessage(`{"type": "object", "properties": {"nickname": {"type": "string", "description": "用户的昵称"}}}`),
// },
// },
// },
// ToolChoice: &openai.ToolChoice{
// Type: openai.ToolTypeFunction,
// Function: openai.ToolFunction{
// Name: "try_get_qqid_by_nickname",
// },
// },
},
)