From 312abbc2739cbf602a15ac59257976e1da05c637 Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Fri, 7 Feb 2025 21:55:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=BC=E5=AE=B9=20O3=20=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 CHANGELOG.md                |  1 +
 api/handler/chat_handler.go | 14 ++++++++------
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index ea3a7ec2..cceb14ef 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -8,6 +8,7 @@
 - 功能优化：优化 Suno 歌曲播放按钮样式，居中显示
 - 功能优化：后台管理新增模型的时候，可以绑定所有的 API KEY，而不只是能绑定 Chat 类型的 API KEY
 - 功能新增：新增每日签到功能，每日签到可以获得算力奖励
+- 功能优化：兼容 OpenAI o3 系列模型
 
 ## v4.1.9
 
diff --git a/api/handler/chat_handler.go b/api/handler/chat_handler.go
index 78a22b3b..79530b69 100644
--- a/api/handler/chat_handler.go
+++ b/api/handler/chat_handler.go
@@ -97,17 +97,19 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
 	}
 
 	var req = types.ApiRequest{
-		Model: session.Model.Value,
+		Model:       session.Model.Value,
+		Stream:      session.Stream,
+		Temperature: session.Model.Temperature,
 	}
-	// 兼容 GPT-O1 模型
-	if strings.HasPrefix(session.Model.Value, "o1-") {
+	// 兼容 OpenAI 模型
+	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.Stream = session.Stream
+		req.MaxCompletionTokens = session.Model.MaxTokens
 		session.Start = time.Now().Unix()
 	} else {
 		req.MaxTokens = session.Model.MaxTokens
-		req.Temperature = session.Model.Temperature
-		req.Stream = session.Stream
 	}
 
 	if len(session.Tools) > 0 && !strings.HasPrefix(session.Model.Value, "o1-") {