claude-brain 0.30.2 → 0.30.3

package/src/memory/chroma/index.ts

@@ -1,82 +1,82 @@
-import type { Logger } from 'pino'
-import type { ChromaConfig } from './config'
-import { ChromaClientManager } from './client'
-import { CollectionManager } from './collection-manager'
-import { ChromaMemoryStore } from './store'
-import { ChromaSearchEngine } from './search'
-import { createEmbeddingProvider, type EmbeddingProvider } from './embeddings'
-
-export * from './config'
-export * from './schemas'
-export * from './client'
-export * from './collection-manager'
-export * from './store'
-export * from './search'
-export * from './embeddings'
-export * from './migration'
-
-export class ChromaManager {
-  private logger: Logger
-  private config: ChromaConfig
-
-  public client: ChromaClientManager
-  public collections: CollectionManager
-  public store: ChromaMemoryStore
-  public search: ChromaSearchEngine
-  public embeddings: EmbeddingProvider | undefined
-
-  private initialized: boolean = false
-
-  constructor(logger: Logger, config: ChromaConfig, customEmbeddingProvider?: EmbeddingProvider) {
-    this.logger = logger.child({ component: 'chroma-manager' })
-    this.config = config
-
-    this.client = new ChromaClientManager(logger, config)
-    this.collections = new CollectionManager(logger, this.client)
-    this.embeddings = customEmbeddingProvider || createEmbeddingProvider(logger, config)
-    this.store = new ChromaMemoryStore(logger, this.collections, this.embeddings)
-    this.search = new ChromaSearchEngine(logger, this.collections, this.embeddings)
-  }
-
-  async initialize(): Promise<void> {
-    if (this.initialized) {
-      this.logger.warn('ChromaDB already initialized')
-      return
-    }
-
-    this.logger.info('Initializing ChromaDB...')
-
-    try {
-      await this.client.initialize()
-      await this.collections.initialize()
-
-      this.initialized = true
-      this.logger.info('ChromaDB initialized successfully')
-
-    } catch (error) {
-      this.logger.warn({ error }, 'ChromaDB unavailable')
-      throw error
-    }
-  }
-
-  getStatus() {
-    return {
-      initialized: this.initialized,
-      client: this.client.getStatus(),
-      config: {
-        mode: this.config.mode,
-        embeddingProvider: this.config.embeddingProvider
-      }
-    }
-  }
-
-  async healthCheck(): Promise<boolean> {
-    return this.client.healthCheck()
-  }
-
-  async close(): Promise<void> {
-    await this.client.close()
-    this.initialized = false
-    this.logger.info('ChromaDB closed')
-  }
-}
+import type { Logger } from 'pino'
+import type { ChromaConfig } from './config'
+import { ChromaClientManager } from './client'
+import { CollectionManager } from './collection-manager'
+import { ChromaMemoryStore } from './store'
+import { ChromaSearchEngine } from './search'
+import { createEmbeddingProvider, type EmbeddingProvider } from './embeddings'
+
+export * from './config'
+export * from './schemas'
+export * from './client'
+export * from './collection-manager'
+export * from './store'
+export * from './search'
+export * from './embeddings'
+export * from './migration'
+
+export class ChromaManager {
+  private logger: Logger
+  private config: ChromaConfig
+
+  public client: ChromaClientManager
+  public collections: CollectionManager
+  public store: ChromaMemoryStore
+  public search: ChromaSearchEngine
+  public embeddings: EmbeddingProvider | undefined
+
+  private initialized: boolean = false
+
+  constructor(logger: Logger, config: ChromaConfig, customEmbeddingProvider?: EmbeddingProvider) {
+    this.logger = logger.child({ component: 'chroma-manager' })
+    this.config = config
+
+    this.client = new ChromaClientManager(logger, config)
+    this.collections = new CollectionManager(logger, this.client)
+    this.embeddings = customEmbeddingProvider || createEmbeddingProvider(logger, config)
+    this.store = new ChromaMemoryStore(logger, this.collections, this.embeddings)
+    this.search = new ChromaSearchEngine(logger, this.collections, this.embeddings)
+  }
+
+  async initialize(): Promise<void> {
+    if (this.initialized) {
+      this.logger.warn('ChromaDB already initialized')
+      return
+    }
+
+    this.logger.info('Initializing ChromaDB...')
+
+    try {
+      await this.client.initialize()
+      await this.collections.initialize()
+
+      this.initialized = true
+      this.logger.info('ChromaDB initialized successfully')
+
+    } catch (error) {
+      this.logger.warn({ error }, 'ChromaDB unavailable')
+      throw error
+    }
+  }
+
+  getStatus() {
+    return {
+      initialized: this.initialized,
+      client: this.client.getStatus(),
+      config: {
+        mode: this.config.mode,
+        embeddingProvider: this.config.embeddingProvider
+      }
+    }
+  }
+
+  async healthCheck(): Promise<boolean> {
+    return this.client.healthCheck()
+  }
+
+  async close(): Promise<void> {
+    await this.client.close()
+    this.initialized = false
+    this.logger.info('ChromaDB closed')
+  }
+}