refactor(prompts-core): 重构代码结构和优化工具函数

2026-06-10 14:51:25 +08:00
parent 1c1db7e30c
commit b69e7386e2
10 changed files with 164 additions and 432 deletions
--- a/service/prompt/prompt_build_service.go
+++ b/service/prompt/prompt_build_service.go
@@ -13,11 +13,10 @@ import (

 	"gitea.com/red-future/common/utils"
 	"github.com/gogf/gf/v2/encoding/gjson"
-	"github.com/gogf/gf/v2/util/gconv"
 )

 // buildPromptTypeRequest 构建提示词类型请求（BuildType=1）
-func buildPromptTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, aiModel *gateway.AsynchModel, chatModel *gateway.AsynchModel, ir *PromptIR, totalBatches int) (map[string]any, error) {
+func buildPromptTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, aiModel *gateway.AsynchModel, chatModel *gateway.AsynchModel, ir *IR) (map[string]any, error) {
 	//1) 构建系统提示词
 	systemPrompt := promptBuildWithRounds(ctx, chatModel, aiModel)
 	ir.AddSystem(systemPrompt)
@@ -32,29 +31,21 @@ func buildPromptTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, ai
 }

 // buildNodeTypeRequest 构建节点类型请求（BuildType=2）
-func buildNodeTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, chatModel *gateway.AsynchModel, ir *PromptIR) (map[string]any, error) {
+func buildNodeTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, chatModel *gateway.AsynchModel, ir *IR) (map[string]any, error) {
 	ir.AddUser(NodeBuild(ctx, req))
 	return compileToProviderRequest(ctx, ir, chatModel, req)
 }

 // buildStructTypeRequest 构建结构体类型请求（BuildType=3）
