autosnippet 3.3.6 → 3.3.8

package/dist/lib/infrastructure/database/drizzle/schema.d.ts

@@ -4,15 +4,17 @@
  * 所有表定义从 migration 001-003 忠实翻译。
  * DB 列名与 migration 保持一致；实体映射由 repository 层处理。
  *
- * 表清单 (16):
+ * 表清单 (15):
  *   001: knowledge_entries, knowledge_edges, guard_violations, audit_logs,
  *        sessions, token_usage, semantic_memories, bootstrap_snapshots,
  *        bootstrap_dim_files, code_entities
- *   002: tasks, task_dependencies, task_events
  *   003: remote_commands
  *   004: evolution_proposals (+ knowledge_entries.staging_deadline)
+ *   005: recipe_source_refs
  *   内联: remote_state
  *   内部: schema_migrations
+ *
+ * 注: Task 系统为纯内存 + JSONL 信号架构，不使用数据库表。
  */
 export declare const schemaMigrations: import("drizzle-orm/sqlite-core").SQLiteTableWithColumns<{
     name: "schema_migrations";
@@ -3411,3 +3413,179 @@ export declare const recipeSourceRefs: import("drizzle-orm/sqlite-core").SQLiteT
     };
     dialect: "sqlite";
 }>;
+export declare const lifecycleTransitionEvents: import("drizzle-orm/sqlite-core").SQLiteTableWithColumns<{
+    name: "lifecycle_transition_events";
+    schema: undefined;
+    columns: {
+        id: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "id";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: true;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        recipeId: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "recipe_id";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        fromState: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "from_state";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        toState: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "to_state";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        trigger: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "trigger";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        operatorId: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "operator_id";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        evidenceJson: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "evidence_json";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        proposalId: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "proposal_id";
+            tableName: "lifecycle_transition_events";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
+        createdAt: import("drizzle-orm/sqlite-core").SQLiteColumn<{
+            name: "created_at";
+            tableName: "lifecycle_transition_events";
+            dataType: "number";
+            columnType: "SQLiteInteger";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+    };
+    dialect: "sqlite";
+}>;

package/dist/lib/infrastructure/database/drizzle/schema.js

@@ -4,17 +4,19 @@
  * 所有表定义从 migration 001-003 忠实翻译。
  * DB 列名与 migration 保持一致；实体映射由 repository 层处理。
  *
- * 表清单 (16):
+ * 表清单 (15):
  *   001: knowledge_entries, knowledge_edges, guard_violations, audit_logs,
  *        sessions, token_usage, semantic_memories, bootstrap_snapshots,
  *        bootstrap_dim_files, code_entities
- *   002: tasks, task_dependencies, task_events
  *   003: remote_commands
  *   004: evolution_proposals (+ knowledge_entries.staging_deadline)
+ *   005: recipe_source_refs
  *   内联: remote_state
  *   内部: schema_migrations
+ *
+ * 注: Task 系统为纯内存 + JSONL 信号架构，不使用数据库表。
  */
-import { index, integer, real, sqliteTable, text, uniqueIndex } from 'drizzle-orm/sqlite-core';
+import { index, integer, real, sqliteTable, text, uniqueIndex, } from 'drizzle-orm/sqlite-core';
 // ═══════════════════════════════════════════════════════════════
 // 内部 — schema_migrations
 // ═══════════════════════════════════════════════════════════════
@@ -339,3 +341,21 @@ export const recipeSourceRefs = sqliteTable('recipe_source_refs', {
     newPath: text('new_path'),
     verifiedAt: integer('verified_at').notNull(),
 }, (table) => [index('idx_rsr_path').on(table.sourcePath), index('idx_rsr_status').on(table.status)]);
+// ═══════════════════════════════════════════════════════════════
+// 18. lifecycle_transition_events — Recipe 生命周期转移事件 (migration 006)
+// ═══════════════════════════════════════════════════════════════
+export const lifecycleTransitionEvents = sqliteTable('lifecycle_transition_events', {
+    id: text('id').primaryKey(),
+    recipeId: text('recipe_id').notNull(),
+    fromState: text('from_state').notNull(),
+    toState: text('to_state').notNull(),
+    trigger: text('trigger').notNull(),
+    operatorId: text('operator_id').notNull().default('system'),
+    evidenceJson: text('evidence_json'),
+    proposalId: text('proposal_id'),
+    createdAt: integer('created_at').notNull(),
+}, (table) => [
+    index('idx_lte_recipe_id').on(table.recipeId),
+    index('idx_lte_created_at').on(table.createdAt),
+    index('idx_lte_trigger').on(table.trigger),
+]);

package/dist/lib/injection/ServiceContainer.d.ts

@@ -32,11 +32,12 @@ export declare class ServiceContainer {
     /**
      * 热重载 AI Provider（API Key 变更后调用，无需重启进程）
      *
-     * 流程：
-     *  1. 替换 singletons.aiProvider
-     *  2. 重新创建 _embedProvider（如果主 provider 不支持 embedding）
-     *  3. 清除已缓存的依赖 AI 的 singleton（SearchEngine 等），
-     *     下次 get() 时会用新 provider 重新创建
+     * 委托给 AiProviderManager.switchProvider() — 原子操作:
+     *  1. 替换 provider 引用 + DI 数据管道同步
+     *  2. Token 追踪 AOP 重新挂载
+     *  3. Embedding fallback 重建
+     *  4. 清除已缓存的依赖 AI 的 singleton（SearchEngine 等）
+     *  5. 监听器回调通知
      */
     reloadAiProvider(newProvider: Record<string, unknown> | null): void;
     /** 获取当前默认 UI 语言偏好 */

package/dist/lib/injection/ServiceContainer.js

@@ -9,6 +9,7 @@ import { CacheCoordinator } from '../infrastructure/cache/CacheCoordinator.js';
 import { GraphCache } from '../infrastructure/cache/GraphCache.js';
 // ─── P3: Infrastructure ──────────────────────────────
 import Logger from '../infrastructure/logging/Logger.js';
+import { unwrapRawDb } from '../repository/search/SearchRepoAdapter.js';
 import * as AgentModule from './modules/AgentModule.js';
 import * as AiModule from './modules/AiModule.js';
 import * as AppModule from './modules/AppModule.js';
@@ -147,36 +148,20 @@ export class ServiceContainer {
     /**
      * 热重载 AI Provider（API Key 变更后调用，无需重启进程）
      *
-     * 流程：
-     *  1. 替换 singletons.aiProvider
-     *  2. 重新创建 _embedProvider（如果主 provider 不支持 embedding）
-     *  3. 清除已缓存的依赖 AI 的 singleton（SearchEngine 等），
-     *     下次 get() 时会用新 provider 重新创建
+     * 委托给 AiProviderManager.switchProvider() — 原子操作:
+     *  1. 替换 provider 引用 + DI 数据管道同步
+     *  2. Token 追踪 AOP 重新挂载
+     *  3. Embedding fallback 重建
+     *  4. 清除已缓存的依赖 AI 的 singleton（SearchEngine 等）
+     *  5. 监听器回调通知
      */
     reloadAiProvider(newProvider) {
-        const old = this.singletons.aiProvider;
-        this.singletons.aiProvider = newProvider;
-        // 重新创建 embedding fallback provider（委托 AiModule）
-        this.singletons._embedProvider = null;
-        if (newProvider &&
-            typeof newProvider.supportsEmbedding === 'function' &&
-            !newProvider.supportsEmbedding()) {
-            AiModule.initEmbeddingFallback(this);
+        if (!newProvider) {
+            this.logger.warn('[ServiceContainer] reloadAiProvider called with null — ignored');
+            return;
         }
-        // 清除持有旧 aiProvider 引用的 singleton 缓存
-        // 下次调用 container.get() 时会使用新 provider 重建
-        const cleared = [];
-        for (const key of this._aiDependentSingletons || []) {
-            if (this.singletons[key]) {
-                this.singletons[key] = null;
-                cleared.push(key);
-            }
-        }
-        this.logger.info('AI provider hot-reloaded', {
-            old: old?.constructor?.name || 'none',
-            new: newProvider?.constructor?.name || 'none',
-            clearedSingletons: cleared,
-        });
+        const manager = this.singletons._aiProviderManager;
+        manager.switchProvider(newProvider);
     }
     // ─── 跨进程缓存协调 ─────
     /**
@@ -193,7 +178,9 @@ export class ServiceContainer {
     #initCacheCoordinator() {
         try {
             const db = this.singletons.database;
-            const rawDb = db?.getDb ? db.getDb() : null;
+            const rawDb = db
+                ? unwrapRawDb(db)
+                : null;
             if (!rawDb) {
                 return;
             }
@@ -316,7 +303,7 @@ export class ServiceContainer {
                 if (changedPaths.length === 0 && deletedPaths.length === 0) {
                     // 完全命中
                     this.singletons.projectGraph = graph;
-                    this.logger.info(`[ServiceContainer] ProjectGraph ⚡ 缓存命中 (${graph.getOverview()?.totalClasses} classes, ` +
+                    this.logger.info(`[ServiceContainer] ProjectGraph ⚡ 缓存命中 (${(await graph.getOverview())?.totalClasses} classes, ` +
                         `${Date.now() - startTime}ms)`);
                     return graph;
                 }
@@ -325,7 +312,7 @@ export class ServiceContainer {
                 this.singletons.projectGraph = graph;
                 // 写回缓存
                 cache.save('project-graph', graph.toJSON(), { fileHashes: newHashes });
-                const overview = graph.getOverview();
+                const overview = (await graph.getOverview());
                 this.logger.info(`[ServiceContainer] ProjectGraph 增量更新: +${diff.added} ~${diff.updated} -${diff.deleted} ` +
                     `(${overview.totalClasses} classes, ${Date.now() - startTime}ms)`);
                 return graph;
@@ -333,7 +320,7 @@ export class ServiceContainer {
             // ── 无缓存，全量构建 ──
             const graph = await ProjectGraph.build(projectRoot, options);
             this.singletons.projectGraph = graph;
-            const overview = graph.getOverview();
+            const overview = (await graph.getOverview());
             // 计算文件 hash 并写入缓存
             const currentFiles = this.#collectSourceFilePaths(projectRoot, options);
             const fileHashes = cache.computeFileHashes(currentFiles, projectRoot);

package/dist/lib/injection/ServiceMap.d.ts

@@ -14,6 +14,7 @@ import type ProjectGraph from '../core/ast/ProjectGraph.js';
 import type Constitution from '../core/constitution/Constitution.js';
 import type Gateway from '../core/gateway/Gateway.js';
 import type { AiProvider } from '../external/ai/AiProvider.js';
+import type { AiProviderManager } from '../external/ai/AiProviderManager.js';
 import type AuditLogger from '../infrastructure/audit/AuditLogger.js';
 import type AuditStore from '../infrastructure/audit/AuditStore.js';
 import type { CacheCoordinator } from '../infrastructure/cache/CacheCoordinator.js';
@@ -23,7 +24,17 @@ import type Logger from '../infrastructure/logging/Logger.js';
 import type { SignalBus } from '../infrastructure/signal/SignalBus.js';
 import type { IndexingPipeline } from '../infrastructure/vector/IndexingPipeline.js';
 import type { VectorStore } from '../infrastructure/vector/VectorStore.js';
+import type { AuditRepositoryImpl } from '../repository/audit/AuditRepository.js';
+import type { BootstrapRepositoryImpl } from '../repository/bootstrap/BootstrapRepository.js';
+import type { CodeEntityRepositoryImpl } from '../repository/code/CodeEntityRepository.js';
+import type { ProposalRepository } from '../repository/evolution/ProposalRepository.js';
+import type { GuardViolationRepositoryImpl } from '../repository/guard/GuardViolationRepository.js';
+import type { KnowledgeEdgeRepositoryImpl } from '../repository/knowledge/KnowledgeEdgeRepository.js';
 import type { KnowledgeRepositoryImpl } from '../repository/knowledge/KnowledgeRepository.impl.js';
+import type { MemoryRepositoryImpl } from '../repository/memory/MemoryRepository.js';
+import type { RemoteCommandRepository } from '../repository/remote/RemoteCommandRepository.js';
+import type { SessionRepositoryImpl } from '../repository/session/SessionRepository.js';
+import type { RecipeSourceRefRepositoryImpl } from '../repository/sourceref/RecipeSourceRefRepository.js';
 import type { TokenUsageStore } from '../repository/token/TokenUsageStore.js';
 import type { BootstrapTaskManager } from '../service/bootstrap/BootstrapTaskManager.js';
 import type { CursorDeliveryPipeline } from '../service/delivery/CursorDeliveryPipeline.js';
@@ -73,6 +84,16 @@ export interface ServiceMap {
     eventBus: EventBus;
     bootstrapTaskManager: BootstrapTaskManager;
     knowledgeRepository: KnowledgeRepositoryImpl;
+    knowledgeEdgeRepository: KnowledgeEdgeRepositoryImpl;
+    codeEntityRepository: CodeEntityRepositoryImpl;
+    bootstrapRepository: BootstrapRepositoryImpl;
+    guardViolationRepository: GuardViolationRepositoryImpl;
+    auditRepository: AuditRepositoryImpl;
+    memoryRepository: MemoryRepositoryImpl;
+    sessionRepository: SessionRepositoryImpl;
+    proposalRepository: ProposalRepository;
+    remoteCommandRepository: RemoteCommandRepository;
+    recipeSourceRefRepository: RecipeSourceRefRepositoryImpl;
     knowledgeFileWriter: KnowledgeFileWriter;
     knowledgeSyncService: KnowledgeSyncService;
     qualityScorer: QualityScorer;
@@ -98,6 +119,7 @@ export interface ServiceMap {
     dimensionCopy: typeof DimensionCopy;
     constitution: Constitution | null;
     aiProvider: AiProvider | null;
+    aiProviderManager: AiProviderManager;
     projectGraph: ProjectGraph | null;
     vectorService: VectorService;
     contextualEnricher: ContextualEnricher | null;

package/dist/lib/injection/modules/AiModule.d.ts

@@ -6,6 +6,7 @@
  *
  * 职责:
  *   - AI Provider 自动探测与创建
+ *   - AiProviderManager 统一管理层
  *   - Embedding fallback provider 管理
  *   - AiFactory 实例注入
  *
@@ -17,19 +18,15 @@ import type { ServiceContainer } from '../ServiceContainer.js';
  *
  * 1. 动态导入 AiFactory
  * 2. 自动探测可用 AI Provider
- * 3. 创建 Embedding fallback（若主 provider 不支持 embedding）
+ * 3. 创建 AiProviderManager（统一管理层）
+ * 4. 绑定 Token 追踪、Embedding fallback、DI 级联清理
  */
 export declare function initialize(c: ServiceContainer): Promise<void>;
-/**
- * 创建/刷新 Embedding fallback provider
- *
- * 若主 provider 不支持 embedding（如 Claude），尝试从其他可用 provider 创建备用。
- */
-export declare function initEmbeddingFallback(c: ServiceContainer): void;
 /**
  * 注册 AI 相关的服务到容器
  *
- * 当前 AI Provider 和 AiFactory 通过 singletons 直接管理，
- * 此方法注册便于其他模块通过 container.get() 获取的快捷服务。
+ * - 标记 AI 模块就绪
+ * - 注册 aiProviderManager 服务
+ * - 延迟注入 TokenRecorder（tokenUsageStore 此时已可用）
  */
 export declare function register(c: ServiceContainer): void;

package/dist/lib/injection/modules/AiModule.js

@@ -6,17 +6,20 @@
  *
  * 职责:
  *   - AI Provider 自动探测与创建
+ *   - AiProviderManager 统一管理层
  *   - Embedding fallback provider 管理
  *   - AiFactory 实例注入
  *
  * @module AiModule
  */
+import { AiProviderManager } from '../../external/ai/AiProviderManager.js';
 /**
  * 初始化 AI Provider（在模块注册前调用）
  *
  * 1. 动态导入 AiFactory
  * 2. 自动探测可用 AI Provider
- * 3. 创建 Embedding fallback（若主 provider 不支持 embedding）
+ * 3. 创建 AiProviderManager（统一管理层）
+ * 4. 绑定 Token 追踪、Embedding fallback、DI 级联清理
  */
 export async function initialize(c) {
     const logger = c.logger;
@@ -43,56 +46,96 @@ export async function initialize(c) {
             c.singletons.aiProvider = null;
         }
     }
-    // Embedding fallback provider
-    initEmbeddingFallback(c);
+    // ── 创建 AiProviderManager（统一管理层）──
+    const manager = new AiProviderManager(c.singletons.aiProvider || { name: 'mock', model: 'mock-fallback' });
+    c.singletons._aiProviderManager = manager;
+    // 绑定: DI 数据管道同步（切换时更新 singletons 中的 provider 引用，供工厂函数读取）
+    manager._bindDiSync((provider, embed) => {
+        c.singletons.aiProvider = provider;
+        c.singletons._embedProvider = embed;
+    });
+    // 绑定: DI 级联清理回调
+    manager._bindDependentClearer(() => {
+        const cleared = [];
+        for (const key of c._aiDependentSingletons || []) {
+            if (c.singletons[key]) {
+                c.singletons[key] = null;
+                cleared.push(key);
+            }
+        }
+        return cleared;
+    });
+    // 绑定: Embedding fallback 初始化器
+    manager._bindEmbedFallbackInit((currentProvider) => {
+        return createEmbedFallback(c, currentProvider);
+    });
+    // Token 追踪 AOP（manager 自身已在构造时 wire，此处延迟注入 recorder）
+    // recorder 注入放到 register() 之后（tokenUsageStore 需先注册）
+    // Embedding fallback: manager 的 embedFallbackInit 回调已绑定，初始化时主动触发一次
+    const initialEmbed = createEmbedFallback(c, c.singletons.aiProvider);
+    if (initialEmbed) {
+        manager.setEmbedProvider(initialEmbed);
+        c.singletons._embedProvider = initialEmbed;
+    }
 }
 /**
- * 创建/刷新 Embedding fallback provider
- *
- * 若主 provider 不支持 embedding（如 Claude），尝试从其他可用 provider 创建备用。
+ * 纯函数: 尝试为给定 provider 创建 Embedding fallback
+ * 被 initEmbeddingFallback() 和 AiProviderManager 的 embedFallbackInit 回调共用
  */
-export function initEmbeddingFallback(c) {
-    const currentProvider = c.singletons.aiProvider;
-    if ((currentProvider &&
-        typeof currentProvider
-            .supportsEmbedding !== 'function') ||
-        (currentProvider &&
-            !currentProvider.supportsEmbedding?.())) {
-        try {
-            const aiFactory = (c.singletons._aiFactory || {});
-            const providerName = (currentProvider?.name || '').replace('-', '');
-            const fbCandidates = typeof aiFactory.getAvailableFallbacks === 'function'
-                ? aiFactory.getAvailableFallbacks(providerName)
-                : [];
-            for (const fb of fbCandidates) {
-                try {
-                    const fbProvider = aiFactory.createProvider({ provider: fb });
-                    if (typeof fbProvider.supportsEmbedding === 'function' &&
-                        fbProvider.supportsEmbedding()) {
-                        c.singletons._embedProvider = fbProvider;
-                        c.logger.info('Embedding fallback provider created', { provider: fb });
-                        break;
-                    }
-                }
-                catch {
-                    /* skip */
+function createEmbedFallback(c, currentProvider) {
+    if (!currentProvider ||
+        (typeof currentProvider.supportsEmbedding === 'function' && currentProvider.supportsEmbedding())) {
+        return null; // 主 provider 已支持 embedding，无需 fallback
+    }
+    try {
+        const aiFactory = (c.singletons._aiFactory || {});
+        const providerName = (currentProvider.name || '').replace('-', '');
+        const fbCandidates = typeof aiFactory.getAvailableFallbacks === 'function'
+            ? aiFactory.getAvailableFallbacks(providerName)
+            : [];
+        for (const fb of fbCandidates) {
+            try {
+                const fbProvider = aiFactory.createProvider?.({ provider: fb });
+                if (fbProvider &&
+                    typeof fbProvider.supportsEmbedding === 'function' &&
+                    fbProvider.supportsEmbedding()) {
+                    c.logger.info('Embedding fallback provider created', { provider: fb });
+                    return fbProvider;
                 }
             }
-        }
-        catch {
-            /* no embed fallback available */
+            catch {
+                /* skip */
+            }
         }
     }
+    catch {
+        /* no embed fallback available */
+    }
+    return null;
 }
 /**
  * 注册 AI 相关的服务到容器
  *
- * 当前 AI Provider 和 AiFactory 通过 singletons 直接管理，
- * 此方法注册便于其他模块通过 container.get() 获取的快捷服务。
+ * - 标记 AI 模块就绪
+ * - 注册 aiProviderManager 服务
+ * - 延迟注入 TokenRecorder（tokenUsageStore 此时已可用）
  */
 export function register(c) {
-    // aiProvider 和 _aiFactory 已通过 initialize() 写入 singletons
-    // KnowledgeModule 中已注册 'aiProvider' 的 register 工厂
-    // 此处仅标记 AI 模块已就绪
     c.singletons._aiModuleReady = true;
+    // 注册 aiProviderManager（消费者通过 container.get('aiProviderManager') 获取）
+    c.register('aiProviderManager', () => c.singletons._aiProviderManager);
+    // 延迟注入 TokenRecorder 到 manager（tokenUsageStore 在 AppModule 中注册）
+    const manager = c.singletons._aiProviderManager;
+    const containerRef = c;
+    manager.setTokenRecorder({
+        record(r) {
+            try {
+                const store = containerRef.get('tokenUsageStore');
+                store.record(r);
+            }
+            catch {
+                /* tokenUsageStore not available yet */
+            }
+        },
+    });
 }

package/dist/lib/injection/modules/AppModule.js

@@ -8,6 +8,7 @@
  *   - primeSearchPipeline (for prime multi-query search — no DB dependency)
  */
 import { resolveProjectRoot } from '#shared/resolveProjectRoot.js';
+import { unwrapRawDb } from '../../repository/search/SearchRepoAdapter.js';
 import { TokenUsageStore } from '../../repository/token/TokenUsageStore.js';
 import { CursorDeliveryPipeline } from '../../service/delivery/CursorDeliveryPipeline.js';
 import { RecipeExtractor } from '../../service/knowledge/RecipeExtractor.js';
@@ -29,7 +30,7 @@ export function register(c) {
     });
     c.singleton('tokenUsageStore', (ct) => {
         const db = ct.get('database');
-        return new TokenUsageStore(db.getDb(), db.getDrizzle());
+        return new TokenUsageStore(unwrapRawDb(db), db.getDrizzle());
     });
     // ═══ Module ═══
     c.singleton('moduleService', (ct) => {

package/dist/lib/injection/modules/GuardModule.js

@@ -9,6 +9,7 @@
 import fs from 'node:fs';
 import path from 'node:path';
 import { resolveProjectRoot } from '#shared/resolveProjectRoot.js';
+import { unwrapRawDb } from '../../repository/search/SearchRepoAdapter.js';
 import { ComplianceReporter } from '../../service/guard/ComplianceReporter.js';
 import { CoverageAnalyzer } from '../../service/guard/CoverageAnalyzer.js';
 import { ExclusionManager } from '../../service/guard/ExclusionManager.js';
@@ -66,6 +67,7 @@ export function register(c) {
         return new GuardCheckEngine(ct.get('database'), {
             guardConfig: merged,
             signalBus: ct.singletons.signalBus || undefined,
+            knowledgeRepo: ct.get('knowledgeRepository'),
         });
     });
     c.singleton('exclusionManager', (ct) => {
@@ -80,7 +82,7 @@ export function register(c) {
     });
     c.singleton('violationsStore', (ct) => {
         const db = ct.get('database');
-        return new ViolationsStore(db.getDb(), db.getDrizzle());
+        return new ViolationsStore(unwrapRawDb(db), db.getDrizzle());
     });
     c.singleton('complianceReporter', (ct) => {
         const config = ct.singletons._config || {};
@@ -91,13 +93,11 @@ export function register(c) {
         signalBus: ct.singletons.signalBus || undefined,
     }));
     c.singleton('reverseGuard', (ct) => {
-        const db = ct.get('database');
-        return new ReverseGuard(db.getDb(), {
+        return new ReverseGuard(ct.get('knowledgeRepository'), ct.get('codeEntityRepository'), ct.get('recipeSourceRefRepository'), {
             signalBus: ct.singletons.signalBus || undefined,
         });
     });
     c.singleton('coverageAnalyzer', (ct) => {
-        const db = ct.get('database');
         let ruleLearner;
         try {
             ruleLearner = { ruleLearner: ct.get('ruleLearner') };
@@ -105,6 +105,6 @@ export function register(c) {
         catch {
             /* ruleLearner not yet available */
         }
-        return new CoverageAnalyzer(db.getDb(), ruleLearner);
+        return new CoverageAnalyzer(ct.get('knowledgeRepository'), ct.get('guardViolationRepository'), ruleLearner);
     });
 }