claude-brain 0.14.2 → 0.14.4

package/src/config/loader.ts

@@ -1,166 +1,166 @@
-import { existsSync, readFileSync } from 'node:fs'
-import { resolve } from 'node:path'
-import { config as loadEnv } from 'dotenv'
-import { ConfigSchema, type Config, type PartialConfig } from './schema'
-import { defaultConfig } from './defaults'
-import { getClaudeBrainHome, resolveHomePath } from './home'
-
-const CONFIG_FILE_NAME = '.claudebrainrc.json'
-
-/** Load configuration from a JSON file if it exists */
-function loadFromFile(basePath: string): PartialConfig {
-  // Check basePath first, then Claude Brain home
-  for (const dir of [basePath, getClaudeBrainHome()]) {
-    const configPath = resolve(dir, CONFIG_FILE_NAME)
-    if (!existsSync(configPath)) {
-      continue
-    }
-
-    try {
-      const content = readFileSync(configPath, 'utf-8')
-      return JSON.parse(content) as PartialConfig
-    } catch (error) {
-      console.warn(`Warning: Failed to parse config file at ${configPath}:`, error)
-      return {}
-    }
-  }
-
-  return {}
-}
-
-/** Load configuration from environment variables */
-function loadFromEnv(): PartialConfig {
-  // Skip .env loading if SKIP_DOTENV is set (for MCP servers where env is provided externally)
-  if (!process.env.SKIP_DOTENV) {
-    // Load .env from Claude Brain home directory
-    loadEnv({ path: resolve(getClaudeBrainHome(), '.env'), debug: false })
-  }
-
-  const config: PartialConfig = {}
-
-  if (process.env.VAULT_PATH) {
-    config.vaultPath = process.env.VAULT_PATH
-  }
-  if (process.env.MCP_SERVER_NAME) {
-    config.serverName = process.env.MCP_SERVER_NAME
-  }
-  if (process.env.MCP_SERVER_VERSION) {
-    config.serverVersion = process.env.MCP_SERVER_VERSION
-  }
-  if (process.env.LOG_LEVEL) {
-    config.logLevel = process.env.LOG_LEVEL as Config['logLevel']
-  }
-  if (process.env.LOG_FILE_PATH) {
-    config.logFilePath = process.env.LOG_FILE_PATH
-  }
-  if (process.env.DB_PATH) {
-    config.dbPath = process.env.DB_PATH
-  }
-  if (process.env.PORT) {
-    config.port = parseInt(process.env.PORT, 10)
-  }
-  if (process.env.ENABLE_FILE_WATCH) {
-    config.enableFileWatch = process.env.ENABLE_FILE_WATCH === 'true'
-  }
-  if (process.env.CACHE_SIZE) {
-    config.cacheSize = parseInt(process.env.CACHE_SIZE, 10)
-  }
-  if (process.env.NODE_ENV) {
-    config.nodeEnv = process.env.NODE_ENV as Config['nodeEnv']
-  }
-
-  // Retrieval configuration
-  if (process.env.RETRIEVAL_ENABLED) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.enabled = process.env.RETRIEVAL_ENABLED === 'true'
-  }
-  if (process.env.RETRIEVAL_DENSE_WEIGHT) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.dense = config.retrieval.dense || {}
-    config.retrieval.dense.weight = parseFloat(process.env.RETRIEVAL_DENSE_WEIGHT)
-  }
-  if (process.env.RETRIEVAL_SPARSE_ENABLED) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.sparse = config.retrieval.sparse || {}
-    config.retrieval.sparse.enabled = process.env.RETRIEVAL_SPARSE_ENABLED === 'true'
-  }
-  if (process.env.RETRIEVAL_SPARSE_WEIGHT) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.sparse = config.retrieval.sparse || {}
-    config.retrieval.sparse.weight = parseFloat(process.env.RETRIEVAL_SPARSE_WEIGHT)
-  }
-  if (process.env.RETRIEVAL_FUSION_METHOD) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.fusion = config.retrieval.fusion || {}
-    config.retrieval.fusion.method = process.env.RETRIEVAL_FUSION_METHOD as 'rrf' | 'linear' | 'max'
-  }
-  if (process.env.RETRIEVAL_RRF_K) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.fusion = config.retrieval.fusion || {}
-    config.retrieval.fusion.rrfK = parseInt(process.env.RETRIEVAL_RRF_K, 10)
-  }
-  if (process.env.RETRIEVAL_RERANKER_ENABLED) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.reranker = config.retrieval.reranker || {}
-    config.retrieval.reranker.enabled = process.env.RETRIEVAL_RERANKER_ENABLED === 'true'
-  }
-  if (process.env.RETRIEVAL_RERANKER_MODEL) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.reranker = config.retrieval.reranker || {}
-    config.retrieval.reranker.model = process.env.RETRIEVAL_RERANKER_MODEL
-  }
-  if (process.env.RETRIEVAL_FEEDBACK_ENABLED) {
-    config.retrieval = config.retrieval || {}
-    config.retrieval.feedback = config.retrieval.feedback || {}
-    config.retrieval.feedback.enabled = process.env.RETRIEVAL_FEEDBACK_ENABLED === 'true'
-  }
-
-  return config
-}
-
-/** Merge multiple partial configs, later values override earlier */
-function mergeConfigs(...configs: PartialConfig[]): PartialConfig {
-  return configs.reduce((acc, config) => {
-    return { ...acc, ...Object.fromEntries(Object.entries(config).filter(([_, v]) => v !== undefined)) }
-  }, {})
-}
-
-/** Load and validate configuration from all sources */
-export async function loadConfig(basePath: string = process.cwd()): Promise<Config> {
-  const fileConfig = loadFromFile(basePath)
-  const envConfig = loadFromEnv()
-
-  const merged = mergeConfigs(defaultConfig, fileConfig, envConfig)
-
-  const result = ConfigSchema.safeParse(merged)
-
-  if (!result.success) {
-    const errors = result.error.errors
-      .map(e => `  - ${e.path.join('.')}: ${e.message}`)
-      .join('\n')
-    throw new Error(`Configuration validation failed:\n${errors}`)
-  }
-
-  // Resolve relative paths against Claude Brain home
-  const data = result.data
-  data.logFilePath = resolveHomePath(data.logFilePath)
-  data.dbPath = resolveHomePath(data.dbPath)
-  if (data.chroma.path) {
-    data.chroma.path = resolveHomePath(data.chroma.path)
-  }
-  if (data.knowledge?.graph?.persistPath) {
-    data.knowledge.graph.persistPath = resolveHomePath(data.knowledge.graph.persistPath)
-  }
-
-  return data
-}
-
-/** Validate a partial configuration object */
-export function validateConfig(config: unknown): config is Config {
-  return ConfigSchema.safeParse(config).success
-}
-
-/** Get the path to the config file */
-export function getConfigFilePath(basePath: string = process.cwd()): string {
-  return resolve(basePath, CONFIG_FILE_NAME)
-}
+import { existsSync, readFileSync } from 'node:fs'
+import { resolve } from 'node:path'
+import { config as loadEnv } from 'dotenv'
+import { ConfigSchema, type Config, type PartialConfig } from './schema'
+import { defaultConfig } from './defaults'
+import { getClaudeBrainHome, resolveHomePath } from './home'
+
+const CONFIG_FILE_NAME = '.claudebrainrc.json'
+
+/** Load configuration from a JSON file if it exists */
+function loadFromFile(basePath: string): PartialConfig {
+  // Check basePath first, then Claude Brain home
+  for (const dir of [basePath, getClaudeBrainHome()]) {
+    const configPath = resolve(dir, CONFIG_FILE_NAME)
+    if (!existsSync(configPath)) {
+      continue
+    }
+
+    try {
+      const content = readFileSync(configPath, 'utf-8')
+      return JSON.parse(content) as PartialConfig
+    } catch (error) {
+      console.warn(`Warning: Failed to parse config file at ${configPath}:`, error)
+      return {}
+    }
+  }
+
+  return {}
+}
+
+/** Load configuration from environment variables */
+function loadFromEnv(): PartialConfig {
+  // Skip .env loading if SKIP_DOTENV is set (for MCP servers where env is provided externally)
+  if (!process.env.SKIP_DOTENV) {
+    // Load .env from Claude Brain home directory
+    loadEnv({ path: resolve(getClaudeBrainHome(), '.env'), debug: false })
+  }
+
+  const config: PartialConfig = {}
+
+  if (process.env.VAULT_PATH) {
+    config.vaultPath = process.env.VAULT_PATH
+  }
+  if (process.env.MCP_SERVER_NAME) {
+    config.serverName = process.env.MCP_SERVER_NAME
+  }
+  if (process.env.MCP_SERVER_VERSION) {
+    config.serverVersion = process.env.MCP_SERVER_VERSION
+  }
+  if (process.env.LOG_LEVEL) {
+    config.logLevel = process.env.LOG_LEVEL as Config['logLevel']
+  }
+  if (process.env.LOG_FILE_PATH) {
+    config.logFilePath = process.env.LOG_FILE_PATH
+  }
+  if (process.env.DB_PATH) {
+    config.dbPath = process.env.DB_PATH
+  }
+  if (process.env.PORT) {
+    config.port = parseInt(process.env.PORT, 10)
+  }
+  if (process.env.ENABLE_FILE_WATCH) {
+    config.enableFileWatch = process.env.ENABLE_FILE_WATCH === 'true'
+  }
+  if (process.env.CACHE_SIZE) {
+    config.cacheSize = parseInt(process.env.CACHE_SIZE, 10)
+  }
+  if (process.env.NODE_ENV) {
+    config.nodeEnv = process.env.NODE_ENV as Config['nodeEnv']
+  }
+
+  // Retrieval configuration
+  if (process.env.RETRIEVAL_ENABLED) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.enabled = process.env.RETRIEVAL_ENABLED === 'true'
+  }
+  if (process.env.RETRIEVAL_DENSE_WEIGHT) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.dense = config.retrieval.dense || {}
+    config.retrieval.dense.weight = parseFloat(process.env.RETRIEVAL_DENSE_WEIGHT)
+  }
+  if (process.env.RETRIEVAL_SPARSE_ENABLED) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.sparse = config.retrieval.sparse || {}
+    config.retrieval.sparse.enabled = process.env.RETRIEVAL_SPARSE_ENABLED === 'true'
+  }
+  if (process.env.RETRIEVAL_SPARSE_WEIGHT) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.sparse = config.retrieval.sparse || {}
+    config.retrieval.sparse.weight = parseFloat(process.env.RETRIEVAL_SPARSE_WEIGHT)
+  }
+  if (process.env.RETRIEVAL_FUSION_METHOD) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.fusion = config.retrieval.fusion || {}
+    config.retrieval.fusion.method = process.env.RETRIEVAL_FUSION_METHOD as 'rrf' | 'linear' | 'max'
+  }
+  if (process.env.RETRIEVAL_RRF_K) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.fusion = config.retrieval.fusion || {}
+    config.retrieval.fusion.rrfK = parseInt(process.env.RETRIEVAL_RRF_K, 10)
+  }
+  if (process.env.RETRIEVAL_RERANKER_ENABLED) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.reranker = config.retrieval.reranker || {}
+    config.retrieval.reranker.enabled = process.env.RETRIEVAL_RERANKER_ENABLED === 'true'
+  }
+  if (process.env.RETRIEVAL_RERANKER_MODEL) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.reranker = config.retrieval.reranker || {}
+    config.retrieval.reranker.model = process.env.RETRIEVAL_RERANKER_MODEL
+  }
+  if (process.env.RETRIEVAL_FEEDBACK_ENABLED) {
+    config.retrieval = config.retrieval || {}
+    config.retrieval.feedback = config.retrieval.feedback || {}
+    config.retrieval.feedback.enabled = process.env.RETRIEVAL_FEEDBACK_ENABLED === 'true'
+  }
+
+  return config
+}
+
+/** Merge multiple partial configs, later values override earlier */
+function mergeConfigs(...configs: PartialConfig[]): PartialConfig {
+  return configs.reduce((acc, config) => {
+    return { ...acc, ...Object.fromEntries(Object.entries(config).filter(([_, v]) => v !== undefined)) }
+  }, {})
+}
+
+/** Load and validate configuration from all sources */
+export async function loadConfig(basePath: string = process.cwd()): Promise<Config> {
+  const fileConfig = loadFromFile(basePath)
+  const envConfig = loadFromEnv()
+
+  const merged = mergeConfigs(defaultConfig, fileConfig, envConfig)
+
+  const result = ConfigSchema.safeParse(merged)
+
+  if (!result.success) {
+    const errors = result.error.errors
+      .map(e => `  - ${e.path.join('.')}: ${e.message}`)
+      .join('\n')
+    throw new Error(`Configuration validation failed:\n${errors}`)
+  }
+
+  // Resolve relative paths against Claude Brain home
+  const data = result.data
+  data.logFilePath = resolveHomePath(data.logFilePath)
+  data.dbPath = resolveHomePath(data.dbPath)
+  if (data.chroma.path) {
+    data.chroma.path = resolveHomePath(data.chroma.path)
+  }
+  if (data.knowledge?.graph?.persistPath) {
+    data.knowledge.graph.persistPath = resolveHomePath(data.knowledge.graph.persistPath)
+  }
+
+  return data
+}
+
+/** Validate a partial configuration object */
+export function validateConfig(config: unknown): config is Config {
+  return ConfigSchema.safeParse(config).success
+}
+
+/** Get the path to the config file */
+export function getConfigFilePath(basePath: string = process.cwd()): string {
+  return resolve(basePath, CONFIG_FILE_NAME)
+}

