claude-brain 0.14.2 → 0.14.4

package/src/server/handlers/tools/get-corrections.ts

@@ -1,154 +1,154 @@
-/**
- * Get Corrections Handler
- * Retrieves corrections and lessons learned from ChromaDB semantic memory
- */
-
-import type { Logger } from 'pino'
-import type { GetCorrectionsInput, 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 { withMemoryIndicator, formatMemoryStats } from '@/server/utils/memory-indicator'
-import { ErrorHandler } from '@/server/utils/error-handler'
-import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
-
-export async function handleGetCorrections(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  try {
-    const input = args as GetCorrectionsInput
-    const {
-      project_name,
-      limit = 10,
-      query
-    } = input
-
-    logger.debug(
-      { project_name, limit, has_query: !!query },
-      'Getting corrections'
-    )
-
-    if (!isServicesInitialized()) {
-      throw new McpError(
-        ErrorCode.InternalError,
-        'Memory service not initialized. The server may still be starting up.'
-      )
-    }
-
-    const memory = getMemoryService()
-    const vault = getVaultService()
-
-    let results: any[]
-
-    if (query) {
-      results = await memory.searchCorrections(query, {
-        project: project_name,
-        limit,
-        minSimilarity: 0.5
-      })
-    } else {
-      results = await memory.getCorrections(project_name, {
-        limit
-      })
-    }
-
-    if (results.length === 0) {
-      const message = project_name
-        ? `No corrections found for project "${project_name}".`
-        : 'No corrections found.'
-
-      logger.warn({ project_name }, 'No corrections found')
-
-      return ResponseFormatter.text(
-        `No corrections found.\n\n${message}\n\nTips:\n` +
-        `- Use \`record_correction\` to document mistakes and lessons learned\n` +
-        `- Review past corrections to avoid repeating mistakes\n` +
-        `- Check project documentation for best practices\n`
-      )
-    }
-
-    logger.info(
-      { project_name, count: results.length },
-      'Corrections retrieved successfully'
-    )
-
-    const projectInfo = project_name ? ` for **${project_name}**` : ''
-
-    let response = `Found **${results.length} correction(s)${projectInfo}:\n\n`
-
-    results.forEach((correction: any, index: number) => {
-      const num = index + 1
-      const metadata = correction.metadata
-
-      response += `## ${num}. What went wrong\n\n`
-      response += `**ID:** ${correction.id}\n`
-      response += `**Confidence:** ${metadata.confidence.toFixed(2)}\n`
-      response += `**Project:** ${metadata.project}\n\n`
-      response += `**Original:** ${metadata.original}\n\n`
-      response += `**Correction:** ${metadata.correction}\n\n`
-
-      if (metadata.reasoning) {
-        response += `**Reasoning:** ${metadata.reasoning}\n\n`
-      }
-
-      if (metadata.context) {
-        response += `**Context:** ${metadata.context}\n\n`
-      }
-
-      response += `**Recorded:** ${new Date(metadata.created_at).toLocaleDateString()}\n\n`
-
-      if (index < results.length) {
-        response += `---\n\n`
-      }
-    })
-
-    if (results.length === 0) {
-      response += `No corrections documented yet.\n\n` +
-        `Use \`record_correction\` to document mistakes and lessons learned.\n`
-    }
-
-    response += `\n**Next Steps:**\n`
-    response += `- Learn from these corrections to avoid mistakes\n`
-    response += `- Document new lessons as you discover them\n`
-
-    // Add memory indicator to show Claude Brain helped
-    const statsLine = formatMemoryStats({ corrections: results.length })
-    return ResponseFormatter.text(withMemoryIndicator(response, results.length) + '\n\n' + statsLine)
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'get_corrections', 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 retrieval 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.`
-        )
-      }
-      if (error.message.includes('requires ChromaDB')) {
-        throw new McpError(
-          ErrorCode.InternalError,
-          `Correction retrieval requires ChromaDB backend.\n\n` +
-          `ChromaDB may not be enabled or may have failed to initialize.\n` +
-          `Check your configuration and ensure ChromaDB is running.`
-        )
-      }
-    }
-
-    throw error
-  }
-}
+/**
+ * Get Corrections Handler
+ * Retrieves corrections and lessons learned from ChromaDB semantic memory
+ */
+
+import type { Logger } from 'pino'
+import type { GetCorrectionsInput, 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 { withMemoryIndicator, formatMemoryStats } from '@/server/utils/memory-indicator'
+import { ErrorHandler } from '@/server/utils/error-handler'
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
+
+export async function handleGetCorrections(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  try {
+    const input = args as GetCorrectionsInput
+    const {
+      project_name,
+      limit = 10,
+      query
+    } = input
+
+    logger.debug(
+      { project_name, limit, has_query: !!query },
+      'Getting corrections'
+    )
+
+    if (!isServicesInitialized()) {
+      throw new McpError(
+        ErrorCode.InternalError,
+        'Memory service not initialized. The server may still be starting up.'
+      )
+    }
+
+    const memory = getMemoryService()
+    const vault = getVaultService()
+
+    let results: any[]
+
+    if (query) {
+      results = await memory.searchCorrections(query, {
+        project: project_name,
+        limit,
+        minSimilarity: 0.5
+      })
+    } else {
+      results = await memory.getCorrections(project_name, {
+        limit
+      })
+    }
+
+    if (results.length === 0) {
+      const message = project_name
+        ? `No corrections found for project "${project_name}".`
+        : 'No corrections found.'
+
+      logger.warn({ project_name }, 'No corrections found')
+
+      return ResponseFormatter.text(
+        `No corrections found.\n\n${message}\n\nTips:\n` +
+        `- Use \`record_correction\` to document mistakes and lessons learned\n` +
+        `- Review past corrections to avoid repeating mistakes\n` +
+        `- Check project documentation for best practices\n`
+      )
+    }
+
+    logger.info(
+      { project_name, count: results.length },
+      'Corrections retrieved successfully'
+    )
+
+    const projectInfo = project_name ? ` for **${project_name}**` : ''
+
+    let response = `Found **${results.length} correction(s)${projectInfo}:\n\n`
+
+    results.forEach((correction: any, index: number) => {
+      const num = index + 1
+      const metadata = correction.metadata
+
+      response += `## ${num}. What went wrong\n\n`
+      response += `**ID:** ${correction.id}\n`
+      response += `**Confidence:** ${metadata.confidence.toFixed(2)}\n`
+      response += `**Project:** ${metadata.project}\n\n`
+      response += `**Original:** ${metadata.original}\n\n`
+      response += `**Correction:** ${metadata.correction}\n\n`
+
+      if (metadata.reasoning) {
+        response += `**Reasoning:** ${metadata.reasoning}\n\n`
+      }
+
+      if (metadata.context) {
+        response += `**Context:** ${metadata.context}\n\n`
+      }
+
+      response += `**Recorded:** ${new Date(metadata.created_at).toLocaleDateString()}\n\n`
+
+      if (index < results.length) {
+        response += `---\n\n`
+      }
+    })
+
+    if (results.length === 0) {
+      response += `No corrections documented yet.\n\n` +
+        `Use \`record_correction\` to document mistakes and lessons learned.\n`
+    }
+
+    response += `\n**Next Steps:**\n`
+    response += `- Learn from these corrections to avoid mistakes\n`
+    response += `- Document new lessons as you discover them\n`
+
+    // Add memory indicator to show Claude Brain helped
+    const statsLine = formatMemoryStats({ corrections: results.length })
+    return ResponseFormatter.text(withMemoryIndicator(response, results.length) + '\n\n' + statsLine)
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'get_corrections', 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 retrieval 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.`
+        )
+      }
+      if (error.message.includes('requires ChromaDB')) {
+        throw new McpError(
+          ErrorCode.InternalError,
+          `Correction retrieval requires ChromaDB backend.\n\n` +
+          `ChromaDB may not be enabled or may have failed to initialize.\n` +
+          `Check your configuration and ensure ChromaDB is running.`
+        )
+      }
+    }
+
+    throw error
+  }
+}

package/src/server/handlers/tools/get-decision-timeline.ts

@@ -1,172 +1,172 @@
-/**
- * Get Decision Timeline Handler
- * Returns a chronological timeline of decisions for a project/topic
- */
-
-import type { Logger } from 'pino'
-import type { ToolResponse } from '@/tools/types'
-import { getMemoryService } from '@/server/services'
-import { ToolValidator } from '@/server/utils/validators'
-import { ResponseFormatter } from '@/server/utils/response-formatter'
-import { withMemoryIndicator, formatMemoryStats } from '@/server/utils/memory-indicator'
-import { ErrorHandler } from '@/server/utils/error-handler'
-import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
-import { GetDecisionTimelineSchema } from './schemas'
-import { TimelineBuilder } from '@/intelligence/temporal/timeline'
-import { TemporalQueryProcessor } from '@/intelligence/temporal/query-processor'
-
-export async function handleGetDecisionTimeline(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  try {
-    const input = ToolValidator.validate(args, GetDecisionTimelineSchema)
-    const { project_name, topic, time_range, limit } = input
-
-    const memory = getMemoryService()
-
-    // Parse temporal expression if provided
-    let startDate: string | undefined
-    let endDate: string | undefined
-
-    if (time_range) {
-      const temporalProcessor = new TemporalQueryProcessor(logger)
-      const parsed = temporalProcessor.parse(time_range)
-      startDate = parsed.startDate
-      endDate = parsed.endDate
-    }
-
-    if (!memory.isChromaDBEnabled()) {
-      // SQLite fallback: fetch all data and build timeline
-      const [decisions, patterns, corrections] = await Promise.all([
-        memory.fetchAllDecisions(project_name),
-        memory.fetchAllPatterns(project_name),
-        memory.fetchAllCorrections(project_name)
-      ])
-
-      // Combine into timeline entries
-      let entries: Array<{ date: string; type: string; content: string; reasoning?: string }> = []
-
-      for (const d of decisions) {
-        entries.push({
-          date: d.date,
-          type: 'decision',
-          content: d.decision || d.content,
-          reasoning: d.reasoning
-        })
-      }
-      for (const p of patterns) {
-        entries.push({
-          date: p.date,
-          type: 'pattern',
-          content: p.description || p.content
-        })
-      }
-      for (const c of corrections) {
-        entries.push({
-          date: c.date,
-          type: 'correction',
-          content: c.correction || c.content,
-          reasoning: c.reasoning
-        })
-      }
-
-      // Filter by topic (substring match)
-      if (topic) {
-        const topicLower = topic.toLowerCase()
-        entries = entries.filter(e =>
-          e.content.toLowerCase().includes(topicLower) ||
-          (e.reasoning && e.reasoning.toLowerCase().includes(topicLower))
-        )
-      }
-
-      // Filter by date range
-      if (startDate) {
-        entries = entries.filter(e => e.date >= startDate!)
-      }
-      if (endDate) {
-        entries = entries.filter(e => e.date <= endDate!)
-      }
-
-      // Sort chronologically and limit
-      entries.sort((a, b) => new Date(a.date).getTime() - new Date(b.date).getTime())
-      const limited = entries.slice(0, limit || 20)
-
-      if (limited.length === 0) {
-        return ResponseFormatter.text('No decisions found for the specified criteria.')
-      }
-
-      const parts: string[] = []
-      parts.push(`## Decision Timeline (SQLite)`)
-      if (project_name) parts.push(`**Project:** ${project_name}`)
-      if (topic) parts.push(`**Topic:** ${topic}`)
-      if (limited.length > 0) {
-        parts.push(`**Period:** ${new Date(limited[0].date).toLocaleDateString()} - ${new Date(limited[limited.length - 1].date).toLocaleDateString()}`)
-      }
-      parts.push(`**Total entries:** ${limited.length}`)
-      parts.push('')
-
-      for (const entry of limited) {
-        const date = new Date(entry.date).toLocaleDateString()
-        const icon = entry.type === 'decision' ? '📋' : entry.type === 'pattern' ? '🔄' : '⚠️'
-        parts.push(`### ${icon} ${date} [${entry.type}]`)
-        parts.push(entry.content.slice(0, 200))
-        if (entry.reasoning) {
-          parts.push(`*Reasoning:* ${entry.reasoning.slice(0, 100)}`)
-        }
-        parts.push('')
-      }
-
-      const statsLine = formatMemoryStats({ recalled: limited.length })
-      return ResponseFormatter.text(withMemoryIndicator(parts.join('\n'), limited.length) + '\n\n' + statsLine)
-    }
-
-    const timelineBuilder = new TimelineBuilder(
-      logger,
-      memory.chroma.collections,
-      memory.chroma.embeddings
-    )
-
-    const timeline = await timelineBuilder.buildTimeline({
-      project: project_name,
-      topic,
-      startDate,
-      endDate,
-      limit
-    })
-
-    if (timeline.entries.length === 0) {
-      return ResponseFormatter.text('No decisions found for the specified criteria.')
-    }
-
-    // Format timeline
-    const parts: string[] = []
-    parts.push(`## Decision Timeline`)
-    if (project_name) parts.push(`**Project:** ${project_name}`)
-    if (topic) parts.push(`**Topic:** ${topic}`)
-    if (timeline.startDate && timeline.endDate) {
-      parts.push(`**Period:** ${new Date(timeline.startDate).toLocaleDateString()} - ${new Date(timeline.endDate).toLocaleDateString()}`)
-    }
-    parts.push(`**Total entries:** ${timeline.totalCount}`)
-    parts.push('')
-
-    for (const entry of timeline.entries) {
-      const date = new Date(entry.date).toLocaleDateString()
-      const icon = entry.type === 'decision' ? '📋' : entry.type === 'pattern' ? '🔄' : '⚠️'
-      parts.push(`### ${icon} ${date} [${entry.type}]`)
-      parts.push(entry.content.slice(0, 200))
-      if (entry.metadata.reasoning) {
-        parts.push(`*Reasoning:* ${entry.metadata.reasoning.slice(0, 100)}`)
-      }
-      parts.push('')
-    }
-
-    const statsLine = formatMemoryStats({ recalled: timeline.totalCount })
-    return ResponseFormatter.text(withMemoryIndicator(parts.join('\n'), timeline.totalCount) + '\n\n' + statsLine)
-
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'get_decision_timeline' })
-    if (error instanceof McpError) throw error
-    throw new McpError(ErrorCode.InternalError, `Timeline failed: ${error instanceof Error ? error.message : 'Unknown error'}`)
-  }
-}
+/**
+ * Get Decision Timeline Handler
+ * Returns a chronological timeline of decisions for a project/topic
+ */
+
+import type { Logger } from 'pino'
+import type { ToolResponse } from '@/tools/types'
+import { getMemoryService } from '@/server/services'
+import { ToolValidator } from '@/server/utils/validators'
+import { ResponseFormatter } from '@/server/utils/response-formatter'
+import { withMemoryIndicator, formatMemoryStats } from '@/server/utils/memory-indicator'
+import { ErrorHandler } from '@/server/utils/error-handler'
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
+import { GetDecisionTimelineSchema } from './schemas'
+import { TimelineBuilder } from '@/intelligence/temporal/timeline'
+import { TemporalQueryProcessor } from '@/intelligence/temporal/query-processor'
+
+export async function handleGetDecisionTimeline(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  try {
+    const input = ToolValidator.validate(args, GetDecisionTimelineSchema)
+    const { project_name, topic, time_range, limit } = input
+
+    const memory = getMemoryService()
+
+    // Parse temporal expression if provided
+    let startDate: string | undefined
+    let endDate: string | undefined
+
+    if (time_range) {
+      const temporalProcessor = new TemporalQueryProcessor(logger)
+      const parsed = temporalProcessor.parse(time_range)
+      startDate = parsed.startDate
+      endDate = parsed.endDate
+    }
+
+    if (!memory.isChromaDBEnabled()) {
+      // SQLite fallback: fetch all data and build timeline
+      const [decisions, patterns, corrections] = await Promise.all([
+        memory.fetchAllDecisions(project_name),
+        memory.fetchAllPatterns(project_name),
+        memory.fetchAllCorrections(project_name)
+      ])
+
+      // Combine into timeline entries
+      let entries: Array<{ date: string; type: string; content: string; reasoning?: string }> = []
+
+      for (const d of decisions) {
+        entries.push({
+          date: d.date,
+          type: 'decision',
+          content: d.decision || d.content,
+          reasoning: d.reasoning
+        })
+      }
+      for (const p of patterns) {
+        entries.push({
+          date: p.date,
+          type: 'pattern',
+          content: p.description || p.content
+        })
+      }
+      for (const c of corrections) {
+        entries.push({
+          date: c.date,
+          type: 'correction',
+          content: c.correction || c.content,
+          reasoning: c.reasoning
+        })
+      }
+
+      // Filter by topic (substring match)
+      if (topic) {
+        const topicLower = topic.toLowerCase()
+        entries = entries.filter(e =>
+          e.content.toLowerCase().includes(topicLower) ||
+          (e.reasoning && e.reasoning.toLowerCase().includes(topicLower))
+        )
+      }
+
+      // Filter by date range
+      if (startDate) {
+        entries = entries.filter(e => e.date >= startDate!)
+      }
+      if (endDate) {
+        entries = entries.filter(e => e.date <= endDate!)
+      }
+
+      // Sort chronologically and limit
+      entries.sort((a, b) => new Date(a.date).getTime() - new Date(b.date).getTime())
+      const limited = entries.slice(0, limit || 20)
+
+      if (limited.length === 0) {
+        return ResponseFormatter.text('No decisions found for the specified criteria.')
+      }
+
+      const parts: string[] = []
+      parts.push(`## Decision Timeline (SQLite)`)
+      if (project_name) parts.push(`**Project:** ${project_name}`)
+      if (topic) parts.push(`**Topic:** ${topic}`)
+      if (limited.length > 0) {
+        parts.push(`**Period:** ${new Date(limited[0].date).toLocaleDateString()} - ${new Date(limited[limited.length - 1].date).toLocaleDateString()}`)
+      }
+      parts.push(`**Total entries:** ${limited.length}`)
+      parts.push('')
+
+      for (const entry of limited) {
+        const date = new Date(entry.date).toLocaleDateString()
+        const icon = entry.type === 'decision' ? '📋' : entry.type === 'pattern' ? '🔄' : '⚠️'
+        parts.push(`### ${icon} ${date} [${entry.type}]`)
+        parts.push(entry.content.slice(0, 200))
+        if (entry.reasoning) {
+          parts.push(`*Reasoning:* ${entry.reasoning.slice(0, 100)}`)
+        }
+        parts.push('')
+      }
+
+      const statsLine = formatMemoryStats({ recalled: limited.length })
+      return ResponseFormatter.text(withMemoryIndicator(parts.join('\n'), limited.length) + '\n\n' + statsLine)
+    }
+
+    const timelineBuilder = new TimelineBuilder(
+      logger,
+      memory.chroma.collections,
+      memory.chroma.embeddings
+    )
+
+    const timeline = await timelineBuilder.buildTimeline({
+      project: project_name,
+      topic,
+      startDate,
+      endDate,
+      limit
+    })
+
+    if (timeline.entries.length === 0) {
+      return ResponseFormatter.text('No decisions found for the specified criteria.')
+    }
+
+    // Format timeline
+    const parts: string[] = []
+    parts.push(`## Decision Timeline`)
+    if (project_name) parts.push(`**Project:** ${project_name}`)
+    if (topic) parts.push(`**Topic:** ${topic}`)
+    if (timeline.startDate && timeline.endDate) {
+      parts.push(`**Period:** ${new Date(timeline.startDate).toLocaleDateString()} - ${new Date(timeline.endDate).toLocaleDateString()}`)
+    }
+    parts.push(`**Total entries:** ${timeline.totalCount}`)
+    parts.push('')
+
+    for (const entry of timeline.entries) {
+      const date = new Date(entry.date).toLocaleDateString()
+      const icon = entry.type === 'decision' ? '📋' : entry.type === 'pattern' ? '🔄' : '⚠️'
+      parts.push(`### ${icon} ${date} [${entry.type}]`)
+      parts.push(entry.content.slice(0, 200))
+      if (entry.metadata.reasoning) {
+        parts.push(`*Reasoning:* ${entry.metadata.reasoning.slice(0, 100)}`)
+      }
+      parts.push('')
+    }
+
+    const statsLine = formatMemoryStats({ recalled: timeline.totalCount })
+    return ResponseFormatter.text(withMemoryIndicator(parts.join('\n'), timeline.totalCount) + '\n\n' + statsLine)
+
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'get_decision_timeline' })
+    if (error instanceof McpError) throw error
+    throw new McpError(ErrorCode.InternalError, `Timeline failed: ${error instanceof Error ? error.message : 'Unknown error'}`)
+  }
+}