101 lines
4.1 KiB
Go
101 lines
4.1 KiB
Go
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
|
|
}
|
|
|
|
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",
|
|
}
|
|
|
|
// 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"`
|
|
}
|