package entity

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

type asynchModelCol struct {
	beans.SQLBaseCol
	ModelName          string
	ModelType          string
	BaseURL            string
	HttpMethod         string
	HeadMsg            string
	FormJSON           string
	RequestMapping     string
	ResponseMapping    string
	ResponseBody       string
	ResponseTokenField string
	Prompt             string
	IsPrivate          string
	IsChatModel        string
	ApiKey             string
	Enabled            string
	MaxConcurrency     string
	QueueLimit         string
	TimeoutSeconds     string
	ExpectedSeconds    string
	RetryTimes         string
	RetryQueueMaxSecs  string
	AutoCleanSeconds   string
	Remark             string
	IsOwner            string
	OperatorName       string
	TokenConfig        string
	ExtendMapping      string
	QueryConfig        string
	StreamConfig       string
}

var AsynchModelCol = asynchModelCol{
	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",
	Prompt:             "prompt",
	IsPrivate:          "is_private",
	IsChatModel:        "is_chat_model",
	ApiKey:             "api_key",
	Enabled:            "enabled",
	MaxConcurrency:     "max_concurrency",
	QueueLimit:         "queue_limit",
	TimeoutSeconds:     "timeout_seconds",
	ExpectedSeconds:    "expected_seconds",
	RetryTimes:         "retry_times",
	RetryQueueMaxSecs:  "retry_queue_max_seconds",
	AutoCleanSeconds:   "auto_clean_seconds",
	Remark:             "remark",
	IsOwner:            "is_owner",
	OperatorName:       "operator_name",
	TokenConfig:        "token_config",
	ExtendMapping:      "extend_mapping",
	QueryConfig:        "query_config",
	StreamConfig:       "stream_config",
}

// AsynchModel 异步模型配置
type AsynchModel 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              string           `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         map[string]any   `orm:"response_body" json:"responseBody"`
	ResponseTokenField   string           `orm:"response_token_field" json:"responseTokenField"`
	Prompt               string           `orm:"prompt" json:"prompt"`
	IsPrivate            *int             `orm:"is_private" json:"isPrivate"`
	IsChatModel          *int             `orm:"is_chat_model" json:"isChatModel"`
	ApiKey               string           `orm:"api_key" json:"apiKey"`
	Enabled              *int             `orm:"enabled" json:"enabled"`
	MaxConcurrency       int              `orm:"max_concurrency" json:"maxConcurrency"`
	QueueLimit           int              `orm:"queue_limit" json:"queueLimit"`
	TimeoutSeconds       int              `orm:"timeout_seconds" json:"timeoutSeconds"`
	ExpectedSeconds      int              `orm:"expected_seconds" json:"expectedSeconds"`
	RetryTimes           int              `orm:"retry_times" json:"retryTimes"`
	RetryQueueMaxSeconds int              `orm:"retry_queue_max_seconds" json:"retryQueueMaxSeconds"`
	AutoCleanSeconds     int              `orm:"auto_clean_seconds" json:"autoCleanSeconds"`
	Remark               string           `orm:"remark" json:"remark"`
	IsOwner              *int             `json:"isOwner" orm:"is_owner"`
	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"`
}