claude-brain 0.15.2 → 0.17.0

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

@@ -1,131 +1,131 @@
-/**
- * Update Progress Handler
- * Updates project progress tracking with completed tasks and next steps
- */
-
-import type { Logger } from 'pino'
-import type { ToolResponse } from '@/tools/types'
-import { getVaultService, getContextService } from '@/server/services'
-import { ToolValidator } from '@/server/utils/validators'
-import { ResponseFormatter } from '@/server/utils/response-formatter'
-import { ErrorHandler } from '@/server/utils/error-handler'
-import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
-import { UpdateProgressSchema } from './schemas'
-
-export async function handleUpdateProgress(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  try {
-    const input = ToolValidator.validate(args, UpdateProgressSchema)
-    const { project_name, completed_task, next_steps, notes } = input
-
-    logger.debug({
-      project_name,
-      completed_task,
-      next_steps,
-      hasNotes: !!notes
-    }, 'Updating progress')
-
-    const vault = getVaultService()
-    const context = getContextService()
-
-    const projectPaths = vault.getProjectPaths(project_name)
-
-    try {
-      const file = await vault.reader.readMarkdownFile(projectPaths.progress)
-    } catch {
-      throw new McpError(
-        ErrorCode.InvalidParams,
-        `Project "${project_name}" not found. Progress file does not exist. Use list_projects to see available projects.`
-      )
-    }
-
-    const taskId = generateTaskId(completed_task)
-
-    // Step 1: Append completed task to progress file
-    await context.progress.addCompletedTask(project_name, {
-      id: taskId,
-      title: completed_task,
-      status: 'done',
-      completedAt: new Date()
-    })
-
-    // Step 2: Read file bypassing cache, update next steps, write content
-    const progressFile = await vault.reader.readMarkdownFile(projectPaths.progress, false)
-    const updatedContent = updateNextStepsSection(progressFile.content, next_steps)
-
-    await vault.writer.writeMarkdownFile(
-      projectPaths.progress,
-      progressFile.frontmatter,
-      updatedContent,
-      true
-    )
-
-    // Step 3: Notes appended if present
-    if (notes) {
-      const notesEntry = `\n## Notes (${new Date().toLocaleDateString()})\n${notes}\n`
-      await vault.writer.appendContent(projectPaths.progress, notesEntry)
-    }
-
-    // Step 4: Calculate and update frontmatter LAST (so nothing overwrites it)
-    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'
-    })
-
-    // Step 5: Invalidate cache so subsequent reads get fresh data
-    vault.reader.invalidateCache(projectPaths.progress)
-    context.invalidateContext(project_name)
-
-    logger.info({ projectName: project_name, taskId }, 'Progress updated successfully')
-
-    return ResponseFormatter.text(
-      `✅ Progress updated for ${project_name}\n\n` +
-      `**Completed:** ${completed_task}\n` +
-      `**Next Steps:** ${next_steps}\n` +
-      (notes ? `**Notes:** ${notes}\n` : '') +
-      `\nThe progress has been saved to your vault and context cache has been refreshed.`
-    )
-
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'update_progress', args })
-
-    if (error instanceof McpError) {
-      throw error
-    }
-
-    throw new McpError(
-      ErrorCode.InternalError,
-      `Failed to update progress: ${error instanceof Error ? error.message : 'Unknown error'}`
-    )
-  }
-}
-
-function generateTaskId(title: string): string {
-  return title
-    .toLowerCase()
-    .replace(/[^a-z0-9]+/g, '-')
-    .replace(/^-+|-+$/g, '')
-    .slice(0, 50)
-}
-
-function updateNextStepsSection(content: string, nextSteps: string): string {
-  const nextStepsRegex = /##\s+Next Steps\s+(.+?)(?=\n##|$)/s
-  const match = nextStepsRegex.exec(content)
-
-  if (match) {
-    return content.replace(
-      nextStepsRegex,
-      `## Next Steps\n\n- ${nextSteps}\n`
-    )
-  } else {
-    return content + `\n\n## Next Steps\n\n- ${nextSteps}\n`
-  }
-}
+/**
+ * Update Progress Handler
+ * Updates project progress tracking with completed tasks and next steps
+ */
+
+import type { Logger } from 'pino'
+import type { ToolResponse } from '@/tools/types'
+import { getVaultService, getContextService } from '@/server/services'
+import { ToolValidator } from '@/server/utils/validators'
+import { ResponseFormatter } from '@/server/utils/response-formatter'
+import { ErrorHandler } from '@/server/utils/error-handler'
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
+import { UpdateProgressSchema } from './schemas'
+
+export async function handleUpdateProgress(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  try {
+    const input = ToolValidator.validate(args, UpdateProgressSchema)
+    const { project_name, completed_task, next_steps, notes } = input
+
+    logger.debug({
+      project_name,
+      completed_task,
+      next_steps,
+      hasNotes: !!notes
+    }, 'Updating progress')
+
+    const vault = getVaultService()
+    const context = getContextService()
+
+    const projectPaths = vault.getProjectPaths(project_name)
+
+    try {
+      const file = await vault.reader.readMarkdownFile(projectPaths.progress)
+    } catch {
+      throw new McpError(
+        ErrorCode.InvalidParams,
+        `Project "${project_name}" not found. Progress file does not exist. Use list_projects to see available projects.`
+      )
+    }
+
+    const taskId = generateTaskId(completed_task)
+
+    // Step 1: Append completed task to progress file
+    await context.progress.addCompletedTask(project_name, {
+      id: taskId,
+      title: completed_task,
+      status: 'done',
+      completedAt: new Date()
+    })
+
+    // Step 2: Read file bypassing cache, update next steps, write content
+    const progressFile = await vault.reader.readMarkdownFile(projectPaths.progress, false)
+    const updatedContent = updateNextStepsSection(progressFile.content, next_steps)
+
+    await vault.writer.writeMarkdownFile(
+      projectPaths.progress,
+      progressFile.frontmatter,
+      updatedContent,
+      true
+    )
+
+    // Step 3: Notes appended if present
+    if (notes) {
+      const notesEntry = `\n## Notes (${new Date().toLocaleDateString()})\n${notes}\n`
+      await vault.writer.appendContent(projectPaths.progress, notesEntry)
+    }
+
+    // Step 4: Calculate and update frontmatter LAST (so nothing overwrites it)
+    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'
+    })
+
+    // Step 5: Invalidate cache so subsequent reads get fresh data
+    vault.reader.invalidateCache(projectPaths.progress)
+    context.invalidateContext(project_name)
+
+    logger.info({ projectName: project_name, taskId }, 'Progress updated successfully')
+
+    return ResponseFormatter.text(
+      `✅ Progress updated for ${project_name}\n\n` +
+      `**Completed:** ${completed_task}\n` +
+      `**Next Steps:** ${next_steps}\n` +
+      (notes ? `**Notes:** ${notes}\n` : '') +
+      `\nThe progress has been saved to your vault and context cache has been refreshed.`
+    )
+
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'update_progress', args })
+
+    if (error instanceof McpError) {
+      throw error
+    }
+
+    throw new McpError(
+      ErrorCode.InternalError,
+      `Failed to update progress: ${error instanceof Error ? error.message : 'Unknown error'}`
+    )
+  }
+}
+
+function generateTaskId(title: string): string {
+  return title
+    .toLowerCase()
+    .replace(/[^a-z0-9]+/g, '-')
+    .replace(/^-+|-+$/g, '')
+    .slice(0, 50)
+}
+
+function updateNextStepsSection(content: string, nextSteps: string): string {
+  const nextStepsRegex = /##\s+Next Steps\s+(.+?)(?=\n##|$)/s
+  const match = nextStepsRegex.exec(content)
+
+  if (match) {
+    return content.replace(
+      nextStepsRegex,
+      `## Next Steps\n\n- ${nextSteps}\n`
+    )
+  } else {
+    return content + `\n\n## Next Steps\n\n- ${nextSteps}\n`
+  }
+}

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

