Havoc412
81cd287109
- 新增 RAG 聊天模式常量和前端字段设定 - 修改 Encounters Create 方法中的 ES 同步逻辑 - 更新 Rag ChatSSE 和 ChatWebSocket 方法，支持新的聊天模式 - 重构 NlpWebSocketResult 创建函数，使用新增的常量 - 新增 Encounter 的 TopK 方法，用于 ES 向量搜索 - 更新 DocResult 结构，实现 DocInterface 接口 - 修改 prompts.yml，增加 Diary 模式的提示模板
38 lines
887 B
Go
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
|
|
}
|