claude-brain 0.15.2 → 0.16.0

package/src/server/handlers/tools/record-correction.ts

@@ -1,125 +1,125 @@
-/**
- * Record Correction Handler
- * Stores corrections and lessons learned in long-term memory
- */
-
-import type { Logger } from 'pino'
-import type { ToolResponse } from '@/tools/types'
-import { getMemoryService, getVaultService, isServicesInitialized } 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 { RecordCorrectionSchema } from './schemas'
-
-export async function handleRecordCorrection(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  try {
-    const input = ToolValidator.validate(args, RecordCorrectionSchema)
-    const {
-      project_name,
-      original,
-      correction,
-      reasoning,
-      context,
-      confidence
-    } = input
-
-    logger.debug(
-      {
-        project_name,
-        original_length: original.length,
-        correction_length: correction.length
-      },
-      'Recording correction'
-    )
-
-    if (!isServicesInitialized()) {
-      throw new McpError(
-        ErrorCode.InternalError,
-        'Memory service not initialized. The server may still be starting up.'
-      )
-    }
-
-    const memory = getMemoryService()
-    const vault = getVaultService()
-
-    const correctionId = await memory.storeCorrection({
-      project: project_name,
-      original,
-      correction,
-      reasoning,
-      context: context || '',
-      confidence: confidence || 0.9
-    })
-
-    const projectPaths = vault.getProjectPaths(project_name)
-
-    try {
-      await vault.writer.appendContent(
-        projectPaths.corrections,
-        `\n## Correction: ${original.substring(0, 50)}...\n\n**ID:** ${correctionId}\n\n**What was wrong:**\n${original}\n\n**Fix:**\n${correction}\n\n**Reasoning:**\n${reasoning}\n\n**Context:** ${context || 'N/A'}\n\n---\n`,
-        '\n'
-      )
-    } catch {
-      throw new McpError(
-        ErrorCode.InvalidParams,
-        `Project "${project_name}" not found. Corrections file does not exist. Use list_projects to see available projects.`
-      )
-    }
-
-    const correctionsFile = await vault.reader.readMarkdownFile(projectPaths.corrections)
-    const currentCount = (correctionsFile.frontmatter?.correction_count as number) || 0
-
-    await vault.writer.updateFrontmatter(projectPaths.corrections, {
-      correction_count: currentCount + 1,
-      last_updated: new Date().toISOString().split('T')[0]
-    })
-
-    logger.info(
-      { projectName: project_name, correctionId },
-      'Correction stored successfully'
-    )
-
-    return ResponseFormatter.text(
-      `✅ Correction recorded successfully!\n\n` +
-      `**ID:** ${correctionId}\n\n` +
-      `**Project:** ${project_name}\n\n` +
-      `**Original:** ${original}\n\n` +
-      `**Correction:** ${correction}\n\n` +
-      `This correction has been:\n` +
-      `- Stored in semantic memory for learning from mistakes\n` +
-      `- Added to ${project_name}/corrections.md in your vault\n` +
-      `- Can be retrieved via \`get_corrections\` or semantic search\n`
-    )
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'record_correction', args })
-
-    // Provide detailed error messages based on error type
-    if (error instanceof Error) {
-      if (error.message.includes('ChromaDB') || error.message.includes('collection')) {
-        throw new McpError(
-          ErrorCode.InternalError,
-          `ChromaDB correction storage failed: ${error.message}\n\n` +
-          `This may be caused by:\n` +
-          `- ChromaDB not running (check if ChromaDB server is started)\n` +
-          `- Collections not initialized (try restarting the MCP server)\n` +
-          `- Connection issues (verify ChromaDB configuration)\n\n` +
-          `Error details: ${error.stack || error.message}`
-        )
-      }
-      if (error.message.includes('not initialized')) {
-        throw new McpError(
-          ErrorCode.InternalError,
-          `Memory service not initialized properly.\n\n` +
-          `The server may still be starting up. Please wait a moment and try again.\n` +
-          `If the issue persists, restart the MCP server.`
-        )
-      }
-    }
-
-    throw error
-  }
-}
+/**
+ * Record Correction Handler
+ * Stores corrections and lessons learned in long-term memory
+ */
+
+import type { Logger } from 'pino'
+import type { ToolResponse } from '@/tools/types'
+import { getMemoryService, getVaultService, isServicesInitialized } 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 { RecordCorrectionSchema } from './schemas'
+
+export async function handleRecordCorrection(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  try {
+    const input = ToolValidator.validate(args, RecordCorrectionSchema)
+    const {
+      project_name,
+      original,
+      correction,
+      reasoning,
+      context,
+      confidence
+    } = input
+
+    logger.debug(
+      {
+        project_name,
+        original_length: original.length,
+        correction_length: correction.length
+      },
+      'Recording correction'
+    )
+
+    if (!isServicesInitialized()) {
+      throw new McpError(
+        ErrorCode.InternalError,
+        'Memory service not initialized. The server may still be starting up.'
+      )
+    }
+
+    const memory = getMemoryService()
+    const vault = getVaultService()
+
+    const correctionId = await memory.storeCorrection({
+      project: project_name,
+      original,
+      correction,
+      reasoning,
+      context: context || '',
+      confidence: confidence || 0.9
+    })
+
+    const projectPaths = vault.getProjectPaths(project_name)
+
+    try {
+      await vault.writer.appendContent(
+        projectPaths.corrections,
+        `\n## Correction: ${original.substring(0, 50)}...\n\n**ID:** ${correctionId}\n\n**What was wrong:**\n${original}\n\n**Fix:**\n${correction}\n\n**Reasoning:**\n${reasoning}\n\n**Context:** ${context || 'N/A'}\n\n---\n`,
+        '\n'
+      )
+    } catch {
+      throw new McpError(
+        ErrorCode.InvalidParams,
+        `Project "${project_name}" not found. Corrections file does not exist. Use list_projects to see available projects.`
+      )
+    }
+
+    const correctionsFile = await vault.reader.readMarkdownFile(projectPaths.corrections)
+    const currentCount = (correctionsFile.frontmatter?.correction_count as number) || 0
+
+    await vault.writer.updateFrontmatter(projectPaths.corrections, {
+      correction_count: currentCount + 1,
+      last_updated: new Date().toISOString().split('T')[0]
+    })
+
+    logger.info(
+      { projectName: project_name, correctionId },
+      'Correction stored successfully'
+    )
+
+    return ResponseFormatter.text(
+      `✅ Correction recorded successfully!\n\n` +
+      `**ID:** ${correctionId}\n\n` +
+      `**Project:** ${project_name}\n\n` +
+      `**Original:** ${original}\n\n` +
+      `**Correction:** ${correction}\n\n` +
+      `This correction has been:\n` +
+      `- Stored in semantic memory for learning from mistakes\n` +
+      `- Added to ${project_name}/corrections.md in your vault\n` +
+      `- Can be retrieved via \`get_corrections\` or semantic search\n`
+    )
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'record_correction', args })
+
+    // Provide detailed error messages based on error type
+    if (error instanceof Error) {
+      if (error.message.includes('ChromaDB') || error.message.includes('collection')) {
+        throw new McpError(
+          ErrorCode.InternalError,
+          `ChromaDB correction storage failed: ${error.message}\n\n` +
+          `This may be caused by:\n` +
+          `- ChromaDB not running (check if ChromaDB server is started)\n` +
+          `- Collections not initialized (try restarting the MCP server)\n` +
+          `- Connection issues (verify ChromaDB configuration)\n\n` +
+          `Error details: ${error.stack || error.message}`
+        )
+      }
+      if (error.message.includes('not initialized')) {
+        throw new McpError(
+          ErrorCode.InternalError,
+          `Memory service not initialized properly.\n\n` +
+          `The server may still be starting up. Please wait a moment and try again.\n` +
+          `If the issue persists, restart the MCP server.`
+        )
+      }
+    }
+
+    throw error
+  }
+}

