Havoc412 91073fdf7f refactor(app): 重构 WebSocket 聊天流程并优化文档查询功能
- 调整了 ES TopK 查询逻辑,增加了文档类型筛选
- 优化了 WebSocket 的关闭流程,增加了文档和 token 信息的发送
- 新增了 Doc 模型的 ShowById 和 ShowByIds 方法,用于查询特定文档
- 更新了 prompts.yml 文件中的提示语,将"上下文"改为"知识库"
2024-11-19 13:06:39 +08:00

24 lines
430 B
Go

package model_res
import (
"catface/app/model"
"catface/app/model_es"
)
// BUG 存在 依賴循環
func NewDocResult(doc *model.Doc, doc_es *model_es.Doc) *DocResult {
return &DocResult{
Type: "doc",
Id: doc.Id,
Name: doc.Name,
Content: doc_es.Content,
}
}
type DocResult struct {
Type string `json:"type"`
Id int64 `json:"id"`
Name string `json:"name"`
Content string `json:"content"`
}