O1 模型支持流式输出

api/core/types/config.go

@@ -133,7 +133,7 @@ type SystemConfig struct {
 	AdminTitle    string `json:"admin_title,omitempty"`     // 管理后台标题
 	Logo          string `json:"logo,omitempty"`            // 方形 Logo
 	InitPower     int    `json:"init_power,omitempty"`      // 新用户注册赠送算力值
-	DailyPower int `json:"daily_power,omitempty"`           // 每日签到赠送算力
+	DailyPower    int    `json:"daily_power,omitempty"`     // 每日签到赠送算力
 	InvitePower   int    `json:"invite_power,omitempty"`    // 邀请新用户赠送算力值
 	VipMonthPower int    `json:"vip_month_power,omitempty"` // VIP 会员每月赠送的算力值
 
@@ -146,7 +146,7 @@ type SystemConfig struct {
 	MjPower           int `json:"mj_power,omitempty"`            // MJ 绘画消耗算力
 	MjActionPower     int `json:"mj_action_power,omitempty"`     // MJ 操作（放大，变换）消耗算力
 	SdPower           int `json:"sd_power,omitempty"`            // SD 绘画消耗算力
-	DallPower int `json:"dall_power,omitempty"`                  // DALL-E-3 绘图消耗算力
+	DallPower         int `json:"dall_power,omitempty"`          // DALL-E-3 绘图消耗算力
 	SunoPower         int `json:"suno_power,omitempty"`          // Suno 生成歌曲消耗算力
 	LumaPower         int `json:"luma_power,omitempty"`          // Luma 生成视频消耗算力
 	AdvanceVoicePower int `json:"advance_voice_power,omitempty"` // 高级语音对话消耗算力
@@ -160,7 +160,6 @@ type SystemConfig struct {
 	SdNegPrompt string `json:"sd_neg_prompt"` // SD 默认反向提示词
 	MjMode      string `json:"mj_mode"`       // midjourney 默认的API模式，relax, fast, turbo
 
-	IndexBgURL  string `json:"index_bg_url"`  // 前端首页背景图片
 	IndexNavs   []int  `json:"index_navs"`    // 首页显示的导航菜单
 	Copyright   string `json:"copyright"`     // 版权信息
 	MarkMapText string `json:"mark_map_text"` // 思维导入的默认文本

api/handler/chat_handler.go

@@ -101,8 +101,8 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
 	}
 	// 兼容 GPT-O1 模型
 	if strings.HasPrefix(session.Model.Value, "o1-") {
-		utils.SendChunkMsg(ws, "AI 正在思考...\n")
-		req.Stream = false
+		utils.SendChunkMsg(ws, "> AI 正在思考...\n")
+		req.Stream = session.Stream
 		session.Start = time.Now().Unix()
 	} else {
 		req.MaxTokens = session.Model.MaxTokens

api/handler/chat_openai_handler.go

@@ -63,7 +63,7 @@ func (h *ChatHandler) sendOpenAiMessage(
 	start := time.Now()
 	var apiKey = model.ApiKey{}
 	response, err := h.doRequest(ctx, req, session, &apiKey)
-	logger.Info("HTTP请求完成，耗时：", time.Now().Sub(start))
+	logger.Info("HTTP请求完成，耗时：", time.Since(start))
 	if err != nil {
 		if strings.Contains(err.Error(), "context canceled") {
 			return fmt.Errorf("用户取消了请求：%s", prompt)
@@ -89,6 +89,12 @@ func (h *ChatHandler) sendOpenAiMessage(
 		var function model.Function
 		var toolCall = false
 		var arguments = make([]string, 0)
+
+		if strings.HasPrefix(req.Model, "o1-") {
+			content := fmt.Sprintf("AI 思考结束，耗时：%d 秒。\n\n", time.Now().Unix()-session.Start)
+			utils.SendChunkMsg(ws, content)
+		}
+
 		scanner := bufio.NewScanner(response.Body)
 		for scanner.Scan() {
 			line := scanner.Text()
@@ -150,10 +156,10 @@ func (h *ChatHandler) sendOpenAiMessage(
 			// output stopped
 			if responseBody.Choices[0].FinishReason != "" {
 				break // 输出完成或者输出中断了
-			} else {
+			} else { // 正常输出结果
 				content := responseBody.Choices[0].Delta.Content
 				contents = append(contents, utils.InterfaceToString(content))
-				utils.SendChunkMsg(ws, responseBody.Choices[0].Delta.Content)
+				utils.SendChunkMsg(ws, content)
 			}
 		} // end for