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