AskaEth
a6cb44b8a8
主人要求「链路流程应该有本地数据，不然每次重载之后页面就没数据了」： - 落盘：迁移 0022_flow_graphs（session_id/turn_id/graph_json/时间戳 + 唯一键 upsert）+ internal/runtime/flowgraph（RecordRound/Graph/Latest/Turns/Turn + 保留期 GC） - 写入：loop.EndRound **每轮一条**（锁外深拷贝后写；失败 WARN 且不打印图内容）；只落发布会话白名单（复用 flowSessionAllowed） - 读回：端点加 turnId + source(live/persisted/none) + turns[]（最新在前，空为 []）；**内存优先、miss 才落盘**（最新一轮永远最新） - 前端：flowGraphs 纳入 persistOptions.partialize（**刷新即刻有图**，version 3→4 + migrate）+ 历史轮选择器 + 诚实文案；顺带收 L1 转交的 flow/NodeDetail.tsx 节点日志倒序 - 新 env 键：FLOW_GRAPH_RETENTION=168h / FLOW_GRAPH_GC_INTERVAL=1h - 变异实测：每 span 一行 / 保留期守卫失效 / partialize 去掉 flowGraphs / 去掉内存优先 / 白名单失效 ⇒ 全部打红 - 3 个禁动面文件的纯追加改动（Flow.tsx 挂选择器、FlowSnapshotHint 纠已成假话的文案、其渲染断言随之更新）由主会话裁定认可 - 登记存疑：图桶持久化无体积上限；useFlowTurns 的 effect 无自动化证据（项目无 jsdom）
221 lines
9.3 KiB
Go
221 lines
9.3 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"fmt"
|
||
"log/slog"
|
||
"time"
|
||
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/bus"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/capability/mcp"
|
||
pluginmgr "git.yeij.top/AskaEth/Cyrene_v2/internal/capability/plugin"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/capability/registry"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/capability/transcode"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/channel/session"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/config"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/db"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/iot"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/persona/emotion"
|
||
personareg "git.yeij.top/AskaEth/Cyrene_v2/internal/persona/registry"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/embedding"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/exec"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/events"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/flowgraph"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/history"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/loop"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/memory"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/memory/projectindex"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/memory/vector"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/multimodal/voice"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/orchestration"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/task"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/tool"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/services/automations"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/services/backthink"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/services/channelbridge"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/services/gateway"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/user"
|
||
)
|
||
|
||
// defaultOwnerPath 与 defaultPersonasDir 指向随仓库分发的磁盘上人格/主人定义，
|
||
// 路径相对工作目录。
|
||
const (
|
||
defaultOwnerPath = "configs/users/owner.yaml"
|
||
defaultPersonasDir = "configs/personas"
|
||
defaultSQLitePath = "./data/cyrene.db"
|
||
fallbackInstruction = "You are Cyrene, a warm and attentive companion."
|
||
|
||
// 上界与旋钮常量。编译期定值，运行时路径上不出现裸魔法数字（项目规范 §10）。
|
||
defaultMaxRounds = 8
|
||
defaultMaxParallel = 4
|
||
defaultAgentID = "orchestrator"
|
||
defaultToolTimeout = 30 * time.Second
|
||
)
|
||
|
||
// app 持有单二进制 Cyrene 组装的全部子系统：由 newApp 构建，关闭时由 close 拆除。
|
||
type app struct {
|
||
cfg *config.Config
|
||
log *slog.Logger
|
||
db *sql.DB
|
||
dbPath string
|
||
bus bus.Bus
|
||
modelAdapter model.Adapter
|
||
llm *model.RuntimeLLM
|
||
personas *personareg.Registry
|
||
caps *registry.Registry
|
||
// transcoder 是本地转码引擎（design/25）：进程内、同步、纯本地，供渠道与其它
|
||
// 模块经依赖注入调用；同一个引擎也以 `transcode` 工具注册进 caps。装配只告警、
|
||
// 绝不阻塞启动，见 wire_transcode.go。
|
||
transcoder transcode.Transcoder
|
||
dispatcher *tool.Dispatcher
|
||
memStore *memory.Store
|
||
retriever *memory.Retriever
|
||
// embedder 与 memoryTidy 属于记忆整理（design/23）：整理只在 embedder
|
||
// 上算召回相似度，绝不接触会话/历史读数。
|
||
embedder embedding.Embedder
|
||
memoryTidy *memory.TidyService
|
||
projStore *projectindex.Store
|
||
loop *loop.Loop
|
||
scheduler *orchestration.Scheduler
|
||
tasks *task.Manager
|
||
execProv exec.Provider
|
||
iot *iot.Manager
|
||
voice *voice.Chain
|
||
mcpRegs []*mcp.Registry
|
||
plugins *pluginmgr.Manager
|
||
sessions *session.SessionManager
|
||
automation *automations.Service
|
||
backthink *backthink.Backthink
|
||
vindex vector.Index
|
||
owner *user.OwnerProfile
|
||
history *history.Store
|
||
gw *gateway.Server
|
||
channelBridge *channelbridge.Bridge
|
||
channels gateway.ChannelConfigStore
|
||
// docGC 是文档回收的常驻任务（切片 S7）：未被引用且超过保留期的文档连同
|
||
// sidecar 与元数据行一起回收。GC 被配置关闭时为 nil（见 wire_document_gc.go）。
|
||
docGC *documentGC
|
||
// channelEvents 是渠道事件留存面（design/26 §4；切片 E3+E4）：非消息事件逐条
|
||
// 落行 / T1 计数都经它，桥接层由 wire_channels.go 注入。库未接时为 nil
|
||
// （桥接层随即走内存计数降级，见 wire_channel_events.go）。
|
||
channelEvents *events.SQLStore
|
||
// channelEventKeeper 是事件留存的常驻维护任务（保留期清扫 + 计数落盘），
|
||
// 由 main 的 startChannelEvents 拉起（见 wire_channel_events.go）。
|
||
channelEventKeeper *channelEventKeeper
|
||
// usage 是 LLM 用量的异步落库出口（切片 U1）：注入模型栈后，每次成功模型
|
||
// 调用都会把非敏感计数样本塞进它的有界队列。库未接时为 nil（= 只记日志）。
|
||
usage *model.UsageWriter
|
||
// usageGC 是用量样本的保留期清理常驻任务。`LLM_USAGE_RETENTION ≤ 0` 时为
|
||
// nil（清理关闭，见 wire_usage.go）。
|
||
usageGC *usageRetentionGC
|
||
// capacitySink 把 capacity_awaken 状态迁移广播给 Web UI（design/14 §5）。
|
||
// 由 *gateway.Server 满足（BroadcastCapacityAwaken）；以接口形式持有，
|
||
// 组合根才能用假出口做单元测试。
|
||
capacitySink capacityEmitter
|
||
// emotion 把每一轮落定的对话施加到所绑人格的情绪机与好感度上（design/01 §2），
|
||
// 并把变化交给 moodSink。没有注册人格时为 nil，此时不追踪、不持久化任何心情。
|
||
emotion *emotion.Observer
|
||
// moodSink 把心情变化广播给 Web UI（design/07 §3）。由 *gateway.Server 满足
|
||
// （BroadcastMood）；以接口形式持有，组合根才能用假出口做单元测试。
|
||
moodSink moodEmitter
|
||
// traceWriter 把完成的 trace span 扇出到 JSONL stderr 记录器，网关出现后也送往
|
||
// 网关的可观测性缓冲（挂在 GET /api/v1/admin/observability/trace 后面，
|
||
// design/19 §2）。持有它是为了让 wireFrontend 能往已在运行的 tracer 上注册
|
||
// 网关出口。
|
||
traceWriter *traceFanout
|
||
// flow 是 loop 的实时链路图发布器（design/19 §1）。持有它是因为网关还要一个
|
||
// **只读**出口：`GET /api/v1/admin/flow/graph` 回「最近一轮」的快照，补上
|
||
// 「轮次结束后才打开的页面」看不到图的那个洞（契约 v0.71）。组合根把**同一个**
|
||
// 发布器同时交给 loop/orchestration（写入面）与网关（查询面），因此查询到的
|
||
// 就是实时发布的那张图，不会出现两份状态。
|
||
flow *loop.FlowPublisher
|
||
// flowStore 是链路图的**本地落盘**（design/19 §4 P3；切片 F2；迁移 0022）：发布器
|
||
// 每轮收尾经它落一条，网关按 turnId / turns 读回。库未接时为 nil（发布器随即走
|
||
// 纯内存路径，见 wire_flow_graphs.go）。它与 flow 是两件事：flow 是实时内存图，
|
||
// flowStore 是收尾存档。
|
||
flowStore *flowgraph.SQLStore
|
||
// flowGraphKeeper 是链路图留存的常驻维护任务（保留期清扫），由 main 的
|
||
// startFlowGraphs 拉起（见 wire_flow_graphs.go）。
|
||
flowGraphKeeper *flowGraphKeeper
|
||
}
|
||
|
||
// newApp 依据 cfg 与 log 组装完整的 Cyrene 运行时：打开数据库、迁移表结构，接线
|
||
// 人格注册表、模型适配器、工具分发器、记忆栈、agent 循环、编排、执行、IoT、
|
||
// 渠道会话/适配器与网关，最后返回就绪的 app。
|
||
//
|
||
// 致命的接线失败（数据库打开/迁移、必需的认证、网关）以错误返回；可选子系统则
|
||
// 优雅降级，使得未配置模型密钥或外部渠道时二进制仍能启动。
|
||
func newApp(ctx context.Context, cfg *config.Config, log *slog.Logger) (*app, error) {
|
||
dataDir := resolveDataDir(cfg)
|
||
databasePath := resolveSQLitePath(cfg, dataDir)
|
||
dbCfg := cfg.DB
|
||
dbCfg.SQLitePath = databasePath
|
||
|
||
sqlDB, err := db.Open(ctx, databasePath, log)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("open database: %w", err)
|
||
}
|
||
if err := db.MigrateUp(ctx, sqlDB, log); err != nil {
|
||
_ = sqlDB.Close()
|
||
return nil, fmt.Errorf("migrate database: %w", err)
|
||
}
|
||
|
||
eventBus := bus.New()
|
||
a := &app{
|
||
cfg: cfg,
|
||
log: log,
|
||
db: sqlDB,
|
||
dbPath: databasePath,
|
||
bus: eventBus,
|
||
caps: registry.New(),
|
||
}
|
||
|
||
// 后续任一接线步骤失败时，释放已经打开的资源，但不打整套服务关闭日志
|
||
// （半途启动不等同于一次干净退出）。
|
||
completed := false
|
||
defer func() {
|
||
if completed {
|
||
return
|
||
}
|
||
if a.vindex != nil {
|
||
_ = a.vindex.Close()
|
||
}
|
||
if a.tasks != nil {
|
||
_ = a.tasks.Close()
|
||
}
|
||
if a.bus != nil {
|
||
_ = a.bus.Close()
|
||
}
|
||
if a.db != nil {
|
||
_ = a.db.Close()
|
||
}
|
||
}()
|
||
|
||
if err := a.buildPersonas(); err != nil {
|
||
return a, err
|
||
}
|
||
if err := a.wireMemory(ctx); err != nil {
|
||
return a, err
|
||
}
|
||
if err := a.wireRuntime(); err != nil {
|
||
return a, err
|
||
}
|
||
if err := a.wireAuxiliary(); err != nil {
|
||
return a, err
|
||
}
|
||
a.wirePlugins(ctx)
|
||
// 渠道事件留存面（design/26 §4；切片 E3+E4）：必须在 wireChannels 之前构造 ——
|
||
// 桥接层在构造时就把它注入（Options.EventStore），运行时不再重取。
|
||
a.wireChannelEvents()
|
||
if err := a.wireChannels(); err != nil {
|
||
return a, err
|
||
}
|
||
if err := a.wireFrontend(); err != nil {
|
||
return a, err
|
||
}
|
||
completed = true
|
||
return a, nil
|
||
}
|