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
	ResponseTokenField 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
	CallbackUrl        string
	MaxTokens          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",
	ResponseBody:       "response_body",
	ResponseTokenField: "response_token_field",
	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",
	CallbackUrl:        "callback_url",
	MaxTokens:          "max_tokens",
}

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"`
	ResponseBody       string           `orm:"response_body" json:"responseBody"`
	ResponseTokenField string           `orm:"response_token_field" json:"tokenField"`
	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"`
	CallbackUrl        string           `orm:"callback_url" json:"callbackUrl"`
	MaxTokens          int              `orm:"max_tokens" json:"maxTokens"`
}