From 5f98e52b3461397e6d5f870a98af20eadc39dba6 Mon Sep 17 00:00:00 2001 From: WangLiZhao <1838393649@qq.com> Date: Fri, 15 May 2026 15:16:26 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E9=87=8D=E6=9E=84=E6=8F=90?= =?UTF-8?q?=E7=A4=BA=E8=AF=8D=E6=9E=84=E5=BB=BA=E9=80=BB=E8=BE=91=E5=B9=B6?= =?UTF-8?q?=E7=A7=BB=E9=99=A4=E4=B8=8D=E5=86=8D=E4=BD=BF=E7=94=A8=E7=9A=84?= =?UTF-8?q?=E6=96=87=E4=BB=B6=E5=A4=84=E7=90=86=E6=9C=8D=E5=8A=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config.yml | 2 +- controller/prompt_controller.go | 20 ++----------------- service/{build_service.go => build_prompt.go} | 1 - service/compose_service.go | 2 ++ service/{files_service.go => files_handle.go} | 0 5 files changed, 5 insertions(+), 20 deletions(-) rename service/{build_service.go => build_prompt.go} (99%) rename service/{files_service.go => files_handle.go} (100%) diff --git a/config.yml b/config.yml index f30876e..fd4b23d 100644 --- a/config.yml +++ b/config.yml @@ -1,5 +1,5 @@ server: - address: ":3005" + address: ":3009" name: "prompts-core" workerId: 1 # 雪花算法 worker ID(用于 common/db/gfdb) diff --git a/controller/prompt_controller.go b/controller/prompt_controller.go index cd34458..00a9c91 100644 --- a/controller/prompt_controller.go +++ b/controller/prompt_controller.go @@ -14,7 +14,7 @@ type prompt struct{} // 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) { return service.Prompt.ComposeMessages(ctx, req) } @@ -58,23 +58,7 @@ func (c *prompt) GetPrompt(ctx context.Context, req *dto.GetPromptReq) (res *dto // ListPrompt 配置列表 func (c *prompt) ListPrompt(ctx context.Context, req *dto.ListPromptReq) (res *dto.ListPromptRes, err error) { - pageNum, pageSize := 1, 10 - 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) + list, total, err := service.Prompt.List(ctx, int(req.Page.PageNum), int(req.Page.PageSize), req.ModelTypeId, req.ModelType) if err != nil { return nil, err } diff --git a/service/build_service.go b/service/build_prompt.go similarity index 99% rename from service/build_service.go rename to service/build_prompt.go index 87ac326..4a4e32d 100644 --- a/service/build_service.go +++ b/service/build_prompt.go @@ -34,7 +34,6 @@ func buildInferenceRequest(ctx context.Context, req *dto.ComposeMessagesReq, cha "role": "system", "content": promptBuild(ctx, req, model), }) - // 2. 构建历史会话提示词 for _, msg := range history { role := gconv.String(msg["role"]) diff --git a/service/compose_service.go b/service/compose_service.go index ba300cf..9fd7779 100644 --- a/service/compose_service.go +++ b/service/compose_service.go @@ -114,7 +114,9 @@ func (s *promptService) ComposeMessages(ctx context.Context, req *dto.ComposeMes if err != nil { return nil, err } + fmt.Println("构建节点前", taskRecord) message = s.parseNodeBuild(taskRecord) + fmt.Println("构建节点后", message) default: epicycleId, err = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{ SessionId: req.SessionId, diff --git a/service/files_service.go b/service/files_handle.go similarity index 100% rename from service/files_service.go rename to service/files_handle.go