Havoc412
91073fdf7f
- 调整了 ES TopK 查询逻辑，增加了文档类型筛选 - 优化了 WebSocket 的关闭流程，增加了文档和 token 信息的发送 - 新增了 Doc 模型的 ShowById 和 ShowByIds 方法，用于查询特定文档 - 更新了 prompts.yml 文件中的提示语，将"上下文"改为"知识库"
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package model
|
|
|
|
import (
|
|
"catface/app/global/variable"
|
|
"catface/app/utils/data_bind"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// INFO @brief 这个 model 是便于宏观管理知识库文件的。
|
|
|
|
func CreateDocFactory(sqlType string) *Doc {
|
|
return &Doc{BaseModel: BaseModel{DB: UseDbConn(sqlType)}}
|
|
}
|
|
|
|
type Doc struct {
|
|
BaseModel
|
|
Name string `gorm:"name" json:"name"` // 文件名保存原本的设定，但是实际存储的【真名】还是借助 Snow + MD5 防止冲突；
|
|
Path string `gorm:"path" json:"path"`
|
|
}
|
|
|
|
func (d *Doc) TableName() string { return "docs" }
|
|
|
|
func (d *Doc) InsertDocumentData(c *gin.Context) (int64, bool) {
|
|
var tmp Doc
|
|
if err := data_bind.ShouldBindFormDataToModel(c, &tmp); err == nil {
|
|
if res := d.Create(&tmp); res.Error == nil {
|
|
return tmp.Id, true
|
|
} else {
|
|
variable.ZapLog.Error("Doc 数据新增出错", zap.Error(res.Error))
|
|
}
|
|
} else {
|
|
variable.ZapLog.Error("Doc 数据绑定出错", zap.Error(err))
|
|
}
|
|
return 0, false
|
|
}
|
|
|
|
func (d *Doc) ShowById(id int64, attrs ...string) *Doc {
|
|
var temp Doc
|
|
db := d.DB.Table(d.TableName()).Where("id = ?", id)
|
|
|
|
if len(attrs) > 0 {
|
|
db.Select(attrs)
|
|
}
|
|
|
|
err := db.First(&temp)
|
|
if err.Error != nil {
|
|
variable.ZapLog.Error("Doc ShowById Error", zap.Error(err.Error))
|
|
}
|
|
return &temp
|
|
}
|
|
|
|
func (d *Doc) ShowByIds(ids []int64, attrs ...string) (temp []Doc) {
|
|
db := d.DB.Table(d.TableName()).Where("id in (?)", ids)
|
|
if len(attrs) > 0 {
|
|
db.Select(attrs)
|
|
}
|
|
err := db.Find(&temp)
|
|
if err.Error != nil {
|
|
variable.ZapLog.Error("Doc ShowByIds Error", zap.Error(err.Error))
|
|
}
|
|
return
|
|
}
|