domain

type CheckModelReq struct {
	Owner   consts.ModelOwner `json:"owner" validate:"required"`      // 提供商
	ModelID string            `json:"model_name" validate:"required"` // 模型名称
	APIKey  string            `json:"api_key" validate:"required"`    // 接口密钥
	SubType consts.ModelType  `json:"sub_type" validate:"required"`   // 模型类型
}

type CheckModelResp struct {
	Error   string `json:"error"`
	Content string `json:"content"`
}

type GetProviderModelListReq struct {
	Provider  string `` /* 166-byte string literal not displayed */
	BaseURL   string `json:"base_url" query:"base_url" validate:"required"`
	APIKey    string `json:"api_key" query:"api_key"`
	APIHeader string `json:"api_header" query:"api_header"`
	Type      string `json:"type" query:"type" validate:"required,oneof=chat embedding rerank"`
}

type GetProviderModelListResp struct {
	Models []ProviderModelListItem `json:"models"`
}

type ListModelReq struct {
	OwnedBy consts.ModelOwner `json:"owned_by" query:"owned_by"` // 提供商
	SubType consts.ModelType  `json:"sub_type" query:"sub_type"` // 模型类型
}

type Model struct {
	ID        string            `json:"id"`         // 模型的名字
	Object    string            `json:"object"`     // 总是model
	Created   int               `json:"created"`    // 创建时间
	OwnedBy   consts.ModelOwner `json:"owned_by"`   // 提供商
	ModelType consts.ModelType  `json:"model_type"` // 模型类型
}

type ModelKit interface {
	CheckModel(ctx context.Context, req *CheckModelReq) (*Model, error)
	ListModel(ctx context.Context, req *ListModelReq) ([]*Model, error)
	PandaModelList(ctx context.Context, req *GetProviderModelListReq) (*GetProviderModelListResp, error)
}

type ModelOwner struct {
	OwnerName  consts.ModelOwner `json:"owner_name"`  // 提供商
	APIBase    string            `json:"api_base"`    // 接口地址 如：https://api.qwen.com
	APIKey     string            `json:"api_key"`     // 接口密钥 如：sk-xxxx
	APIVersion string            `json:"api_version"` // 接口版本 如：2023-05-15
	APIHeader  string            `json:"api_header"`  // 接口头 如：Authorization: Bearer sk-xxxx
	Models     []*Model          `json:"models"`      // 模型列表
}

type ProviderModelListItem struct {
	Model string `json:"model"`
}

type Resp struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Data    any    `json:"data,omitempty"`
}

type Response struct {
	Message string `json:"message"`
	Success bool   `json:"success"`
	Data    any    `json:"data,omitempty"`
}