prompts-core

model/entity/asynch_model.go

@@ -0,0 +1,85 @@
+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"`
+}

model/entity/compose_session.go

@@ -0,0 +1,27 @@
+package entity
+
+import "gitea.com/red-future/common/beans"
+
+type composeSessionCol struct {
+	beans.SQLBaseCol
+	SessionId       string
+	RequestContent  string
+	ResponseContent string
+	Remark          string
+}
+
+var ComposeSessionCol = composeSessionCol{
+	SQLBaseCol:      beans.DefSQLBaseCol,
+	SessionId:       "session_id",
+	RequestContent:  "request_content",
+	ResponseContent: "response_content",
+	Remark:          "remark",
+}
+
+type ComposeSession struct {
+	beans.SQLBaseDO `orm:",inline"`
+	SessionId       string `orm:"session_id" json:"sessionId"`
+	RequestContent  any    `orm:"request_content" json:"requestContent"`
+	ResponseContent any    `orm:"response_content" json:"responseContent"`
+	Remark          string `orm:"remark" json:"remark"`
+}

model/entity/compose_task.go

@@ -0,0 +1,45 @@
+package entity
+
+import "gitea.com/red-future/common/beans"
+
+type composeTaskCol struct {
+	beans.SQLBaseCol
+	TaskId          string
+	ModelName       string
+	SkillName       string
+	LimitWords      string
+	RequestPayload  string
+	CallbackPayload string
+	ModelResult     string
+	Messages        string
+	Status          string
+	ErrorMessage    string
+}
+
+var ComposeTaskCol = composeTaskCol{
+	SQLBaseCol:      beans.DefSQLBaseCol,
+	TaskId:          "task_id",
+	ModelName:       "model_name",
+	SkillName:       "skill_name",
+	LimitWords:      "limit_words",
+	RequestPayload:  "request_payload",
+	CallbackPayload: "callback_payload",
+	ModelResult:     "model_result",
+	Messages:        "messages",
+	Status:          "status",
+	ErrorMessage:    "error_message",
+}
+
+type ComposeTask struct {
+	beans.SQLBaseDO `orm:",inline"`
+	TaskId          string `orm:"task_id" json:"taskId"`
+	ModelName       string `orm:"model_name" json:"modelName"`
+	SkillName       string `orm:"skill_name" json:"skillName"`
+	LimitWords      int    `orm:"limit_words" json:"limitWords"`
+	RequestPayload  any    `orm:"request_payload" json:"requestPayload"`
+	CallbackPayload any    `orm:"callback_payload" json:"callbackPayload"`
+	ModelResult     any    `orm:"model_result" json:"modelResult"`
+	Messages        any    `orm:"messages" json:"messages"`
+	Status          string `orm:"status" json:"status"`
+	ErrorMessage    string `orm:"error_message" json:"errorMessage"`
+}

model/entity/prompt_config.go

@@ -0,0 +1,39 @@
+package entity
+
+import "gitea.com/red-future/common/beans"
+
+type promptConfigCol struct {
+	beans.SQLBaseCol
+	ModelTypeId        string
+	ModelType          string
+	PromptInfo         string
+	ResponseJsonSchema string
+	Enabled            string
+	Version            string
+}
+
+var PromptConfigCol = promptConfigCol{
+	SQLBaseCol:         beans.DefSQLBaseCol,
+	ModelTypeId:        "model_type_id",
+	ModelType:          "model_type",
+	PromptInfo:         "prompt_info",
+	ResponseJsonSchema: "response_json_schema",
+	Enabled:            "enabled",
+	Version:            "version",
+}
+
+// PromptConfig 模型提示词配置
+//
+// 说明：
+// - prompt_info 使用 JSONB 保存（对外用 json 传输）
+// - response_json_schema 为模型返回 JSON 格式约束
+// - enabled：1启用/0禁用
+type PromptConfig struct {
+	beans.SQLBaseDO `orm:",inline"`
+	ModelTypeId        int    `orm:"model_type_id" json:"modelTypeId"`
+	ModelType          string `orm:"model_type" json:"modelType"`
+	PromptInfo         any    `orm:"prompt_info" json:"promptInfo"`
+	ResponseJsonSchema any    `orm:"response_json_schema" json:"responseJsonSchema"`
+	Enabled            int    `orm:"enabled" json:"enabled"`
+	Version            string `orm:"version" json:"version"`
+}