xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-12-07 16:56:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复公式解析的 Bug

Browse Source
This commit is contained in:
RockYang
2025-02-20 11:17:12 +08:00
parent dd675c9a9b
commit 03d33c784c
8 changed files with 69 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/handler/chat_handler.go
View File

@@ -90,7 +90,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
}
// 检查 prompt 长度是否超过了当前模型允许的最大上下文长度
promptTokens, err := utils.CalcTokens(prompt, session.Model.Value)
promptTokens, _ := utils.CalcTokens(prompt, session.Model.Value)
if promptTokens > session.Model.MaxContext {
return errors.New("对话内容超出了当前模型允许的最大上下文长度！")
@@ -105,7 +105,6 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
if strings.HasPrefix(session.Model.Value, "o1-") ||
strings.HasPrefix(session.Model.Value, "o3-") ||
strings.HasPrefix(session.Model.Value, "gpt") {
utils.SendChunkMsg(ws, "> AI 正在思考...\n")
req.MaxCompletionTokens = session.Model.MaxTokens
session.Start = time.Now().Unix()
} else {