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