fae162162a
后端： 1. Memory Day1 链路打通（chat_history -> outbox -> memory_jobs） - 更新 service/events/chat_history_persist.go：聊天消息落库同事务追加 memory.extract.requested 事件（仅 user 消息，失败回滚后由 outbox 重试） - 新建 service/events/memory_extract_requested.go：消费 memory.extract.requested 并幂等入队 memory_jobs，补齐 payload 校验、文本截断与 idempotency key - 更新 cmd/start.go：注册 RegisterMemoryExtractRequestedHandler 2. Memory 模块骨架落地（先跑通状态机，再接入真实抽取） - 新建 memory/model、repo、service、orchestrator、worker、utils 目录与 Day1 mock 抽取执行链 - 新建 model/memory.go：补齐 memory_items / memory_jobs / memory_audit_logs / memory_user_settings 与事件 payload 模型 - 更新 inits/mysql.go：接入 4 张 memory 相关表 AutoMigrate 3. RAG 复用基础设施预埋（依赖可替换） - 新建 infra/rag：core pipeline + chunk/embed/retrieve/rerank/store/corpus/config 分层实现 - 默认接入 MockEmbedder + InMemoryStore，预留 Milvus / Eino 适配实现 - 新增 infra/rag/RAG复用接口实施计划.md 4. 本地依赖与交接文档同步 - 更新 docker-compose.yml：新增 etcd / minio / milvus / attu 服务与数据卷 - 删除 newAgent/HANDOFF_工具研究与运行态重置.md、newAgent/阶段3_上下文瘦身设计.md - 新增 newAgent/HANDOFF_WebSearch两阶段实施计划.md、memory/HANDOFF-RAG复用后续实施计划.md、memory/README.md 前端：无 仓库：无
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package inits
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/LoveLosita/smartflow/backend/model"
|
|
"github.com/spf13/viper"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func autoMigrateModels(db *gorm.DB) error {
|
|
models := []any{
|
|
&model.User{},
|
|
&model.AgentChat{},
|
|
&model.ChatHistory{},
|
|
&model.Task{},
|
|
&model.TaskClass{},
|
|
&model.TaskClassItem{},
|
|
&model.ScheduleEvent{},
|
|
&model.Schedule{},
|
|
&model.AgentOutboxMessage{},
|
|
&model.AgentScheduleState{},
|
|
&model.AgentStateSnapshotRecord{},
|
|
&model.MemoryItem{},
|
|
&model.MemoryJob{},
|
|
&model.MemoryAuditLog{},
|
|
&model.MemoryUserSetting{},
|
|
}
|
|
|
|
for _, m := range models {
|
|
if err := db.AutoMigrate(m); err != nil {
|
|
return fmt.Errorf("auto migrate failed for %T: %w", m, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ConnectDB() (*gorm.DB, error) {
|
|
host := viper.GetString("database.host")
|
|
port := viper.GetString("database.port")
|
|
user := viper.GetString("database.user")
|
|
password := viper.GetString("database.password")
|
|
dbname := viper.GetString("database.dbname")
|
|
|
|
dsn := fmt.Sprintf(
|
|
"%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
|
user, password, host, port, dbname,
|
|
)
|
|
|
|
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err = autoMigrateModels(db); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.Println("Database connected successfully")
|
|
log.Println("Database auto migration completed")
|
|
return db, nil
|
|
}
|