prompts-core

service/build_service.go

@@ -3,6 +3,7 @@ package service
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"prompts-core/model/dto"
 	"prompts-core/model/entity"
@@ -12,45 +13,61 @@ import (
 	"github.com/gogf/gf/v2/util/gconv"
 )
 
-// 获取系统模型提示词
-func getConfPrompt(ctx context.Context, modelType int) string {
-	return g.Cfg().MustGet(ctx, "modelPrompts.types."+gconv.String(modelType), "").String()
+// 获取请求模型的提示词
+func GetModelPrompt(ctx context.Context, Type int) string {
+	return g.Cfg().MustGet(ctx, "modelPrompts.types."+gconv.String(Type), "").String()
 }
 
-func buildInferenceRequest(ctx context.Context, req *dto.ComposeMessagesReq, sessionModel *entity.AsynchModel, model *entity.AsynchModel, historyMessages []Message) (map[string]any, error) {
-	// 读取 task 相关的配置
-	// 构建消息数组
-	// 1. 系统提示词（不动）
-
-	fmt.Println("打印sessionModel结果", sessionModel)
-	fmt.Println("打印model结果", model)
+// 获取构建提示词
+func GetBuildPrompt(ctx context.Context, Type int) string {
+	return g.Cfg().MustGet(ctx, "buildProject.types."+gconv.String(Type), "").String()
+}
 
+// buildInferenceRequest 构建返回请求
+func buildInferenceRequest(ctx context.Context, req *dto.ComposeMessagesReq, chatModel *entity.AsynchModel, model *entity.AsynchModel, history []map[string]any) (map[string]any, error) {
 	messages := []map[string]any{}
-	messages = append(messages, map[string]any{
-		"role":    "system",
-		"content": GetSystemPrompt(ctx, req, model),
-	})
-
-	// 2. 历史对话 - 动态添加（新增部分）
-	for _, msg := range historyMessages {
+	switch req.BuildType {
+	//构建提示词请求
+	case 1:
+		//1. 构建系统提示词
 		messages = append(messages, map[string]any{
-			"role":    msg.Role,
-			"content": msg.GetContentString(),
+			"role":    "system",
+			"content": promptBuild(ctx, req, model),
 		})
-	}
-	// 3. 当前用户问题（原来的最后一条）
-	messages = append(messages, map[string]any{
-		"role":    "user",
-		"content": buildCombinedInput(req, getConfPrompt(ctx, model.ModelsType)),
-	})
 
+		// 2. 构建历史会话提示词
+		for _, msg := range history {
+			role := gconv.String(msg["role"])
+			content := gconv.String(msg["content"])
+			if role != "user" && role != "assistant" {
+				continue
+			}
+			messages = append(messages, map[string]any{
+				"role":    role,
+				"content": content,
+			})
+		}
+		// 3. 当前用户问题（原来的最后一条）
+		messages = append(messages, map[string]any{
+			"role":    "user",
+			"content": buildUserPrompt(ctx, req, GetModelPrompt(ctx, model.ModelType)),
+		})
+	//构建节点请求
+	case 2:
+		messages = append(messages, map[string]any{
+			"role":    "user",
+			"content": NodeBuid(ctx, req),
+		})
+	default:
+		return nil, errors.New("不支持的构建类型")
+	}
 	// 构建请求体
 	return map[string]any{
-		"modelName":   sessionModel.ModelName,
+		"modelName":   chatModel.ModelName,
 		"bizName":     "prompts-core",
 		"callbackUrl": "/prompt/callback",
 		"requestPayload": map[string]any{
-			"model":    sessionModel.ModelName,
+			"model":    chatModel.ModelName,
 			"messages": messages,
 			"stream":   false,
 		},
@@ -58,10 +75,11 @@ func buildInferenceRequest(ctx context.Context, req *dto.ComposeMessagesReq, ses
 }
 
 // ============================================
-// 输入构建
+// 构建用户提示词
 // ============================================
-func buildCombinedInput(req *dto.ComposeMessagesReq, prompt string) string {
+func buildUserPrompt(ctx context.Context, req *dto.ComposeMessagesReq, prompt string) string {
 	payload := map[string]any{
+		"model": req.ModelName,
 		//数据库提示信息
 		"promptInfo": prompt,
 		// 系统表单
@@ -71,20 +89,20 @@ func buildCombinedInput(req *dto.ComposeMessagesReq, prompt string) string {
 		//文件url
 		"userFiles": req.UserFiles,
 		//解读文件（只支持可读类型 如：xml，json,yaml）
-		"userFilesText": fetchFileTexts(context.Background(), req.UserFiles),
+		"userFilesText": FetchFileTexts(ctx, req.UserFiles),
+		//skill 相关（根据传入的 skillName 获取 zip 内所有 md 文件拼接内容）
+		"skills": SkillMdContent(ctx, req.SkillName),
 	}
 	return mustMarshal(payload)
 }
 
-// GetSystemPrompt 定义获取系统提示词的函数
-// GetSystemPrompt 从配置文件读取提示词 + 格式化变量
-func GetSystemPrompt(ctx context.Context, req *dto.ComposeMessagesReq, model *entity.AsynchModel) string {
+// promptBuild 提示词构建
+func promptBuild(ctx context.Context, req *dto.ComposeMessagesReq, model *entity.AsynchModel) string {
 	// 1. 从配置文件读取提示词模板
-	promptTpl := g.Cfg().MustGet(ctx, "projectPrompts", "").String()
+	promptTpl := GetBuildPrompt(ctx, req.BuildType)
 	if promptTpl == "" {
 		return ""
 	}
-
 	// 2. 构建字段映射说明
 	mappingBytes, _ := json.Marshal(model.RequestMapping)
 	mappingStr := string(mappingBytes)
@@ -115,11 +133,13 @@ func GetSystemPrompt(ctx context.Context, req *dto.ComposeMessagesReq, model *en
 	return fmt.Sprintf(promptTpl, mappingStr, fieldDesc.String(), formInfo)
 }
 
-// formToJSON 工具函数不变
-func formToJSON(form map[string]any) string {
-	if form == nil {
-		return "{}"
+// NodeBuid 节点构建
+func NodeBuid(ctx context.Context, req *dto.ComposeMessagesReq) string {
+	promptTpl := GetBuildPrompt(ctx, req.BuildType)
+	if promptTpl == "" {
+		return ""
 	}
-	b, _ := json.Marshal(form)
-	return string(b)
+	formStr := formToJSON(req.Form)
+	userFormStr := formToJSON(req.UserForm)
+	return fmt.Sprintf(promptTpl, formStr, userFormStr)
 }