claude-brain 0.27.2 → 0.28.0

package/src/cli/commands/status.ts

@@ -123,6 +123,50 @@ export async function runStatus() {
     status: installedHooks.length > 0 ? 'success' : 'warning'
   })
 
+  // SLM Models
+  const modelsDir = join(home, 'models')
+  const manifestPath = join(modelsDir, 'manifest.json')
+  if (existsSync(manifestPath)) {
+    try {
+      const manifest = JSON.parse(readFileSync(manifestPath, 'utf-8'))
+      const tasks = Object.keys(manifest.models || {})
+      const available = tasks.filter(t => existsSync(join(modelsDir, manifest.models[t].file)))
+      items.push({
+        label: 'SLM Models',
+        value: `${available.length}/${tasks.length} available (${available.join(', ') || 'none'})`,
+        status: available.length > 0 ? 'success' : 'warning'
+      })
+    } catch {
+      items.push({ label: 'SLM Models', value: 'Manifest error', status: 'error' })
+    }
+  } else {
+    items.push({ label: 'SLM Models', value: 'Not installed', status: 'warning' })
+  }
+
+  // Training data
+  if (existsSync(dbPath)) {
+    try {
+      const { Database } = await import('bun:sqlite')
+      const db2 = new Database(dbPath, { readonly: true })
+      const hasTable = db2.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='training_data'").get()
+      if (hasTable) {
+        const trainingTotal = (db2.prepare('SELECT COUNT(*) as cnt FROM training_data').get() as any)?.cnt ?? 0
+        if (trainingTotal > 0) {
+          const taskCounts = db2.prepare('SELECT task, COUNT(*) as cnt FROM training_data GROUP BY task').all() as any[]
+          const breakdown = taskCounts.map((r: any) => `${r.cnt} ${r.task}`).join(', ')
+          items.push({
+            label: 'Training Data',
+            value: `${trainingTotal} examples (${breakdown})`,
+            status: 'success'
+          })
+        }
+      }
+      db2.close()
+    } catch {
+      // ignore
+    }
+  }
+
   // Server status
   const port = process.env.CLAUDE_BRAIN_PORT || '3000'
   try {

package/src/config/home.ts

@@ -51,5 +51,6 @@ export function getHomePaths() {
     vault: join(root, 'vault'),
     vaultProjects: join(root, 'vault', 'Projects'),
     vaultGlobal: join(root, 'vault', 'Global'),
+    models: join(root, 'models'),
   }
 }

package/src/config/schema.ts

@@ -356,6 +356,36 @@ export const ConfigSchema = z.object({
     autoRestart: z.boolean().default(true),
   }).default({} as any),
 
+  /** SLM Upgrade: Local model inference configuration */
+  slm: z.object({
+    /** Master switch for local model inference */
+    enabled: z.boolean().default(false),
+    /** Directory containing ONNX model files */
+    modelsDir: z.string().default('~/.claude-brain/models'),
+    /** Minimum model confidence to use model prediction (below → regex fallback) */
+    confidenceThreshold: z.number().min(0).max(1).default(0.7),
+    /** Per-task mode: 'model' uses model only, 'regex' uses regex only, 'both' runs both and logs comparison */
+    tasks: z.object({
+      intent: z.enum(['model', 'regex', 'both']).default('regex'),
+      entity: z.enum(['model', 'regex', 'both']).default('regex'),
+      query: z.enum(['model', 'regex', 'both']).default('regex'),
+      knowledge: z.enum(['model', 'regex', 'both']).default('regex'),
+      compress: z.enum(['model', 'api', 'both']).default('api'),
+      pattern: z.enum(['model', 'regex', 'both']).default('regex'),
+    }).default({} as any),
+    /** Phase 6B: Automated retraining configuration */
+    retrain: z.object({
+      /** Minimum new feedback entries before triggering retrain */
+      minFeedbackCount: z.number().int().min(1).default(100),
+      /** Maximum disagreement rate threshold to trigger retrain */
+      maxDisagreementRate: z.number().min(0).max(1).default(0.15),
+      /** Python executable path */
+      pythonPath: z.string().default('python3'),
+      /** Path to SLM training directory */
+      trainingDir: z.string().default('~/slm-training'),
+    }).default({} as any),
+  }).default({} as any),
+
   /** Phase 30: Optional LLM compression for observations */
   compression: z.object({
     /** Enable LLM-based compression of long observations */