Havoc412
ae7edb5e8d
- 重构了 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 文件，实现了结构体到解释字符串的转换
28 lines
443 B
Go
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
|
|
}
|