diff --git a/service/imService.go b/service/imService.go index 3033f6a..607a62c 100644 --- a/service/imService.go +++ b/service/imService.go @@ -92,7 +92,10 @@ func WSReceiveMessageService(userID, sessionID int, channel string, msg proto.WS log.Println("get ollama client error:", err) } prompt := msg.Msg - actx, _ := actxMap[sessionID] + actx, ok := actxMap[sessionID] + if ok { + log.Println("actx has get will use:", len(actx)) + } req := &api.GenerateRequest{ Model: "qwen2.5:0.5b", Context: actx, @@ -112,6 +115,7 @@ func WSReceiveMessageService(userID, sessionID int, channel string, msg proto.WS actxMap = make(map[int][]int) } actxMap[sessionID] = actx + log.Println("actx has save:", len(actxMap[sessionID])) //创建消息 err2, msgID := CreateMessage(2, sessionID, 2, userID, robotMsg, 1) //机器人id为2,消息类型为2,状态为1 if err2 != nil {