115 lines
4.1 KiB
Go
115 lines
4.1 KiB
Go
package dto
|
||
|
||
import (
|
||
"rag/consts/model"
|
||
"time"
|
||
|
||
"gitea.com/red-future/common/beans"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
)
|
||
|
||
type GetModelAllEnumsReq struct {
|
||
g.Meta `path:"/getAllEnums" method:"get" tags:"模型配置管理" summary:"获取全量模型枚举（类型+配置）"`
|
||
}
|
||
|
||
type GetModelEnumRes struct {
|
||
Options []ModelEnumOption `json:"options"`
|
||
}
|
||
|
||
// ModelEnumOption 主类型：模型类型（vector/chat）
|
||
type ModelEnumOption struct {
|
||
Key interface{} `json:"key"`
|
||
Value interface{} `json:"value"`
|
||
ConfigTypes []ModelKeyValue `json:"configTypes"` // 这里统一！
|
||
}
|
||
|
||
// ModelKeyValue 统一的 KV 结构 → 给模型类型 + 配置类型共用
|
||
type ModelKeyValue struct {
|
||
Key interface{} `json:"key"`
|
||
Value interface{} `json:"value"`
|
||
}
|
||
|
||
// GetModelConfigFormFieldsReq 获取模型配置表单请求
|
||
type GetModelConfigFormFieldsReq struct {
|
||
g.Meta `path:"/getModelFormField" method:"get" tags:"模型配置管理" summary:"获取模型表单" dc:"获取模型表单列表"`
|
||
|
||
ModelType model.ModelType `json:"modelType"` // 模型类型 vector/chat
|
||
ConfigType model.ModelConfigType `json:"configType"` // 配置类型 ark/ollama/openai...
|
||
}
|
||
|
||
// GetModelConfigFormFieldsRes 获取模型配置表单响应
|
||
type GetModelConfigFormFieldsRes struct {
|
||
ModelType model.ModelType `json:"modelType"`
|
||
ConfigType model.ModelConfigType `json:"configType"`
|
||
Fields []map[string]interface{} `json:"fields"`
|
||
}
|
||
|
||
// CreateModelReq 创建模型请求
|
||
type CreateModelReq struct {
|
||
g.Meta `path:"/create" method:"post" tags:"模型配置管理" summary:"创建模型配置" dc:"创建模型配置"`
|
||
|
||
ModelType model.ModelType `json:"modelType" v:"required#模型类型不能为空"`
|
||
ModelName string `json:"modelName" v:"required#模型名称不能为空"`
|
||
ModelDesc string `json:"modelDesc"`
|
||
ConfigType model.ModelConfigType `json:"configType"`
|
||
ConfigContent map[string]interface{} `json:"configContent"`
|
||
}
|
||
|
||
// CreateModelRes 创建模型响应
|
||
type CreateModelRes struct {
|
||
Id int64 `json:"id,string"`
|
||
}
|
||
|
||
// UpdateModelReq 更新模型请求
|
||
type UpdateModelReq struct {
|
||
g.Meta `path:"/update" method:"put" tags:"模型配置管理" summary:"更新模型配置" dc:"更新模型配置"`
|
||
|
||
Id int64 `json:"id" v:"required#ID不能为空"`
|
||
ModelType model.ModelType `json:"modelType"`
|
||
ModelName string `json:"modelName"`
|
||
ModelDesc string `json:"modelDesc"`
|
||
ConfigType model.ModelConfigType `json:"configType"`
|
||
ConfigContent map[string]interface{} `json:"configContent"`
|
||
}
|
||
|
||
// DeleteModelReq 删除模型请求
|
||
type DeleteModelReq struct {
|
||
g.Meta `path:"/delete" method:"delete" tags:"模型配置管理" summary:"删除模型配置" dc:"删除模型配置"`
|
||
|
||
Id int64 `json:"id" v:"required#ID不能为空"`
|
||
}
|
||
|
||
// GetModelReq 获取模型请求
|
||
type GetModelReq struct {
|
||
g.Meta `path:"/get" method:"get" tags:"模型配置管理" summary:"获取模型配置详情" dc:"获取模型配置详情"`
|
||
|
||
Id int64 `json:"id"`
|
||
ModelType model.ModelType `json:"modelType"`
|
||
}
|
||
|
||
// ListModelReq 获取模型列表请求
|
||
type ListModelReq struct {
|
||
g.Meta `path:"/list" method:"get" tags:"模型配置管理" summary:"获取模型配置列表" dc:"分页查询模型配置列表，支持多条件筛选"`
|
||
|
||
Page *beans.Page `json:"page"`
|
||
ModelType model.ModelType `json:"modelType"`
|
||
ModelName string `json:"modelName"`
|
||
}
|
||
|
||
// ListModelRes 获取模型列表响应
|
||
type ListModelRes struct {
|
||
List []*ModelVO `json:"list"`
|
||
Total int `json:"total"`
|
||
}
|
||
|
||
type ModelVO struct {
|
||
Id int64 `json:"id,string"`
|
||
ModelType model.ModelType `json:"modelType"`
|
||
ModelName string `json:"modelName"`
|
||
ModelDesc string `json:"modelDesc"`
|
||
ConfigType model.ModelConfigType `json:"configType"`
|
||
ConfigContent map[string]interface{} `json:"configContent"`
|
||
CreateTime time.Time `json:"createTime"`
|
||
UpdateTime time.Time `json:"updateTime"`
|
||
}
|