package entity

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

type asynchModelCol struct {
	beans.SQLBaseCol
	ModelName         string
	ModelsType        string
	BaseURL           string
	HttpMethod        string
	HeadMsg           string
	FormJSON          string
	RequestMapping    string
	ResponseMapping   string
	ResponseBody      string
	TokenMapping      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
}

var AsynchModelCol = asynchModelCol{
	SQLBaseCol:        beans.DefSQLBaseCol,
	ModelName:         "model_name",
	ModelsType:        "models_type",
	BaseURL:           "base_url",
	HttpMethod:        "http_method",
	HeadMsg:           "head_msg",
	FormJSON:          "form_json",
	RequestMapping:    "request_mapping",
	ResponseMapping:   "response_mapping",
	ResponseBody:      "response_body",
	TokenMapping:      "token_mapping",
	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",
}

// AsynchModel 异步模型配置
type AsynchModel struct {
	beans.SQLBaseDO      `orm:",inline"`
	ModelName            string `orm:"model_name" json:"modelName"`
	ModelsType           int    `orm:"models_type" json:"modelsType"`
	BaseURL              string `orm:"base_url" json:"baseUrl"`
	HttpMethod           string `orm:"http_method" json:"httpMethod"`
	HeadMsg              string `orm:"head_msg" json:"headMsg"`
	Form                 any    `orm:"form_json" json:"form"`
	RequestMapping       any    `orm:"request_mapping" json:"requestMapping"`
	ResponseMapping      any    `orm:"response_mapping" json:"responseMapping"`
	ResponseBody         any    `orm:"response_body" json:"responseBody"`
	TokenMapping         string `orm:"token_mapping" json:"tokenMapping"`
	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"`
}