89 lines
3.4 KiB
Go
89 lines
3.4 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
|
||
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=超级管理员的
|
||
}
|