-func buildStructTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, chatModel *gateway.AsynchModel, ir *PromptIR) (map[string]any, error) {
-	// 提取 userForm 中的 prompt 作为自定义提示词
-	var customPrompt string
-	for _, item := range req.UserForm {
-		if prompt, ok := item["prompt"]; ok && gconv.String(prompt) != "" {
-			customPrompt = gconv.String(prompt)
-			break
-		}
-	}
-	// 用户消息
+func buildStructTypeRequest(ctx context.Context, req *dto.ComposeMessagesReq, chatModel *gateway.AsynchModel, ir *IR) (map[string]any, error) {
+	customPrompt := gjson.New(req.UserForm).Get("0.prompt").String()
 	ir.AddSystem(customPrompt)
 	ir.AddUser(buildUserPrompt(ctx, req, ""))
 	return compileToProviderRequest(ctx, ir, chatModel, req, customPrompt)
 }

 // compileToProviderRequest 编译为 Provider 请求
-func compileToProviderRequest(ctx context.Context, ir *PromptIR, chatModel *gateway.AsynchModel, req *dto.ComposeMessagesReq, customPrompt ...string) (map[string]any, error) {
+func compileToProviderRequest(ctx context.Context, ir *IR, chatModel *gateway.AsynchModel, req *dto.ComposeMessagesReq, customPrompt ...string) (map[string]any, error) {
 	protocol, err := GetProtocolByProvider(ctx, chatModel.OperatorName)
 	if err != nil || protocol == nil {
 		return nil, fmt.Errorf("协议配置不存在或获取失败: %w", err)
@@ -78,6 +69,7 @@ func compileToProviderRequest(ctx context.Context, ir *PromptIR, chatModel *gate
 	}, nil
 }

+// promptBuildWithRounds 构建提示词
 func promptBuildWithRounds(ctx context.Context, chatModel *gateway.AsynchModel, aiModel *gateway.AsynchModel) string {
 	providerProtocol, err := dao.ProviderProtocol.Get(ctx, &entity.ProviderProtocol{
 		ProviderName: chatModel.OperatorName,
@@ -86,7 +78,7 @@ func promptBuildWithRounds(ctx context.Context, chatModel *gateway.AsynchModel,
 	if err != nil || providerProtocol == nil {
 		return ""
 	}
-	outputJSON := util.JSONPretty(util.ReverseMap(aiModel.RequestMapping, map[string]any{}))
+	outputJSON := gjson.New(util.ReverseMap(aiModel.RequestMapping, map[string]any{})).MustToJsonIndentString()

 	return fmt.Sprintf(providerProtocol.SystemPromptTemplate,
 		outputJSON, //【输出结构】 %s
@@ -94,7 +86,7 @@ func promptBuildWithRounds(ctx context.Context, chatModel *gateway.AsynchModel,
 }

 // checkOverallContent 检查整体内容是否超出窗口
-func checkOverallContent(ir *PromptIR, model *gateway.AsynchModel) bool {
+func checkOverallContent(ir *IR, model *gateway.AsynchModel) bool {
 	fullContent := ir.String()
 	return util.CountToken(fullContent, model.TokenConfig)
 }
@@ -124,7 +116,6 @@ func buildUserPrompt(ctx context.Context, req *dto.ComposeMessagesReq, prompt st
 	return b.String()
 }

-// buildUserFormText 构建用户表单内容字符串
 func buildUserFormText(form []map[string]any) string {
 	if len(form) == 0 {
 		return ""
@@ -132,32 +123,22 @@ func buildUserFormText(form []map[string]any) string {
 	var builder strings.Builder
 	for _, item := range form {
 		for k, v := range item {
+			builder.WriteString(fmt.Sprintf("%s：\n", k))
 			switch val := v.(type) {
 			case []any:
-				// 数组类型：逐条列出
-				builder.WriteString(fmt.Sprintf("%s：\n", k))
 				for i, elem := range val {
+					builder.WriteString(fmt.Sprintf("  %d. ", i+1))
 					if m, ok := elem.(map[string]any); ok {
-						builder.WriteString(fmt.Sprintf("  %d. ", i+1))
 						for mk, mv := range m {
 							builder.WriteString(fmt.Sprintf("%s：%v ", mk, mv))
 						}
-						builder.WriteString("\n")
 					} else {
-						builder.WriteString(fmt.Sprintf("  %d. %v\n", i+1, elem))
-					}
-				}
-			case []map[string]any:
-				builder.WriteString(fmt.Sprintf("%s：\n", k))
-				for i, m := range val {
-					builder.WriteString(fmt.Sprintf("  %d. ", i+1))
-					for mk, mv := range m {
-						builder.WriteString(fmt.Sprintf("%s：%v ", mk, mv))
+						builder.WriteString(fmt.Sprint(elem))
 					}
 					builder.WriteString("\n")
 				}
 			default:
-				builder.WriteString(fmt.Sprintf("%s：%v\n", k, v))
+				builder.WriteString(fmt.Sprintf("  %v\n", v))
 			}
 		}
 	}
@@ -170,9 +151,8 @@ func NodeBuild(ctx context.Context, req *dto.ComposeMessagesReq) string {
 	if promptTpl == "" {
 		return ""
 	}
-
-	formStr := util.FormToJSON(req.Form)
-	userFormStr := util.UserFormToJSON(req.UserForm)
-
-	return fmt.Sprintf(promptTpl, formStr, userFormStr)
+	return fmt.Sprintf(promptTpl,
+		gjson.New(req.Form).MustToJsonString(),
+		gjson.New(req.UserForm).MustToJsonString(),
+	)
 }
--- a/service/prompt/prompt_compose_service.go
+++ b/service/prompt/prompt_compose_service.go
@@ -2,7 +2,6 @@ package prompt

 import (
 	"context"
-	"encoding/json"
 	"errors"
 	"fmt"
 	"prompts-core/service/session"
@@ -80,7 +79,7 @@ func validateUserForm(req *dto.ComposeMessagesReq, model *gateway.AsynchModel) e
 // handleBuild 通用构建处理
 func handleBuild(ctx context.Context, req *dto.ComposeMessagesReq, chatModel, aiModel *gateway.AsynchModel) (*dto.ComposeMessagesRes, error) {
 	// 1) 处理表单分批
-	processedReq, totalBatches, err := ProcessUserFormBatches(ctx, req, aiModel)
+	processedReq, _, err := ProcessUserFormBatches(ctx, req, aiModel)
 	if err != nil {
 		return nil, fmt.Errorf("处理用户表单分批失败: %w", err)
 	}
@@ -90,7 +89,7 @@ func handleBuild(ctx context.Context, req *dto.ComposeMessagesReq, chatModel, ai
 	var taskReq map[string]any
 	switch req.BuildType {
 	case public.BuildTypePrompt:
-		taskReq, err = buildPromptTypeRequest(ctx, processedReq, aiModel, chatModel, ir, totalBatches)
+		taskReq, err = buildPromptTypeRequest(ctx, processedReq, aiModel, chatModel, ir)
 	case public.BuildTypeNode:
 		taskReq, err = buildNodeTypeRequest(ctx, req, chatModel, ir)
 	case public.BuildTypeStruct:
@@ -118,7 +117,7 @@ func handleBuild(ctx context.Context, req *dto.ComposeMessagesReq, chatModel, ai
 		SkillName:      req.SkillName,
 		BuildType:      req.BuildType,
 		CallbackUrl:    req.CallbackUrl,
-		RequestPayload: util.MustMarshalToMap(req),
+		RequestPayload: gconv.Map(req),
 		Status:         public.ComposeStatusPending,
 	}); err != nil {
 		return nil, err
@@ -164,6 +163,7 @@ func handleCallbackFailed(ctx context.Context, req *dto.CallbackReq, composeTask
 	return err
 }

+// handleCallbackSuccess 处理回调成功
 func handleCallbackSuccess(ctx context.Context, req *dto.CallbackReq, composeTask *entity.ComposeTask) error {
 	// 1) 获取模型配置
 	model, err := gateway.GetModelConfig(ctx, &gateway.AsynchModel{
@@ -180,12 +180,15 @@ func handleCallbackSuccess(ctx context.Context, req *dto.CallbackReq, composeTas
 		Status:       1,
 	})

-	// 3) 获取历史消息
+	// 3) 获取历史消息 + 保存当前轮
 	payload := composeTask.RequestPayload
 	sessionId := gconv.String(payload["sessionId"])
 	nodeId := gconv.String(payload["nodeId"])
 	var history []dto.FlatMessage
-	if sessionId != "" && nodeId != "" {
+	var epicycleId int64
+
+	if sessionId != "" && nodeId != "" && model.ModelType == public.ModelTypeInference {
+		// 3.1 获取历史
 		h, _ := session.GetHistoryMessages(ctx, &dto.GetHistoryMessagesReq{
 			SessionId: sessionId,
 			NodeId:    nodeId,
@@ -193,12 +196,21 @@ func handleCallbackSuccess(ctx context.Context, req *dto.CallbackReq, composeTas
 		if h != nil {
 			history = h.Messages
 		}
+
+		// 3.2 保存当前轮（先存，下次查询就能拿到）
+		if userMsg := util.ExtractUserText(req.Messages); userMsg != nil {
+			epicycleId, _ = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{
+				NodeId:         nodeId,
+				SessionId:      sessionId,
+				RequestContent: userMsg,
+			})
+		}
 	}

 	// 4) 合并附加结构
 	messages := util.MergeConsult(composeTask.RequestPayload, req.Messages, model.ExtendMapping)
-	// 5) 注入历史到 rounds 中
-	if protocol != nil && len(history) > 0 {
+	// 5) 注入历史
+	if len(history) > 0 {
 		messages = InjectHistory(messages, history, protocol)
 	}

@@ -215,18 +227,6 @@ func handleCallbackSuccess(ctx context.Context, req *dto.CallbackReq, composeTas
 		return err
 	}

-	// 7) 存储历史
-	var epicycleId int64
-	if sessionId != "" && nodeId != "" {
-		if userMsg := util.ExtractUserText(req.Messages); userMsg != nil {
-			epicycleId, _ = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{
-				NodeId:         nodeId,
-				SessionId:      sessionId,
-				RequestContent: userMsg,
-			})
-		}
-	}
-
 	// 8) 回调业务方
 	if composeTask.CallbackUrl != "" {
 		composeTask.Status = public.ComposeStatusSuccess
@@ -237,77 +237,7 @@ func handleCallbackSuccess(ctx context.Context, req *dto.CallbackReq, composeTas
 	return nil
 }

-// GetComposeTask 查询任务结果
-func GetComposeTask(ctx context.Context, taskID string) (*dto.GetComposeTaskRes, error) {
-	record, err := dao.ComposeTask.Get(ctx, &entity.ComposeTask{
-		TaskId: taskID,
-	})
-	if err != nil {
-		return nil, fmt.Errorf("查询任务失败: %w", err)
-	}
-	if record == nil {
-		return nil, fmt.Errorf("未找到任务(taskId=%s)", taskID)
-	}
-
-	messages := parseMessagesForResponse(record.ResultJson)
-
-	return &dto.GetComposeTaskRes{
-		TaskId:       record.TaskId,
-		Status:       record.Status,
-		ErrorMessage: record.ErrorMessage,
-		Messages:     messages,
-	}, nil
-}
-
-// parseMessagesForResponse 解析用于响应的消息
-func parseMessagesForResponse(messages any) any {
-	str, ok := messages.(string)
-	if !ok || str == "" {
-		return messages
-	}
-
-	var parsed any
-	if err := json.Unmarshal([]byte(str), &parsed); err == nil {
-		return parsed
-	}
-
-	return messages
-}
-
-func GetPromptText(ctx context.Context, req *dto.GetPromptTextReq) (*dto.GetPromptTextRes, error) {
-	// 1) 获取协议配置
-	protocol, err := dao.ProviderProtocol.Get(ctx, &entity.ProviderProtocol{
-		ProviderName: "火山引擎",
-		Status:       1,
-	})
-	if err != nil {
-		return nil, err
-	}
-
-	// 2) 获取历史消息
-	history, err := session.GetHistoryMessages(ctx, &dto.GetHistoryMessagesReq{
-		SessionId: "88888888",
-		NodeId:    "node1",
-	})
-	if err != nil {
-		return nil, err
-	}
-
-	// 3) 模拟roundsData数据
-	task, err := dao.ComposeTask.Get(ctx, &entity.ComposeTask{
-		TaskId: "0e1872f0-0e73-42f1-9aa8-63d317300ffc",
-	})
-	if err != nil {
-		return nil, err
-	}
-	fmt.Println("[打印数据]", task.ResultJson)
-	fmt.Println("[打印历史]", history.Messages)
-	fmt.Println("[打印协议]", protocol)
-	return &dto.GetPromptTextRes{
-		Messages: InjectHistory(task.ResultJson, history.Messages, protocol),
-	}, nil
-}
-
+// InjectHistory 插入历史会话
 func InjectHistory(roundsData map[string]any, history []dto.FlatMessage, protocol *entity.ProviderProtocol) map[string]any {
 	if protocol == nil || len(history) == 0 {
 		return roundsData
@@ -363,3 +293,19 @@ func InjectHistory(roundsData map[string]any, history []dto.FlatMessage, protoco
 	firstRound["messages"] = result
 	return roundsData
 }
+
+// GetComposeTask 查询任务结果
+func GetComposeTask(ctx context.Context, taskID string) (*dto.GetComposeTaskRes, error) {
+	record, err := dao.ComposeTask.Get(ctx, &entity.ComposeTask{
+		TaskId: taskID,
+	})
+	if err != nil {
+		return nil, fmt.Errorf("查询任务失败: %w", err)
+	}
+	return &dto.GetComposeTaskRes{
+		TaskId:       record.TaskId,
+		Status:       record.Status,
+		ErrorMessage: record.ErrorMessage,
+		Messages:     record.ResultJson,
+	}, nil
+}
--- a/service/prompt/prompt_files_handle_service.go
+++ b/service/prompt/prompt_files_handle_service.go
@@ -190,6 +190,9 @@ func fetchFileContent(ctx context.Context, client *http.Client, url string) (str
 }

 func SkillMdContent(ctx context.Context, skillName string) string {
+	if skillName == "" {
+		return ""
+	}
 	skillResp, err := gateway.GetSkillUser(ctx, skillName)
 	if err != nil {
 		g.Log().Warningf(ctx, "[SkillMd] GetSkillUser 失败: %v", err)
--- a/service/prompt/prompt_ir_service.go
+++ b/service/prompt/prompt_ir_service.go
@@ -2,9 +2,7 @@ package prompt

 import (
 	"context"
-	"encoding/json"
 	"fmt"
-	"prompts-core/common/util"
 	"prompts-core/service/gateway"
 	"strings"

@@ -14,8 +12,8 @@ import (
 	"github.com/gogf/gf/v2/util/gconv"
 )

-// PromptIR 统一 Prompt 中间表示
-type PromptIR struct {
+// IR 统一 Prompt 中间表示
+type IR struct {
 	System  []Segment `json:"system"`
 	History []Segment `json:"history"`
 	User    []Segment `json:"user"`
@@ -46,8 +44,8 @@ type ContentMapping struct {
 }

 // NewPromptIR 创建空 PromptIR
-func NewPromptIR() *PromptIR {
-	return &PromptIR{
+func NewPromptIR() *IR {
+	return &IR{
 		System:  make([]Segment, 0),
 		History: make([]Segment, 0),
 		User:    make([]Segment, 0),
@@ -55,7 +53,7 @@ func NewPromptIR() *PromptIR {
 }

 // String 返回 PromptIR 的完整内容字符串（用于 token 计算）
-func (ir *PromptIR) String() string {
+func (ir *IR) String() string {
 	var builder strings.Builder

 	for _, seg := range ir.System {
@@ -81,7 +79,7 @@ func (ir *PromptIR) String() string {
 }

 // GetTotalContent 获取所有内容的拼接字符串（更精确的 token 计算）
-func (ir *PromptIR) GetTotalContent() string {
+func (ir *IR) GetTotalContent() string {
 	var builder strings.Builder

 	for _, seg := range ir.System {
@@ -103,7 +101,7 @@ func (ir *PromptIR) GetTotalContent() string {
 }

 // AddSystem 添加系统提示
-func (ir *PromptIR) AddSystem(content string) *PromptIR {
+func (ir *IR) AddSystem(content string) *IR {
 	if content != "" {
 		ir.System = append(ir.System, Segment{Type: "text", Content: content})
 	}
@@ -111,7 +109,7 @@ func (ir *PromptIR) AddSystem(content string) *PromptIR {
 }

 // AddUser 添加用户消息
-func (ir *PromptIR) AddUser(content string) *PromptIR {
+func (ir *IR) AddUser(content string) *IR {
 	if content != "" {
 		ir.User = append(ir.User, Segment{Type: "text", Content: content})
 	}
@@ -119,7 +117,7 @@ func (ir *PromptIR) AddUser(content string) *PromptIR {
 }

 // AddHistory 添加历史消息
-func (ir *PromptIR) AddHistory(role, content string) *PromptIR {
+func (ir *IR) AddHistory(role, content string) *IR {
 	if content != "" {
 		ir.History = append(ir.History, Segment{Type: "text", Content: content, Role: role})
 	}
@@ -127,7 +125,7 @@ func (ir *PromptIR) AddHistory(role, content string) *PromptIR {
 }

 // ToMessages 转换为 OpenAI 兼容的 messages 格式（MVP 默认）
-func (ir *PromptIR) ToMessages() []map[string]any {
+func (ir *IR) ToMessages() []map[string]any {
 	var messages []map[string]any

 	for _, seg := range ir.System {
@@ -168,22 +166,22 @@ func GetProtocolByProvider(ctx context.Context, providerName string) (*ProviderP

 // parseProtocol 将 DB entity 转为编译用协议配置
 func parseProtocol(e *entity.ProviderProtocol) *ProviderProtocol {
-	p := &ProviderProtocol{
+	return &ProviderProtocol{
 		TargetField:          e.TargetField,
 		SystemPromptTemplate: e.SystemPromptTemplate,
+		MergeOrder:           e.MergeOrder,
+		RoleMapping:          gconv.MapStrStr(e.RoleMapping),
+		ContentMapping: ContentMapping{
+			Type:  gconv.String(e.ContentMapping["type"]),
+			Field: gconv.String(e.ContentMapping["field"]),
+		},
+		RequestTemplate: e.RequestTemplate,
+		Capabilities:    e.Capabilities,
 	}
-
-	// 使用通用解析方法处理各个字段
-	util.ParseJSONFieldFromGvar(e.MergeOrder, &p.MergeOrder)
-	util.ParseJSONFieldFromGvar(e.RoleMapping, &p.RoleMapping)
-	util.ParseJSONFieldFromGvar(e.ContentMapping, &p.ContentMapping)
-	util.ParseJSONFieldFromGvar(e.RequestTemplate, &p.RequestTemplate)
-	util.ParseJSONFieldFromGvar(e.Capabilities, &p.Capabilities)
-	return p
 }

 // Compile 将 PromptIR 按协议配置编译为 Provider Request
-func Compile(ir *PromptIR, p *ProviderProtocol, chatModel *gateway.AsynchModel) (map[string]any, error) {
+func Compile(ir *IR, p *ProviderProtocol, chatModel *gateway.AsynchModel) (map[string]any, error) {
 	if ir == nil || p == nil {
 		return nil, fmt.Errorf("ir and protocol are required")
 	}
@@ -195,35 +193,25 @@ func Compile(ir *PromptIR, p *ProviderProtocol, chatModel *gateway.AsynchModel)
 }

 // mergeByOrder 按协议配置顺序拼接消息
-func mergeByOrder(ir *PromptIR, order []string) []map[string]any {
-	var messages []map[string]any
-
-	for _, part := range order {
-		switch part {
-		case "system":
-			for _, seg := range ir.System {
-				messages = append(messages, map[string]any{
-					"role":    "system",
-					"content": seg.Content,
-				})
-			}
-		case "history":
-			for _, seg := range ir.History {
-				messages = append(messages, map[string]any{
-					"role":    seg.Role,
-					"content": seg.Content,
-				})
-			}
-		case "user":
-			for _, seg := range ir.User {
-				messages = append(messages, map[string]any{
-					"role":    "user",
-					"content": seg.Content,
-				})
-			}
-		}
+func mergeByOrder(ir *IR, order []string) []map[string]any {
+	roleMap := map[string][]Segment{
+		"system":  ir.System,
+		"history": ir.History,
+		"user":    ir.User,
 	}

+	var messages []map[string]any
+	for _, part := range order {
+		for _, seg := range roleMap[part] {
+			msg := map[string]any{"content": seg.Content}
+			if part == "history" {
+				msg["role"] = seg.Role
+			} else {
+				msg["role"] = part
+			}
+			messages = append(messages, msg)
+		}
+	}
 	return messages
 }

@@ -247,22 +235,22 @@ func mapRoles(messages []map[string]any, mapping map[string]string) []map[string
 	return messages
 }

-// mapContent 内容字段映射
 func mapContent(messages []map[string]any, cm ContentMapping) []map[string]any {
-	for _, msg := range messages {
-		content := msg["content"]
-		delete(msg, "content")
-
-		switch cm.Type {
-		case "parts":
-			msg["parts"] = []map[string]any{
-				{cm.Field: content},
-			}
-		default:
-			msg[cm.Field] = content
-		}
+	if cm.Field == "" || cm.Field == "content" {
+		return messages
 	}

+	for i, msg := range messages {
+		if content, ok := msg["content"]; ok {
+			delete(msg, "content")
+			switch cm.Type {
+			case "parts":
+				messages[i]["parts"] = []map[string]any{{cm.Field: content}}
+			default:
+				messages[i][cm.Field] = content
+			}
+		}
+	}
 	return messages
 }

@@ -277,20 +265,17 @@ func buildRequest(messages []map[string]any, p *ProviderProtocol, chatModel *gat
 	}
 }

-// renderTemplate 简单的 {{key}} 模板替换
+// renderTemplate 模板渲染
 func renderTemplate(p *ProviderProtocol, messages []map[string]any, chatModel *gateway.AsynchModel) map[string]any {
-	b, _ := json.Marshal(p.RequestTemplate)
-	str := string(b)
-
-	if chatModel != nil {
-		str = strings.ReplaceAll(str, `"{{model}}"`, `"`+chatModel.ModelName+`"`)
+	result := make(map[string]any, len(p.RequestTemplate)+1)
+	for k, v := range p.RequestTemplate {
+		result[k] = v
 	}

-	msgBytes, _ := json.Marshal(messages)
-	str = strings.ReplaceAll(str, `"{{messages}}"`, string(msgBytes))
-
-	var result map[string]any
-	_ = json.Unmarshal([]byte(str), &result)
+	if chatModel != nil {
+		result["model"] = chatModel.ModelName
+	}
+	result["messages"] = messages

 	if maxTokens := gconv.Int(p.Capabilities["max_tokens"]); maxTokens > 0 {
 		result["max_tokens"] = maxTokens
--- a/service/session/prompt_session_service.go
+++ b/service/session/prompt_session_service.go
@@ -21,8 +21,8 @@ import (

 // Callback 会话回调
 func Callback(ctx context.Context, req *dto.SessionCallbackReq) (*dto.SessionCallbackRes, error) {
+	fmt.Println("打印会话回调", req)
 	req.Messages["role"] = "assistant"
-
 	// 1) 更新 DB
 	_, err := dao.ComposeSession.Update(ctx, &entity.ComposeSession{
 		SQLBaseDO:       beans.SQLBaseDO{Id: req.EpicycleId},
@@ -163,23 +163,15 @@ func DeleteSession(ctx context.Context, req *dto.DeleteSessionReq) (*dto.DeleteS

 // entityToHistoryRound entity → HistoryRound
 func entityToHistoryRound(s *entity.ComposeSession) *dto.HistoryRound {
-	reqMsgs := util.ConvertToMessages(s.RequestContent)
-	respMsgs := util.ConvertToMessages(s.ResponseContent)
-
-	round := &dto.HistoryRound{
+	return &dto.HistoryRound{
 		Id:        s.Id,
 		SessionId: s.SessionId,
 		NodeId:    s.NodeId,
 		CreatedAt: gconv.String(s.CreatedAt),
 		UpdatedAt: gconv.String(s.UpdatedAt),
+		User:      s.RequestContent,
+		Assistant: s.ResponseContent,
 	}
-	if len(reqMsgs) > 0 {
-		round.User = reqMsgs[0]
-	}
-	if len(respMsgs) > 0 {
-		round.Assistant = respMsgs[0]
-	}
-	return round
 }

 // sessionsToHistoryRounds 批量转换