refactor: 重构提示词构建逻辑并移除不再使用的文件处理服务

This commit is contained in:
2026-05-15 15:16:26 +08:00
parent 6497629fd0
commit 5f98e52b34
5 changed files with 5 additions and 20 deletions

View File

@@ -1,5 +1,5 @@
server: server:
address: ":3005" address: ":3009"
name: "prompts-core" name: "prompts-core"
workerId: 1 # 雪花算法 worker ID用于 common/db/gfdb workerId: 1 # 雪花算法 worker ID用于 common/db/gfdb

View File

@@ -14,7 +14,7 @@ type prompt struct{}
// Prompt 提示词配置控制器 // Prompt 提示词配置控制器
var Prompt = new(prompt) var Prompt = new(prompt)
// ComposeMessages 调用 model-gateway 异步任务并同步等待结果 // ComposeMessages 调用 model-gateway 异步任务并同步等待结果,
func (c *prompt) ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (res *dto.ComposeMessagesRes, err error) { func (c *prompt) ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (res *dto.ComposeMessagesRes, err error) {
return service.Prompt.ComposeMessages(ctx, req) return service.Prompt.ComposeMessages(ctx, req)
} }
@@ -58,23 +58,7 @@ func (c *prompt) GetPrompt(ctx context.Context, req *dto.GetPromptReq) (res *dto
// ListPrompt 配置列表 // ListPrompt 配置列表
func (c *prompt) ListPrompt(ctx context.Context, req *dto.ListPromptReq) (res *dto.ListPromptRes, err error) { func (c *prompt) ListPrompt(ctx context.Context, req *dto.ListPromptReq) (res *dto.ListPromptRes, err error) {
pageNum, pageSize := 1, 10 list, total, err := service.Prompt.List(ctx, int(req.Page.PageNum), int(req.Page.PageSize), req.ModelTypeId, req.ModelType)
if req != nil && req.Page != nil {
if req.Page.PageNum > 0 {
pageNum = int(req.Page.PageNum)
}
if req.Page.PageSize > 0 {
pageSize = int(req.Page.PageSize)
}
}
var modelTypeID *int
modelType := ""
if req != nil {
modelTypeID = req.ModelTypeId
modelType = req.ModelType
}
list, total, err := service.Prompt.List(ctx, pageNum, pageSize, modelTypeID, modelType)
if err != nil { if err != nil {
return nil, err return nil, err
} }

View File

@@ -34,7 +34,6 @@ func buildInferenceRequest(ctx context.Context, req *dto.ComposeMessagesReq, cha
"role": "system", "role": "system",
"content": promptBuild(ctx, req, model), "content": promptBuild(ctx, req, model),
}) })
// 2. 构建历史会话提示词 // 2. 构建历史会话提示词
for _, msg := range history { for _, msg := range history {
role := gconv.String(msg["role"]) role := gconv.String(msg["role"])

View File

@@ -114,7 +114,9 @@ func (s *promptService) ComposeMessages(ctx context.Context, req *dto.ComposeMes
if err != nil { if err != nil {
return nil, err return nil, err
} }
fmt.Println("构建节点前", taskRecord)
message = s.parseNodeBuild(taskRecord) message = s.parseNodeBuild(taskRecord)
fmt.Println("构建节点后", message)
default: default:
epicycleId, err = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{ epicycleId, err = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{
SessionId: req.SessionId, SessionId: req.SessionId,