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,
 	})