embedding_config.go 1.4 KB

1234567891011121314151617181920
  1. package models
  2. import "time"
  3. // EmbeddingConfig 知识库向量模型配置(平台级单例,仅一条有效记录)
  4. // 用于文档向量化与 RAG 检索,在前端「设置 - 知识库向量模型」中配置
  5. type EmbeddingConfig struct {
  6. ID uint `json:"id" gorm:"primaryKey"`
  7. EmbeddingType string `json:"embedding_type" gorm:"type:varchar(50);default:'openai'"` // openai / bge / local
  8. APIURL string `json:"api_url" gorm:"type:varchar(500)"` // API 地址
  9. APIKey string `json:"-" gorm:"type:varchar(1000)"` // API Key(加密存储,不返回给前端)
  10. Model string `json:"model" gorm:"type:varchar(100)"` // 模型名称
  11. CustomerCanUseKB bool `json:"customer_can_use_kb" gorm:"default:true"` // 是否开放知识库给客服使用(创建/上传/RAG)
  12. // 访客端是否显示「本回合联网搜索」选项(由配置页控制)
  13. VisitorWebSearchEnabled bool `json:"visitor_web_search_enabled" gorm:"default:false"`
  14. // 联网方式:vendor(厂商内置 web_search)/ custom(自建 Serper,后端执行)
  15. WebSearchSource string `json:"web_search_source" gorm:"type:varchar(20);default:'custom'"`
  16. CreatedAt time.Time `json:"created_at"`
  17. UpdatedAt time.Time `json:"updated_at"`
  18. }