@prmichaelsen/remember-mcp 2.5.2 → 2.6.1

package/src/weaviate/client.ts

@@ -1,5 +1,6 @@
 import weaviate, { WeaviateClient } from 'weaviate-client';
 import { config } from '../config.js';
+import { logger } from '../utils/logger.js';
 
 let client: WeaviateClient | null = null;
 
@@ -25,7 +26,10 @@ export async function initWeaviateClient(): Promise<WeaviateClient> {
   // Use appropriate connection method
   if (!isLocal) {
     // Use connectToWeaviateCloud() for ALL remote instances (cloud or self-hosted)
-    console.log('[Weaviate] Connecting to remote Weaviate:', weaviateUrl);
+    logger.info('Connecting to remote Weaviate', {
+      module: 'weaviate-client',
+      url: weaviateUrl,
+    });
     client = await weaviate.connectToWeaviateCloud(weaviateUrl, {
       authCredentials: config.weaviate.apiKey
         ? new weaviate.ApiKey(config.weaviate.apiKey)
@@ -36,7 +40,10 @@ export async function initWeaviateClient(): Promise<WeaviateClient> {
     });
   } else {
     // connectToLocal() for localhost only
-    console.log('[Weaviate] Connecting to local Weaviate:', weaviateUrl);
+    logger.info('Connecting to local Weaviate', {
+      module: 'weaviate-client',
+      url: weaviateUrl,
+    });
     const localConfig: any = {
       host: weaviateUrl.replace(/^https?:\/\//, '').split(':')[0],
       port: weaviateUrl.includes(':')
@@ -56,7 +63,10 @@ export async function initWeaviateClient(): Promise<WeaviateClient> {
     client = await weaviate.connectToLocal(localConfig);
   }
 
-  console.log('[Weaviate] Client initialized successfully');
+  logger.info('Weaviate client initialized successfully', {
+    module: 'weaviate-client',
+    isLocal,
+  });
   return client;
 }
 
@@ -77,10 +87,16 @@ export async function testWeaviateConnection(): Promise<boolean> {
   try {
     const weaviateClient = getWeaviateClient();
     const isReady = await weaviateClient.isReady();
-    console.log('[Weaviate] Connection successful, ready:', isReady);
+    logger.info('Weaviate connection test successful', {
+      module: 'weaviate-client',
+      isReady,
+    });
     return isReady;
   } catch (error) {
-    console.error('[Weaviate] Connection failed:', error);
+    logger.error('Weaviate connection test failed', {
+      module: 'weaviate-client',
+      error: error instanceof Error ? error.message : String(error),
+    });
     return false;
   }
 }
@@ -132,7 +148,11 @@ export async function collectionExists(collectionName: string): Promise<boolean>
     const exists = await weaviateClient.collections.exists(collectionName);
     return exists;
   } catch (error) {
-    console.error(`[Weaviate] Error checking collection ${collectionName}:`, error);
+    logger.error('Error checking collection existence', {
+      module: 'weaviate-client',
+      collectionName,
+      error: error instanceof Error ? error.message : String(error),
+    });
     return false;
   }
 }
@@ -144,6 +164,8 @@ export async function closeWeaviateClient(): Promise<void> {
   if (client) {
     // Weaviate client doesn't have explicit close method
     client = null;
-    console.log('[Weaviate] Client closed');
+    logger.info('Weaviate client closed', {
+      module: 'weaviate-client',
+    });
   }
 }

package/src/weaviate/schema.ts

@@ -6,6 +6,7 @@
 import weaviate, { WeaviateClient } from 'weaviate-client';
 import { getWeaviateClient, sanitizeUserId } from './client.js';
 import { config } from '../config.js';
+import { logger } from '../utils/logger.js';
 
 /**
  * Create Memory collection schema for a user
@@ -25,11 +26,17 @@ export async function createMemoryCollection(userId: string): Promise<void> {
   // Check if collection already exists
   const exists = await client.collections.exists(collectionName);
   if (exists) {
-    console.log(`[Weaviate] Collection ${collectionName} already exists`);
+    logger.debug('Collection already exists', {
+      module: 'weaviate-schema',
+      collectionName,
+    });
     return;
   }
 
-  console.log(`[Weaviate] Creating collection ${collectionName}...`);
+  logger.info('Creating memory collection', {
+    module: 'weaviate-schema',
+    collectionName,
+  });
 
   // Create collection with schema
   await client.collections.create({
@@ -243,10 +250,30 @@ export async function createMemoryCollection(userId: string): Promise<void> {
         dataType: 'number' as any,
         description: 'Calculated effective weight',
       },
+
+      // Comment/threading fields (for threaded discussions in shared spaces)
+      {
+        name: 'parent_id',
+        dataType: 'text' as any,
+        description: 'ID of parent memory or comment (for threading)',
+      },
+      {
+        name: 'thread_root_id',
+        dataType: 'text' as any,
+        description: 'Root memory ID for fetching entire thread',
+      },
+      {
+        name: 'moderation_flags',
+        dataType: 'text[]' as any,
+        description: 'Per-space moderation flags (format: "{space_id}:{flag_type}")',
+      },
     ],
   });
 
-  console.log(`[Weaviate] Collection ${collectionName} created successfully`);
+  logger.info('Memory collection created successfully', {
+    module: 'weaviate-schema',
+    collectionName,
+  });
 }
 
 /**
@@ -283,6 +310,9 @@ export async function deleteMemoryCollection(userId: string): Promise<void> {
   
   if (exists) {
     await client.collections.delete(collectionName);
-    console.log(`[Weaviate] Collection ${collectionName} deleted`);
+    logger.info('Memory collection deleted', {
+      module: 'weaviate-schema',
+      collectionName,
+    });
   }
 }

package/src/weaviate/space-schema.ts

@@ -1,6 +1,6 @@
 /**
  * Weaviate space collection schema and utilities
- * 
+ *
  * Manages shared space collections where users can publish memories
  * for discovery by other users.
  */
@@ -8,6 +8,7 @@
 import weaviate, { type WeaviateClient, type Collection } from 'weaviate-client';
 import { config } from '../config.js';
 import { SUPPORTED_SPACES, type SpaceId } from '../types/space-memory.js';
+import { logger } from '../utils/logger.js';
 
 /**
  * Unified public collection name for all public spaces
@@ -83,7 +84,11 @@ async function createSpaceCollection(
     ? PUBLIC_COLLECTION_NAME
     : getSpaceCollectionName(spaceId);
 
-  console.log(`[Weaviate] Creating space collection ${collectionName}...`);
+  logger.info('Creating space collection', {
+    module: 'weaviate-space-schema',
+    collectionName,
+    spaceId,
+  });
 
   // Create collection with schema (same as Memory schema but for spaces)
   await client.collections.create({
@@ -251,10 +256,30 @@ async function createSpaceCollection(
         dataType: 'number' as any,
         description: 'Version number (increments on update)',
       },
+
+      // Comment/threading fields (for threaded discussions in shared spaces)
+      {
+        name: 'parent_id',
+        dataType: 'text' as any,
+        description: 'ID of parent memory or comment (for threading)',
+      },
+      {
+        name: 'thread_root_id',
+        dataType: 'text' as any,
+        description: 'Root memory ID for fetching entire thread',
+      },
+      {
+        name: 'moderation_flags',
+        dataType: 'text[]' as any,
+        description: 'Per-space moderation flags (format: "{space_id}:{flag_type}")',
+      },
     ],
   });
 
-  console.log(`[Weaviate] Space collection ${collectionName} created successfully`);
+  logger.info('Space collection created successfully', {
+    module: 'weaviate-space-schema',
+    collectionName,
+  });
 }
 
 /**