package entity

import "gitea.redpowerfuture.com/red-future/common/beans"

type modelGatewayModelCol struct {
	beans.SQLBaseCol
	ModelName        string
	ModelType        string
	BaseURL          string
	HttpMethod       string
	HeadMsg          string
	FormJSON         string
	RequestMapping   string
	ResponseMapping  string
	ResponseBody     string
	RequiredFields   string
	IsPrivate        string
	IsChatModel      string
	CallMode         string
	ApiKey           string
	Enabled          string
	MaxConcurrency   string
	TimeoutSeconds   string
	RetryTimes       string
	AutoCleanSeconds string
	IsOwner          string
	OperatorName     string
	TokenConfig      string
	ExtendMapping    string
	QueryConfig      string
	StreamConfig     string
	FirstFrame       string
	LastFrame        string
}

var ModelGatewayModelCol = modelGatewayModelCol{
	SQLBaseCol:       beans.DefSQLBaseCol,
	ModelName:        "model_name",
	ModelType:        "model_type",
	BaseURL:          "base_url",
	HttpMethod:       "http_method",
	HeadMsg:          "head_msg",
	FormJSON:         "form_json",
	RequestMapping:   "request_mapping",
	ResponseMapping:  "response_mapping",
	RequiredFields:   "required_fields",
	IsPrivate:        "is_private",
	IsChatModel:      "is_chat_model",
	CallMode:         "call_mode",
	ApiKey:           "api_key",
	Enabled:          "enabled",
	MaxConcurrency:   "max_concurrency",
	TimeoutSeconds:   "timeout_seconds",
	RetryTimes:       "retry_times",
	AutoCleanSeconds: "auto_clean_seconds",
	IsOwner:          "is_owner",
	OperatorName:     "operator_name",
	TokenConfig:      "token_config",
	ExtendMapping:    "extend_mapping",
	QueryConfig:      "query_config",
	StreamConfig:     "stream_config",
	FirstFrame:       "first_frame",
	LastFrame:        "last_frame",
}

type ModelGatewayModel struct {
	beans.SQLBaseDO  `orm:",inline"`
	ModelName        string           `orm:"model_name" json:"modelName"`
	ModelType        int              `orm:"model_type" json:"modelType"`
	BaseURL          string           `orm:"base_url" json:"baseUrl"`
	HttpMethod       string           `orm:"http_method" json:"httpMethod"`
	HeadMsg          map[string]any   `orm:"head_msg" json:"headMsg"`
	Form             []map[string]any `orm:"form_json" json:"form"`
	RequestMapping   map[string]any   `orm:"request_mapping" json:"requestMapping"`
	ResponseMapping  map[string]any   `orm:"response_mapping" json:"responseMapping"`
	RequiredFields   []string         `orm:"required_fields" json:"requiredFields"`
	IsPrivate        *int             `orm:"is_private" json:"isPrivate"`
	IsChatModel      *int             `orm:"is_chat_model" json:"isChatModel"`
	CallMode         *int             `orm:"call_mode" json:"callMode"`
	ApiKey           string           `orm:"api_key" json:"apiKey"`
	Enabled          *int             `orm:"enabled" json:"enabled"`
	MaxConcurrency   int              `orm:"max_concurrency" json:"maxConcurrency"`
	TimeoutSeconds   int              `orm:"timeout_seconds" json:"timeoutSeconds"`
	RetryTimes       int              `orm:"retry_times" json:"retryTimes"`
	AutoCleanSeconds int              `orm:"auto_clean_seconds" json:"autoCleanSeconds"`
	IsOwner          *int             `orm:"is_owner" json:"isOwner"`
	OperatorName     string           `orm:"operator_name" json:"operatorName"`
	TokenConfig      map[string]any   `orm:"token_config" json:"tokenConfig"`
	ExtendMapping    map[string]any   `orm:"extend_mapping" json:"extendMapping"`
	QueryConfig      map[string]any   `orm:"query_config" json:"queryConfig"`
	StreamConfig     map[string]any   `orm:"stream_config" json:"streamConfig"`
	FirstFrame       string           `orm:"first_frame" json:"firstFrame"`
	LastFrame        string           `orm:"last_frame" json:"lastFrame"`
}

const ( //ResponseMapping 下的字段
	ResponseBody = "response_body" //返回主体
	TotalTokens  = "total_tokens"  //总token数
)