refactor: 重构模型 HTTP 客户端并移除冗余代码

2026-08-29 11:53:02 +08:00
parent e079f4ea28
commit 4a9ae2d412
43 changed files with 719 additions and 4295 deletions
@@ -1,18 +0,0 @@
-package controller
-
-import (
-	"context"
-	statService "model-gateway/service/stat"
-
-	"model-gateway/model/dto"
-)
-
-// ModelGatewayLogsStat 统计控制器
-var ModelGatewayLogsStat = new(stat)
-
-type stat struct{}
-
-// ListModelStat 统计列表
-func (c *stat) ListModelStat(ctx context.Context, req *dto.ListModelStatReq) (res *dto.ListModelStatRes, err error) {
-	return statService.ModelGatewayLogsStat.List(ctx, req)
-}
@@ -1,66 +0,0 @@
-package controller
-
-import (
-	"context"
-	"model-gateway/model/dto"
-	modelService "model-gateway/service/model"
-	"model-gateway/service/queue"
-)
-
-// ModelGatewayModels 模型配置控制器
-var ModelGatewayModels = new(model)
-
-type model struct{}
-
-// CreateModel 添加配置
-func (c *model) CreateModel(ctx context.Context, req *dto.CreateModelReq) (res *dto.CreateModelRes, err error) {
-	return modelService.ModelGatewayModels.Create(ctx, req)
-}
-
-// UpdateModel 更改配置
-func (c *model) UpdateModel(ctx context.Context, req *dto.UpdateModelReq) (res *dto.UpdateModelRes, err error) {
-	err = modelService.ModelGatewayModels.Update(ctx, req)
-	return
-}
-
-// DeleteModel 删除配置
-func (c *model) DeleteModel(ctx context.Context, req *dto.DeleteModelReq) (res *dto.DeleteModelRes, err error) {
-	err = modelService.ModelGatewayModels.Delete(ctx, req)
-	return
-}
-
-// GetModel 获取配置详情
-func (c *model) GetModel(ctx context.Context, req *dto.GetModelReq) (res *dto.GetModelRes, err error) {
-	return modelService.ModelGatewayModels.Get(ctx, req)
-}
-
-// ListModel 配置列表
-func (c *model) ListModel(ctx context.Context, req *dto.ListModelReq) (res *dto.ListModelRes, err error) {
-	return modelService.ModelGatewayModels.List(ctx, req)
-}
-
-// AutoTune 动态调参（由上层定时任务每小时触发一次）
-func (c *model) AutoTune(ctx context.Context, req *dto.AutoTuneReq) (res *dto.AutoTuneRes, err error) {
-	return queue.AutoTune(ctx, req)
-}
-
-// ListType 模型类型列表
-func (c *model) ListType(ctx context.Context, req *dto.ListTypeReq) (res *dto.TypeItem, err error) {
-	return modelService.GetModelTypesFromConfig()
-}
-
-// ListOperator 运营商列表
-func (c *model) ListOperator(ctx context.Context, req *dto.ListOperatorReq) (res *dto.ListOperatorRes, err error) {
-	return modelService.GetOperatorList()
-}
-
-// UpdateChatModel 更新是否为聊天模型
-func (c *model) UpdateChatModel(ctx context.Context, req *dto.UpdateChatModelReq) (res *dto.UpdateChatModelRes, err error) {
-	err = modelService.ModelGatewayModels.UpdateChatModel(ctx, req)
-	return
-}
-
-// GetIsChatModel 获取当前会话模型
-func (c *model) GetIsChatModel(ctx context.Context, req *dto.GetIsChatModelReq) (res *dto.GetIsChatModelRes, err error) {
-	return modelService.ModelGatewayModels.GetIsChatModel(ctx)
-}
@@ -1,43 +0,0 @@
-package controller
-
-import (
-	"context"
-	taskService "model-gateway/service/task"
-
-	"model-gateway/model/dto"
-)
-
-// ModelGatewayTask 任务控制器
-var ModelGatewayTask = new(task)
-
-type task struct{}
-
-// CreateTask 根据 modelName 创建异步任务，返回 taskId
-func (c *task) CreateTask(ctx context.Context, req *dto.CreateTaskReq) (res *dto.CreateTaskRes, err error) {
-	return taskService.ModelGatewayTask.Create(ctx, req)
-}
-
-// GetTaskResult 获取单条任务结果（返回 *dto.GetTaskResultRes）
-func (c *task) GetTaskResult(ctx context.Context, req *dto.GetTaskResultReq) (res *dto.GetTaskResultRes, err error) {
-	return taskService.ModelGatewayTask.GetResult(ctx, req.TaskID)
-}
-
-// GetTaskBatch 批量查询任务（返回 *[]dto.GetTaskBatchItem）
-func (c *task) GetTaskBatch(ctx context.Context, req *dto.GetTaskBatchReq) (res *dto.GetTaskBatchRes, err error) {
-	return taskService.ModelGatewayTask.GetBatch(ctx, req)
-}
-
-// ListTask 任务列表分页查询
-func (c *task) ListTask(ctx context.Context, req *dto.ListTaskReq) (res *dto.ListTaskRes, err error) {
-	return taskService.ModelGatewayTask.List(ctx, req)
-}
-
-// ModelTaskCallback 接收模型异步任务的回调通知 —— 待调整
-func (c *task) ModelTaskCallback(ctx context.Context, req *dto.ModelTaskCallbackReq) (res *dto.ModelTaskCallbackRes, err error) {
-	return taskService.ModelGatewayTask.ModelTaskCallback(ctx, req)
-}
-
-// QueryPendingTasks 批量轮询进行中的异步任务 —— 待调整
-func (c *task) QueryPendingTasks(ctx context.Context, req *dto.QueryPendingTasksReq) (res *dto.QueryPendingTasksRes, err error) {
-	return taskService.ModelGatewayTask.QueryPendingTasks(ctx, req)
-}