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
	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
	IsOwner           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",
	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",
	IsOwner:           "is_owner",
}

// 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                 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"`
	IsOwner              *int   `json:"isOwner" orm:"is_owner"` // 1=当前用户创建的，0=超级管理员的
}