279 lines
14 KiB
Go
279 lines
14 KiB
Go
package v1
|
||
|
||
import "github.com/gogf/gf/v2/frame/g"
|
||
|
||
// ModelListReq 模型列表请求
|
||
type ModelListReq struct {
|
||
g.Meta `path:"/models" method:"get" mime:"json" tags:"管理后台-模型" summary:"模型列表"`
|
||
Page int `json:"page" d:"1" v:"min:1" dc:"页码"`
|
||
PageSize int `json:"page_size" d:"20" v:"min:1|max:100" dc:"每页数量"`
|
||
Category string `json:"category" dc:"模型分类筛选：chat/embedding/image/audio/rerank/video"`
|
||
Status string `json:"status" dc:"状态筛选：active/deprecated/offline"`
|
||
Search string `json:"search" dc:"搜索关键词（模型名或显示名）"`
|
||
PricingStatus string `json:"pricing_status" dc:"定价状态筛选：priced/unpriced"`
|
||
}
|
||
|
||
// ModelListRes 模型列表响应
|
||
type ModelListRes struct {
|
||
List []ModelItem `json:"list"`
|
||
Total int `json:"total"`
|
||
Page int `json:"page"`
|
||
PageSize int `json:"page_size"`
|
||
}
|
||
|
||
// ModelChannelInfo 模型关联的渠道信息
|
||
type ModelChannelInfo struct {
|
||
ChannelID int64 `json:"channel_id"`
|
||
ChannelName string `json:"channel_name"`
|
||
Type int `json:"type"`
|
||
}
|
||
|
||
// ModelItem 模型信息
|
||
type ModelItem struct {
|
||
ID int64 `json:"id"`
|
||
ModelId string `json:"model_id"`
|
||
ModelName string `json:"model_name"`
|
||
Category string `json:"category"`
|
||
Status string `json:"status"`
|
||
MaxContext int `json:"max_context_tokens"`
|
||
MaxOutput int `json:"max_output_tokens"`
|
||
Capabilities map[string]bool `json:"capabilities"`
|
||
Description string `json:"description"`
|
||
Tags []string `json:"tags"`
|
||
CreatedAt string `json:"created_at"`
|
||
UpdatedAt string `json:"updated_at"`
|
||
DeprecatedAt *string `json:"deprecated_at"`
|
||
SunsetDate *string `json:"sunset_date"`
|
||
ReplacementModel string `json:"replacement_model"`
|
||
// 定价摘要（来自 mdl_pricing，min_tokens=0 的基准行）
|
||
PricingMode string `json:"pricing_mode"` // "" | "token" | "per_request" | "tiered"
|
||
InputPrice float64 `json:"input_price"` // $/1M tokens
|
||
OutputPrice float64 `json:"output_price"` // $/1M tokens
|
||
PerRequestPrice float64 `json:"per_request_price"` // $/request（按次计费模式）
|
||
// 可用渠道列表（来自 chn_abilities + chn_channels）
|
||
Channels []ModelChannelInfo `json:"channels"`
|
||
}
|
||
|
||
// ModelCreateReq 创建模型请求
|
||
type ModelCreateReq struct {
|
||
g.Meta `path:"/models" method:"post" mime:"json" tags:"管理后台-模型" summary:"创建模型"`
|
||
ModelId string `json:"model_id" v:"required|length:1,100#请输入模型标识|模型标识长度1-100" dc:"模型唯一标识"`
|
||
ModelName string `json:"model_name" dc:"模型显示名称"`
|
||
Category string `json:"category" v:"required|in:chat,embedding,image,audio,rerank,video#请选择分类|分类必须是 chat/embedding/image/audio/rerank/video" dc:"模型分类"`
|
||
MaxContext int `json:"max_context_tokens" dc:"最大上下文 token 数"`
|
||
MaxOutput int `json:"max_output_tokens" dc:"最大输出 token 数"`
|
||
Capabilities map[string]bool `json:"capabilities" dc:"模型能力特性"`
|
||
Description string `json:"description" dc:"模型描述"`
|
||
Tags []string `json:"tags" dc:"标签列表"`
|
||
}
|
||
|
||
// ModelCreateRes 创建模型响应
|
||
type ModelCreateRes struct {
|
||
ID int64 `json:"id"`
|
||
}
|
||
|
||
// ModelUpdateReq 更新模型请求
|
||
type ModelUpdateReq struct {
|
||
g.Meta `path:"/models/{id}" method:"put" mime:"json" tags:"管理后台-模型" summary:"更新模型"`
|
||
ID int64 `json:"id" in:"path" v:"required" dc:"模型ID"`
|
||
ModelName string `json:"model_name" dc:"模型显示名称"`
|
||
Category string `json:"category" v:"in:chat,embedding,image,audio,rerank,video" dc:"模型分类"`
|
||
MaxContext int `json:"max_context_tokens" dc:"最大上下文 token 数"`
|
||
MaxOutput int `json:"max_output_tokens" dc:"最大输出 token 数"`
|
||
Capabilities map[string]bool `json:"capabilities" dc:"模型能力特性"`
|
||
Description string `json:"description" dc:"模型描述"`
|
||
Tags []string `json:"tags" dc:"标签列表"`
|
||
Status string `json:"status" v:"in:active,deprecated,offline" dc:"状态"`
|
||
SunsetDate *string `json:"sunset_date" dc:"下线日期（格式：YYYY-MM-DD，仅 deprecated 状态有效）"`
|
||
ReplacementModel string `json:"replacement_model" dc:"推荐替代模型名"`
|
||
}
|
||
|
||
// ModelDeleteReq 删除模型请求
|
||
type ModelDeleteReq struct {
|
||
g.Meta `path:"/models/{id}" method:"delete" mime:"json" tags:"管理后台-模型" summary:"删除模型"`
|
||
ID int64 `json:"id" in:"path" v:"required" dc:"模型ID"`
|
||
}
|
||
|
||
// PricingItem 定价项（支持按次/按量/阶梯）
|
||
type PricingItem struct {
|
||
BillingMode string `json:"billing_mode" v:"required|in:token,per_request,tiered" dc:"计费模式"`
|
||
MinTokens int64 `json:"min_tokens" dc:"阶梯起始 token 数"`
|
||
MaxTokens *int64 `json:"max_tokens" dc:"阶梯结束 token 数（NULL=无上限）"`
|
||
InputPrice float64 `json:"input_price" dc:"每 1M input token 价格"`
|
||
OutputPrice float64 `json:"output_price" dc:"每 1M output token 价格"`
|
||
PerRequestPrice *float64 `json:"per_request_price" dc:"按次单价（仅 per_request）"`
|
||
CacheReadPrice float64 `json:"cache_read_price" dc:"缓存读取每 1M token 价格"`
|
||
CacheCreationPrice float64 `json:"cache_creation_price" dc:"缓存创建每 1M token 价格"`
|
||
}
|
||
|
||
// PricingGetReq 获取模型定价
|
||
type PricingGetReq struct {
|
||
g.Meta `path:"/models/{model_id}/pricing" method:"get" mime:"json" tags:"管理后台-模型" summary:"获取模型定价"`
|
||
ModelID int64 `json:"model_id" in:"path" v:"required" dc:"模型ID"`
|
||
}
|
||
|
||
type PricingGetRes struct {
|
||
List []PricingItem `json:"list"`
|
||
}
|
||
|
||
// PricingSetReq 设置模型定价（全量替换）
|
||
type PricingSetReq struct {
|
||
g.Meta `path:"/models/{model_id}/pricing" method:"put" mime:"json" tags:"管理后台-模型" summary:"设置模型定价"`
|
||
ModelID int64 `json:"model_id" in:"path" v:"required" dc:"模型ID"`
|
||
Items []PricingItem `json:"items" v:"required" dc:"定价列表"`
|
||
}
|
||
|
||
type PricingSetRes struct{}
|
||
|
||
// ModelUpdateRes 更新模型响应
|
||
type ModelUpdateRes struct{}
|
||
|
||
// ModelDeleteRes 删除模型响应
|
||
type ModelDeleteRes struct{}
|
||
|
||
// ModelOptionsReq 模型选项列表请求（下拉选择专用，不分页）
|
||
type ModelOptionsReq struct {
|
||
g.Meta `path:"/models/options" method:"get" mime:"json" tags:"管理后台-模型" summary:"模型选项列表（不分页）"`
|
||
Status string `json:"status" in:"query" dc:"状态筛选：active/deprecated/offline"`
|
||
Category string `json:"category" in:"query" dc:"模型分类筛选：chat/embedding/image/audio/rerank/video"`
|
||
}
|
||
|
||
// ModelOptionsRes 模型选项列表响应
|
||
type ModelOptionsRes struct {
|
||
List []ModelOptionItem `json:"list"`
|
||
}
|
||
|
||
// ModelOptionItem 模型选项项（精简字段）
|
||
type ModelOptionItem struct {
|
||
ID int64 `json:"id"`
|
||
ModelId string `json:"model_id"`
|
||
ModelName string `json:"model_name"`
|
||
Category string `json:"category"`
|
||
}
|
||
|
||
// ModelExportReq 导出模型列表请求
|
||
type ModelExportReq struct {
|
||
g.Meta `path:"/models/export" method:"get" mime:"json" tags:"管理后台-模型" summary:"导出模型列表"`
|
||
Format string `json:"format" in:"query" d:"csv" v:"in:csv,xlsx" dc:"导出格式：csv / xlsx"`
|
||
Category string `json:"category" in:"query" dc:"模型分类筛选：chat/embedding/image/audio/rerank"`
|
||
Status string `json:"status" in:"query" dc:"状态筛选：active/deprecated/offline"`
|
||
Search string `json:"search" in:"query" dc:"搜索关键词（模型名或显示名）"`
|
||
}
|
||
|
||
type ModelExportRes struct{}
|
||
|
||
// PricingFetchOfficialReq 拉取模型官方定价
|
||
type PricingFetchOfficialReq struct {
|
||
g.Meta `path:"/models/{model_id}/official-pricing" method:"get" mime:"json" tags:"管理后台-模型" summary:"拉取模型官方定价"`
|
||
ModelID int64 `json:"model_id" in:"path" v:"required" dc:"模型ID"`
|
||
}
|
||
|
||
// PricingFetchOfficialRes 拉取官方定价响应
|
||
type PricingFetchOfficialRes struct {
|
||
ModelName string `json:"model_name"` // 数据库中的模型名
|
||
Sources []*OfficialPricingSource `json:"sources"` // 多来源定价数据
|
||
}
|
||
|
||
// OfficialPricingSource 单个来源的官方定价数据
|
||
type OfficialPricingSource struct {
|
||
Source string `json:"source"` // 数据来源（"litellm" / "models.dev"）
|
||
Found bool `json:"found"` // 是否找到
|
||
Error string `json:"error,omitempty"` // 获取远程数据失败时的错误信息
|
||
Provider string `json:"provider,omitempty"` // 供应商（anthropic/openai/...）
|
||
Mode string `json:"mode,omitempty"` // 模型类型（chat/embedding/...）
|
||
MaxContext int `json:"max_context_tokens,omitempty"` // 最大上下文
|
||
MaxOutput int `json:"max_output_tokens,omitempty"` // 最大输出
|
||
Pricing *OfficialPricingItem `json:"pricing,omitempty"` // 定价信息
|
||
}
|
||
|
||
// OfficialPricingItem 官方定价项
|
||
type OfficialPricingItem struct {
|
||
InputPrice float64 `json:"input_price"` // $/1M tokens
|
||
OutputPrice float64 `json:"output_price"` // $/1M tokens
|
||
CacheReadPrice float64 `json:"cache_read_price"` // $/1M tokens
|
||
CacheCreationPrice float64 `json:"cache_creation_price"` // $/1M tokens
|
||
BillingMode string `json:"billing_mode"` // 建议计费模式
|
||
}
|
||
|
||
// ModelFetchOfficialInfoReq 按名称拉取官方模型信息（上下文长度+能力特性）
|
||
type ModelFetchOfficialInfoReq struct {
|
||
g.Meta `path:"/models/official-info" method:"get" mime:"json" tags:"管理后台-模型" summary:"拉取官方模型信息"`
|
||
ModelName string `json:"model_name" in:"query" v:"required" dc:"模型名称"`
|
||
}
|
||
|
||
// ModelFetchOfficialInfoRes 官方模型信息响应
|
||
type ModelFetchOfficialInfoRes struct {
|
||
Found bool `json:"found"`
|
||
Error string `json:"error,omitempty"` // 获取远程数据失败时的错误信息
|
||
Provider string `json:"provider,omitempty"`
|
||
MaxContextTokens int `json:"max_context_tokens,omitempty"`
|
||
MaxOutputTokens int `json:"max_output_tokens,omitempty"`
|
||
Capabilities map[string]bool `json:"capabilities,omitempty"`
|
||
}
|
||
|
||
// ModelExportJsonReq 导出模型配置（JSON，用于跨环境迁移）
|
||
type ModelExportJsonReq struct {
|
||
g.Meta `path:"/models/export-json" method:"post" mime:"json" tags:"管理后台-模型" summary:"导出模型配置JSON"`
|
||
ModelIds []string `json:"model_ids" v:"required#请至少选择一个模型" dc:"要导出的model_id列表"`
|
||
}
|
||
|
||
// ModelExportJsonRes 导出模型配置JSON响应（直接写入响应流）
|
||
type ModelExportJsonRes struct{}
|
||
|
||
// ModelImportPreviewReq 导入模型预览请求
|
||
type ModelImportPreviewReq struct {
|
||
g.Meta `path:"/models/import-preview" method:"post" tags:"管理后台-模型" summary:"导入模型预览"`
|
||
}
|
||
|
||
// ModelImportPreviewRes 导入模型预览响应
|
||
type ModelImportPreviewRes struct {
|
||
Models []ModelImportPreviewItem `json:"models"`
|
||
}
|
||
|
||
// ModelImportPreviewItem 导入预览项（模型信息 + 冲突标记）
|
||
type ModelImportPreviewItem struct {
|
||
ModelId string `json:"model_id"`
|
||
ModelName string `json:"model_name"`
|
||
Category string `json:"category"`
|
||
Status string `json:"status"`
|
||
MaxContextTokens int `json:"max_context_tokens"`
|
||
MaxOutputTokens int `json:"max_output_tokens"`
|
||
Capabilities map[string]bool `json:"capabilities"`
|
||
Description string `json:"description"`
|
||
Tags []string `json:"tags"`
|
||
SunsetDate string `json:"sunset_date"`
|
||
ReplacementModel string `json:"replacement_model"`
|
||
Pricing []PricingItem `json:"pricing"`
|
||
Conflict string `json:"conflict"` // "" 或 "exists"
|
||
}
|
||
|
||
// ModelImportReq 确认导入模型请求
|
||
type ModelImportReq struct {
|
||
g.Meta `path:"/models/import" method:"post" mime:"json" tags:"管理后台-模型" summary:"确认导入模型"`
|
||
Models []ModelImportItem `json:"models" v:"required#请至少导入一个模型"`
|
||
}
|
||
|
||
// ModelImportItem 导入模型项
|
||
type ModelImportItem struct {
|
||
ModelId string `json:"model_id" v:"required" dc:"模型唯一标识"`
|
||
ModelName string `json:"model_name" dc:"模型显示名称"`
|
||
Category string `json:"category" v:"required|in:chat,embedding,image,audio,rerank,video#请选择分类|分类无效" dc:"模型分类"`
|
||
Status string `json:"status" dc:"状态"`
|
||
MaxContextTokens int `json:"max_context_tokens" dc:"最大上下文 token 数"`
|
||
MaxOutputTokens int `json:"max_output_tokens" dc:"最大输出 token 数"`
|
||
Capabilities map[string]bool `json:"capabilities" dc:"模型能力特性"`
|
||
Description string `json:"description" dc:"模型描述"`
|
||
Tags []string `json:"tags" dc:"标签列表"`
|
||
SunsetDate string `json:"sunset_date" dc:"下线日期"`
|
||
ReplacementModel string `json:"replacement_model" dc:"推荐替代模型名"`
|
||
Pricing []PricingItem `json:"pricing" dc:"定价列表"`
|
||
ConflictAction string `json:"conflict_action" v:"in:skip,overwrite" dc:"冲突处理策略：skip/overwrite"`
|
||
}
|
||
|
||
// ModelImportRes 导入结果响应
|
||
type ModelImportRes struct {
|
||
Imported int `json:"imported"`
|
||
Skipped int `json:"skipped"`
|
||
Errors []string `json:"errors,omitempty"`
|
||
}
|