catface_backend_go/app/model_res/rag_websocket.go
Havoc412 ae7edb5e8d 🎨 refactor(rag): 重构 RAG 模型相关代码
- 重构了 rag_controller.go 中的逻辑,使用新的 DocumentHub 结构
- 修改了 encounter.go 中的 Encounter 结构,增加了 explain 标签
- 重写了 rag_websocket.go 中的逻辑,使用新的 DocumentHub 结构
- 新增了 curd_es/encounter_es_curd.go 文件,实现了 Encounter 的 CURD 操作
- 更新了 nlp/func.go 中的 ChatRAG 函数,使用新的 DocumentHub 结构
- 新增了 curd/docs_hub.go 文件,实现了 DocumentHub 的 TopK 方法
- 新增了 utils/data_explain/data_explain_rag.go 文件,实现了结构体到解释字符串的转换
2024-11-20 19:30:11 +08:00

28 lines
443 B
Go

package model_res
import (
"catface/app/global/consts"
"encoding/json"
)
func CreateNlpWebSocketResult(t string, data any) *NlpWebSocketResult {
if t == "" {
t = consts.AiMessageTypeText
}
return &NlpWebSocketResult{
Type: t,
Data: data,
}
}
type NlpWebSocketResult struct {
Type string `json:"type"`
Data any `json:"data"`
}
func (n *NlpWebSocketResult) JsonMarshal() []byte {
data, _ := json.Marshal(n)
return data
}