refactor(util): 重构映射工具函数并优化异步任务轮询逻辑

2026-06-03 13:30:39 +08:00
parent c11a9ad5c8
commit 3fa2896fc3
9 changed files with 80 additions and 75 deletions
--- a/controller/prompt_compose_controller.go
+++ b/controller/prompt_compose_controller.go
@@ -6,6 +6,7 @@ import (
 	"prompts-core/dao"
 	"prompts-core/model/dto"
 	"prompts-core/model/entity"
+	"prompts-core/service/gateway"

 	promptService "prompts-core/service/prompt"

@@ -42,7 +43,7 @@ func (c *prompt) Text(ctx context.Context, req *dto.TextReq) (res *dto.TextRes,
 	if err != nil {
 		return
 	}
-	model, err := dao.Model.Get(ctx, &entity.AsynchModel{
+	model, err := gateway.GetModelConfig(ctx, &gateway.AsynchModel{
 		SQLBaseDO: beans.SQLBaseDO{Creator: composeTask.Creator},
 		ModelName: composeTask.ModelName,
 	})
--- a/controller/prompt_session_controller.go
+++ b/controller/prompt_session_controller.go
@@ -16,8 +16,3 @@ var Session = new(session)
 func (c *session) SessionCallback(ctx context.Context, req *dto.SessionCallbackReq) (res *dto.SessionCallbackRes, err error) {
 	return sessionService.Callback(ctx, req)
 }
-
-//TODO:后期历史相关服务可能拆分(三个接口)
-// 1. 添加历史会话
-// 2. 获取历史会话
-// 3. 更新历史信息