@@ -1,135 +1,135 @@
-/**
- * What-If Analysis Handler
- * Counterfactual analysis using knowledge graph and decision history
- */
-
-import type { Logger } from 'pino'
-import type { ToolResponse } from '@/tools/types'
-import { getMemoryService, getKnowledgeGraphService } from '@/server/services'
-import { ToolValidator } from '@/server/utils/validators'
-import { ResponseFormatter } from '@/server/utils/response-formatter'
-import { withMemoryIndicator } from '@/server/utils/memory-indicator'
-import { ErrorHandler } from '@/server/utils/error-handler'
-import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
-import { WhatIfAnalysisSchema } from './schemas'
-import { CounterfactualAnalyzer } from '@/intelligence/reasoning/counterfactual'
-
-export async function handleWhatIfAnalysis(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  try {
-    const input = ToolValidator.validate(args, WhatIfAnalysisSchema)
-    const { change, project_name, max_results } = input
-
-    const memory = getMemoryService()
-
-    if (!memory.isChromaDBEnabled()) {
-      // SQLite fallback: use searchRaw for semantic search + knowledge graph if available
-      const results = await memory.searchRaw(change, {
-        project: project_name,
-        limit: max_results || 10,
-        minSimilarity: 0.2
-      })
-
-      const parts: string[] = []
-      parts.push(`## What-If: "${change}" (SQLite)`)
-      parts.push('')
-
-      if (results.length === 0) {
-        parts.push('No related decisions found to assess impact.')
-        return ResponseFormatter.text(parts.join('\n'))
-      }
-
-      // Assess impact based on similarity
-      const affectedDecisions = results.map(r => {
-        const similarity = r.similarity || 0
-        const impact = similarity > 0.7 ? 'high' : similarity > 0.4 ? 'medium' : 'low'
-        return {
-          decision: r.decision?.decision || r.memory?.content?.slice(0, 150) || '',
-          impact,
-          similarity,
-          reason: `${Math.round(similarity * 100)}% similarity — ${r.decision?.reasoning?.slice(0, 100) || 'related context found'}`
-        }
-      })
-
-      parts.push(`Analyzed ${results.length} related decisions.\n`)
-
-      parts.push('### Affected Decisions')
-      for (const d of affectedDecisions) {
-        const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
-        parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
-        parts.push(`  *${d.reason}*`)
-      }
-
-      // Try knowledge graph if available
-      try {
-        const kgService = getKnowledgeGraphService()
-        if (kgService?.graph) {
-          const graphResults = await kgService.graph.search(change, { limit: 5 })
-          if (graphResults && graphResults.length > 0) {
-            const technologies = graphResults
-              .filter((n: any) => n.type === 'technology')
-              .map((n: any) => n.name || n.label)
-            if (technologies.length > 0) {
-              parts.push(`\n### Connected Technologies`)
-              parts.push(technologies.map((t: string) => `- ${t}`).join('\n'))
-            }
-          }
-        }
-      } catch {
-        // Knowledge graph not available, skip
-      }
-
-      const content = parts.join('\n')
-      return ResponseFormatter.text(withMemoryIndicator(content, results.length))
-    }
-
-    const kgService = getKnowledgeGraphService()
-    const graph = kgService?.graph || null
-
-    const analyzer = new CounterfactualAnalyzer(
-      logger,
-      graph,
-      memory.chroma.collections,
-      memory.chroma.embeddings
-    )
-
-    const scenario = await analyzer.whatIf(change, {
-      project: project_name,
-      maxResults: max_results
-    })
-
-    const parts: string[] = []
-    parts.push(scenario.summary)
-    parts.push('')
-
-    if (scenario.affectedDecisions.length > 0) {
-      parts.push('### Affected Decisions')
-      for (const d of scenario.affectedDecisions.slice(0, 10)) {
-        const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
-        parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
-        parts.push(`  *${d.reason}*`)
-      }
-    }
-
-    if (scenario.affectedTechnologies.length > 0) {
-      parts.push(`\n### Connected Technologies`)
-      parts.push(scenario.affectedTechnologies.map(t => `- ${t}`).join('\n'))
-    }
-
-    const content = parts.join('\n')
-    const totalAffected = scenario.affectedDecisions.length
-
-    if (totalAffected > 0) {
-      return ResponseFormatter.text(withMemoryIndicator(content, totalAffected))
-    }
-
-    return ResponseFormatter.text(content)
-
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'what_if_analysis' })
-    if (error instanceof McpError) throw error
-    throw new McpError(ErrorCode.InternalError, `What-if analysis failed: ${error instanceof Error ? error.message : 'Unknown error'}`)
-  }
-}
+/**
+ * What-If Analysis Handler
+ * Counterfactual analysis using knowledge graph and decision history
+ */
+
+import type { Logger } from 'pino'
+import type { ToolResponse } from '@/tools/types'
+import { getMemoryService, getKnowledgeGraphService } from '@/server/services'
+import { ToolValidator } from '@/server/utils/validators'
+import { ResponseFormatter } from '@/server/utils/response-formatter'
+import { withMemoryIndicator } from '@/server/utils/memory-indicator'
+import { ErrorHandler } from '@/server/utils/error-handler'
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
+import { WhatIfAnalysisSchema } from './schemas'
+import { CounterfactualAnalyzer } from '@/intelligence/reasoning/counterfactual'
+
+export async function handleWhatIfAnalysis(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  try {
+    const input = ToolValidator.validate(args, WhatIfAnalysisSchema)
+    const { change, project_name, max_results } = input
+
+    const memory = getMemoryService()
+
+    if (!memory.isChromaDBEnabled()) {
+      // SQLite fallback: use searchRaw for semantic search + knowledge graph if available
+      const results = await memory.searchRaw(change, {
+        project: project_name,
+        limit: max_results || 10,
+        minSimilarity: 0.2
+      })
+
+      const parts: string[] = []
+      parts.push(`## What-If: "${change}" (SQLite)`)
+      parts.push('')
+
+      if (results.length === 0) {
+        parts.push('No related decisions found to assess impact.')
+        return ResponseFormatter.text(parts.join('\n'))
+      }
+
+      // Assess impact based on similarity
+      const affectedDecisions = results.map(r => {
+        const similarity = r.similarity || 0
+        const impact = similarity > 0.7 ? 'high' : similarity > 0.4 ? 'medium' : 'low'
+        return {
+          decision: r.decision?.decision || r.memory?.content?.slice(0, 150) || '',
+          impact,
+          similarity,
+          reason: `${Math.round(similarity * 100)}% similarity — ${r.decision?.reasoning?.slice(0, 100) || 'related context found'}`
+        }
+      })
+
+      parts.push(`Analyzed ${results.length} related decisions.\n`)
+
+      parts.push('### Affected Decisions')
+      for (const d of affectedDecisions) {
+        const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
+        parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
+        parts.push(`  *${d.reason}*`)
+      }
+
+      // Try knowledge graph if available
+      try {
+        const kgService = getKnowledgeGraphService()
+        if (kgService?.graph) {
+          const graphResults = await kgService.graph.search(change, { limit: 5 })
+          if (graphResults && graphResults.length > 0) {
+            const technologies = graphResults
+              .filter((n: any) => n.type === 'technology')
+              .map((n: any) => n.name || n.label)
+            if (technologies.length > 0) {
+              parts.push(`\n### Connected Technologies`)
+              parts.push(technologies.map((t: string) => `- ${t}`).join('\n'))
+            }
+          }
+        }
+      } catch {
+        // Knowledge graph not available, skip
+      }
+
+      const content = parts.join('\n')
+      return ResponseFormatter.text(withMemoryIndicator(content, results.length))
+    }
+
+    const kgService = getKnowledgeGraphService()
+    const graph = kgService?.graph || null
+
+    const analyzer = new CounterfactualAnalyzer(
+      logger,
+      graph,
+      memory.chroma.collections,
+      memory.chroma.embeddings
+    )
+
+    const scenario = await analyzer.whatIf(change, {
+      project: project_name,
+      maxResults: max_results
+    })
+
+    const parts: string[] = []
+    parts.push(scenario.summary)
+    parts.push('')
+
+    if (scenario.affectedDecisions.length > 0) {
+      parts.push('### Affected Decisions')
+      for (const d of scenario.affectedDecisions.slice(0, 10)) {
+        const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
+        parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
+        parts.push(`  *${d.reason}*`)
+      }
+    }
+
+    if (scenario.affectedTechnologies.length > 0) {
+      parts.push(`\n### Connected Technologies`)
+      parts.push(scenario.affectedTechnologies.map(t => `- ${t}`).join('\n'))
+    }
+
+    const content = parts.join('\n')
+    const totalAffected = scenario.affectedDecisions.length
+
+    if (totalAffected > 0) {
+      return ResponseFormatter.text(withMemoryIndicator(content, totalAffected))
+    }
+
+    return ResponseFormatter.text(content)
+
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'what_if_analysis' })
+    if (error instanceof McpError) throw error
+    throw new McpError(ErrorCode.InternalError, `What-if analysis failed: ${error instanceof Error ? error.message : 'Unknown error'}`)
+  }
+}