package/src/config/migration.ts

@@ -1,76 +1,76 @@
-import fs from 'fs/promises'
-import type { Logger } from 'pino'
-
-export interface ConfigVersion {
-  version: string
-  config: Record<string, any>
-}
-
-export class ConfigMigration {
-  private logger: Logger
-  private currentVersion = '1.0.0'
-
-  constructor(logger: Logger) {
-    this.logger = logger.child({ component: 'config-migration' })
-  }
-
-  async migrate(config: any): Promise<any> {
-    const configVersion = config.version || '0.0.0'
-
-    if (configVersion === this.currentVersion) {
-      return config
-    }
-
-    this.logger.info(
-      { from: configVersion, to: this.currentVersion },
-      'Migrating configuration'
-    )
-
-    await this.backup(config)
-
-    let migrated = { ...config }
-
-    if (this.compareVersions(configVersion, '1.0.0') < 0) {
-      migrated = await this.migrateTo_1_0_0(migrated)
-    }
-
-    migrated.version = this.currentVersion
-
-    return migrated
-  }
-
-  private async migrateTo_1_0_0(config: any): Promise<any> {
-    this.logger.info('Migrating to version 1.0.0')
-
-    return {
-      ...config,
-      enableFileWatch: config.enableFileWatch ?? true,
-      serverVersion: config.serverVersion || '1.0.0'
-    }
-  }
-
-  private async backup(config: any): Promise<void> {
-    const timestamp = new Date().toISOString().replace(/[:.]/g, '-')
-    const backupPath = `./data/config-backup-${timestamp}.json`
-
-    await fs.writeFile(
-      backupPath,
-      JSON.stringify(config, null, 2),
-      'utf-8'
-    )
-
-    this.logger.info({ backupPath }, 'Configuration backed up')
-  }
-
-  private compareVersions(v1: string, v2: string): number {
-    const parts1 = v1.split('.').map(Number)
-    const parts2 = v2.split('.').map(Number)
-
-    for (let i = 0; i < 3; i++) {
-      if (parts1[i] > parts2[i]) return 1
-      if (parts1[i] < parts2[i]) return -1
-    }
-
-    return 0
-  }
-}
+import fs from 'fs/promises'
+import type { Logger } from 'pino'
+
+export interface ConfigVersion {
+  version: string
+  config: Record<string, any>
+}
+
+export class ConfigMigration {
+  private logger: Logger
+  private currentVersion = '1.0.0'
+
+  constructor(logger: Logger) {
+    this.logger = logger.child({ component: 'config-migration' })
+  }
+
+  async migrate(config: any): Promise<any> {
+    const configVersion = config.version || '0.0.0'
+
+    if (configVersion === this.currentVersion) {
+      return config
+    }
+
+    this.logger.info(
+      { from: configVersion, to: this.currentVersion },
+      'Migrating configuration'
+    )
+
+    await this.backup(config)
+
+    let migrated = { ...config }
+
+    if (this.compareVersions(configVersion, '1.0.0') < 0) {
+      migrated = await this.migrateTo_1_0_0(migrated)
+    }
+
+    migrated.version = this.currentVersion
+
+    return migrated
+  }
+
+  private async migrateTo_1_0_0(config: any): Promise<any> {
+    this.logger.info('Migrating to version 1.0.0')
+
+    return {
+      ...config,
+      enableFileWatch: config.enableFileWatch ?? true,
+      serverVersion: config.serverVersion || '1.0.0'
+    }
+  }
+
+  private async backup(config: any): Promise<void> {
+    const timestamp = new Date().toISOString().replace(/[:.]/g, '-')
+    const backupPath = `./data/config-backup-${timestamp}.json`
+
+    await fs.writeFile(
+      backupPath,
+      JSON.stringify(config, null, 2),
+      'utf-8'
+    )
+
+    this.logger.info({ backupPath }, 'Configuration backed up')
+  }
+
+  private compareVersions(v1: string, v2: string): number {
+    const parts1 = v1.split('.').map(Number)
+    const parts2 = v2.split('.').map(Number)
+
+    for (let i = 0; i < 3; i++) {
+      if (parts1[i] > parts2[i]) return 1
+      if (parts1[i] < parts2[i]) return -1
+    }
+
+    return 0
+  }
+}