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

43 lines
981 B
Go

package curd
import (
"catface/app/model"
"catface/app/model_es"
"catface/app/model_res"
)
func CreateEncounterCurdFactory() *EncounterCurd {
return &EncounterCurd{
enc: model.CreateEncounterFactory(""),
enc_es: model_es.CreateEncounterESFactory(nil),
}
}
type EncounterCurd struct {
enc *model.Encounter
enc_es *model_es.Encounter
}
func (e *EncounterCurd) TopK(embedding []float64, k int) (temp []model_res.EncounterResult, err error) {
// ES: TopK
encounters_es, err := e.enc_es.TopK(embedding, k)
if err != nil {
return
}
// MySQL 补充信息
var ids []int64
for _, encounter := range encounters_es {
ids = append(ids, encounter.Id)
}
encounters := e.enc.ShowByIDs(ids, "id", "title", "content", "updated_at")
for _, encounter := range encounters {
for _, encounter_es := range encounters_es {
if encounter.Id == encounter_es.Id {
temp = append(temp, *model_res.NewEncounterResult(&encounter, &encounter_es))
}
}
}
return
}