claude-brain 0.3.7 → 0.4.1

package/src/server/handlers/tools/update-progress.ts

@@ -50,6 +50,18 @@ export async function handleUpdateProgress(
       completedAt: new Date()
     })
 
+    // Calculate and update completion percentage
+    const progressState = await context.progress.getProgress(project_name)
+    const totalTasks = progressState.completedTasks.length + progressState.currentTasks.length
+    const completionPercentage = totalTasks > 0
+      ? Math.round((progressState.completedTasks.length / totalTasks) * 100)
+      : 0
+
+    await context.progress.updateProgress(project_name, {
+      completionPercentage,
+      currentPhase: completionPercentage >= 100 ? 'complete' : 'active'
+    })
+
     const progressFile = await vault.reader.readMarkdownFile(projectPaths.progress)
     const updatedContent = updateNextStepsSection(progressFile.content, next_steps)
 

package/src/server/handlers/tools/what-if-analysis.ts

@@ -23,6 +23,17 @@ export async function handleWhatIfAnalysis(
     const { change, project_name, max_results } = input
 
     const memory = getMemoryService()
+
+    if (!memory.isChromaDBEnabled()) {
+      return ResponseFormatter.text(
+        `## What-If: "${change}"\n\n` +
+        'Unable to perform what-if analysis.\n\n' +
+        'Note: ChromaDB is not connected. What-if analysis requires ChromaDB for semantic search across decisions. ' +
+        'Decisions stored via SQLite fallback are available through recall_similar. ' +
+        'To enable what-if analysis, start a ChromaDB server or configure persistent mode.'
+      )
+    }
+
     const kgService = getKnowledgeGraphService()
     const graph = kgService?.graph || null
 

package/src/server/services.ts

@@ -112,9 +112,9 @@ export async function initializeServices(config: Config, logger: Logger): Promis
     await phase12.initialize()
     serviceLogger.info('Phase 12 service initialized')
 
-    // Initialize Retrieval Service (Phase 13)
+    // Initialize Retrieval Service (Phase 13) — requires ChromaDB
     let retrieval: RetrievalService | null = null
-    if (config.retrieval?.feedback?.enabled || config.retrieval?.enabled) {
+    if ((config.retrieval?.feedback?.enabled || config.retrieval?.enabled) && memory.isChromaDBEnabled()) {
       retrieval = new RetrievalService(
         logger,
         memory.chroma.collections,
@@ -123,6 +123,8 @@ export async function initializeServices(config: Config, logger: Logger): Promis
       )
       await retrieval.initialize()
       serviceLogger.info('Retrieval service initialized')
+    } else if (config.retrieval?.enabled && !memory.isChromaDBEnabled()) {
+      serviceLogger.warn('Retrieval service requires ChromaDB, skipping initialization')
     }
 
     // Initialize Knowledge Graph Service (Phase 14)
@@ -146,18 +148,21 @@ export async function initializeServices(config: Config, logger: Logger): Promis
 
         knowledgeGraph = { graph, search, builder, linker }
 
-        // Migrate existing decisions if graph is empty
-        if (graph.getNodeCount() === 0) {
+        // Migrate existing decisions if graph is empty (requires ChromaDB)
+        if (graph.getNodeCount() === 0 && memory.isChromaDBEnabled()) {
           serviceLogger.info('Empty graph detected, migrating existing decisions...')
           const migrationResult = await builder.migrateExistingDecisions(memory.chroma.collections)
           serviceLogger.info(
             { processed: migrationResult.processed, errors: migrationResult.errors },
             'Initial graph migration complete'
           )
+        } else if (graph.getNodeCount() === 0) {
+          serviceLogger.info('Empty graph detected, but ChromaDB unavailable — graph will populate as new decisions are stored')
         }
 
         // Hook builder into decision storage for real-time graph population
-        memory.chroma.store.addDecisionStoredListener((input) => {
+        // Uses MemoryManager listener so it works for both ChromaDB and SQLite paths
+        memory.addDecisionStoredListener((input) => {
           builder.processDecision(input)
         })
 
@@ -170,9 +175,9 @@ export async function initializeServices(config: Config, logger: Logger): Promis
       }
     }
 
-    // Initialize Episode Manager (Phase 14)
+    // Initialize Episode Manager (Phase 14) — requires ChromaDB
     let episodeManager: EpisodeManager | null = null
-    if (config.knowledge?.episodic?.enabled !== false) {
+    if (config.knowledge?.episodic?.enabled !== false && memory.isChromaDBEnabled()) {
       try {
         episodeManager = new EpisodeManager(
           logger,
@@ -186,10 +191,10 @@ export async function initializeServices(config: Config, logger: Logger): Promis
       }
     }
 
-    // Initialize Semantic Cache & Precompute (Phase 15)
+    // Initialize Semantic Cache & Precompute (Phase 15) — requires ChromaDB
     let semanticCache: SemanticCache | null = null
     let precompute: PrecomputeEngine | null = null
-    if (config.advancedIntelligence?.enabled !== false && config.advancedIntelligence?.cache?.enabled !== false) {
+    if (config.advancedIntelligence?.enabled !== false && config.advancedIntelligence?.cache?.enabled !== false && memory.isChromaDBEnabled()) {
       try {
         const cacheConfig = config.advancedIntelligence?.cache || {}
         semanticCache = new SemanticCache(logger, {

package/src/setup/wizard.ts

@@ -1,6 +1,7 @@
 import prompts from 'prompts'
 import fs from 'fs/promises'
 import { existsSync } from 'fs'
+import { execSync } from 'child_process'
 import path from 'path'
 import os from 'os'
 import { fileURLToPath } from 'url'
@@ -21,6 +22,7 @@ export interface SetupAnswers {
   vaultPath: string
   logLevel: string
   enableFileWatch: boolean
+  enableChromaDB: boolean
   createSampleProject: boolean
   installClaudeMd: boolean
 }
@@ -34,7 +36,7 @@ export class SetupWizard {
 
   async run(): Promise<SetupAnswers> {
     // Step 1: Detect Vaults
-    console.log(stepIndicator(1, 5, 'Detecting Obsidian Vaults'))
+    console.log(stepIndicator(1, 6, 'Detecting Obsidian Vaults'))
     await transition()
 
     const suggestedPaths = await withSpinner('Scanning for Obsidian vaults', () =>
@@ -49,7 +51,7 @@ export class SetupWizard {
     console.log()
 
     // Step 2: Vault Configuration
-    console.log(stepIndicator(2, 5, 'Vault Configuration'))
+    console.log(stepIndicator(2, 6, 'Vault Configuration'))
     await transition()
 
     const vaultAnswers = await prompts([
@@ -86,7 +88,7 @@ export class SetupWizard {
     const finalVaultPath = vaultAnswers.vaultPath || vaultAnswers.vaultPathChoice
 
     // Step 3: Logging
-    console.log(stepIndicator(3, 5, 'Logging Configuration'))
+    console.log(stepIndicator(3, 6, 'Logging Configuration'))
     await transition()
 
     const loggingAnswers = await prompts({
@@ -103,7 +105,7 @@ export class SetupWizard {
     })
 
     // Step 4: Features
-    console.log(stepIndicator(4, 5, 'Feature Selection'))
+    console.log(stepIndicator(4, 6, 'Feature Selection'))
     await transition()
 
     const featureAnswers = await prompts([
@@ -127,14 +129,40 @@ export class SetupWizard {
       }
     ])
 
-    // Step 5: Review
-    console.log(stepIndicator(5, 5, 'Review Configuration'))
+    // Step 5: ChromaDB
+    console.log(stepIndicator(5, 6, 'ChromaDB (Vector Database)'))
+    await transition()
+
+    console.log(dimText('ChromaDB enables semantic search, knowledge graph,'))
+    console.log(dimText('and advanced intelligence features. Requires Python 3.'))
+    console.log()
+
+    const chromaInstalled = this.isChromaInstalled()
+    if (chromaInstalled) {
+      console.log(successText('ChromaDB CLI detected'))
+    } else {
+      console.log(warningText('ChromaDB CLI not found — you can install it later'))
+    }
+    console.log()
+
+    const chromaAnswers = await prompts({
+      type: 'confirm',
+      name: 'enableChromaDB',
+      message: chromaInstalled
+        ? 'Enable ChromaDB for advanced features?'
+        : 'Enable ChromaDB? (install later: pip install chromadb)',
+      initial: chromaInstalled
+    })
+
+    // Step 6: Review
+    console.log(stepIndicator(6, 6, 'Review Configuration'))
     await transition()
 
     const answers: SetupAnswers = {
       vaultPath: finalVaultPath,
       logLevel: loggingAnswers.logLevel ?? 'warn',
       enableFileWatch: featureAnswers.enableFileWatch ?? true,
+      enableChromaDB: chromaAnswers.enableChromaDB ?? false,
       createSampleProject: featureAnswers.createSampleProject ?? true,
       installClaudeMd: featureAnswers.installClaudeMd ?? true,
     }
@@ -143,6 +171,7 @@ export class SetupWizard {
       { label: 'Vault Path', value: answers.vaultPath, status: 'success' },
       { label: 'Log Level', value: answers.logLevel, status: 'info' },
       { label: 'File Watching', value: answers.enableFileWatch ? 'Enabled' : 'Disabled', status: answers.enableFileWatch ? 'success' : 'warning' },
+      { label: 'ChromaDB', value: answers.enableChromaDB ? 'Enabled' : 'Disabled', status: answers.enableChromaDB ? 'success' : 'warning' },
       { label: 'Sample Project', value: answers.createSampleProject ? 'Yes' : 'No', status: 'info' },
       { label: 'Install CLAUDE.md', value: answers.installClaudeMd ? 'Yes' : 'No', status: 'info' },
     ]))
@@ -191,11 +220,31 @@ export class SetupWizard {
     return locations
   }
 
+  private isChromaInstalled(): boolean {
+    try {
+      execSync('chroma --version', { stdio: 'pipe', timeout: 5000 })
+      return true
+    } catch {
+      return false
+    }
+  }
+
   async applyConfiguration(answers: SetupAnswers): Promise<void> {
     console.log('\n' + heading('Applying configuration...') + '\n')
 
     const homePaths = getHomePaths()
 
+    const chromaEnvLines = answers.enableChromaDB
+      ? [
+          '',
+          '# ChromaDB Configuration',
+          'CHROMA_MODE=client-server',
+          'CHROMA_HOST=localhost',
+          'CHROMA_PORT=8000',
+          'CHROMA_EMBEDDING_PROVIDER=transformers',
+        ].join('\n')
+      : ''
+
     const envContent = `# Claude Brain Configuration
 VAULT_PATH=${answers.vaultPath}
 LOG_LEVEL=${answers.logLevel}
@@ -203,6 +252,7 @@ ENABLE_FILE_WATCH=${answers.enableFileWatch}
 DB_PATH=./data/memory.db
 LOG_FILE_PATH=./logs/claude-brain.log
 SERVER_NAME=claude-brain
+${chromaEnvLines}
 `
 
     await withSpinner('Writing .env configuration', async () => {
@@ -211,6 +261,7 @@ SERVER_NAME=claude-brain
 
     await withSpinner('Creating data and log directories', async () => {
       await fs.mkdir(homePaths.data, { recursive: true })
+      await fs.mkdir(homePaths.chroma, { recursive: true })
       await fs.mkdir(homePaths.logs, { recursive: true })
     })
 
@@ -229,16 +280,35 @@ SERVER_NAME=claude-brain
       }
     }
 
-    console.log()
-    console.log(box([
+    const nextSteps: string[] = [
       heading('Setup complete!'),
       '',
       dimText('Next steps:'),
       `  ${theme.primary('1.')} ${theme.bold('claude-brain install')}`,
       `     ${dimText('Register as MCP server')}`,
-      `  ${theme.primary('2.')} ${theme.bold('claude-brain health')}`,
-      `     ${dimText('Verify everything works')}`,
-    ].join('\n'), 'Done'))
+    ]
+
+    if (answers.enableChromaDB) {
+      nextSteps.push(
+        `  ${theme.primary('2.')} ${theme.bold('claude-brain chroma install')}`,
+        `     ${dimText('Install ChromaDB (if not already installed)')}`,
+        `  ${theme.primary('3.')} ${theme.bold('claude-brain chroma start')}`,
+        `     ${dimText('Start the ChromaDB server')}`,
+        `  ${theme.primary('4.')} ${theme.bold('claude-brain health')}`,
+        `     ${dimText('Verify everything works')}`,
+      )
+    } else {
+      nextSteps.push(
+        `  ${theme.primary('2.')} ${theme.bold('claude-brain health')}`,
+        `     ${dimText('Verify everything works')}`,
+        '',
+        dimText('Optional: Enable ChromaDB later for advanced features:'),
+        `  ${dimText('claude-brain chroma install && claude-brain chroma start')}`,
+      )
+    }
+
+    console.log()
+    console.log(box(nextSteps.join('\n'), 'Done'))
     console.log()
   }