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

38 lines
887 B
Go

package curd
import (
"catface/app/global/consts"
"catface/app/model_res"
"fmt"
)
func TopK(mode string, embedding []float64, k int) (temp []model_res.DocInterface, err error) {
switch mode {
case consts.RagChatModeKnowledge:
results, err := CreateDocCurdFactory().TopK(embedding, k)
if err != nil {
return nil, fmt.Errorf("TopK: 获取知识库TopK失败: %w", err)
}
for _, result := range results {
temp = append(temp, result)
}
case consts.RagChatModeDiary:
results, err := CreateEncounterCurdFactory().TopK(embedding, k)
if err != nil {
return nil, fmt.Errorf("TopK: 获取路遇笔记TopK失败: %w", err)
}
for _, result := range results {
temp = append(temp, result)
}
default:
if mode == "" {
err = fmt.Errorf("TopK: mode不能为空")
} else {
err = fmt.Errorf("TopK: 不支持的mode: %s", mode)
}
}
return temp, err
}