Havoc412 81cd287109 feat(api): 新增 RAG 聊天模式和优化 ES 查询功能
- 新增 RAG 聊天模式常量和前端字段设定
- 修改 Encounters Create 方法中的 ES 同步逻辑
- 更新 Rag ChatSSE 和 ChatWebSocket 方法,支持新的聊天模式
- 重构 NlpWebSocketResult 创建函数,使用新增的常量
- 新增 Encounter 的 TopK 方法,用于 ES 向量搜索
- 更新 DocResult 结构,实现 DocInterface 接口
- 修改 prompts.yml,增加 Diary 模式的提示模板
2024-11-20 17:32:10 +08:00

19 lines
549 B
Go

package consts
const (
// TAG animal/ 接口模式
AnimalModePrefer string = "prefer" // 根据用户行为记录优先返回【偏好目标】
// TAG rag/chat 接口模式; 配合 yml 文件的书写习惯。
RagChatModeKnowledge string = "Knowledge"
RagChatModeDiary string = "Diary" // 查询路遇资料等
RagChatModeDetect string = "Detect" // 辅助 catface 的辨认功能;
)
// 前端的字段设定
const (
AiMessageTypeText string = "text"
AiMessageTypeDoc string = "doc"
AiMessageTypeToken string = "token"
)