package/src/server/handlers/tools/remember-decision.ts

@@ -1,153 +1,153 @@
-/**
- * Remember Decision Handler
- * Stores important architectural or technical decisions in long-term memory
- */
-
-import type { Logger } from 'pino'
-import type { ToolResponse } from '@/tools/types'
-import { getMemoryService, getVaultService, isServicesInitialized } 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 { RememberDecisionSchema } from './schemas'
-
-export async function handleRememberDecision(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  logger.debug({ args }, 'Remembering decision - raw args received')
-
-  try {
-    // Handle both old MCP format (project) and new format (project_name)
-    const normalizedArgs = {
-      ...(args as Record<string, unknown>),
-      project_name: (args as any).project_name || (args as any).project
-    }
-
-    const input = ToolValidator.validate(normalizedArgs, RememberDecisionSchema)
-
-    const {
-      project_name,
-      context,
-      decision,
-      reasoning,
-      alternatives_considered,
-      tags
-    } = input
-
-    logger.debug(
-      {
-        project_name,
-        contextLength: context.length,
-        decisionLength: decision.length,
-        tags
-      },
-      'Remembering decision'
-    )
-
-    if (!isServicesInitialized()) {
-      throw new McpError(
-        ErrorCode.InternalError,
-        'Memory service not initialized. The server may still be starting up.'
-      )
-    }
-
-    const memory = getMemoryService()
-    const vault = getVaultService()
-
-    // Use rememberDecision which routes to ChromaDB when enabled
-    const decisionId = await memory.rememberDecision(
-      project_name,
-      context,
-      decision,
-      reasoning,
-      {
-        alternatives: alternatives_considered,
-        tags
-      }
-    )
-
-    const projectPaths = vault.getProjectPaths(project_name)
-
-    try {
-      await vault.reader.readMarkdownFile(projectPaths.decisions)
-    } catch {
-      throw new McpError(
-        ErrorCode.InvalidParams,
-        `Project "${project_name}" not found. Decisions file does not exist. Use list_projects to see available projects.`
-      )
-    }
-
-    const decisionEntry = formatDecisionEntry({
-      id: decisionId,
-      project: project_name,
-      title: decision,
-      context,
-      decision,
-      reasoning,
-      alternatives: alternatives_considered,
-      tags,
-      timestamp: new Date()
-    })
-
-    await vault.writer.appendContent(projectPaths.decisions, decisionEntry, '\n')
-
-    const decisionsFile = await vault.reader.readMarkdownFile(projectPaths.decisions)
-    const currentCount = (decisionsFile.frontmatter.decision_count as number) || 0
-
-    await vault.writer.updateFrontmatter(projectPaths.decisions, {
-      decision_count: currentCount + 1,
-      last_updated: new Date().toISOString().split('T')[0]
-    })
-
-    logger.info({ projectName: project_name, decisionId }, 'Decision stored successfully')
-
-    return ResponseFormatter.text(
-      `✅ Decision stored successfully!\n\n` +
-      `**ID:** ${decisionId}\n` +
-      `**Project:** ${project_name}\n` +
-      `**Decision:** ${decision}\n\n` +
-      `This decision has been:\n` +
-      `- Stored in semantic memory for future recall\n` +
-      `- Added to ${project_name}/decisions.md in your vault\n` +
-      `- Tagged with: ${tags.length > 0 ? tags.join(', ') : 'none'}\n\n` +
-      `You can find similar decisions later using the recall_similar tool.`
-    )
-
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'remember_decision', args })
-
-    if (error instanceof McpError) {
-      throw error
-    }
-
-    throw new McpError(
-      ErrorCode.InternalError,
-      `Failed to store decision: ${error instanceof Error ? error.message : 'Unknown error'}`
-    )
-  }
-}
-
-function formatDecisionEntry(decision: any): string {
-  const date = decision.timestamp.toISOString().split('T')[0]
-
-  let entry = `### Decision: ${decision.title}\n\n`
-  entry += `**Date:** ${date}\n\n`
-  entry += `**Context:** ${decision.context}\n\n`
-  entry += `**Decision:** ${decision.decision}\n\n`
-  entry += `**Reasoning:** ${decision.reasoning}\n\n`
-
-  if (decision.alternatives) {
-    entry += `**Alternatives Considered:** ${decision.alternatives}\n\n`
-  }
-
-  if (decision.tags) {
-    const tags = Array.isArray(decision.tags) 
-      ? decision.tags.join(', ')
-      : decision.tags
-    entry += `**Tags:** ${tags}\n\n`
-  }
-
-  return entry + '---\n\n'
-}
+/**
+ * Remember Decision Handler
+ * Stores important architectural or technical decisions in long-term memory
+ */
+
+import type { Logger } from 'pino'
+import type { ToolResponse } from '@/tools/types'
+import { getMemoryService, getVaultService, isServicesInitialized } 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 { RememberDecisionSchema } from './schemas'
+
+export async function handleRememberDecision(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  logger.debug({ args }, 'Remembering decision - raw args received')
+
+  try {
+    // Handle both old MCP format (project) and new format (project_name)
+    const normalizedArgs = {
+      ...(args as Record<string, unknown>),
+      project_name: (args as any).project_name || (args as any).project
+    }
+
+    const input = ToolValidator.validate(normalizedArgs, RememberDecisionSchema)
+
+    const {
+      project_name,
+      context,
+      decision,
+      reasoning,
+      alternatives_considered,
+      tags
+    } = input
+
+    logger.debug(
+      {
+        project_name,
+        contextLength: context.length,
+        decisionLength: decision.length,
+        tags
+      },
+      'Remembering decision'
+    )
+
+    if (!isServicesInitialized()) {
+      throw new McpError(
+        ErrorCode.InternalError,
+        'Memory service not initialized. The server may still be starting up.'
+      )
+    }
+
+    const memory = getMemoryService()
+    const vault = getVaultService()
+
+    // Use rememberDecision which routes to ChromaDB when enabled
+    const decisionId = await memory.rememberDecision(
+      project_name,
+      context,
+      decision,
+      reasoning,
+      {
+        alternatives: alternatives_considered,
+        tags
+      }
+    )
+
+    const projectPaths = vault.getProjectPaths(project_name)
+
+    try {
+      await vault.reader.readMarkdownFile(projectPaths.decisions)
+    } catch {
+      throw new McpError(
+        ErrorCode.InvalidParams,
+        `Project "${project_name}" not found. Decisions file does not exist. Use list_projects to see available projects.`
+      )
+    }
+
+    const decisionEntry = formatDecisionEntry({
+      id: decisionId,
+      project: project_name,
+      title: decision,
+      context,
+      decision,
+      reasoning,
+      alternatives: alternatives_considered,
+      tags,
+      timestamp: new Date()
+    })
+
+    await vault.writer.appendContent(projectPaths.decisions, decisionEntry, '\n')
+
+    const decisionsFile = await vault.reader.readMarkdownFile(projectPaths.decisions)
+    const currentCount = (decisionsFile.frontmatter.decision_count as number) || 0
+
+    await vault.writer.updateFrontmatter(projectPaths.decisions, {
+      decision_count: currentCount + 1,
+      last_updated: new Date().toISOString().split('T')[0]
+    })
+
+    logger.info({ projectName: project_name, decisionId }, 'Decision stored successfully')
+
+    return ResponseFormatter.text(
+      `✅ Decision stored successfully!\n\n` +
+      `**ID:** ${decisionId}\n` +
+      `**Project:** ${project_name}\n` +
+      `**Decision:** ${decision}\n\n` +
+      `This decision has been:\n` +
+      `- Stored in semantic memory for future recall\n` +
+      `- Added to ${project_name}/decisions.md in your vault\n` +
+      `- Tagged with: ${tags.length > 0 ? tags.join(', ') : 'none'}\n\n` +
+      `You can find similar decisions later using the recall_similar tool.`
+    )
+
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'remember_decision', args })
+
+    if (error instanceof McpError) {
+      throw error
+    }
+
+    throw new McpError(
+      ErrorCode.InternalError,
+      `Failed to store decision: ${error instanceof Error ? error.message : 'Unknown error'}`
+    )
+  }
+}
+
+function formatDecisionEntry(decision: any): string {
+  const date = decision.timestamp.toISOString().split('T')[0]
+
+  let entry = `### Decision: ${decision.title}\n\n`
+  entry += `**Date:** ${date}\n\n`
+  entry += `**Context:** ${decision.context}\n\n`
+  entry += `**Decision:** ${decision.decision}\n\n`
+  entry += `**Reasoning:** ${decision.reasoning}\n\n`
+
+  if (decision.alternatives) {
+    entry += `**Alternatives Considered:** ${decision.alternatives}\n\n`
+  }
+
+  if (decision.tags) {
+    const tags = Array.isArray(decision.tags) 
+      ? decision.tags.join(', ')
+      : decision.tags
+    entry += `**Tags:** ${tags}\n\n`
+  }
+
+  return entry + '---\n\n'
+}