From b179fab2a114811997cb5c933734eea788360647 Mon Sep 17 00:00:00 2001 From: WangLiZhao <1838393649@qq.com> Date: Tue, 12 May 2026 13:59:15 +0800 Subject: [PATCH] prompts-core --- .todo | 11 + README.md | 30 ++ config.yml | 72 +++++ consts/public/public.go | 7 + consts/public/table_name.go | 8 + controller/compose_session.go | 20 ++ controller/prompt_controller.go | 85 +++++ dao/compose_session_dao.go | 100 ++++++ dao/compose_task_dao.go | 49 +++ dao/model_dao.go | 57 ++++ dao/prompt_dao.go | 97 ++++++ go.mod | 95 ++++++ go.sum | 467 +++++++++++++++++++++++++++ main.go | 38 +++ model/dto/compose_messages_dto.go | 51 +++ model/dto/compose_session_dto.go | 9 + model/dto/prompt_dto.go | 63 ++++ model/entity/asynch_model.go | 85 +++++ model/entity/compose_session.go | 27 ++ model/entity/compose_task.go | 45 +++ model/entity/prompt_config.go | 39 +++ service/build_service.go | 162 ++++++++++ service/compose_parser.go | 362 +++++++++++++++++++++ service/compose_service.go | 510 ++++++++++++++++++++++++++++++ service/files_service.go | 131 ++++++++ service/form_processor.go | 325 +++++++++++++++++++ service/headers.go | 69 ++++ service/http_service.go | 54 ++++ service/model_output_parser.go | 43 +++ service/prompt_service.go | 92 ++++++ service/session_redis_service.go | 181 +++++++++++ service/session_service.go | 54 ++++ service/utils.go | 337 ++++++++++++++++++++ update.sql | 117 +++++++ 34 files changed, 3892 insertions(+) create mode 100644 .todo create mode 100644 README.md create mode 100644 config.yml create mode 100644 consts/public/public.go create mode 100644 consts/public/table_name.go create mode 100644 controller/compose_session.go create mode 100644 controller/prompt_controller.go create mode 100644 dao/compose_session_dao.go create mode 100644 dao/compose_task_dao.go create mode 100644 dao/model_dao.go create mode 100644 dao/prompt_dao.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go create mode 100644 model/dto/compose_messages_dto.go create mode 100644 model/dto/compose_session_dto.go create mode 100644 model/dto/prompt_dto.go create mode 100644 model/entity/asynch_model.go create mode 100644 model/entity/compose_session.go create mode 100644 model/entity/compose_task.go create mode 100644 model/entity/prompt_config.go create mode 100644 service/build_service.go create mode 100644 service/compose_parser.go create mode 100644 service/compose_service.go create mode 100644 service/files_service.go create mode 100644 service/form_processor.go create mode 100644 service/headers.go create mode 100644 service/http_service.go create mode 100644 service/model_output_parser.go create mode 100644 service/prompt_service.go create mode 100644 service/session_redis_service.go create mode 100644 service/session_service.go create mode 100644 service/utils.go create mode 100644 update.sql diff --git a/.todo b/.todo new file mode 100644 index 0000000..0b9aae8 --- /dev/null +++ b/.todo @@ -0,0 +1,11 @@ +{ + "tasks": [ + { + "id": "1778047497735wxayc", + "title": "Review blocked/reverted work in update.sql", + "category": "Blocked", + "dueDate": null, + "completed": false + } + ] +} \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..4c0d610 --- /dev/null +++ b/README.md @@ -0,0 +1,30 @@ +# prompts-core（提示词服务）[2026.5.12前，暂时弃置] + +## 1. 功能范围（当前阶段） +- 仅做提示词配置的基础 CRUD（最小可用版本） +- 表：`prompts_model_prompt` + +## 2. 接口 +> 路由注册方式与参考项目一致：使用 `common/http.RouteRegister` 注册 controller。 + +- `POST /composeMessages`：按 `modelTypeId` 读取 `prompt_info + response_json_schema`，`modelName` 作为实际调用的网关模型；结合前端 `form(role/value)` 与 `userfiles` 调用 `model-gateway /task/createTask`，同步等待回调后直接返回最终 `messages` +- `GET /composeMessagesCallback/prompts-core`：`model-gateway` 成功回调接口（真实地址由 `callbackUrl + /bizName` 组成） +- `GET /getComposeTask`：按 `taskId` 查询拼接任务状态和结果 +- `POST /createPrompt`：创建（默认启用） +- `PUT /updatePrompt`：更新 +- `DELETE /deletePrompt`：删除 +- `GET /getPrompt`：详情 +- `POST /listPrompt`：列表分页 + +## 3. 数据库初始化 +执行根目录 `update.sql`。 + +## 4. 运行配置 +配置文件：`config.yml` + +### 新增说明 +- `prompts_model_prompt` 去除了 `limit_length` +- 新增 `response_json_schema` +- 新增任务记录表 `prompts_compose_task` +- `callbackUrl` 必须填写 prompts-core 的绝对地址基路径，例如：`http://127.0.0.1:8002/composeMessagesCallback` +- `model-gateway` 实际回调地址为：`callbackUrl/{bizName}`，本项目固定为：`/composeMessagesCallback/prompts-core` diff --git a/config.yml b/config.yml new file mode 100644 index 0000000..4beb4e4 --- /dev/null +++ b/config.yml @@ -0,0 +1,72 @@ +server: + address: ":8002" + name: "prompts-core" + workerId: 1 # 雪花算法 worker ID（用于 common/db/gfdb） + +# PostgreSQL（GoFrame driver pgsql） +database: + default: + - type: "pgsql" + host: "116.204.74.41" + port: "15432" + user: "postgres" + pass: "Bjang09@686^*^" + name: "model-gateway" + prefix: "" # (可选)表名前缀 + role: "master" # (可选)数据库主从角色(master/slave)，默认为master。如果不使用应用主从机制请不配置或留空即可。 + debug: true # (可选)开启调试模式 + dryRun: false # (可选)ORM空跑(只读不写) + charset: "utf8" # (可选)数据库编码(如: utf8mb4/utf8/gbk/gb2312)，一般设置为utf8mb4。默认为utf8。 + timezone: "Asia/Shanghai" # (可选)时区配置，例如:Local + maxIdle: 5 # (可选)连接池最大闲置的连接数(默认10) + maxOpen: 20 # (可选)连接池最大打开的连接数(默认无限制) + maxLifetime: "30s" # (可选)连接对象可重复使用的时间长度(默认30秒) + maxIdleConnTime: "30s" # (可选，v2.10新增)连接池中空闲连接的最大生存时间(默认30秒)。可以通过配置文件或SetConnMaxIdleTime方法设置，避免长时间空闲连接占用资源。 + createdAt: "created_at" # (可选)自动创建时间字段名称 + updatedAt: "updated_at" # (可选)自动更新时间字段名称 + deletedAt: "deleted_at" # (可选)软删除时间字段名称 + timeMaintainDisabled: false # (可选)是否完全关闭时间更新特性，为true时CreatedAt/UpdatedAt/DeletedAt都将失效 + +redis: + default: + address: 192.168.3.30:6379 + db: 0 + +consul: + address: 192.168.3.30:8500 + +jaeger: + addr: 192.168.3.30:4318 + +task: + modelKey: "" # 动态请求头；会透传给 model-gateway，用于覆盖/补充模型配置中的 head_msg，没有可留空 + waitTimeoutSeconds: 300 # /composeMessages 同步等待最终结果的最长时间（秒） + pollIntervalMillis: 500 # 同步等待期间，轮询本地任务表 / 网关状态的时间间隔（毫秒） + +session: + maxRounds: 10 # 最大轮数 + expireTime: 1800 # 过期时间（秒），30分钟 + +modelPrompts: + types: + 1: | + 你是一个智能文字处理助手，专注于文本理解、文本创作、文本优化与语言表达任务，能够根据不同场景完成文章撰写、商业文案、报告总结、邮件通知、脚本创作、内容改写、信息提炼、语言翻译等多种文字处理工作，并能够理解上下文语义关系，保持内容逻辑完整、结构清晰、表达自然。 + 在执行文本任务时，你需要以专业内容创作者、编辑顾问、语言优化专家的身份完成输出，严格保证语言准确性、逻辑连贯性、表达一致性与阅读体验，根据不同用户场景自动适配正式、口语化、专业化、营销化等表达风格，同时避免空洞表达、重复描述与机械化生成内容。 + 当用户提供具体需求时，需要结合用户输入、上下文信息、参数条件与目标场景生成最终文本结果；若涉及改写、扩写、摘要、总结、标题、营销内容等任务，需要保证核心语义不偏离，并根据用户真实目的完成结构化输出。 + 2: | + 你是一个智能图片处理助手，专注于视觉内容生成、图像编辑、画面分析与风格控制任务，能够根据文字描述生成不同风格的图片内容，包括写实、插画、动漫、水彩、电影感、商业海报等多种视觉形式，并支持图片局部修改、风格迁移、画面扩展、背景处理与视觉增强等操作。 + 在执行图片相关任务时，你需要以专业视觉设计师、插画师、摄影指导、美术导演的身份进行画面构建，重点关注主体构图、色彩关系、光影氛围、镜头语言、视觉层次与整体风格统一性，确保生成结果具备明确视觉主题与稳定审美表现，而不是简单关键词堆砌。 + 当用户提供图片需求时，需要结合用户描述、场景用途、风格方向、尺寸比例、主体元素、氛围要求等信息生成完整视觉方案；若存在图片编辑任务，则必须保留原图核心特征，仅对用户指定区域或效果进行修改。 + 3: | + 你是一个智能音频处理助手，专注于语音生成、语音识别、音频分析与声音编辑任务，能够完成文字转语音、语音转文本、多语言识别、音频降噪、音色处理、混音剪辑、情绪识别与声音特征分析等多种音频相关工作，并能够根据不同场景匹配对应语音风格与声音表现形式。 + 在执行音频任务时，你需要以专业配音导演、声音工程师、语音分析专家、后期音频制作人员的身份进行处理，重点保证语音自然度、情绪一致性、识别准确率、音频清晰度与输出稳定性，同时确保不同格式、采样率与播放场景下具备良好兼容性。 + 当用户提供具体音频需求时，需要结合音色、语速、语言类型、情绪风格、背景环境、输出格式等参数完成对应处理；若涉及语音识别或音频分析，则需要尽可能保留原始语义与声音特征，并明确标注不确定内容。 + 4: | + 你是一个智能向量化处理助手，专注于文本向量化、语义检索、知识索引、相似度计算与语义聚类任务，能够将文本内容转换为高维语义向量，并基于向量相似度完成语义搜索、知识召回、内容聚类、文档匹配与知识库构建等处理流程。 + + 在执行向量化任务时，你需要以语义检索工程师、知识库架构师、AI检索系统专家的身份进行处理，重点保证语义表达准确性、向量一致性、检索稳定性与召回有效性，同时确保不同文本之间的语义关系能够被正确表达与计算。 + 当用户提供文本集合、知识内容或检索需求时，需要结合文本上下文、主题方向、检索目标、相似度要求与业务场景生成最终结果；若涉及聚类或知识库构建，则必须明确类别关系、索引结构与召回逻辑。 + 5: | + 你是一个全模态智能处理助手，能够同时理解、分析与生成文本、图片、音频、视频等多种模态内容，并支持跨模态转换、多模态融合推理、联合内容生成与复杂场景交互，能够根据不同输入形式自动匹配最合理的处理策略与输出方式。 + 在执行多模态任务时，你需要以全链路AI内容架构师、多模态交互专家、综合内容生成系统的身份完成处理，重点保证不同模态之间的语义一致性、风格统一性、信息完整性与交互连贯性，避免出现跨模态语义断裂或输出不一致的问题。 + 当用户提供混合输入内容时，需要结合文本、图片、音频、视频等多种信息共同分析用户真实目标，并根据任务场景自动决定最终输出形式；若涉及跨模态生成，则必须保证生成结果能够准确映射原始语义与核心信息。 \ No newline at end of file diff --git a/consts/public/public.go b/consts/public/public.go new file mode 100644 index 0000000..b0936bf --- /dev/null +++ b/consts/public/public.go @@ -0,0 +1,7 @@ +package public + +const ( + ComposeStatusPending = "pending" + ComposeStatusSuccess = "success" + ComposeStatusFailed = "failed" +) diff --git a/consts/public/table_name.go b/consts/public/table_name.go new file mode 100644 index 0000000..933be53 --- /dev/null +++ b/consts/public/table_name.go @@ -0,0 +1,8 @@ +package public + +const ( + TableNameModel = "asynch_models" // 模型表 + TableNamePromptConfig = "prompts_model_prompt" // 模型提示词配置表（prompts-core） + TableNameComposeTask = "prompts_compose_task" // 拼接提示词任务记录表 + TableNameComposeSession = "prompts_compose_session" // 拼接提示词会话记录表 +) diff --git a/controller/compose_session.go b/controller/compose_session.go new file mode 100644 index 0000000..5b3048b --- /dev/null +++ b/controller/compose_session.go @@ -0,0 +1,20 @@ +package controller + +import ( + "context" + + "prompts-core/model/dto" + "prompts-core/service" + + "gitea.com/red-future/common/beans" +) + +type session struct{} + +// Prompt 提示词配置控制器 +var Session = new(session) + +// SessionCallback 会话回调 +func (c *session) SessionCallback(ctx context.Context, req *dto.SessionCallbackReq) (res *beans.ResponseEmpty, err error) { + return service.Session.SessionCallback(ctx, req) +} diff --git a/controller/prompt_controller.go b/controller/prompt_controller.go new file mode 100644 index 0000000..cd34458 --- /dev/null +++ b/controller/prompt_controller.go @@ -0,0 +1,85 @@ +package controller + +import ( + "context" + + "prompts-core/model/dto" + "prompts-core/service" + + "gitea.com/red-future/common/beans" +) + +type prompt struct{} + +// Prompt 提示词配置控制器 +var Prompt = new(prompt) + +// ComposeMessages 调用 model-gateway 异步任务并同步等待结果 +func (c *prompt) ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (res *dto.ComposeMessagesRes, err error) { + return service.Prompt.ComposeMessages(ctx, req) +} + +// ComposeMessagesCallback model-gateway 提示词回调 +func (c *prompt) Callback(ctx context.Context, req *dto.CallbackReq) (res *beans.ResponseEmpty, err error) { + err = service.Prompt.Callback(ctx, req) + return +} + +// GetComposeTask 查询拼接任务结果 +func (c *prompt) GetComposeTask(ctx context.Context, req *dto.GetComposeTaskReq) (res *dto.GetComposeTaskRes, err error) { + return service.Prompt.GetComposeTask(ctx, req.TaskId) +} + +// CreatePrompt 添加配置（默认启用） +func (c *prompt) CreatePrompt(ctx context.Context, req *dto.CreatePromptReq) (res *dto.CreatePromptRes, err error) { + return service.Prompt.Create(ctx, req) +} + +// UpdatePrompt 更新配置 +func (c *prompt) UpdatePrompt(ctx context.Context, req *dto.UpdatePromptReq) (res *beans.ResponseEmpty, err error) { + err = service.Prompt.Update(ctx, req) + return +} + +// DeletePrompt 删除配置 +func (c *prompt) DeletePrompt(ctx context.Context, req *dto.DeletePromptReq) (res *beans.ResponseEmpty, err error) { + err = service.Prompt.Delete(ctx, req.ID) + return +} + +// GetPrompt 获取配置详情 +func (c *prompt) GetPrompt(ctx context.Context, req *dto.GetPromptReq) (res *dto.GetPromptRes, err error) { + m, err := service.Prompt.Get(ctx, req.ID) + if err != nil { + return nil, err + } + return &dto.GetPromptRes{Prompt: m}, nil +} + +// ListPrompt 配置列表 +func (c *prompt) ListPrompt(ctx context.Context, req *dto.ListPromptReq) (res *dto.ListPromptRes, err error) { + pageNum, pageSize := 1, 10 + if req != nil && req.Page != nil { + if req.Page.PageNum > 0 { + pageNum = int(req.Page.PageNum) + } + if req.Page.PageSize > 0 { + pageSize = int(req.Page.PageSize) + } + } + var modelTypeID *int + modelType := "" + if req != nil { + modelTypeID = req.ModelTypeId + modelType = req.ModelType + } + + list, total, err := service.Prompt.List(ctx, pageNum, pageSize, modelTypeID, modelType) + if err != nil { + return nil, err + } + return &dto.ListPromptRes{ + List: list, + Total: total, + }, nil +} diff --git a/dao/compose_session_dao.go b/dao/compose_session_dao.go new file mode 100644 index 0000000..f3b8737 --- /dev/null +++ b/dao/compose_session_dao.go @@ -0,0 +1,100 @@ +package dao + +import ( + "context" + + "prompts-core/consts/public" + "prompts-core/model/entity" + + "gitea.com/red-future/common/db/gfdb" +) + +var ComposeSession = &composeSessionDao{} + +type composeSessionDao struct{} + +func (d *composeSessionDao) Insert(ctx context.Context, m *entity.ComposeSession) (id int64, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).Data(m).Insert() + if err != nil { + return 0, err + } + return r.LastInsertId() +} + +func (d *composeSessionDao) GetById(ctx context.Context, Id int64) (m *entity.ComposeSession, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession). + Where("deleted_at IS NULL"). + Where(entity.ComposeSessionCol.Id, Id). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +func (d *composeSessionDao) GetBySessionId(ctx context.Context, sessionId string) (m *entity.ComposeSession, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession). + Where("deleted_at IS NULL"). + Where(entity.ComposeSessionCol.SessionId, sessionId). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +func (d *composeSessionDao) UpdateById(ctx context.Context, id int64, data map[string]any) (rows int64, err error) { + data[entity.ComposeSessionCol.Updater] = "" + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession). + Where(entity.ComposeSessionCol.Id, id). + Data(data). + Update() + if err != nil { + return 0, err + } + return r.RowsAffected() +} + +func (d *composeSessionDao) List(ctx context.Context, page, size int, where map[string]any) (list []*entity.ComposeSession, total int, err error) { + model := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession). + Where("deleted_at IS NULL") + + // 动态拼接查询条件 + for k, v := range where { + model = model.Where(k, v) + } + + // 查询总数 + total, err = model.Count() + if err != nil { + return nil, 0, err + } + + // 分页查询 + err = model.Order("created_at DESC"). + Page(page, size). + Scan(&list) + + return +} + +func (d *composeSessionDao) DeleteBySessionId(ctx context.Context, sessionId string) (rows int64, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession). + Where(entity.ComposeSessionCol.SessionId, sessionId). + Data(map[string]any{ + entity.ComposeSessionCol.DeletedAt: "NOW()", + }). + Update() + if err != nil { + return 0, err + } + return r.RowsAffected() +} diff --git a/dao/compose_task_dao.go b/dao/compose_task_dao.go new file mode 100644 index 0000000..79c6347 --- /dev/null +++ b/dao/compose_task_dao.go @@ -0,0 +1,49 @@ +package dao + +import ( + "context" + + "prompts-core/consts/public" + "prompts-core/model/entity" + + "gitea.com/red-future/common/db/gfdb" +) + +var ComposeTask = &composeTaskDao{} + +type composeTaskDao struct{} + +func (d *composeTaskDao) Insert(ctx context.Context, m *entity.ComposeTask) (id int64, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).Data(m).Insert() + if err != nil { + return 0, err + } + return r.LastInsertId() +} + +func (d *composeTaskDao) GetByTaskId(ctx context.Context, taskId string) (m *entity.ComposeTask, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask). + Where("deleted_at IS NULL"). + Where(entity.ComposeTaskCol.TaskId, taskId). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +func (d *composeTaskDao) UpdateByTaskId(ctx context.Context, taskId string, data map[string]any) (rows int64, err error) { + data[entity.ComposeTaskCol.Updater] = "" + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask). + Where(entity.ComposeTaskCol.TaskId, taskId). + Data(data). + Update() + if err != nil { + return 0, err + } + return r.RowsAffected() +} diff --git a/dao/model_dao.go b/dao/model_dao.go new file mode 100644 index 0000000..de10541 --- /dev/null +++ b/dao/model_dao.go @@ -0,0 +1,57 @@ +package dao + +import ( + "context" + "fmt" + "prompts-core/consts/public" + "prompts-core/model/entity" + + "gitea.com/red-future/common/db/gfdb" +) + +var Model = &modelDao{} + +type modelDao struct{} + +func (d *modelDao) GetByModelName(ctx context.Context, modelName string) (m *entity.AsynchModel, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel). + Where(entity.AsynchModelCol.ModelName, modelName). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +func (d *modelDao) GetByIsChatModel(ctx context.Context) (m *entity.AsynchModel, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel). + Where(entity.AsynchModelCol.IsChatModel, 1). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +// GetBySuperAdmin 查询超级管理员（tenant_id=1）的模型 +func (d *modelDao) GetBySuperAdmin(ctx context.Context, modelName string) (m *entity.AsynchModel, err error) { + sql := fmt.Sprintf("SELECT * FROM %s WHERE model_name = ? AND tenant_id = 1 AND deleted_at IS NULL LIMIT 1", public.TableNameModel) + r, err := gfdb.DB(ctx).GetAll(ctx, sql, modelName) + if err != nil { + return nil, err + } + if len(r) == 0 { + return nil, nil + } + + err = r[0].Struct(&m) + return +} diff --git a/dao/prompt_dao.go b/dao/prompt_dao.go new file mode 100644 index 0000000..aa5a884 --- /dev/null +++ b/dao/prompt_dao.go @@ -0,0 +1,97 @@ +package dao + +import ( + "context" + + "prompts-core/consts/public" + "prompts-core/model/entity" + + "gitea.com/red-future/common/db/gfdb" + "github.com/gogf/gf/v2/util/gconv" +) + +var Prompt = &promptDao{} + +type promptDao struct{} + +func (d *promptDao) Insert(ctx context.Context, m *entity.PromptConfig) (id int64, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).Data(m).Insert() + if err != nil { + return 0, err + } + return r.LastInsertId() +} + +func (d *promptDao) UpdateByID(ctx context.Context, id int64, data map[string]any) (rows int64, err error) { + // 触发 gfdb 的 updateHook 自动填充 updater，需要显式带 updater 字段 + data[entity.PromptConfigCol.Updater] = "" + r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig). + Where(entity.PromptConfigCol.Id, id). + Data(data). + Update() + if err != nil { + return 0, err + } + return r.RowsAffected() +} + +func (d *promptDao) DeleteByID(ctx context.Context, id int64) (rows int64, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig). + Where(entity.PromptConfigCol.Id, id). + Delete() + if err != nil { + return 0, err + } + return r.RowsAffected() +} + +func (d *promptDao) GetByID(ctx context.Context, id int64) (m *entity.PromptConfig, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig). + Where(entity.PromptConfigCol.Id, id). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +func (d *promptDao) GetLatestEnabledByModelTypeID(ctx context.Context, modelTypeID int) (m *entity.PromptConfig, err error) { + r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig). + Where("deleted_at IS NULL"). + Where(entity.PromptConfigCol.ModelTypeId, modelTypeID). + Where(entity.PromptConfigCol.Enabled, 1). + OrderDesc(entity.PromptConfigCol.CreatedAt). + One() + if err != nil { + return nil, err + } + if r.IsEmpty() { + return nil, nil + } + err = r.Struct(&m) + return +} + +func (d *promptDao) List(ctx context.Context, pageNum, pageSize int, modelTypeID *int, modelTypeLike string) (list []*entity.PromptConfig, total int64, err error) { + model := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).Where("deleted_at IS NULL").OrderDesc(entity.PromptConfigCol.CreatedAt) + if modelTypeID != nil && *modelTypeID > 0 { + model = model.Where(entity.PromptConfigCol.ModelTypeId, *modelTypeID) + } + if modelTypeLike != "" { + model = model.WhereLike(entity.PromptConfigCol.ModelType, "%"+modelTypeLike+"%") + } + if pageNum > 0 && pageSize > 0 { + model = model.Page(pageNum, pageSize) + } + r, totalInt, err := model.AllAndCount(false) + if err != nil { + return nil, 0, err + } + total = gconv.Int64(totalInt) + err = r.Structs(&list) + return +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..5dfab77 --- /dev/null +++ b/go.mod @@ -0,0 +1,95 @@ +module prompts-core + +go 1.26.0 + +require ( + gitea.com/red-future/common v0.0.19 + github.com/gogf/gf/contrib/drivers/pgsql/v2 v2.10.0 + github.com/gogf/gf/contrib/nosql/redis/v2 v2.10.0 + github.com/gogf/gf/v2 v2.10.0 +) + +require ( + github.com/tidwall/match v1.1.1 // indirect + github.com/tidwall/pretty v1.2.0 // indirect +) + +require ( + github.com/BurntSushi/toml v1.5.0 // indirect + github.com/armon/go-metrics v0.4.1 // indirect + github.com/bwmarrin/snowflake v0.3.0 // indirect + github.com/cenkalti/backoff/v5 v5.0.3 // indirect + github.com/cespare/xxhash/v2 v2.3.0 // indirect + github.com/clbanning/mxj/v2 v2.7.0 // indirect + github.com/dgraph-io/badger/v4 v4.2.0 // indirect + github.com/dgraph-io/ristretto v0.1.1 // indirect + github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect + github.com/dustin/go-humanize v1.0.1 // indirect + github.com/emirpasic/gods/v2 v2.0.0-alpha // indirect + github.com/fatih/color v1.18.0 // indirect + github.com/fsnotify/fsnotify v1.9.0 // indirect + github.com/go-ego/gse v1.0.2 // indirect + github.com/go-logr/logr v1.4.3 // indirect + github.com/go-logr/stdr v1.2.2 // indirect + github.com/gogf/gf/contrib/registry/consul/v2 v2.9.5 // indirect + github.com/gogf/gf/contrib/trace/otlphttp/v2 v2.9.5 // indirect + github.com/gogo/protobuf v1.3.2 // indirect + github.com/golang-jwt/jwt/v5 v5.3.1 // indirect + github.com/golang/glog v1.2.5 // indirect + github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect + github.com/golang/protobuf v1.5.4 // indirect + github.com/golang/snappy v1.0.0 // indirect + github.com/google/flatbuffers v1.12.1 // indirect + github.com/google/uuid v1.6.0 // indirect + github.com/gorilla/websocket v1.5.4-0.20250319132907-e064f32e3674 // indirect + github.com/grokify/html-strip-tags-go v0.1.0 // indirect + github.com/grpc-ecosystem/grpc-gateway/v2 v2.27.2 // indirect + github.com/hashicorp/consul/api v1.26.1 // indirect + github.com/hashicorp/errwrap v1.1.0 // indirect + github.com/hashicorp/go-cleanhttp v0.5.2 // indirect + github.com/hashicorp/go-hclog v1.5.0 // indirect + github.com/hashicorp/go-immutable-radix v1.3.1 // indirect + github.com/hashicorp/go-multierror v1.1.1 // indirect + github.com/hashicorp/go-rootcerts v1.0.2 // indirect + github.com/hashicorp/golang-lru v1.0.2 // indirect + github.com/hashicorp/serf v0.10.1 // indirect + github.com/klauspost/compress v1.18.0 // indirect + github.com/lib/pq v1.10.9 // indirect + github.com/magiconair/properties v1.8.10 // indirect + github.com/mattn/go-colorable v0.1.14 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/mattn/go-runewidth v0.0.16 // indirect + github.com/mitchellh/go-homedir v1.1.0 // indirect + github.com/mitchellh/mapstructure v1.5.0 // indirect + github.com/olekukonko/errors v1.1.0 // indirect + github.com/olekukonko/ll v0.0.9 // indirect + github.com/olekukonko/tablewriter v1.1.0 // indirect + github.com/pkg/errors v0.9.1 // indirect + github.com/r3labs/diff/v2 v2.15.1 // indirect + github.com/redis/go-redis/v9 v9.12.1 // indirect + github.com/rivo/uniseg v0.4.7 // indirect + github.com/tidwall/gjson v1.19.0 + github.com/tiger1103/gfast-token v1.0.10 // indirect + github.com/vcaesar/cedar v0.30.0 // indirect + github.com/vmihailenco/msgpack v4.0.4+incompatible // indirect + go.mongodb.org/mongo-driver/v2 v2.4.0 // indirect + go.opencensus.io v0.23.0 // indirect + go.opentelemetry.io/auto/sdk v1.1.0 // indirect + go.opentelemetry.io/otel v1.38.0 // indirect + go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.38.0 // indirect + go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.38.0 // indirect + go.opentelemetry.io/otel/metric v1.38.0 // indirect + go.opentelemetry.io/otel/sdk v1.38.0 // indirect + go.opentelemetry.io/otel/trace v1.38.0 // indirect + go.opentelemetry.io/proto/otlp v1.7.1 // indirect + golang.org/x/exp v0.0.0-20250305212735-054e65f0b394 // indirect + golang.org/x/net v0.47.0 // indirect + golang.org/x/sys v0.38.0 // indirect + golang.org/x/text v0.31.0 // indirect + google.golang.org/appengine v1.6.7 // indirect + google.golang.org/genproto/googleapis/api v0.0.0-20250825161204-c5933d9347a5 // indirect + google.golang.org/genproto/googleapis/rpc v0.0.0-20250825161204-c5933d9347a5 // indirect + google.golang.org/grpc v1.75.0 // indirect + google.golang.org/protobuf v1.36.8 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..97621ce --- /dev/null +++ b/go.sum @@ -0,0 +1,467 @@ +cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= +gitea.com/red-future/common v0.0.19 h1:9/WrfCFUCeFUYwuhBYF+JOQi5F5xuOy+gVnf2ZvHZu4= +gitea.com/red-future/common v0.0.19/go.mod h1:6/nqIucVzmjOyqDTIq71feYBXXFNBy0rFwzaQ0/Ueoo= +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/BurntSushi/toml v1.5.0 h1:W5quZX/G/csjUnuI8SUYlsHs9M38FC7znL0lIO+DvMg= +github.com/BurntSushi/toml v1.5.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho= +github.com/DataDog/datadog-go v3.2.0+incompatible/go.mod h1:LButxg5PwREeZtORoXG3tL4fMGNddJ+vMq1mwgfaqoQ= +github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc= +github.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc= +github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0= +github.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0= +github.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e/go.mod h1:3U/XgcO3hCbHZ8TKRvWD2dDTCfh9M9ya+I9JpbB7O8o= +github.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da/go.mod h1:Q73ZrmVTwzkszR9V5SSuryQ31EELlFMUz1kKyl939pY= +github.com/armon/go-metrics v0.4.1 h1:hR91U9KYmb6bLBYLQjyM+3j+rcd/UhE+G78SFnF8gJA= +github.com/armon/go-metrics v0.4.1/go.mod h1:E6amYzXo6aW1tqzoZGT755KkbgrJsSdpwZ+3JqfkOG4= +github.com/armon/go-radix v0.0.0-20180808171621-7fddfc383310/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8= +github.com/armon/go-radix v1.0.0/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8= +github.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q= +github.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8= +github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw= +github.com/bgentry/speakeasy v0.1.0/go.mod h1:+zsyZBPWlz7T6j88CTgSN5bM796AkVf0kBD4zp0CCIs= +github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs= +github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c= +github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA= +github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0= +github.com/bwmarrin/snowflake v0.3.0 h1:xm67bEhkKh6ij1790JB83OujPR5CzNe8QuQqAgISZN0= +github.com/bwmarrin/snowflake v0.3.0/go.mod h1:NdZxfVWX+oR6y2K0o6qAYv6gIOP9rjG0/E9WsDpxqwE= +github.com/cenkalti/backoff/v5 v5.0.3 h1:ZN+IMa753KfX5hd8vVaMixjnqRZ3y8CuJKRKj1xcsSM= +github.com/cenkalti/backoff/v5 v5.0.3/go.mod h1:rkhZdG3JZukswDf7f0cwqPNk4K0sa+F97BxZthm/crw= +github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= +github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs= +github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/circonus-labs/circonus-gometrics v2.3.1+incompatible/go.mod h1:nmEj6Dob7S7YxXgwXpfOuvO54S+tGdZdw9fuRZt25Ag= +github.com/circonus-labs/circonusllhist v0.1.3/go.mod h1:kMXHVDlOchFAehlya5ePtbp5jckzBHf4XRpQvBOLI+I= +github.com/clbanning/mxj/v2 v2.7.0 h1:WA/La7UGCanFe5NpHF0Q3DNtnCsVoxbPKuyBNHWRyME= +github.com/clbanning/mxj/v2 v2.7.0/go.mod h1:hNiWqW14h+kc+MdF9C6/YoRfjEJoR3ou6tn/Qo+ve2s= +github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= +github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= +github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dgraph-io/badger/v4 v4.2.0 h1:kJrlajbXXL9DFTNuhhu9yCx7JJa4qpYWxtE8BzuWsEs= +github.com/dgraph-io/badger/v4 v4.2.0/go.mod h1:qfCqhPoWDFJRx1gp5QwwyGo8xk1lbHUxvK9nK0OGAak= +github.com/dgraph-io/ristretto v0.1.1 h1:6CWw5tJNgpegArSHpNHJKldNeq03FQCwYvfMVWajOK8= +github.com/dgraph-io/ristretto v0.1.1/go.mod h1:S1GPSBCYCIhmVNfcth17y2zZtQT6wzkzgwUve0VDWWA= +github.com/dgryski/go-farm v0.0.0-20190423205320-6a90982ecee2 h1:tdlZCpZ/P9DhczCTSixgIKmwPv6+wP5DGjqLYw5SUiA= +github.com/dgryski/go-farm v0.0.0-20190423205320-6a90982ecee2/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk= +github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY= +github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto= +github.com/emirpasic/gods/v2 v2.0.0-alpha h1:dwFlh8pBg1VMOXWGipNMRt8v96dKAIvBehtCt6OtunU= +github.com/emirpasic/gods/v2 v2.0.0-alpha/go.mod h1:W0y4M2dtBB9U5z3YlghmpuUhiaZT2h6yoeE+C1sCp6A= +github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98= +github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c= +github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4= +github.com/fatih/color v1.9.0/go.mod h1:eQcE1qtQxscV5RaZvpXrrb8Drkc3/DdQ+uUYCNjL+zU= +github.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk= +github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM= +github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU= +github.com/fsnotify/fsnotify v1.9.0 h1:2Ml+OJNzbYCTzsxtv8vKSFD9PbJjmhYF14k/jKC7S9k= +github.com/fsnotify/fsnotify v1.9.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0= +github.com/go-ego/gse v1.0.2 h1:+27lYFPhQEhA9igtdOsJPRKYL/k3TwYsxBF5jr6KFv4= +github.com/go-ego/gse v1.0.2/go.mod h1:Fy35G+q7VV7Et1zIKO8o/sW1kkugV3znXap/lF/11zc= +github.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as= +github.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as= +github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE= +github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk= +github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= +github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI= +github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY= +github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= +github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= +github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY= +github.com/gogf/gf/contrib/drivers/pgsql/v2 v2.10.0 h1:39+jbTenm7KBj4hO2C8ANAxVHpX/7OuRDs1VcGC9ylA= +github.com/gogf/gf/contrib/drivers/pgsql/v2 v2.10.0/go.mod h1:B0s0fVzn0W220E8UTpSGzrrGKsop5KcB90twBeLCiz0= +github.com/gogf/gf/contrib/nosql/redis/v2 v2.10.0 h1:N/F9CuDdUZLoM1nVRqrDE/33pDZuhVxpNY4wYdeIaBs= +github.com/gogf/gf/contrib/nosql/redis/v2 v2.10.0/go.mod h1:x6uoJGfZOtirIRQls8xUlYzC6f7T/eULPUa9er368X0= +github.com/gogf/gf/contrib/registry/consul/v2 v2.9.5 h1:eUqwJ/qNH8lJ6yssiqskazgp1ACQuNU6zXlLOZVuXTQ= +github.com/gogf/gf/contrib/registry/consul/v2 v2.9.5/go.mod h1:sjQyMry9+0POYZCA6lHXBxO77WoNKkruJpRB4xKqk5k= +github.com/gogf/gf/contrib/trace/otlphttp/v2 v2.9.5 h1:tHUEZYB5GTqEYYVDYnlGobf1xISARKDE4KHVlgjwTec= +github.com/gogf/gf/contrib/trace/otlphttp/v2 v2.9.5/go.mod h1:cfzTn2HS9RDX8f5pUVkbGxUWcSosouqfNQ1G6cY0V88= +github.com/gogf/gf/v2 v2.10.0 h1:rzDROlyqGMe/eM6dCalSR8dZOuMIdLhmxKSH1DGhbFs= +github.com/gogf/gf/v2 v2.10.0/go.mod h1:Svl1N+E8G/QshU2DUbh/3J/AJauqCgUnxHurXWR4Qx0= +github.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ= +github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q= +github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q= +github.com/golang-jwt/jwt/v5 v5.3.1 h1:kYf81DTWFe7t+1VvL7eS+jKFVWaUnK9cB1qbwn63YCY= +github.com/golang-jwt/jwt/v5 v5.3.1/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE= +github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= +github.com/golang/glog v1.2.5 h1:DrW6hGnjIhtvhOIiAKT6Psh/Kd/ldepEa81DKeiRJ5I= +github.com/golang/glog v1.2.5/go.mod h1:6AhwSGph0fcJtXVM/PEHPqZlFeoLxhs7/t5UDAwmO+w= +github.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc= +github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE= +github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc= +github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= +github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= +github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= +github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= +github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= +github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8= +github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= +github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= +github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= +github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs= +github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= +github.com/google/btree v0.0.0-20180813153112-4030bb1f1f0c/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ= +github.com/google/btree v1.0.1 h1:gK4Kx5IaGY9CD5sPJ36FHiBJ6ZXl0kilRiiCj+jdYp4= +github.com/google/btree v1.0.1/go.mod h1:xXMiIv4Fb/0kKde4SpL7qlzvu5cMJDRkFDxJfI9uaxA= +github.com/google/flatbuffers v1.12.1 h1:MVlul7pQNoDzWRLTw5imwYsl+usrS1TXG2H4jg6ImGw= +github.com/google/flatbuffers v1.12.1/go.mod h1:1AeVuKshWv4vARoZatz6mlQ0JxURH0Kv5+zNeJKJCa8= +github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8= +github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/gorilla/websocket v1.5.4-0.20250319132907-e064f32e3674 h1:JeSE6pjso5THxAzdVpqr6/geYxZytqFMBCOtn/ujyeo= +github.com/gorilla/websocket v1.5.4-0.20250319132907-e064f32e3674/go.mod h1:r4w70xmWCQKmi1ONH4KIaBptdivuRPyosB9RmPlGEwA= +github.com/grokify/html-strip-tags-go v0.1.0 h1:03UrQLjAny8xci+R+qjCce/MYnpNXCtgzltlQbOBae4= +github.com/grokify/html-strip-tags-go v0.1.0/go.mod h1:ZdzgfHEzAfz9X6Xe5eBLVblWIxXfYSQ40S/VKrAOGpc= +github.com/grpc-ecosystem/grpc-gateway/v2 v2.27.2 h1:8Tjv8EJ+pM1xP8mK6egEbD1OgnVTyacbefKhmbLhIhU= +github.com/grpc-ecosystem/grpc-gateway/v2 v2.27.2/go.mod h1:pkJQ2tZHJ0aFOVEEot6oZmaVEZcRme73eIFmhiVuRWs= +github.com/hashicorp/consul/api v1.26.1 h1:5oSXOO5fboPZeW5SN+TdGFP/BILDgBm19OrPZ/pICIM= +github.com/hashicorp/consul/api v1.26.1/go.mod h1:B4sQTeaSO16NtynqrAdwOlahJ7IUDZM9cj2420xYL8A= +github.com/hashicorp/consul/sdk v0.15.0 h1:2qK9nDrr4tiJKRoxPGhm6B7xJjLVIQqkjiab2M4aKjU= +github.com/hashicorp/consul/sdk v0.15.0/go.mod h1:r/OmRRPbHOe0yxNahLw7G9x5WG17E1BIECMtCjcPSNo= +github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= +github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I= +github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= +github.com/hashicorp/go-cleanhttp v0.5.0/go.mod h1:JpRdi6/HCYpAwUzNwuwqhbovhLtngrth3wmdIIUrZ80= +github.com/hashicorp/go-cleanhttp v0.5.2 h1:035FKYIWjmULyFRBKPs8TBQoi0x6d9G4xc9neXJWAZQ= +github.com/hashicorp/go-cleanhttp v0.5.2/go.mod h1:kO/YDlP8L1346E6Sodw+PrpBSV4/SoxCXGY6BqNFT48= +github.com/hashicorp/go-hclog v1.5.0 h1:bI2ocEMgcVlz55Oj1xZNBsVi900c7II+fWDyV9o+13c= +github.com/hashicorp/go-hclog v1.5.0/go.mod h1:W4Qnvbt70Wk/zYJryRzDRU/4r0kIg0PVHBcfoyhpF5M= +github.com/hashicorp/go-immutable-radix v1.0.0/go.mod h1:0y9vanUI8NX6FsYoO3zeMjhV/C5i9g4Q3DwcSNZ4P60= +github.com/hashicorp/go-immutable-radix v1.3.1 h1:DKHmCUm2hRBK510BaiZlwvpD40f8bJFeZnpfm2KLowc= +github.com/hashicorp/go-immutable-radix v1.3.1/go.mod h1:0y9vanUI8NX6FsYoO3zeMjhV/C5i9g4Q3DwcSNZ4P60= +github.com/hashicorp/go-msgpack v0.5.3/go.mod h1:ahLV/dePpqEmjfWmKiqvPkv/twdG7iPBM1vqhUKIvfM= +github.com/hashicorp/go-msgpack v0.5.5 h1:i9R9JSrqIz0QVLz3sz+i3YJdT7TTSLcfLLzJi9aZTuI= +github.com/hashicorp/go-msgpack v0.5.5/go.mod h1:ahLV/dePpqEmjfWmKiqvPkv/twdG7iPBM1vqhUKIvfM= +github.com/hashicorp/go-multierror v1.0.0/go.mod h1:dHtQlpGsu+cZNNAkkCN/P3hoUDHhCYQXV3UM06sGGrk= +github.com/hashicorp/go-multierror v1.1.0/go.mod h1:spPvp8C1qA32ftKqdAHm4hHTbPw+vmowP0z+KUhOZdA= +github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo= +github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM= +github.com/hashicorp/go-retryablehttp v0.5.3/go.mod h1:9B5zBasrRhHXnJnui7y6sL7es7NDiJgTc6Er0maI1Xs= +github.com/hashicorp/go-rootcerts v1.0.2 h1:jzhAVGtqPKbwpyCPELlgNWhE1znq+qwJtW5Oi2viEzc= +github.com/hashicorp/go-rootcerts v1.0.2/go.mod h1:pqUvnprVnM5bf7AOirdbb01K4ccR319Vf4pU3K5EGc8= +github.com/hashicorp/go-sockaddr v1.0.0/go.mod h1:7Xibr9yA9JjQq1JpNB2Vw7kxv8xerXegt+ozgdvDeDU= +github.com/hashicorp/go-sockaddr v1.0.2 h1:ztczhD1jLxIRjVejw8gFomI1BQZOe2WoVOu0SyteCQc= +github.com/hashicorp/go-sockaddr v1.0.2/go.mod h1:rB4wwRAUzs07qva3c5SdrY/NEtAUjGlgmH/UkBUC97A= +github.com/hashicorp/go-syslog v1.0.0/go.mod h1:qPfqrKkXGihmCqbJM2mZgkZGvKG1dFdvsLplgctolz4= +github.com/hashicorp/go-uuid v1.0.0/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= +github.com/hashicorp/go-uuid v1.0.1/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= +github.com/hashicorp/go-uuid v1.0.3 h1:2gKiV6YVmrJ1i2CKKa9obLvRieoRGviZFL26PcT/Co8= +github.com/hashicorp/go-uuid v1.0.3/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= +github.com/hashicorp/go-version v1.2.1 h1:zEfKbn2+PDgroKdiOzqiE8rsmLqU2uwi5PB5pBJ3TkI= +github.com/hashicorp/go-version v1.2.1/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA= +github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= +github.com/hashicorp/golang-lru v1.0.2 h1:dV3g9Z/unq5DpblPpw+Oqcv4dU/1omnb4Ok8iPY6p1c= +github.com/hashicorp/golang-lru v1.0.2/go.mod h1:iADmTwqILo4mZ8BN3D2Q6+9jd8WM5uGBxy+E8yxSoD4= +github.com/hashicorp/logutils v1.0.0/go.mod h1:QIAnNjmIWmVIIkWDTG1z5v++HQmx9WQRO+LraFDTW64= +github.com/hashicorp/mdns v1.0.4/go.mod h1:mtBihi+LeNXGtG8L9dX59gAEa12BDtBQSp4v/YAJqrc= +github.com/hashicorp/memberlist v0.5.0 h1:EtYPN8DpAURiapus508I4n9CzHs2W+8NZGbmmR/prTM= +github.com/hashicorp/memberlist v0.5.0/go.mod h1:yvyXLpo0QaGE59Y7hDTsTzDD25JYBZ4mHgHUZ8lrOI0= +github.com/hashicorp/serf v0.10.1 h1:Z1H2J60yRKvfDYAOZLd2MU0ND4AH/WDz7xYHDWQsIPY= +github.com/hashicorp/serf v0.10.1/go.mod h1:yL2t6BqATOLGc5HF7qbFkTfXoPIY0WZdWHfEvMqbG+4= +github.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU= +github.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4= +github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w= +github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8= +github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= +github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo= +github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ= +github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= +github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= +github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= +github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw= +github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o= +github.com/magiconair/properties v1.8.10 h1:s31yESBquKXCV9a/ScB3ESkOjUYYv+X0rg8SYxI99mE= +github.com/magiconair/properties v1.8.10/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0= +github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU= +github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE= +github.com/mattn/go-colorable v0.1.6/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc= +github.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc= +github.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4= +github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE= +github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8= +github.com/mattn/go-isatty v0.0.3/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4= +github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s= +github.com/mattn/go-isatty v0.0.11/go.mod h1:PhnuNfih5lzO57/f3n+odYbM4JtupLOxQOAqxQCu2WE= +github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= +github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc= +github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= +github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0= +github.com/miekg/dns v1.1.26/go.mod h1:bPDLeHnStXmXAq1m/Ch/hvfNHr14JKNPMBo3VZKjuso= +github.com/miekg/dns v1.1.41/go.mod h1:p6aan82bvRIyn+zDIv9xYNUpwa73JcSh9BKwknJysuI= +github.com/miekg/dns v1.1.63 h1:8M5aAw6OMZfFXTT7K5V0Eu5YiiL8l7nUAkyN6C9YwaY= +github.com/miekg/dns v1.1.63/go.mod h1:6NGHfjhpmr5lt3XPLuyfDJi5AXbNIPM9PY6H6sF1Nfs= +github.com/mitchellh/cli v1.1.0/go.mod h1:xcISNoH86gajksDmfB23e/pu+B+GeFRMYmoHXxx3xhI= +github.com/mitchellh/go-homedir v1.1.0 h1:lukF9ziXFxDFPkA1vsr5zpc1XuPDn/wFntq5mG+4E0Y= +github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0= +github.com/mitchellh/mapstructure v0.0.0-20160808181253-ca63d7c062ee/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= +github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY= +github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U= +github.com/olekukonko/errors v1.1.0 h1:RNuGIh15QdDenh+hNvKrJkmxxjV4hcS50Db478Ou5sM= +github.com/olekukonko/errors v1.1.0/go.mod h1:ppzxA5jBKcO1vIpCXQ9ZqgDh8iwODz6OXIGKU8r5m4Y= +github.com/olekukonko/ll v0.0.9 h1:Y+1YqDfVkqMWuEQMclsF9HUR5+a82+dxJuL1HHSRpxI= +github.com/olekukonko/ll v0.0.9/go.mod h1:En+sEW0JNETl26+K8eZ6/W4UQ7CYSrrgg/EdIYT2H8g= +github.com/olekukonko/tablewriter v1.1.0 h1:N0LHrshF4T39KvI96fn6GT8HEjXRXYNDrDjKFDB7RIY= +github.com/olekukonko/tablewriter v1.1.0/go.mod h1:5c+EBPeSqvXnLLgkm9isDdzR3wjfBkHR9Nhfp3NWrzo= +github.com/pascaldekloe/goe v0.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc= +github.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY= +github.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc= +github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= +github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= +github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI= +github.com/posener/complete v1.2.3/go.mod h1:WZIdtGGp+qx0sLrYKtIRAruyNpv6hFCicSgv7Sy7s/s= +github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw= +github.com/prometheus/client_golang v1.0.0/go.mod h1:db9x61etRT2tGnBNRi70OPL5FsnadC4Ky3P0J6CfImo= +github.com/prometheus/client_golang v1.4.0/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU= +github.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo= +github.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +github.com/prometheus/client_model v0.2.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +github.com/prometheus/common v0.4.1/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4= +github.com/prometheus/common v0.9.1/go.mod h1:yhUN8i9wzaXS3w1O07YhxHEBxD+W35wd8bs7vj7HSQ4= +github.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk= +github.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA= +github.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A= +github.com/r3labs/diff/v2 v2.15.1 h1:EOrVqPUzi+njlumoqJwiS/TgGgmZo83619FNDB9xQUg= +github.com/r3labs/diff/v2 v2.15.1/go.mod h1:I8noH9Fc2fjSaMxqF3G2lhDdC0b+JXCfyx85tWFM9kc= +github.com/redis/go-redis/v9 v9.12.1 h1:k5iquqv27aBtnTm2tIkROUDp8JBXhXZIVu1InSgvovg= +github.com/redis/go-redis/v9 v9.12.1/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw= +github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ= +github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88= +github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= +github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= +github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts= +github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529 h1:nn5Wsu0esKSJiIVhscUtVbo7ada43DJhG55ua/hjS5I= +github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc= +github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= +github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.5.0 h1:1zr/of2m5FGMsad5YfcqgdqdWrIhu+EBEJRhR1U7z/c= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= +github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.2/go.mod h1:R6va5+xMeoiuVRoj+gSkQ7d3FALtqAAGI1FQKckRals= +github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U= +github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= +github.com/tidwall/gjson v1.19.0 h1:xwxm7n691Uf3u5OFjzngavjGTh55KX5q/9w9xHW88JU= +github.com/tidwall/gjson v1.19.0/go.mod h1:V37/opeE/JbLUOfH0QTXiNez2l0RUjYUhpT4szFQAfc= +github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA= +github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM= +github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs= +github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU= +github.com/tiger1103/gfast-token v1.0.10 h1:fNiBE/Dq5iTHvTGlCx3DmXa2o4hr0NtumFpffZ39k6s= +github.com/tiger1103/gfast-token v1.0.10/go.mod h1:a/21mxmj7zFeNvjhZSC0XpEAFHfb1aT2k6DXnufFU1s= +github.com/tv42/httpunix v0.0.0-20150427012821-b75d8614f926/go.mod h1:9ESjWnEqriFuLhtthL60Sar/7RFoluCcXsuvEwTV5KM= +github.com/vcaesar/cedar v0.30.0 h1:9fSDpM7FTjjUdPiBUUa0MWYMRGSEcqgFXvppZcZ4d7Y= +github.com/vcaesar/cedar v0.30.0/go.mod h1:lyuGvALuZZDPNXwpzv/9LyxW+8Y6faN7zauFezNsnik= +github.com/vcaesar/tt v0.20.1 h1:D/jUeeVCNbq3ad8M7hhtB3J9x5RZ6I1n1eZ0BJp7M+4= +github.com/vcaesar/tt v0.20.1/go.mod h1:cH2+AwGAJm19Wa6xvEa+0r+sXDJBT0QgNQey6mwqLeU= +github.com/vmihailenco/msgpack v4.0.4+incompatible h1:dSLoQfGFAo3F6OoNhwUmLwVgaUXK79GlxNBwueZn0xI= +github.com/vmihailenco/msgpack v4.0.4+incompatible/go.mod h1:fy3FlTQTDXWkZ7Bh6AcGMlsjHatGryHQYUTf1ShIgkk= +github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +go.mongodb.org/mongo-driver/v2 v2.4.0 h1:Oq6BmUAAFTzMeh6AonuDlgZMuAuEiUxoAD1koK5MuFo= +go.mongodb.org/mongo-driver/v2 v2.4.0/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI= +go.opencensus.io v0.23.0 h1:gqCw0LfLxScz8irSi8exQc7fyQ0fKQU/qnC/X8+V/1M= +go.opencensus.io v0.23.0/go.mod h1:XItmlyltB5F7CS4xOC1DcqMoFqwtC6OG2xF7mCv7P7E= +go.opentelemetry.io/auto/sdk v1.1.0 h1:cH53jehLUN6UFLY71z+NDOiNJqDdPRaXzTel0sJySYA= +go.opentelemetry.io/auto/sdk v1.1.0/go.mod h1:3wSPjt5PWp2RhlCcmmOial7AvC4DQqZb7a7wCow3W8A= +go.opentelemetry.io/otel v1.38.0 h1:RkfdswUDRimDg0m2Az18RKOsnI8UDzppJAtj01/Ymk8= +go.opentelemetry.io/otel v1.38.0/go.mod h1:zcmtmQ1+YmQM9wrNsTGV/q/uyusom3P8RxwExxkZhjM= +go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.38.0 h1:GqRJVj7UmLjCVyVJ3ZFLdPRmhDUp2zFmQe3RHIOsw24= +go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.38.0/go.mod h1:ri3aaHSmCTVYu2AWv44YMauwAQc0aqI9gHKIcSbI1pU= +go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.38.0 h1:aTL7F04bJHUlztTsNGJ2l+6he8c+y/b//eR0jjjemT4= +go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.38.0/go.mod h1:kldtb7jDTeol0l3ewcmd8SDvx3EmIE7lyvqbasU3QC4= +go.opentelemetry.io/otel/metric v1.38.0 h1:Kl6lzIYGAh5M159u9NgiRkmoMKjvbsKtYRwgfrA6WpA= +go.opentelemetry.io/otel/metric v1.38.0/go.mod h1:kB5n/QoRM8YwmUahxvI3bO34eVtQf2i4utNVLr9gEmI= +go.opentelemetry.io/otel/sdk v1.38.0 h1:l48sr5YbNf2hpCUj/FoGhW9yDkl+Ma+LrVl8qaM5b+E= +go.opentelemetry.io/otel/sdk v1.38.0/go.mod h1:ghmNdGlVemJI3+ZB5iDEuk4bWA3GkTpW+DOoZMYBVVg= +go.opentelemetry.io/otel/sdk/metric v1.38.0 h1:aSH66iL0aZqo//xXzQLYozmWrXxyFkBJ6qT5wthqPoM= +go.opentelemetry.io/otel/sdk/metric v1.38.0/go.mod h1:dg9PBnW9XdQ1Hd6ZnRz689CbtrUp0wMMs9iPcgT9EZA= +go.opentelemetry.io/otel/trace v1.38.0 h1:Fxk5bKrDZJUH+AMyyIXGcFAPah0oRcT+LuNtJrmcNLE= +go.opentelemetry.io/otel/trace v1.38.0/go.mod h1:j1P9ivuFsTceSWe1oY+EeW3sc+Pp42sO++GHkg4wwhs= +go.opentelemetry.io/proto/otlp v1.7.1 h1:gTOMpGDb0WTBOP8JaO72iL3auEZhVmAQg4ipjOVAtj4= +go.opentelemetry.io/proto/otlp v1.7.1/go.mod h1:b2rVh6rfI/s2pHWNlB7ILJcRALpcNDzKhACevjI+ZnE= +go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= +go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= +golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20190923035154-9ee001bba392/go.mod h1:/lpIB1dKB+9EgE3H3cr1v9wB50oz8l4C4h62xy7jSTY= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= +golang.org/x/exp v0.0.0-20250305212735-054e65f0b394 h1:nDVHiLt8aIbd/VzvPWN6kSOPE7+F/fNFDSXLVYkE/Iw= +golang.org/x/exp v0.0.0-20250305212735-054e65f0b394/go.mod h1:sIifuuw/Yco/y6yb6+bDNfyeQ/MdPUy/hKEMYQV17cM= +golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= +golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= +golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.29.0 h1:HV8lRxZC4l2cr3Zq1LvtOsi/ThTgWnUk/y64QSs8GwA= +golang.org/x/mod v0.29.0/go.mod h1:NyhrlYXJ2H4eJiRy/WDBO6HMqZQ6q9nk4JzS3NuCK+w= +golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks= +golang.org/x/net v0.0.0-20190613194153-d28f0bde5980/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20190923162816-aa69164e4478/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20210410081132-afb366fc7cd1/go.mod h1:9tjilg8BloeKEkVJvy7fQ90B1CfIiPueXVOjqfkSzI8= +golang.org/x/net v0.47.0 h1:Mx+4dIFzqraBXUugkia1OOvlD6LemFo1ALMHjrXDOhY= +golang.org/x/net v0.47.0/go.mod h1:/jNxtkgq5yWUGYkaZGqo27cfGZ1c5Nen03aYrrKpVRU= +golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.18.0 h1:kr88TuHDroi+UVf+0hZnirlk8o8T+4MrK6mr60WkH/I= +golang.org/x/sync v0.18.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI= +golang.org/x/sys v0.0.0-20180823144017-11551d06cbcc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190922100055-0a153f010e69/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190924154521-2837fb4f24fe/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200122134326-e047566fdf82/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210303074136-134d130e1a04/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220728004956-3c1f35247d10/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20221010170243-090e33056c14/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc= +golang.org/x/sys v0.38.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.31.0 h1:aC8ghyu4JhP8VojJ2lEHBnochRno1sgL6nEi9WGFGMM= +golang.org/x/text v0.31.0/go.mod h1:tKRAlv61yKIjGGHX/4tP1LTbc13YSec1pxVEWXzfoeM= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY= +golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= +golang.org/x/tools v0.0.0-20190907020128-2ca718005c18/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE= +golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA= +golang.org/x/tools v0.38.0 h1:Hx2Xv8hISq8Lm16jvBZ2VQf+RLmbd7wVUsALibYI/IQ= +golang.org/x/tools v0.38.0/go.mod h1:yEsQ/d/YK8cjh0L6rZlY8tgtlKiBNTL14pGDJPJpYQs= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk= +gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E= +google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM= +google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= +google.golang.org/appengine v1.6.6/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc= +google.golang.org/appengine v1.6.7 h1:FZR1q0exgwxzPzp/aF+VccGrSfxfPpkBqjIIEq3ru6c= +google.golang.org/appengine v1.6.7/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc= +google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc= +google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= +google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo= +google.golang.org/genproto/googleapis/api v0.0.0-20250825161204-c5933d9347a5 h1:BIRfGDEjiHRrk0QKZe3Xv2ieMhtgRGeLcZQ0mIVn4EY= +google.golang.org/genproto/googleapis/api v0.0.0-20250825161204-c5933d9347a5/go.mod h1:j3QtIyytwqGr1JUDtYXwtMXWPKsEa5LtzIFN1Wn5WvE= +google.golang.org/genproto/googleapis/rpc v0.0.0-20250825161204-c5933d9347a5 h1:eaY8u2EuxbRv7c3NiGK0/NedzVsCcV6hDuU5qPX5EGE= +google.golang.org/genproto/googleapis/rpc v0.0.0-20250825161204-c5933d9347a5/go.mod h1:M4/wBTSeyLxupu3W3tJtOgB14jILAS/XWPSSa3TAlJc= +google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= +google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg= +google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY= +google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk= +google.golang.org/grpc v1.33.2/go.mod h1:JMHMWHQWaTccqQQlmk3MJZS+GWXOdAesneDmEnv2fbc= +google.golang.org/grpc v1.75.0 h1:+TW+dqTd2Biwe6KKfhE5JpiYIBWq865PhKGSXiivqt4= +google.golang.org/grpc v1.75.0/go.mod h1:JtPAzKiq4v1xcAB2hydNlWI2RnF85XXcV0mhKXr2ecQ= +google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= +google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= +google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= +google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= +google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= +google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c= +google.golang.org/protobuf v1.36.8 h1:xHScyCOEuuwZEc6UtSOvPbAT4zRh0xcNRYekJwfqyMc= +google.golang.org/protobuf v1.36.8/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU= +gopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= +gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= diff --git a/main.go b/main.go new file mode 100644 index 0000000..1c49d22 --- /dev/null +++ b/main.go @@ -0,0 +1,38 @@ +package main + +import ( + "context" + "os" + "os/signal" + "syscall" + + "prompts-core/controller" + + "gitea.com/red-future/common/http" + "gitea.com/red-future/common/jaeger" + _ "gitea.com/red-future/common/swagger" + _ "github.com/gogf/gf/contrib/drivers/pgsql/v2" + _ "github.com/gogf/gf/contrib/nosql/redis/v2" + "github.com/gogf/gf/v2/frame/g" +) + +func main() { + ctx, cancel := context.WithCancel(context.Background()) + defer cancel() + defer jaeger.ShutDown(ctx) + + // 注册路由 + http.RouteRegister([]interface{}{ + controller.Prompt, + controller.Session, + }) + + // 监听退出信号，确保 Ctrl+C 能完整退出并关闭 http server + quit := make(chan os.Signal, 1) + signal.Notify(quit, os.Interrupt, syscall.SIGTERM) + <-quit + + g.Log().Infof(ctx, "[main] 收到退出信号，开始优雅退出...") + cancel() + _ = http.Httpserver.Shutdown() +} diff --git a/model/dto/compose_messages_dto.go b/model/dto/compose_messages_dto.go new file mode 100644 index 0000000..a1cd11a --- /dev/null +++ b/model/dto/compose_messages_dto.go @@ -0,0 +1,51 @@ +package dto + +import "github.com/gogf/gf/v2/frame/g" + +type Message struct { + Role string `json:"role" dc:"角色：system/user/assistant"` + Content any `json:"content" dc:"消息内容"` +} + +type ComposeMessagesReq struct { + g.Meta `path:"/composeMessages" method:"post" tags:"提示词处理" summary:"拼接提示词" dc:"按 modelTypeId 读取 prompts_model_prompt.prompt_info 与 response_json_schema；form 作为系统表单，userForm 作为用户表单，结合 userFiles 调用 model-gateway，并直接返回最终 messages"` + ModelName string `p:"modelName" json:"modelName" v:"required#modelName不能为空" dc:"实际请求的网关模型名称"` + SessionId string `p:"sessionId" json:"sessionId" v:"required#sessionId不能为空" dc:"会话ID"` + IsBuilder bool `p:"isBuild" json:"isBuild" v:"required#isBuild不能为空" dc:"是否构建"` + Cause string `p:"cause" json:"cause" v:"required-if:IsBuilder,false#原因不能为空" dc:"原因"` + Form map[string]any `p:"form" json:"form" dc:"系统表单：form 下所有字段都作为系统提示词来源"` + UserForm map[string]any `p:"userForm" json:"userForm" dc:"用户表单：userForm 下所有字段都作为用户提示词来源；若与 form 含义接近则严格覆盖系统字段"` + SkillName string `p:"skillName" json:"skillName" dc:"技能名称"` + UserFiles []string `p:"userFiles" json:"userFiles" dc:"用户附件地址列表"` +} + +type ComposeMessagesRes struct { + Messages any `json:"messages,omitempty" dc:"最终消息数组"` + EpicycleId int64 `json:"epicycleId" dc:"轮次ID"` +} + +type CallbackReq struct { + g.Meta `path:"/callback" method:"post" tags:"提示词处理" summary:"model-gateway 回调" dc:"model-gateway 成功后 POST 回调：callbackUrl/{bizName}"` + TaskId string `json:"task_id" v:"required#task_id不能为空" dc:"网关任务ID"` + State int `json:"state" dc:"网关任务状态"` + OssFile string `json:"oss_file" dc:"结果文件地址"` + FileType string `json:"file_type" dc:"结果文件类型"` + Text string `json:"text" dc:"文本结果"` + ErrorMsg string `json:"error_msg" dc:"错误信息"` + EpicycleId int64 `json:"epicycleId" dc:"轮次ID"` +} + +type GetComposeTaskReq struct { + g.Meta `path:"/getComposeTask" method:"get" tags:"提示词处理" summary:"查询拼接任务" dc:"按 taskId 查询提示词拼接任务结果"` + TaskId string `p:"taskId" json:"taskId" v:"required#taskId不能为空" dc:"任务ID"` +} + +type GetComposeTaskRes struct { + TaskId string `json:"taskId" dc:"任务ID"` + Status string `json:"status" dc:"业务状态"` + GatewayState int `json:"gatewayState" dc:"网关状态"` + ErrorMessage string `json:"errorMessage" dc:"错误信息"` + Messages any `json:"messages" dc:"最终消息数组"` + OssFile string `json:"ossFile" dc:"结果文件地址"` + FileType string `json:"fileType" dc:"结果文件类型"` +} diff --git a/model/dto/compose_session_dto.go b/model/dto/compose_session_dto.go new file mode 100644 index 0000000..c9bcc61 --- /dev/null +++ b/model/dto/compose_session_dto.go @@ -0,0 +1,9 @@ +package dto + +import "github.com/gogf/gf/v2/frame/g" + +type SessionCallbackReq struct { + g.Meta `path:"/sessionCallback" method:"post" tags:"提示词处理"` + Text string `json:"text" dc:"文本结果"` + EpicycleId int64 `json:"epicycleId" dc:"轮次ID"` +} diff --git a/model/dto/prompt_dto.go b/model/dto/prompt_dto.go new file mode 100644 index 0000000..07eb5d1 --- /dev/null +++ b/model/dto/prompt_dto.go @@ -0,0 +1,63 @@ +package dto + +import ( + "gitea.com/red-future/common/beans" + "github.com/gogf/gf/v2/frame/g" +) + +// CreatePromptReq 添加提示词配置（默认启用） +type CreatePromptReq struct { + g.Meta `path:"/createPrompt" method:"post" tags:"提示词管理" summary:"创建提示词配置" dc:"创建新的模型提示词配置（默认启用）"` + ModelTypeId int `p:"modelTypeId" json:"modelTypeId" v:"required#modelTypeId不能为空" dc:"模型分类ID"` + ModelType string `p:"modelType" json:"modelType" v:"required#modelType不能为空" dc:"模型类别/模型类型"` + PromptInfo any `p:"promptInfo" json:"promptInfo" v:"required#promptInfo不能为空" dc:"数据库定义的表单规则数据（JSON）"` + ResponseJsonSchema any `p:"responseJsonSchema" json:"responseJsonSchema" v:"required#responseJsonSchema不能为空" dc:"模型返回表单 JSON 格式约束"` + // Version 预留字段：先不使用，但表结构保留 + Version string `p:"version" json:"version" dc:"版本号（预留）"` +} + +type CreatePromptRes struct { + ID int64 `json:"id,string" dc:"配置ID"` +} + +// UpdatePromptReq 更新提示词配置 +type UpdatePromptReq struct { + g.Meta `path:"/updatePrompt" method:"put" tags:"提示词管理" summary:"更新提示词配置" dc:"更新指定ID的提示词配置"` + ID int64 `p:"id" json:"id,string" v:"required#id不能为空" dc:"配置ID"` + + ModelTypeId *int `p:"modelTypeId" json:"modelTypeId" dc:"模型分类ID（可选更新）"` + ModelType *string `p:"modelType" json:"modelType" dc:"模型类别/模型类型（可选更新）"` + PromptInfo any `p:"promptInfo" json:"promptInfo" dc:"数据库定义的表单规则数据（JSON）（可选更新）"` + ResponseJsonSchema any `p:"responseJsonSchema" json:"responseJsonSchema" dc:"模型返回表单 JSON 格式约束（可选更新）"` + Enabled *int `p:"enabled" json:"enabled" dc:"是否启用：0-禁用，1-启用（可选更新）"` + Version *string `p:"version" json:"version" dc:"版本号（预留，可选更新）"` +} + +// DeletePromptReq 删除提示词配置 +type DeletePromptReq struct { + g.Meta `path:"/deletePrompt" method:"delete" tags:"提示词管理" summary:"删除提示词配置" dc:"删除指定ID的提示词配置"` + ID int64 `p:"id" json:"id,string" v:"required#id不能为空" dc:"配置ID"` +} + +// GetPromptReq 获取提示词配置详情 +type GetPromptReq struct { + g.Meta `path:"/getPrompt" method:"get" tags:"提示词管理" summary:"获取提示词配置" dc:"根据ID获取提示词配置详情"` + ID int64 `p:"id" json:"id,string" v:"required#id不能为空" dc:"配置ID"` +} + +type GetPromptRes struct { + Prompt any `json:"prompt" dc:"提示词配置详情"` +} + +// ListPromptReq 配置列表 +type ListPromptReq struct { + g.Meta `path:"/listPrompt" method:"post" tags:"提示词管理" summary:"提示词配置列表" dc:"分页获取提示词配置列表"` + Page *beans.Page `p:"page" json:"page" dc:"分页参数"` + ModelTypeId *int `p:"modelTypeId" json:"modelTypeId" dc:"模型分类ID（可选）"` + ModelType string `p:"modelType" json:"modelType" dc:"模型类型名称（可选，模糊查询）"` +} + +type ListPromptRes struct { + List any `json:"list" dc:"列表数据"` + Total int64 `json:"total" dc:"总数"` +} diff --git a/model/entity/asynch_model.go b/model/entity/asynch_model.go new file mode 100644 index 0000000..fdca743 --- /dev/null +++ b/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"` +} diff --git a/model/entity/compose_session.go b/model/entity/compose_session.go new file mode 100644 index 0000000..7f9579f --- /dev/null +++ b/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"` +} diff --git a/model/entity/compose_task.go b/model/entity/compose_task.go new file mode 100644 index 0000000..780bd54 --- /dev/null +++ b/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"` +} diff --git a/model/entity/prompt_config.go b/model/entity/prompt_config.go new file mode 100644 index 0000000..7a16dc8 --- /dev/null +++ b/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"` +} diff --git a/service/build_service.go b/service/build_service.go new file mode 100644 index 0000000..ee560b3 --- /dev/null +++ b/service/build_service.go @@ -0,0 +1,162 @@ +package service + +import ( + "context" + "encoding/json" + "fmt" + "prompts-core/model/dto" + "prompts-core/model/entity" + "strings" + + "github.com/gogf/gf/v2/frame/g" + "github.com/gogf/gf/v2/util/gconv" +) + +// 获取系统模型提示词 +func getConfPrompt(ctx context.Context, modelType int) string { + return g.Cfg().MustGet(ctx, "modelPrompts.types."+gconv.String(modelType), "").String() +} + +func buildInferenceRequest(ctx context.Context, req *dto.ComposeMessagesReq, sessionModel *entity.AsynchModel, model *entity.AsynchModel, historyMessages []Message) (map[string]any, error) { + // 读取 task 相关的配置 + // 构建消息数组 + // 1. 系统提示词（不动） + + fmt.Println("打印sessionModel结果", sessionModel) + fmt.Println("打印model结果", model) + + messages := []map[string]any{} + messages = append(messages, map[string]any{ + "role": "system", + "content": GetSystemPrompt(req, model), + }) + + // 2. 历史对话 - 动态添加（新增部分） + for _, msg := range historyMessages { + messages = append(messages, map[string]any{ + "role": msg.Role, + "content": msg.GetContentString(), + }) + } + // 3. 当前用户问题（原来的最后一条） + messages = append(messages, map[string]any{ + "role": "user", + "content": buildCombinedInput(req, getConfPrompt(ctx, model.ModelsType)), + }) + + // 构建请求体 + return map[string]any{ + "modelName": sessionModel.ModelName, + "bizName": "prompts-core", + "callbackUrl": "/prompt/callback", + "requestPayload": map[string]any{ + "model": sessionModel.ModelName, + "messages": messages, + "stream": false, + }, + }, nil +} + +// ============================================ +// 输入构建 +// ============================================ +func buildCombinedInput(req *dto.ComposeMessagesReq, prompt string) string { + payload := map[string]any{ + //数据库提示信息 + "promptInfo": prompt, + // 系统表单 + "form": req.Form, + // 用户表单 + "userForm": req.UserForm, + //文件url + "userFiles": req.UserFiles, + //解读文件（只支持可读类型 如：xml，json,yaml） + "userFilesText": fetchFileTexts(context.Background(), req.UserFiles), + } + return mustMarshal(payload) +} + +// GetSystemPrompt 定义获取系统提示词的函数 +func GetSystemPrompt(req *dto.ComposeMessagesReq, model *entity.AsynchModel) string { + mappingBytes, _ := json.Marshal(model.RequestMapping) + mappingStr := string(mappingBytes) + + // 解析 mapping + var mapping map[string]string + _ = json.Unmarshal(mappingBytes, mapping) + + // 字段映射说明 + var fieldDesc strings.Builder + for key, path := range mapping { + fieldDesc.WriteString(fmt.Sprintf("- %s → %s\n", key, path)) + } + + // ====================== + // 【核心】UserForm 全部内容完整展开，让模型必须全文阅读 + // 严格按你的业务定义：所有字段作为用户提示词来源 + // ====================== + var userFormContent strings.Builder + for k, v := range req.UserForm { + userFormContent.WriteString(fmt.Sprintf("%s=%v；", k, v)) + } + userFormFullText := strings.TrimSuffix(userFormContent.String(), "；") + + // 拼接双表单 + formInfo := fmt.Sprintf(` +【系统表单（系统提示词/参数）】 +%s + +【用户表单全文（必须完整阅读，全部作为用户提示词）】 +%s +`, formToJSON(req.Form), userFormFullText) + + // 最终提示词（严格遵守你所有规则） + systemPrompt := fmt.Sprintf(` +你是【语义理解 + 结构对齐】的JSON生成专家，必须严格遵守以下所有规则。 + +【强制阅读规则 · 必须100%%遵守】 +1. 必须完整通读全部文本、上下文、规则、表单内容，严禁跳读、略读； +2. 未读完全部信息前，禁止输出任何内容； +3. 必须全覆盖所有约束、所有细节、所有字段后再推理； +4. 禁止断章取义，禁止遗漏任何参数； +5. 必须严格区分系统表单、用户表单。 + +【核心语义规则】 +1. Form = 系统提示词、系统参数、默认配置 +2. UserForm = 用户真实输入全文，所有字段都必须作为用户提示词来源 +3. 若 UserForm 字段与 Form 含义相同 → UserForm 严格覆盖 Form +4. 必须完整使用 UserForm 所有内容，不得遗漏任何一个字段 + +【任务】 +根据双表单内容，智能填充JSON结构： +1. 理解意图：图片/文案 +2. 自动推导数量：各2张=4，一共3张=3 +3. 自动补全默认值：size=1024*1024 +4. 严格按结构输出，不修改字段 + +【输出结构】 +%s +【字段映射关系】 +%s +【完整输入信息】 +%s + +【输出铁律】 +1. 只输出单行JSON，无任何多余字符 +2. 禁止换行、禁止转义、禁止解释 +3. 内容准确、无废话、不编造 +4. 必须完整读取 UserForm 全部内容 + +请输出最终JSON： +`, mappingStr, fieldDesc.String(), formInfo) + + return systemPrompt +} + +func formToJSON(form map[string]any) string { + if form == nil { + return "{}" + } + b, _ := json.Marshal(form) + return string(b) +} diff --git a/service/compose_parser.go b/service/compose_parser.go new file mode 100644 index 0000000..c5fec25 --- /dev/null +++ b/service/compose_parser.go @@ -0,0 +1,362 @@ +package service + +import ( + "encoding/json" + "errors" + "fmt" + "strings" + + "prompts-core/model/dto" +) + +// ============================================ +// 类型定义 +// ============================================ + +// modelOutput 推理模型的标准输出格式 +type modelOutput struct { + Messages []dto.Message `json:"messages"` + System any `json:"system"` + User any `json:"user"` +} + +// gatewayResponse 模型网关的标准响应格式 +type gatewayResponse struct { + Choices []choice `json:"choices"` +} + +type choice struct { + Message message `json:"message"` +} + +type message struct { + Content string `json:"content"` +} + +// ============================================ +// 核心解析函数 +// ============================================ + +// ParseModelResponse 解析推理模型的文本响应，返回消息列表 +// 支持三种格式： +// 1. 标准 messages 格式: {"messages": [...]} +// 2. 简化 system/user 格式: {"system": "...", "user": "..."} +// 3. 网关包装格式: {"choices": [{"message": {"content": "..."}}]} +func ParseModelResponse(text string) ([]dto.Message, error) { + text = strings.TrimSpace(text) + if text == "" { + return nil, errors.New("模型响应为空") + } + + // 1. 尝试解包网关响应 + if content := unwrapGatewayResponse(text); content != "" { + text = content + } + + // 2. 解析为标准格式 + output, err := parseAsModelOutput(text) + if err != nil { + return nil, fmt.Errorf("解析模型输出失败: %w", err) + } + + // 3. 优先使用 messages 字段 + if len(output.Messages) > 0 { + messages := normalizeMessageContents(output.Messages) + if err := validateMessageList(messages); err != nil { + return nil, err + } + return messages, nil + } + + // 4. 兼容 system/user 格式 + return buildMessagesFromSystemUser(output) +} + +// ParseStoredMessages 从数据库存储的数据中解析消息列表 +func ParseStoredMessages(data any) []dto.Message { + if data == nil { + return nil + } + + // 统一序列化 + jsonBytes, err := json.Marshal(data) + if err != nil { + return nil + } + + // 尝试直接解析 + var messages []dto.Message + if err := json.Unmarshal(jsonBytes, &messages); err == nil { + return messages + } + + // 尝试解析为 JSON 字符串再解析 + var jsonStr string + if err := json.Unmarshal(jsonBytes, &jsonStr); err != nil { + return nil + } + + if err := json.Unmarshal([]byte(jsonStr), &messages); err != nil { + return nil + } + + return messages +} + +// ============================================ +// 内部解析函数 +// ============================================ + +// parseAsModelOutput 将文本解析为 modelOutput 结构 +func parseAsModelOutput(text string) (*modelOutput, error) { + // 清理可能的 Markdown 代码块标记 + text = cleanMarkdownCodeBlock(text) + + var output modelOutput + if err := json.Unmarshal([]byte(text), &output); err != nil { + return nil, err + } + + return &output, nil +} + +// unwrapGatewayResponse 解包网关的标准响应格式 +func unwrapGatewayResponse(text string) string { + // 快速检查是否可能是网关响应 + if !strings.Contains(text, `"choices"`) { + return "" + } + + var resp gatewayResponse + if err := json.Unmarshal([]byte(text), &resp); err != nil { + return "" + } + + if len(resp.Choices) == 0 { + return "" + } + + content := strings.TrimSpace(resp.Choices[0].Message.Content) + return content +} + +// buildMessagesFromSystemUser 从 system/user 字段构建消息列表 +func buildMessagesFromSystemUser(output *modelOutput) ([]dto.Message, error) { + messages := make([]dto.Message, 0, 2) + + // 添加 user 消息 + if !isEmptyValue(output.User) { + messages = append(messages, dto.Message{ + Role: "user", + Content: normalizeContent(output.User), + }) + } + + // 添加 system 消息 + if !isEmptyValue(output.System) { + messages = append(messages, dto.Message{ + Role: "system", + Content: normalizeContent(output.System), + }) + } + + if len(messages) == 0 { + return nil, errors.New("未解析到有效的 system 或 user 内容") + } + + if err := validateMessageList(messages); err != nil { + return nil, err + } + + return messages, nil +} + +// ============================================ +// 内容规范化 +// ============================================ + +// normalizeMessageContents 规范化消息列表中的所有内容 +func normalizeMessageContents(messages []dto.Message) []dto.Message { + for i := range messages { + messages[i].Content = normalizeContent(messages[i].Content) + } + return messages +} + +// normalizeContent 规范化单个消息内容 +// - 如果是 JSON 字符串，尝试解析为对象/数组 +// - 否则保持原样 +func normalizeContent(content any) any { + switch v := content.(type) { + case string: + return tryUnmarshalJSON(v) + default: + return content + } +} + +// tryUnmarshalJSON 尝试将 JSON 字符串解析为结构化对象 +func tryUnmarshalJSON(s string) any { + s = strings.TrimSpace(s) + if s == "" { + return s + } + + // 只处理看起来像 JSON 的内容 + if !looksLikeJSON(s) { + return s + } + + var result any + if err := json.Unmarshal([]byte(s), &result); err != nil || result == nil { + return s + } + + return result +} + +// looksLikeJSON 判断字符串是否可能是 JSON +func looksLikeJSON(s string) bool { + s = strings.TrimSpace(s) + return strings.HasPrefix(s, "{") || strings.HasPrefix(s, "[") +} + +// cleanMarkdownCodeBlock 清理 Markdown 代码块标记 +func cleanMarkdownCodeBlock(text string) string { + // 去除可能的 ```json 和 ``` 标记 + text = strings.TrimPrefix(text, "```json") + text = strings.TrimPrefix(text, "```JSON") + text = strings.TrimPrefix(text, "```") + text = strings.TrimSuffix(text, "```") + return strings.TrimSpace(text) +} + +// ============================================ +// 验证 +// ============================================ + +// validateMessageList 验证消息列表的合法性 +func validateMessageList(messages []dto.Message) error { + if len(messages) == 0 { + return errors.New("消息列表不能为空") + } + + hasUser := false + for i, msg := range messages { + if err := validateMessage(msg); err != nil { + return fmt.Errorf("消息[%d]验证失败: %w", i, err) + } + if msg.Role == "user" { + hasUser = true + } + } + + // 至少需要一条 user 消息 + if !hasUser { + return errors.New("消息列表必须包含至少一条 user 角色消息") + } + + return nil +} + +// validateMessage 验证单条消息的合法性 +func validateMessage(msg dto.Message) error { + role := strings.TrimSpace(msg.Role) + if role == "" { + return errors.New("role 不能为空") + } + + if !isValidRole(role) { + return fmt.Errorf("role 值非法: %s (仅允许 system/user/assistant)", role) + } + + // user 角色的 content 不能为空 + if role == "user" && isEmptyValue(msg.Content) { + return errors.New("user 角色的 content 不能为空") + } + + return nil +} + +// isValidRole 判断角色是否合法 +func isValidRole(role string) bool { + switch role { + case "system", "user", "assistant": + return true + default: + return false + } +} + +// HasUserMessage 判断消息列表中是否包含非空的 user 消息 +func HasUserMessage(messages []dto.Message) bool { + for _, msg := range messages { + if msg.Role == "user" && !isEmptyValue(msg.Content) { + return true + } + } + return false +} + +// HasSystemMessage 判断消息列表中是否包含非空的 system 消息 +func HasSystemMessage(messages []dto.Message) bool { + for _, msg := range messages { + if msg.Role == "system" && !isEmptyValue(msg.Content) { + return true + } + } + return false +} + +// ExtractUserContent 提取消息列表中第一个 user 角色的内容 +func ExtractUserContent(messages []dto.Message) any { + for _, msg := range messages { + if msg.Role == "user" { + return msg.Content + } + } + return nil +} + +// ExtractSystemContent 提取消息列表中第一个 system 角色的内容 +func ExtractSystemContent(messages []dto.Message) any { + for _, msg := range messages { + if msg.Role == "system" { + return msg.Content + } + } + return nil +} + +// ============================================ +// 测试辅助函数 (可选) +// ============================================ + +// MockModelResponse 创建模拟的模型响应用于测试 +func MockModelResponse(systemContent, userContent string) string { + output := modelOutput{ + Messages: []dto.Message{ + {Role: "system", Content: systemContent}, + {Role: "user", Content: userContent}, + }, + } + + bytes, _ := json.Marshal(output) + return string(bytes) +} + +// MockGatewayResponse 创建模拟的网关响应用于测试 +func MockGatewayResponse(innerJSON string) string { + resp := gatewayResponse{ + Choices: []choice{ + { + Message: message{ + Content: innerJSON, + }, + }, + }, + } + + bytes, _ := json.Marshal(resp) + return string(bytes) +} diff --git a/service/compose_service.go b/service/compose_service.go new file mode 100644 index 0000000..b9e943d --- /dev/null +++ b/service/compose_service.go @@ -0,0 +1,510 @@ +package service + +import ( + "context" + "encoding/json" + "errors" + "fmt" + "strings" + "time" + + "prompts-core/consts/public" + "prompts-core/dao" + "prompts-core/model/dto" + "prompts-core/model/entity" + + "github.com/gogf/gf/v2/container/gvar" + "github.com/gogf/gf/v2/frame/g" +) + +// ============================================ +// 核心业务流程 +// ============================================ + +// ComposeMessages 拼接提示词主流程 +func (s *promptService) ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (*dto.ComposeMessagesRes, error) { + var ( + epicycleId int64 + err error + historyMessages []Message // 用来存放历史会话 + ) + // 1. 如果不需要构建返回记录id + if req.IsBuilder == false { + epicycleId, err = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{ + SessionId: req.SessionId, + Remark: req.Cause, + }) + return &dto.ComposeMessagesRes{ + EpicycleId: epicycleId, + }, nil + } + + // 2. 获取当前用户模型信息 + sessionModel, err := dao.Model.GetByIsChatModel(ctx) //获取会话模型 + if err != nil { + return nil, err + } + if sessionModel == nil { + return nil, errors.New("当前没有对话模型，请添加") + } + model, err := dao.Model.GetByModelName(ctx, req.ModelName) //获取模型信息 + if err != nil { + return nil, err + } + if model == nil { + return nil, fmt.Errorf("模型 %s 不存在", sessionModel.ModelName) + } + + // 3 获取历史会话 + historyMessages, err = Session.GetSessionHistoryForInference(ctx, req.SessionId) + if err != nil { + g.Log().Errorf(ctx, "获取历史会话失败: %v，将不使用历史会话", err) + historyMessages = nil // 出错就用空的，不影响主流程 + } + + // 4. 调用推理模型 + taskID, err := s.callInferenceModel(ctx, req, sessionModel, model, historyMessages) + if err != nil { + return nil, err + } + + // 5. 保存相关记录 + _, err = dao.ComposeTask.Insert(ctx, &entity.ComposeTask{ + TaskId: taskID, + ModelName: req.ModelName, + SkillName: req.SkillName, + RequestPayload: mustMarshal(req), + Status: public.ComposeStatusPending, + }) + if err != nil { + return nil, err + } + + // 6. 等待结果 + taskRecord, err := s.waitForResult(ctx, taskID) + if err != nil { + return nil, err + } + + // 7. 处理返回结果 + messages := s.processResult(taskRecord) + + //8.1 数据库查询当前会话是否存在 + session, err := dao.ComposeSession.GetBySessionId(ctx, req.SessionId) + if err != nil { + return nil, err + } + if session == nil { + //8.2 不存在则创建新会话记录 + epicycleId, err = dao.ComposeSession.Insert(ctx, &entity.ComposeSession{ + SessionId: req.SessionId, + RequestContent: messages, + }) + if err != nil { + return nil, err + } + } + + // 9. 更新历史会话 + _, err = dao.ComposeSession.UpdateById(ctx, epicycleId, map[string]any{ + entity.ComposeSessionCol.RequestContent: messages, + }) + + return &dto.ComposeMessagesRes{ + Messages: messages, + EpicycleId: epicycleId, + }, nil +} + +func (s *promptService) Callback(ctx context.Context, req *dto.CallbackReq) error { + g.Log().Infof(ctx, "[Callback][RECV] taskId=%s state=%d ossFile=%s fileType=%s textLen=%d", + req.TaskId, req.State, req.OssFile, req.FileType, len(req.Text)) + + // ============ 先查任务是否存在 ============ + task, err := dao.ComposeTask.GetByTaskId(ctx, req.TaskId) + if err != nil { + return err + } + if task == nil { + return fmt.Errorf("任务不存在: %s", req.TaskId) + } + // ============ 根据状态区分处理 ============ + if req.State == 3 { + // 失败：直接更新状态 + _, err = dao.ComposeTask.UpdateByTaskId(ctx, req.TaskId, map[string]any{ + entity.ComposeTaskCol.Status: public.ComposeStatusFailed, + entity.ComposeTaskCol.ErrorMessage: req.ErrorMsg, + }) + return err + } + // ====================================== + // 成功：解析模型输出 + result, err := parseModelOutput(req.Text) + if err != nil { + _, updateErr := dao.ComposeTask.UpdateByTaskId(ctx, req.TaskId, map[string]any{ + entity.ComposeTaskCol.Status: public.ComposeStatusFailed, + entity.ComposeTaskCol.ErrorMessage: err.Error(), + }) + if updateErr != nil { + g.Log().Warningf(ctx, "[Callback] 更新失败状态出错 taskId=%s err=%v", req.TaskId, updateErr) + } + return err + } + + // ============ result 可能为 nil ============ + var messages any + if result != nil { + messages = result + } + // ======================================= + + _, err = dao.ComposeTask.UpdateByTaskId(ctx, req.TaskId, map[string]any{ + entity.ComposeTaskCol.Status: public.ComposeStatusSuccess, + entity.ComposeTaskCol.Messages: messages, + }) + if err != nil { + g.Log().Errorf(ctx, "[Callback] 更新任务失败 taskId=%s err=%v", req.TaskId, err) + } + return err +} + +// GetComposeTask 查询任务结果 +func (s *promptService) GetComposeTask(ctx context.Context, taskID string) (*dto.GetComposeTaskRes, error) { + record, err := dao.ComposeTask.GetByTaskId(ctx, taskID) + if err != nil { + return nil, err + } + if record == nil { + return nil, fmt.Errorf("未找到任务(taskId=%s)", taskID) + } + + // 如果 Messages 是字符串，反序列化为 JSON 数组 + messages := record.Messages + if str, ok := messages.(string); ok && str != "" { + var parsed any + if err := json.Unmarshal([]byte(str), &parsed); err == nil { + messages = parsed + } + } + + return &dto.GetComposeTaskRes{ + TaskId: record.TaskId, + Status: record.Status, + ErrorMessage: record.ErrorMessage, + Messages: messages, + }, nil +} + +// ============================================ +// 步骤4：调用推理模型 +// ============================================ + +func (s *promptService) callInferenceModel(ctx context.Context, req *dto.ComposeMessagesReq, sessionModel *entity.AsynchModel, model *entity.AsynchModel, historyMessages []Message) (string, error) { + // 构建推理模型请求 + taskReq, err := buildInferenceRequest(ctx, req, sessionModel, model, historyMessages) + if err != nil { + return "", fmt.Errorf("构建推理请求失败: %w", err) + } + + // 创建网关任务 + taskID, err := createGatewayTask(ctx, taskReq) + if err != nil { + return "", fmt.Errorf("创建网关任务失败: %w", err) + } + + if taskID == "" { + return "", errors.New("网关未返回taskId") + } + + return taskID, nil +} + +// ============================================ +// 步骤6：等待结果 +// ============================================ + +func (s *promptService) waitForResult(ctx context.Context, taskID string) (*entity.ComposeTask, error) { + timeout := time.Duration(getIntConfig(ctx, "task.waitTimeoutSeconds", 30)) * time.Second + pollInterval := time.Duration(getIntConfig(ctx, "task.pollIntervalMillis", 500)) * time.Millisecond + deadline := time.Now().Add(timeout) + + for { + // 1. 查数据库 + record, err := dao.ComposeTask.GetByTaskId(ctx, taskID) + if err != nil { + return nil, err + } + if record != nil { + switch record.Status { + case public.ComposeStatusSuccess: + return record, nil + case public.ComposeStatusFailed: + return nil, formatTaskError(taskID, record.ErrorMessage) + } + } + + // 2. 查网关状态 + state, err := queryGatewayTaskState(ctx, taskID) + if err != nil { + // ============ 网关不可达不终止，继续轮询 ============ + g.Log().Warningf(ctx, "[waitForResult] 查询网关失败 taskId=%s err=%v", taskID, err) + } else { + switch state { + case 2: // 网关成功 + // ============ 网关已成功，主动更新数据库 ============ + if record != nil { + dao.ComposeTask.UpdateByTaskId(ctx, taskID, map[string]any{ + entity.ComposeTaskCol.Status: public.ComposeStatusSuccess, + }) + } + case 3: // 网关失败 + if record != nil { + dao.ComposeTask.UpdateByTaskId(ctx, taskID, map[string]any{ + entity.ComposeTaskCol.Status: public.ComposeStatusFailed, + entity.ComposeTaskCol.ErrorMessage: "model-gateway 任务执行失败", + }) + } + return nil, fmt.Errorf("model-gateway 任务执行失败(taskId=%s)", taskID) + } + } + + // 3. 超时检查 + if time.Now().After(deadline) { + return nil, fmt.Errorf("等待任务回调超时(taskId=%s)", taskID) + } + time.Sleep(pollInterval) + } +} + +// ============================================ +// 步骤6：处理结果 +// ============================================ +func (s *promptService) processResult(taskRecord *entity.ComposeTask) map[string]any { + if taskRecord == nil { + return nil + } + + // 1. 解析 Messages 获取 content + var contentStr string + switch v := taskRecord.Messages.(type) { + case *gvar.Var: + if v != nil { + var mapped map[string]any + json.Unmarshal([]byte(v.String()), &mapped) + if c, ok := mapped["content"].(string); ok { + contentStr = c + } + } + case string: + var mapped map[string]any + json.Unmarshal([]byte(v), &mapped) + if c, ok := mapped["content"].(string); ok { + contentStr = c + } + case map[string]any: + if c, ok := v["content"].(string); ok { + contentStr = c + } + } + + // 2. 清理并解析 + contentStr = cleanJSONString(contentStr) + var innerData map[string]any + json.Unmarshal([]byte(contentStr), &innerData) + + return innerData +} + +// ============================================ +// 消息处理管道 +// ============================================ + +// parseStoredMessages 从数据库存储的数据中解析消息列表 +// 处理多层 JSON 嵌套的情况 +func parseStoredMessages(data any) []dto.Message { + if data == nil { + return nil + } + + // 统一序列化为 JSON + jsonBytes, err := json.Marshal(data) + if err != nil { + return nil + } + + // 第一层解析：尝试直接解析为消息数组 + var messages []dto.Message + if err := json.Unmarshal(jsonBytes, &messages); err == nil { + // 成功解析，但需要处理 content 可能是 JSON 字符串的情况 + return deepNormalizeMessages(messages) + } + + // 第二层解析：可能是 JSON 字符串包裹的数组 + var rawStr string + if err := json.Unmarshal(jsonBytes, &rawStr); err != nil { + return nil + } + + // 尝试解析字符串为消息数组 + if err := json.Unmarshal([]byte(rawStr), &messages); err == nil { + return deepNormalizeMessages(messages) + } + + return nil +} + +// deepNormalizeMessages 深度规范化消息，处理 content 为 JSON 字符串的情况 +func deepNormalizeMessages(messages []dto.Message) []dto.Message { + for i, msg := range messages { + messages[i].Content = deepNormalizeContent(msg.Content) + } + return messages +} + +// deepNormalizeContent 递归处理 content，支持多层 JSON 嵌套 +func deepNormalizeContent(content any) any { + switch v := content.(type) { + case string: + // 尝试解析 JSON 字符串 + v = strings.TrimSpace(v) + if v == "" { + return v + } + + // 如果看起来像 JSON，尝试解析 + if looksLikeJSON(v) { + var parsed any + if err := json.Unmarshal([]byte(v), &parsed); err == nil { + // 递归处理解析后的内容 + return deepNormalizeContent(parsed) + } + } + return v + + case []any: + // 递归处理数组中的每个元素 + result := make([]any, len(v)) + for i, item := range v { + result[i] = deepNormalizeContent(item) + } + return result + + case map[string]any: + // 递归处理 map 中的每个值 + result := make(map[string]any, len(v)) + for k, val := range v { + result[k] = deepNormalizeContent(val) + } + return result + + default: + return content + } +} + +func NormalizeToTwoPart(messages []dto.Message, req *dto.ComposeMessagesReq) []dto.Message { + var result []dto.Message + + // 1. 提取 system + sysContent := extractByRole(messages, "system") + if sysContent == nil { + sysContent = renderFormText(req.Form, false) + } + result = append(result, dto.Message{Role: "system", Content: sysContent}) + + // 2. 提取 form + formContent := extractByRole(messages, "form") + if formContent != nil { + result = append(result, dto.Message{Role: "form", Content: formContent}) + } else if req != nil { + result = append(result, dto.Message{Role: "form", Content: renderFormJSON(req.Form)}) + } + + // 3. 提取 skill + skillContent := extractByRole(messages, "skill") + if skillContent != nil { + result = append(result, dto.Message{Role: "skill", Content: skillContent}) + } else if req != nil && req.SkillName != "" { + result = append(result, dto.Message{Role: "skill", Content: req.SkillName}) + } + + // 4. 提取 history（如果模型返回了压缩后的历史） + historyContent := extractByRole(messages, "history") + if historyContent != nil { + result = append(result, dto.Message{Role: "history", Content: historyContent}) + } + + // 5. 提取 user + usrContent := extractByRole(messages, "user") + if usrContent == nil { + usrContent = renderUserText(req.UserForm, req.Form) + } + result = append(result, dto.Message{Role: "user", Content: usrContent}) + + return result +} + +// ============================================ +// 辅助函数：按 role 提取第一个非空 content +// ============================================ + +func extractByRole(messages []dto.Message, role string) any { + for _, msg := range messages { + if msg.Role == role && !isEmptyValue(msg.Content) { + return msg.Content + } + } + return nil +} + +// ============================================ +// 辅助函数：将 form 渲染为 JSON 对象 +// ============================================ + +func renderFormJSON(form map[string]any) map[string]any { + if form == nil { + return nil + } + result := make(map[string]any) + for k, v := range form { + result[k] = v + } + return result +} + +func enrichSystemMessages(messages []dto.Message, req *dto.ComposeMessagesReq) []dto.Message { + if len(messages) == 0 { + return messages + } + + // 获取系统字段的值映射 + systemValues := extractSystemValues(req) + + for i, msg := range messages { + if msg.Role != "system" { + continue + } + + // 为 schema 数组补充 value + switch content := msg.Content.(type) { + case []any: + messages[i].Content = enrichSchemaWithValues(content, systemValues) + case []map[string]any: + arr := make([]any, len(content)) + for j, item := range content { + arr[j] = item + } + messages[i].Content = enrichSchemaWithValues(arr, systemValues) + case map[string]any: + // 合并但不覆盖已有值 + for k, v := range systemValues { + if _, exists := content[k]; !exists { + content[k] = v + } + } + messages[i].Content = content + } + } + + return messages +} diff --git a/service/files_service.go b/service/files_service.go new file mode 100644 index 0000000..d654605 --- /dev/null +++ b/service/files_service.go @@ -0,0 +1,131 @@ +package service + +import ( + "context" + "fmt" + "io" + "net/http" + "strings" + "time" + + "github.com/gogf/gf/v2/os/glog" +) + +// ============================================ +// 文件处理 +// ============================================ + +func fetchFileTexts(ctx context.Context, urls []string) map[string]string { + result := make(map[string]string) + + if len(urls) == 0 { + return result + } + + client := &http.Client{ + Timeout: 8 * time.Second, + } + + for _, rawURL := range urls { + url := sanitizeURL(rawURL) + if url == "" { + continue + } + + text, err := fetchFileContent(ctx, client, url) + if err != nil { + glog.Warningf(ctx, + "[FetchFile] failed url=%s err=%v", + url, + err, + ) + continue + } + + if text == "" { + continue + } + + result[url] = text + } + + return result +} + +func fetchFileContent( + ctx context.Context, + client *http.Client, + url string, +) (string, error) { + + req, err := http.NewRequestWithContext( + ctx, + http.MethodGet, + url, + nil, + ) + if err != nil { + return "", err + } + + resp, err := client.Do(req) + if err != nil { + return "", err + } + defer resp.Body.Close() + + // HTTP状态检查 + if resp.StatusCode < 200 || resp.StatusCode >= 300 { + return "", fmt.Errorf("HTTP %d", resp.StatusCode) + } + + // Content-Type检查 + contentType := strings.ToLower(resp.Header.Get("Content-Type")) + + if !isTextContentType(contentType) { + return "", fmt.Errorf("unsupported content-type: %s", contentType) + } + + // 最大读取20KB + body, err := io.ReadAll( + io.LimitReader(resp.Body, 20*1024), + ) + if err != nil { + return "", err + } + + return strings.TrimSpace(string(body)), nil +} + +// 判断是否为文本类型 +func isTextContentType(contentType string) bool { + + // text/* + if strings.HasPrefix(contentType, "text/") { + return true + } + + // 常见文本类型 + allowTypes := []string{ + "application/json", + "application/xml", + "application/javascript", + "application/x-yaml", + "application/yaml", + "application/toml", + } + + for _, t := range allowTypes { + if strings.Contains(contentType, t) { + return true + } + } + + return false +} + +func sanitizeURL(raw string) string { + s := strings.TrimSpace(raw) + s = strings.Trim(s, "`\"") + return s +} diff --git a/service/form_processor.go b/service/form_processor.go new file mode 100644 index 0000000..6204892 --- /dev/null +++ b/service/form_processor.go @@ -0,0 +1,325 @@ +package service + +import ( + "fmt" + "strings" +) + +// ============================================ +// 表单处理 +// ============================================ + +// FormProcessor 表单处理器 +type FormProcessor struct { + SystemForm map[string]any + UserForm map[string]any +} + +// NewFormProcessor 创建表单处理器 +func NewFormProcessor(systemForm, userForm map[string]any) *FormProcessor { + return &FormProcessor{ + SystemForm: systemForm, + UserForm: userForm, + } +} + +// Merge 合并表单，用户表单覆盖系统表单 +func (p *FormProcessor) Merge() map[string]any { + if len(p.SystemForm) == 0 { + return p.SystemForm + } + + result := make(map[string]any) + for k, v := range p.SystemForm { + result[k] = v + } + + if len(p.UserForm) == 0 { + return result + } + + // 构建用户表单索引 + userIndex := buildFieldIndex(p.UserForm) + + // 覆盖匹配的字段 + for key, value := range result { + item, ok := value.(map[string]any) + if !ok || len(item) == 0 { + continue + } + + field := getField(item, key) + + if userItem, exists := findInIndex(userIndex, field, getLabel(item)); exists { + if userValue := getValue(userItem); !isNilOrEmpty(userValue) { + result[key] = cloneWithValue(item, userValue) + } + } + } + + return result +} + +// RemoveDuplicates 移除被用户表单覆盖的字段 +func (p *FormProcessor) RemoveDuplicates() map[string]any { + if len(p.SystemForm) == 0 || len(p.UserForm) == 0 { + return p.SystemForm + } + + userFields := buildFieldSet(p.UserForm) + result := make(map[string]any) + + for key, value := range p.SystemForm { + item, ok := value.(map[string]any) + if !ok || len(item) == 0 { + result[key] = value + continue + } + + field := getField(item, key) + label := getLabel(item) + + // 跳过重复字段 + if userFields.contains(field) || userFields.containsLabel(label) { + continue + } + + result[key] = value + } + + return result +} + +// RemoveSemanticDuplicates 语义去重 +func (p *FormProcessor) RemoveSemanticDuplicates() map[string]any { + if len(p.SystemForm) == 0 || len(p.UserForm) == 0 { + return p.SystemForm + } + + userText := renderUserTextOnly(p.UserForm) + if userText == "" { + return p.SystemForm + } + + result := make(map[string]any) + for key, value := range p.SystemForm { + item, ok := value.(map[string]any) + if !ok || len(item) == 0 { + result[key] = value + continue + } + + if isDuplicate(userText, getField(item, key), getLabel(item), getValue(item)) { + continue + } + + result[key] = value + } + + return result +} + +// RenderSystemText 渲染系统提示词文本 +func (p *FormProcessor) RenderSystemText() string { + return renderFormText(p.SystemForm, false) +} + +// RenderUserText 渲染用户提示词文本 +func (p *FormProcessor) RenderUserText() string { + return renderUserText(p.UserForm, p.SystemForm) +} + +// ============================================ +// 表单处理辅助方法 +// ============================================ + +type fieldSet struct { + fields map[string]bool + labels map[string]bool +} + +func buildFieldSet(form map[string]any) *fieldSet { + fs := &fieldSet{ + fields: make(map[string]bool), + labels: make(map[string]bool), + } + + for key, value := range form { + item, ok := value.(map[string]any) + if !ok || len(item) == 0 { + continue + } + + field := strings.ToLower(getField(item, key)) + if field != "" { + fs.fields[field] = true + } + + if label := strings.ToLower(getLabel(item)); label != "" { + fs.labels[label] = true + } + } + + return fs +} + +func (fs *fieldSet) contains(field string) bool { + return fs.fields[strings.ToLower(field)] +} + +func (fs *fieldSet) containsLabel(label string) bool { + return label != "" && fs.labels[strings.ToLower(label)] +} + +func buildFieldIndex(form map[string]any) map[string]map[string]any { + index := make(map[string]map[string]any) + + for key, value := range form { + item, ok := value.(map[string]any) + if !ok || len(item) == 0 { + continue + } + + field := strings.ToLower(getField(item, key)) + if field != "" { + index[field] = item + } + + if label := strings.ToLower(getLabel(item)); label != "" { + if _, exists := index[label]; !exists { + index[label] = item + } + } + } + + return index +} + +func findInIndex(index map[string]map[string]any, field, label string) (map[string]any, bool) { + key := strings.ToLower(field) + if item, ok := index[key]; ok { + return item, true + } + + if label != "" { + key = strings.ToLower(label) + if item, ok := index[key]; ok { + return item, true + } + } + + return nil, false +} + +// ============================================ +// 表单渲染 +// ============================================ + +func renderFormText(form map[string]any, isUserForm bool) string { + if len(form) == 0 { + return "" + } + + // 用户表单只有一个文本字段时，直接返回值 + if isUserForm && len(form) == 1 { + for _, value := range form { + if item, ok := value.(map[string]any); ok { + return strings.TrimSpace(asString(getValue(item))) + } + } + } + + // 拼接渲染 + items := extractFormItems(form) + if isUserForm { + return renderUserFormItems(items) + } + return renderSystemFormItems(items) +} + +type formItem struct { + Key string + Field string + Label string + Value any +} + +func extractFormItems(form map[string]any) []formItem { + var items []formItem + + keys := sortedKeys(form) + for _, key := range keys { + item, ok := form[key].(map[string]any) + if !ok || len(item) == 0 { + continue + } + + field := getField(item, key) + value := getValue(item) + + // 跳过敏感字段和空值 + if isSensitiveField(field) || isNilOrEmpty(value) { + continue + } + + items = append(items, formItem{ + Key: key, + Field: field, + Label: getLabel(item), + Value: value, + }) + } + + return items +} + +func renderUserFormItems(items []formItem) string { + // 只有一个文本类型字段时，直接返回值 + if len(items) == 1 && isTextType(items[0].Field, items[0].Label) { + return formatValue(items[0].Value) + } + + // 拼接 + var parts []string + for _, item := range items { + if isTextType(item.Field, item.Label) { + parts = append(parts, formatValue(item.Value)) + } else { + label := item.Label + if label == "" { + label = item.Field + } + parts = append(parts, fmt.Sprintf("%s：%s", label, formatValue(item.Value))) + } + } + + return strings.Join(parts, "，") +} + +func renderSystemFormItems(items []formItem) string { + var parts []string + for _, item := range items { + label := item.Label + if label == "" { + label = item.Field + } + parts = append(parts, fmt.Sprintf("%s：%s", label, formatValue(item.Value))) + } + + return strings.Join(parts, "，") +} + +func renderUserText(userForm, systemForm map[string]any) string { + if text := renderFormText(userForm, true); text != "" { + return text + } + // 用户表单为空时，使用系统表单生成 + if text := renderFormText(systemForm, false); text != "" { + return "参考系统字段生成用户提示词：" + text + } + return "" +} + +func renderUserTextOnly(userForm map[string]any) string { + return renderFormText(userForm, true) +} diff --git a/service/headers.go b/service/headers.go new file mode 100644 index 0000000..3920ef0 --- /dev/null +++ b/service/headers.go @@ -0,0 +1,69 @@ +package service + +import ( + "context" + + "gitea.com/red-future/common/utils" + "github.com/gogf/gf/v2/frame/g" + "github.com/gogf/gf/v2/util/gconv" +) + +// asyncCtx 固化异步执行所需的 token/user，避免请求结束后丢失（仅在“同请求内起 goroutine”有用）。 +// 本项目当前是“落库 + 后台 worker”模式，因此还会把必要信息持久化到任务表的 request_payload 中。 +func asyncCtx(ctx context.Context) context.Context { + asyncCtx := context.WithoutCancel(ctx) + if r := g.RequestFromCtx(ctx); r != nil { + if token := r.Header.Get("Authorization"); token != "" { + asyncCtx = context.WithValue(asyncCtx, "token", token) + } + if userInfo := r.Header.Get("X-User-Info"); userInfo != "" { + asyncCtx = context.WithValue(asyncCtx, "xUserInfo", userInfo) + } + } + if user, err := utils.GetUserInfo(ctx); err == nil && user != nil { + asyncCtx = context.WithValue(asyncCtx, "user", user) + } + return asyncCtx +} + +// forwardHeaders 透传调用链路中必须的头信息（优先使用 ctx 里固化的 token / xUserInfo）。 +func forwardHeaders(ctx context.Context) map[string]string { + headers := make(map[string]string) + + if token, ok := ctx.Value("token").(string); ok && token != "" { + headers["Authorization"] = token + } + if x, ok := ctx.Value("xUserInfo").(string); ok && x != "" { + headers["X-User-Info"] = x + } + + // 兜底：从请求头拿 + if r := g.RequestFromCtx(ctx); r != nil { + if headers["Authorization"] == "" { + if token := r.Header.Get("Authorization"); token != "" { + headers["Authorization"] = token + } + } + if headers["X-User-Info"] == "" { + if userInfo := r.Header.Get("X-User-Info"); userInfo != "" { + headers["X-User-Info"] = userInfo + } + } + } + return headers +} + +// GetTenantId 获取租户ID +func GetTenantId(ctx context.Context) int64 { + var currentTenantID int64 + if r := g.RequestFromCtx(ctx); r != nil { + currentTenantID = gconv.Int64(r.Header.Get("X-Tenant-Id")) + } + if currentTenantID == 0 { + userInfo, err := utils.GetUserInfo(ctx) + if err == nil && userInfo != nil { + currentTenantID = int64(userInfo.TenantId) + } + } + return currentTenantID +} diff --git a/service/http_service.go b/service/http_service.go new file mode 100644 index 0000000..765d1ca --- /dev/null +++ b/service/http_service.go @@ -0,0 +1,54 @@ +package service + +import ( + "context" + "encoding/json" + "fmt" + + commonHttp "gitea.com/red-future/common/http" +) + +// ============================================ +// model-gateway 网关交互 +// ============================================ + +// CreateTaskReq 创建任务请求 +type CreateTaskReq struct { + TaskId string `json:"task_id"` + State int `json:"state"` + OssFile string `json:"oss_file"` + FileType string `json:"file_type"` + Text string `json:"text"` + ErrorMsg string `json:"error_msg"` +} + +// createGatewayTask 调用 model-gateway 异步任务并同步等待结果 +func createGatewayTask(ctx context.Context, payload map[string]any) (string, error) { + fullURL := "model-gateway/task/createTask" + headers := forwardHeaders(ctx) + var req CreateTaskReq + body, err := json.Marshal(payload) + if err != nil { + return "", err + } + if err := commonHttp.Post(ctx, fullURL, headers, &req, body); err != nil { + return "", err + } + return req.TaskId, nil +} + +type GetTaskResultRes struct { + OssFile string `json:"ossFile" dc:"结果文件OSS地址"` + State int `json:"state" dc:"任务状态"` +} + +// queryGatewayTaskState 查询网关任务状态 +func queryGatewayTaskState(ctx context.Context, taskID string) (int, error) { + fullURL := fmt.Sprintf("model-gateway/task/getTaskResult?taskId=%s", taskID) + headers := forwardHeaders(ctx) + var req GetTaskResultRes + if err := commonHttp.Get(ctx, fullURL, headers, &req, nil); err != nil { + return 0, err + } + return req.State, nil +} diff --git a/service/model_output_parser.go b/service/model_output_parser.go new file mode 100644 index 0000000..7a8d93c --- /dev/null +++ b/service/model_output_parser.go @@ -0,0 +1,43 @@ +package service + +import ( + "encoding/json" + "fmt" + "strings" +) + +// ============================================ +// 消息解析 +// ============================================ +func parseModelOutput(text string) (map[string]any, error) { + var result map[string]any + if err := json.Unmarshal([]byte(text), &result); err != nil { + return nil, fmt.Errorf("解析模型输出失败: %w", err) + } + return result, nil +} + +// cleanJSONString 清理字符串中可能导致JSON解析失败的字符 +func cleanJSONString(s string) string { + s = strings.ReplaceAll(s, "\u2018", "'") // 左单引号 ‘ + s = strings.ReplaceAll(s, "\u2019", "'") // 右单引号 ’ + s = strings.ReplaceAll(s, "\u201c", "\"") // 左双引号 “ + s = strings.ReplaceAll(s, "\u201d", "\"") // 右双引号 ” + return s +} + +func truncateStr(s string, maxLen int) string { + if len(s) <= maxLen { + return s + } + return s[:maxLen] +} + +// sessionParseModelOutput 解析会话模型输出 +func sessionParseModelOutput(text string) (map[string]any, error) { + var result map[string]any + if err := json.Unmarshal([]byte(text), &result); err != nil { + return nil, fmt.Errorf("解析模型输出失败: %w", err) + } + return result, nil +} diff --git a/service/prompt_service.go b/service/prompt_service.go new file mode 100644 index 0000000..5e35080 --- /dev/null +++ b/service/prompt_service.go @@ -0,0 +1,92 @@ +package service + +import ( + "context" + "encoding/json" + "errors" + + "prompts-core/dao" + "prompts-core/model/dto" + "prompts-core/model/entity" +) + +var Prompt = &promptService{} + +type promptService struct{} + +func (s *promptService) Create(ctx context.Context, req *dto.CreatePromptReq) (res *dto.CreatePromptRes, err error) { + // promptInfo 兜底校验：必须可序列化为 JSON + if req.PromptInfo == nil { + return nil, errors.New("promptInfo不能为空") + } + if _, err := json.Marshal(req.PromptInfo); err != nil { + return nil, errors.New("promptInfo不是合法JSON") + } + if req.ResponseJsonSchema == nil { + return nil, errors.New("responseJsonSchema不能为空") + } + if _, err := json.Marshal(req.ResponseJsonSchema); err != nil { + return nil, errors.New("responseJsonSchema不是合法JSON") + } + + m := &entity.PromptConfig{ + ModelTypeId: req.ModelTypeId, + ModelType: req.ModelType, + PromptInfo: req.PromptInfo, + ResponseJsonSchema: req.ResponseJsonSchema, + Enabled: 1, + Version: req.Version, + } + + id, err := dao.Prompt.Insert(ctx, m) + if err != nil { + return nil, err + } + return &dto.CreatePromptRes{ID: id}, nil +} + +func (s *promptService) Update(ctx context.Context, req *dto.UpdatePromptReq) error { + data := map[string]any{} + if req.ModelTypeId != nil && *req.ModelTypeId > 0 { + data[entity.PromptConfigCol.ModelTypeId] = *req.ModelTypeId + } + if req.ModelType != nil && *req.ModelType != "" { + data[entity.PromptConfigCol.ModelType] = *req.ModelType + } + if req.PromptInfo != nil { + if _, err := json.Marshal(req.PromptInfo); err != nil { + return errors.New("promptInfo不是合法JSON") + } + data[entity.PromptConfigCol.PromptInfo] = req.PromptInfo + } + if req.ResponseJsonSchema != nil { + if _, err := json.Marshal(req.ResponseJsonSchema); err != nil { + return errors.New("responseJsonSchema不是合法JSON") + } + data[entity.PromptConfigCol.ResponseJsonSchema] = req.ResponseJsonSchema + } + if req.Enabled != nil { + data[entity.PromptConfigCol.Enabled] = *req.Enabled + } + if req.Version != nil { + data[entity.PromptConfigCol.Version] = *req.Version + } + if len(data) == 0 { + return errors.New("无可更新字段") + } + _, err := dao.Prompt.UpdateByID(ctx, req.ID, data) + return err +} + +func (s *promptService) Delete(ctx context.Context, id int64) error { + _, err := dao.Prompt.DeleteByID(ctx, id) + return err +} + +func (s *promptService) Get(ctx context.Context, id int64) (*entity.PromptConfig, error) { + return dao.Prompt.GetByID(ctx, id) +} + +func (s *promptService) List(ctx context.Context, pageNum, pageSize int, modelTypeID *int, modelTypeLike string) (list []*entity.PromptConfig, total int64, err error) { + return dao.Prompt.List(ctx, pageNum, pageSize, modelTypeID, modelTypeLike) +} diff --git a/service/session_redis_service.go b/service/session_redis_service.go new file mode 100644 index 0000000..ab7bea1 --- /dev/null +++ b/service/session_redis_service.go @@ -0,0 +1,181 @@ +package service + +import ( + "context" + "encoding/json" + "fmt" + "strings" + "time" + + "github.com/gogf/gf/v2/frame/g" +) + +// Message 消息结构（content 支持 string 或 []string） +type Message struct { + Role string `json:"role"` // user / assistant / system + Content any `json:"content"` // 内容：string 或 []string + Type string `json:"type,omitempty"` // text / file（可选扩展） +} + +// GetContentString 获取 Content 的字符串形式 +func (m Message) GetContentString() string { + switch v := m.Content.(type) { + case string: + return v + case []interface{}: + var parts []string + for _, item := range v { + if s, ok := item.(string); ok { + parts = append(parts, s) + } + } + return strings.Join(parts, "\n") + default: + b, _ := json.Marshal(m.Content) + return string(b) + } +} + +// SessionRoundData Redis存储的单轮会话数据 +type SessionRoundData struct { + SessionId string `json:"sessionId"` // 会话ID + RequestContent []Message `json:"requestContent"` // 用户请求会话 + ResponseContent []Message `json:"responseContent"` // AI回调会话 + Timestamp int64 `json:"timestamp"` // 存入时间戳 +} + +// GetSessionHistory 获取多轮会话历史（供推理时使用） +func (s *sessionService) GetSessionHistory(ctx context.Context, sessionId string) ([]SessionRoundData, error) { + return s.getFromRedis(ctx, sessionId) +} + +// BuildMessages 根据Redis历史构建完整的Messages数组 +func (s *sessionService) BuildMessages(ctx context.Context, sessionId string, currentMessages []Message) ([]Message, error) { + // 获取历史会话 + history, err := s.getFromRedis(ctx, sessionId) + if err != nil { + return nil, fmt.Errorf("获取历史会话失败: %w", err) + } + + var allMessages []Message + + // 按时间顺序拼接历史消息 + for _, round := range history { + allMessages = append(allMessages, round.RequestContent...) + allMessages = append(allMessages, round.ResponseContent...) + } + + // 添加当前轮次的请求消息 + allMessages = append(allMessages, currentMessages...) + + return allMessages, nil +} + +// ==================== Redis 操作 ==================== + +// saveToRedis 保存会话数据到Redis +// sessionId: 会话ID作为key +// 最大10轮，超出替换最早的，过期时间30分钟 +func (s *sessionService) saveToRedis(ctx context.Context, sessionId string, requestMessages []Message, responseMessages []Message) error { + key := fmt.Sprintf("chat:session:%s", sessionId) + + // 从配置读取，提供默认值 + maxRounds := g.Cfg().MustGet(ctx, "session.maxRounds", 10).Int() + expireSeconds := g.Cfg().MustGet(ctx, "session.expireTime", 1800).Int64() + expireTime := time.Duration(expireSeconds) * time.Second + + // 构造存储数据 + data := SessionRoundData{ + SessionId: sessionId, + RequestContent: requestMessages, + ResponseContent: responseMessages, + Timestamp: time.Now().Unix(), + } + + // 序列化 + b, err := json.Marshal(data) + if err != nil { + return fmt.Errorf("序列化会话数据失败: %w", err) + } + + // 写入 Redis（LPUSH 添加到最前面，新的在前） + _, err = g.Redis().Do(ctx, "LPUSH", key, string(b)) + if err != nil { + return fmt.Errorf("写入Redis失败: %w", err) + } + + // 裁剪到最新10轮（保留前10条） + _, err = g.Redis().Do(ctx, "LTRIM", key, 0, maxRounds-1) + if err != nil { + return fmt.Errorf("裁剪Redis列表失败: %w", err) + } + + // 重置过期时间 + _, err = g.Redis().Do(ctx, "EXPIRE", key, int64(expireTime.Seconds())) + if err != nil { + return fmt.Errorf("设置过期时间失败: %w", err) + } + + return nil +} + +// getFromRedis 从Redis获取会话历史 +func (s *sessionService) getFromRedis(ctx context.Context, sessionId string) ([]SessionRoundData, error) { + key := fmt.Sprintf("chat:session:%s", sessionId) + + // 获取列表中所有数据（最多10条） + result, err := g.Redis().Do(ctx, "LRANGE", key, 0, -1) + if err != nil { + return nil, fmt.Errorf("从Redis获取数据失败: %w", err) + } + + if result == nil || result.IsNil() { + return []SessionRoundData{}, nil + } + + // 解析数据 + var sessions []SessionRoundData + + // 将结果转换为字符串数组 + values := result.Strings() + for _, str := range values { + var data SessionRoundData + if err := json.Unmarshal([]byte(str), &data); err != nil { + g.Log().Warningf(ctx, "[会话] 解析Redis数据失败 err=%v", err) + continue + } + sessions = append(sessions, data) + } + + // 反转顺序（Redis存储最新在前，使用时按时间正序） + for i, j := 0, len(sessions)-1; i < j; i, j = i+1, j-1 { + sessions[i], sessions[j] = sessions[j], sessions[i] + } + + return sessions, nil +} + +// GetSessionHistoryForInference 获取历史会话，直接返回Message数组（给推理用） +func (s *sessionService) GetSessionHistoryForInference(ctx context.Context, sessionId string) ([]Message, error) { + // 从Redis获取历史会话数据 + historyData, err := s.getFromRedis(ctx, sessionId) + if err != nil { + return nil, fmt.Errorf("获取历史会话失败: %w", err) + } + + // 如果没有任何历史数据，返回空 + if len(historyData) == 0 { + return []Message{}, nil + } + + // 把SessionRoundData转换成扁平的Message数组 + var messages []Message + for _, round := range historyData { + // 先加用户的请求 + messages = append(messages, round.RequestContent...) + // 再加AI的回答 + messages = append(messages, round.ResponseContent...) + } + + return messages, nil +} diff --git a/service/session_service.go b/service/session_service.go new file mode 100644 index 0000000..40534cf --- /dev/null +++ b/service/session_service.go @@ -0,0 +1,54 @@ +package service + +import ( + "context" + "prompts-core/dao" + "prompts-core/model/dto" + "prompts-core/model/entity" + + "gitea.com/red-future/common/beans" + "github.com/gogf/gf/v2/frame/g" +) + +var Session = &sessionService{} + +type sessionService struct{} + +// SessionCallback 会话回调处理 +func (s *sessionService) SessionCallback(ctx context.Context, req *dto.SessionCallbackReq) (res *beans.ResponseEmpty, err error) { + // 1. 解析AI返回的文本 + result, err := sessionParseModelOutput(req.Text) + if err != nil { + g.Log().Errorf(ctx, "[会话回调] 解析模型输出失败 epicycleId=%d err=%v", req.EpicycleId, err) + return nil, err + } + + // 2. 更新数据库（AI回调内容） + _, err = dao.ComposeSession.UpdateById(ctx, req.EpicycleId, map[string]any{ + entity.ComposeSessionCol.ResponseContent: result, + }) + if err != nil { + g.Log().Errorf(ctx, "[会话回调] 更新数据库失败 epicycleId=%d err=%v", req.EpicycleId, err) + return nil, err + } + + // 3. 获取当前轮次的完整数据 + session, err := dao.ComposeSession.GetById(ctx, req.EpicycleId) + if err != nil { + g.Log().Errorf(ctx, "[会话回调] 获取会话数据失败 epicycleId=%d err=%v", req.EpicycleId, err) + return nil, err + } + + // 4. 写入 Redis（多轮记忆） + requestMessages := s.convertToMessages(session.RequestContent) + responseMessages := s.convertToMessages(session.ResponseContent) + if err = s.saveToRedis(ctx, session.SessionId, requestMessages, responseMessages); err != nil { + g.Log().Errorf(ctx, "[会话回调] Redis存储失败 sessionId=%s id=%d err=%v", + session.SessionId, session.Id, err) + return nil, err + } + + g.Log().Infof(ctx, "[会话回调] 存储成功 sessionId=%s id=%d requestLen=%d responseLen=%d", + session.SessionId, session.Id, len(requestMessages), len(responseMessages)) + return &beans.ResponseEmpty{}, nil +} diff --git a/service/utils.go b/service/utils.go new file mode 100644 index 0000000..2ff7e87 --- /dev/null +++ b/service/utils.go @@ -0,0 +1,337 @@ +package service + +import ( + "context" + "encoding/json" + "fmt" + "prompts-core/model/dto" + "sort" + "strings" + + "github.com/gogf/gf/v2/container/gvar" + "github.com/gogf/gf/v2/frame/g" +) + +// ============================================ +// 工具函数 +// ============================================ + +func getField(item map[string]any, fallback string) string { + if field := asString(item["field"]); field != "" { + return field + } + return fallback +} + +func getLabel(item map[string]any) string { + return asString(item["label"]) +} + +func getValue(item map[string]any) any { + return item["value"] +} + +func cloneWithValue(item map[string]any, value any) map[string]any { + cloned := make(map[string]any) + for k, v := range item { + cloned[k] = v + } + cloned["value"] = value + return cloned +} + +func isSensitiveField(field string) bool { + f := strings.ToLower(field) + return f == "apikey" || f == "authorization" +} + +func isAPIKeyField(field string) bool { + f := strings.ToLower(field) + return f == "apikey" || f == "authorization" +} + +func isTextType(field, label string) bool { + f := strings.ToLower(field) + l := strings.ToLower(label) + return f == "prompt" || f == "text" || + l == "提示词" || l == "文本内容" || l == "prompt" || l == "text" +} + +func isDuplicate(userText, field, label string, value any) bool { + lowerText := strings.ToLower(userText) + + if label != "" && strings.Contains(lowerText, strings.ToLower(label)) { + return true + } + if field != "" && strings.Contains(lowerText, strings.ToLower(field)) { + return true + } + + // 检查值 + if v := asString(value); v != "" && strings.Contains(lowerText, strings.ToLower(v)) { + return true + } + + return false +} + +func isEmptyValue(v any) bool { + if v == nil { + return true + } + if s, ok := v.(string); ok { + return strings.TrimSpace(s) == "" + } + return false +} + +func isNilOrEmpty(v any) bool { + if v == nil { + return true + } + if s, ok := v.(string); ok { + return strings.TrimSpace(s) == "" + } + return false +} + +func asString(v any) string { + switch t := v.(type) { + case string: + return t + default: + b, _ := json.Marshal(t) + return strings.Trim(string(b), "\"") + } +} + +func formatValue(v any) string { + return strings.TrimSpace(asString(v)) +} + +func mapToText(m map[string]any) string { + if len(m) == 0 { + return "" + } + + keys := sortedKeys(m) + var parts []string + for _, k := range keys { + if isNilOrEmpty(m[k]) { + continue + } + parts = append(parts, fmt.Sprintf("%s：%s", k, formatValue(m[k]))) + } + + return strings.Join(parts, "，") +} + +func sortedKeys(m map[string]any) []string { + keys := make([]string, 0, len(m)) + for k := range m { + keys = append(keys, k) + } + sort.Strings(keys) + return keys +} + +func mustMarshal(v any) string { + b, err := json.Marshal(v) + if err != nil { + return "{}" + } + return string(b) +} + +func formatTaskError(taskID, errMsg string) error { + if strings.TrimSpace(errMsg) == "" { + return fmt.Errorf("任务失败(taskId=%s)", taskID) + } + return fmt.Errorf("任务失败(taskId=%s): %s", taskID, errMsg) +} + +func getIntConfig(ctx context.Context, key string, fallback int) int { + v := g.Cfg().MustGet(ctx, key) + if v.IsEmpty() { + return fallback + } + return v.Int() +} + +// ============================================ +// Schema 处理 +// ============================================ + +func enrichSchemaWithValues(schema []any, values map[string]any) []any { + if len(schema) == 0 || len(values) == 0 { + return schema + } + + result := make([]any, len(schema)) + copy(result, schema) + + for i, item := range result { + m, ok := item.(map[string]any) + if !ok { + continue + } + + field := getField(m, "") + if field == "" { + continue + } + + // 已有 value 则跳过 + if _, hasValue := m["value"]; hasValue { + continue + } + + // 补充 value + if v, exists := values[field]; exists { + m["value"] = v + result[i] = m + } + } + + return result +} + +// extractContentFromResponse 从模型完整响应中提取 content 字段 +func extractContentFromResponse(text string) string { + // 尝试解析为完整的 choices 响应 + var response struct { + Choices []struct { + Message struct { + Content string `json:"content"` + } `json:"message"` + } `json:"choices"` + } + + if err := json.Unmarshal([]byte(text), &response); err != nil { + return "" + } + + if len(response.Choices) > 0 && response.Choices[0].Message.Content != "" { + return response.Choices[0].Message.Content + } + + return "" +} + +// ============================================ +// 值提取 +// ============================================ + +func extractSystemValues(req *dto.ComposeMessagesReq) map[string]any { + if req == nil { + return nil + } + + values := make(map[string]any) + + for _, value := range req.Form { + item, ok := value.(map[string]any) + if !ok || len(item) == 0 { + continue + } + + field := getField(item, "") + if field == "" || isSensitiveField(field) { + continue + } + + if v := getValue(item); !isNilOrEmpty(v) { + values[field] = v + } + } + + return values +} + +func extractModelKey(form map[string]any) string { + for _, value := range form { + item, ok := value.(map[string]any) + if !ok { + continue + } + + field := getField(item, "") + if isAPIKeyField(field) { + key := strings.TrimSpace(asString(getValue(item))) + if key != "" { + if strings.Contains(key, ":") { + return key + } + return "Authorization:" + key + } + } + } + + return "" +} + +// ==================== 工具方法 ==================== + +// convertToMessages 将数据库 any 类型转换为 []Message +// 支持：JSON字符串、[]byte、[]interface{}、以及 content 为字符串数组的格式 +func (s *sessionService) convertToMessages(data any) []Message { + if data == nil { + return []Message{} + } + + // 处理 *gvar.Var + if v, ok := data.(*gvar.Var); ok { + if v == nil || v.IsNil() || v.IsEmpty() { + return []Message{} + } + data = v.Val() + } + + var rawList []any + + switch v := data.(type) { + case string: + if err := json.Unmarshal([]byte(v), &rawList); err != nil { + g.Log().Warningf(context.Background(), "[会话] 解析JSON字符串失败 err=%v data=%.200s", err, v) + return []Message{} + } + case []byte: + if err := json.Unmarshal(v, &rawList); err != nil { + g.Log().Warningf(context.Background(), "[会话] 解析字节数组失败 err=%v", err) + return []Message{} + } + case []interface{}: + rawList = v + default: + b, _ := json.Marshal(v) + if err := json.Unmarshal(b, &rawList); err != nil { + g.Log().Warningf(context.Background(), "[会话] 解析未知类型失败 err=%v type=%T", err, v) + return []Message{} + } + } + + // 转换每个元素为 Message + var messages []Message + for _, item := range rawList { + var msg Message + switch val := item.(type) { + case string: + if err := json.Unmarshal([]byte(val), &msg); err != nil { + g.Log().Warningf(context.Background(), "[会话] 解析消息元素失败 err=%v data=%s", err, val) + continue + } + case map[string]interface{}: + b, _ := json.Marshal(val) + json.Unmarshal(b, &msg) + default: + b, _ := json.Marshal(val) + json.Unmarshal(b, &msg) + } + messages = append(messages, msg) + } + + if messages == nil { + messages = []Message{} + } + return messages +} diff --git a/update.sql b/update.sql new file mode 100644 index 0000000..3e1aff1 --- /dev/null +++ b/update.sql @@ -0,0 +1,117 @@ +-- prompts-core 核心表（pgsql） +-- 说明：字段风格尽量与参考项目一致（tenant/creator/updater/created_at/updated_at/deleted_at） + +-- prompts_model_prompt 模型提示词配置表 +CREATE TABLE IF NOT EXISTS prompts_model_prompt ( + -- 基础字段（与 common/db/gfdb 的 Hook 约定保持一致） + id BIGINT PRIMARY KEY, -- 主键ID（非自增） + tenant_id BIGINT NOT NULL DEFAULT 0, -- 租户ID + creator VARCHAR(64) NOT NULL, -- 创建人 + created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, -- 创建时间 + updater VARCHAR(64) NOT NULL, -- 更新人 + updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, -- 更新时间 + deleted_at TIMESTAMP(6), -- 删除时间（软删） + + -- 业务字段（按你当前的最小字段集） + model_type_id INT NOT NULL DEFAULT 0, -- 模型分类ID + model_type VARCHAR(64) NOT NULL, -- 模型类别 + prompt_info JSONB NOT NULL DEFAULT '{}'::jsonb, -- 提示词信息（JSON） + response_json_schema JSONB NOT NULL DEFAULT '{}'::jsonb, -- 模型返回表单 JSON 格式约束 + enabled SMALLINT NOT NULL DEFAULT 1, -- 是否启用：1启用/0禁用 + version VARCHAR(64) NOT NULL DEFAULT '' -- 版本号（预留） +); + +CREATE INDEX IF NOT EXISTS idx_prompts_model_prompt_tenant_id ON prompts_model_prompt(tenant_id); +CREATE INDEX IF NOT EXISTS idx_prompts_model_prompt_model_type_id ON prompts_model_prompt(model_type_id); +CREATE INDEX IF NOT EXISTS idx_prompts_model_prompt_model_type ON prompts_model_prompt(model_type); +CREATE INDEX IF NOT EXISTS idx_prompts_model_prompt_enabled ON prompts_model_prompt(enabled); +CREATE INDEX IF NOT EXISTS idx_prompts_model_prompt_deleted_at ON prompts_model_prompt(deleted_at); + +COMMENT ON TABLE prompts_model_prompt IS '模型提示词配置表'; +COMMENT ON COLUMN prompts_model_prompt.id IS '主键ID（非自增）'; +COMMENT ON COLUMN prompts_model_prompt.tenant_id IS '租户ID'; +COMMENT ON COLUMN prompts_model_prompt.creator IS '创建人'; +COMMENT ON COLUMN prompts_model_prompt.created_at IS '创建时间'; +COMMENT ON COLUMN prompts_model_prompt.updater IS '更新人'; +COMMENT ON COLUMN prompts_model_prompt.updated_at IS '更新时间'; +COMMENT ON COLUMN prompts_model_prompt.deleted_at IS '删除时间（软删）'; +COMMENT ON COLUMN prompts_model_prompt.model_type_id IS '模型分类ID'; +COMMENT ON COLUMN prompts_model_prompt.model_type IS '模型类别'; +COMMENT ON COLUMN prompts_model_prompt.prompt_info IS '提示词信息（JSON）'; +COMMENT ON COLUMN prompts_model_prompt.response_json_schema IS '模型返回表单 JSON 格式约束'; +COMMENT ON COLUMN prompts_model_prompt.enabled IS '是否启用：1启用/0禁用'; +COMMENT ON COLUMN prompts_model_prompt.version IS '版本号（预留）'; + +-- prompts_compose_task 拼接提示词任务记录表 +CREATE TABLE IF NOT EXISTS prompts_compose_task ( + id BIGINT PRIMARY KEY, + tenant_id BIGINT NOT NULL DEFAULT 0, + creator VARCHAR(64) NOT NULL, + created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, + updater VARCHAR(64) NOT NULL, + updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, + deleted_at TIMESTAMP(6), + + task_id VARCHAR(64) NOT NULL, + model_name VARCHAR(128) NOT NULL DEFAULT '', + skill_name VARCHAR(128) NOT NULL DEFAULT '', + gateway_state INT NOT NULL DEFAULT 0, + limit_words INT NOT NULL DEFAULT 0, + request_payload JSONB NOT NULL DEFAULT '{}'::jsonb, + result_text TEXT NOT NULL DEFAULT '', + messages JSONB NOT NULL DEFAULT '[]'::jsonb, + status VARCHAR(32) NOT NULL DEFAULT 'pending', + error_message TEXT NOT NULL DEFAULT '', + oss_file VARCHAR(1024) NOT NULL DEFAULT '', + file_type VARCHAR(64) NOT NULL DEFAULT '' +); + +CREATE UNIQUE INDEX IF NOT EXISTS uk_prompts_compose_task_task_id ON prompts_compose_task(task_id); +CREATE INDEX IF NOT EXISTS idx_prompts_compose_task_status ON prompts_compose_task(status); +CREATE INDEX IF NOT EXISTS idx_prompts_compose_task_deleted_at ON prompts_compose_task(deleted_at); + +COMMENT ON TABLE prompts_compose_task IS '拼接提示词任务记录表'; +COMMENT ON COLUMN prompts_compose_task.task_id IS 'model-gateway 任务ID'; +COMMENT ON COLUMN prompts_compose_task.model_name IS '业务模型名称'; +COMMENT ON COLUMN prompts_compose_task.skill_name IS '技能名称'; +COMMENT ON COLUMN prompts_compose_task.gateway_state IS 'model-gateway 状态：0排队/1执行/2成功/3失败/4已下载'; +COMMENT ON COLUMN prompts_compose_task.limit_words IS '提示词限制字数'; +COMMENT ON COLUMN prompts_compose_task.request_payload IS '发给 model-gateway 的请求内容'; +COMMENT ON COLUMN prompts_compose_task.result_text IS '回调返回的文本结果'; +COMMENT ON COLUMN prompts_compose_task.messages IS '最终解析后的 messages'; +COMMENT ON COLUMN prompts_compose_task.status IS '业务状态：pending/success/failed'; +COMMENT ON COLUMN prompts_compose_task.error_message IS '业务错误信息'; +COMMENT ON COLUMN prompts_compose_task.oss_file IS '网关返回的结果文件地址'; +COMMENT ON COLUMN prompts_compose_task.file_type IS '结果文件类型'; + +-- prompts_compose_session 提示词历史会话表 +CREATE TABLE IF NOT EXISTS prompts_compose_session ( + id BIGINT PRIMARY KEY, + tenant_id BIGINT NOT NULL DEFAULT 0, + creator VARCHAR(64) NOT NULL, + created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, + updater VARCHAR(64) NOT NULL, + updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, + deleted_at TIMESTAMP(6), + + session_id VARCHAR(64) NOT NULL, + request_content JSONB NOT NULL DEFAULT '[]'::jsonb, + response_content JSONB NOT NULL DEFAULT '[]'::jsonb, + remark VARCHAR(500) NOT NULL DEFAULT '' +); + +CREATE INDEX IF NOT EXISTS idx_prompts_compose_session_session_id ON prompts_compose_session(session_id); +CREATE INDEX IF NOT EXISTS idx_prompts_compose_session_deleted_at ON prompts_compose_session(deleted_at); + +COMMENT ON TABLE prompts_compose_session IS '提示词历史会话表'; +COMMENT ON COLUMN prompts_compose_session.id IS '主键ID（非自增）'; +COMMENT ON COLUMN prompts_compose_session.tenant_id IS '租户ID'; +COMMENT ON COLUMN prompts_compose_session.creator IS '创建人'; +COMMENT ON COLUMN prompts_compose_session.created_at IS '创建时间'; +COMMENT ON COLUMN prompts_compose_session.updater IS '更新人'; +COMMENT ON COLUMN prompts_compose_session.updated_at IS '更新时间'; +COMMENT ON COLUMN prompts_compose_session.deleted_at IS '删除时间（软删）'; +COMMENT ON COLUMN prompts_compose_session.session_id IS '会话ID'; +COMMENT ON COLUMN prompts_compose_session.request_content IS '请求内容（JSON格式）'; +COMMENT ON COLUMN prompts_compose_session.response_content IS '返回内容（JSON格式）'; +COMMENT ON COLUMN prompts_compose_session.remark IS '备注'; \ No newline at end of file