Blizzard
3b54e59ecf
embedding 从 env 改为控制面驱动(持久化+可视化)，复用 chat 模型同套范式： 配置控制面泛化为按 kind(chat/embedding)，加 embedding kind。 - shared: 配置 subjects 泛化 sundynix.config.<kind>.get/.updated；bus 方法改 kind 参数 (RequestConfig/ServeConfig/PublishConfigUpdated/SubscribeConfigUpdated) - gateway: sundynix_model 加 kind 列(每 kind 唯一激活)+旧行回填 chat；admin 按 kind 增删改/激活/列表，测试连接 embedding 走 POST /embeddings；main 按 kind ServeConfig； 变更广播各 kind - dispatcher: 取 chat 配置(kind 化) - mcp-go: rag.Engine.SetEmbedding 热更新(RWMutex)；main 取/订阅 embedding 控制面配置 (覆盖 env) - admin 控制台: api 按 kind；抽出复用 ModelManager；ModelsPage(chat)+新 DatasourcesPage (embedding + 向量/图库占位)；routes 数据源页就绪 - 验证: 全模块 build✓ + e2e PASS + 控制台 npm build✓；live 全跑通——chat(DeepSeek 回填 kind 仍工作)；mcp-go 不带 EMBED env 启动→控制台配 embedding(百炼)→测试连接✓→激活 →NATS 热更新 mcp-go→入库+语义检索'存向量的数据库'→Milvus；浏览器数据源页拉到激活配置 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// LLMModel 是一个模型后端配置（控制面：管理员在此登记可用模型）。
|
|
// 表名 sundynix_model（遵守前缀约定）。每个 kind 同一时刻仅一条 Active=true。
|
|
type LLMModel struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
Kind string `gorm:"size:16;index"` // chat / embedding
|
|
Provider string `gorm:"size:32"` // openai-compatible / vllm
|
|
BaseURL string `gorm:"size:255"` // 如 https://api.deepseek.com
|
|
APIKey string `gorm:"size:255"`
|
|
Model string `gorm:"size:64"` // 如 deepseek-chat / text-embedding-v3
|
|
Active bool
|
|
}
|
|
|
|
func (LLMModel) TableName() string { return "sundynix_model" }
|
|
|
|
// ListModels 列出某 kind 的模型配置（kind 空则全部）。
|
|
func (p *Postgres) ListModels(ctx context.Context, kind string) ([]LLMModel, error) {
|
|
if p.db == nil {
|
|
return nil, nil
|
|
}
|
|
var rows []LLMModel
|
|
q := p.db.WithContext(ctx).Order("id")
|
|
if kind != "" {
|
|
q = q.Where("kind = ?", kind)
|
|
}
|
|
err := q.Find(&rows).Error
|
|
return rows, err
|
|
}
|
|
|
|
// SaveModel 新增或更新一条模型配置（ID==0 新增）。
|
|
func (p *Postgres) SaveModel(ctx context.Context, m *LLMModel) error {
|
|
if p.db == nil {
|
|
return errStoreDisabled
|
|
}
|
|
return p.db.WithContext(ctx).Save(m).Error
|
|
}
|
|
|
|
// SetActiveModel 把指定模型设为激活（同 kind 内其余取消），事务保证每 kind 唯一激活。
|
|
func (p *Postgres) SetActiveModel(ctx context.Context, id uint) error {
|
|
if p.db == nil {
|
|
return errStoreDisabled
|
|
}
|
|
return p.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
var m LLMModel
|
|
if err := tx.First(&m, id).Error; err != nil {
|
|
return err
|
|
}
|
|
if err := tx.Model(&LLMModel{}).Where("kind = ? AND active = ?", m.Kind, true).Update("active", false).Error; err != nil {
|
|
return err
|
|
}
|
|
return tx.Model(&LLMModel{}).Where("id = ?", id).Update("active", true).Error
|
|
})
|
|
}
|
|
|
|
// GetActiveModel 返回某 kind 当前激活模型（无则 nil）。
|
|
func (p *Postgres) GetActiveModel(ctx context.Context, kind string) (*LLMModel, error) {
|
|
if p.db == nil {
|
|
return nil, nil
|
|
}
|
|
var m LLMModel
|
|
err := p.db.WithContext(ctx).Where("kind = ? AND active = ?", kind, true).First(&m).Error
|
|
if err != nil {
|
|
return nil, nil // 未配置激活模型
|
|
}
|
|
return &m, nil
|
|
}
|
|
|
|
// DeleteModel 删除一条模型配置。
|
|
func (p *Postgres) DeleteModel(ctx context.Context, id uint) error {
|
|
if p.db == nil {
|
|
return errStoreDisabled
|
|
}
|
|
return p.db.WithContext(ctx).Delete(&LLMModel{}, id).Error
|
|
}
|