@prmichaelsen/remember-mcp 2.5.2 → 2.6.2

package/src/tools/query-space.ts

@@ -17,7 +17,7 @@ import { handleToolError } from '../utils/error-handler.js';
  */
 export const querySpaceTool: Tool = {
   name: 'remember_query_space',
-  description: 'Ask natural language questions about memories in shared spaces. Works like remember_query_memory but queries shared spaces.',
+  description: 'Ask natural language questions about memories in shared spaces. By default, excludes comments to focus on original content. Set include_comments: true to include discussions in answers.',
   inputSchema: {
     type: 'object',
     properties: {
@@ -25,11 +25,15 @@ export const querySpaceTool: Tool = {
         type: 'string',
         description: 'Natural language question',
       },
-      space: {
-        type: 'string',
-        description: 'Which space to query',
-        enum: SUPPORTED_SPACES,
-        default: 'the_void',
+      spaces: {
+        type: 'array',
+        items: {
+          type: 'string',
+          enum: SUPPORTED_SPACES,
+        },
+        description: 'Spaces to query (e.g., ["the_void", "dogs"])',
+        minItems: 1,
+        default: ['the_void'],
       },
       content_type: {
         type: 'string',
@@ -54,6 +58,11 @@ export const querySpaceTool: Tool = {
         type: 'string',
         description: 'Filter memories created before this date (ISO 8601)',
       },
+      include_comments: {
+        type: 'boolean',
+        description: 'Include comments in query results (default: false)',
+        default: false,
+      },
       limit: {
         type: 'number',
         default: 10,
@@ -78,6 +87,7 @@ interface QuerySpaceArgs {
   min_weight?: number;
   date_from?: string;
   date_to?: string;
+  include_comments?: boolean;
   limit?: number;
   format?: 'detailed' | 'compact';
 }
@@ -134,6 +144,13 @@ export async function handleQuerySpace(
       filterList.push(publicCollection.filter.byProperty('type').equal(args.content_type));
     }
 
+    // Exclude comments by default (unless explicitly included)
+    if (!args.include_comments && !args.content_type) {
+      // Only exclude comments if not filtering by content_type
+      // (if content_type is set, user has explicit control)
+      filterList.push(publicCollection.filter.byProperty('type').notEqual('comment'));
+    }
+
     // Apply tags filter
     if (args.tags && args.tags.length > 0) {
       args.tags.forEach(tag => {

package/src/tools/search-space.ts

@@ -18,7 +18,7 @@ import type { SearchFilters } from '../types/memory.js';
  */
 export const searchSpaceTool: Tool = {
   name: 'remember_search_space',
-  description: 'Search one or more shared spaces to discover thoughts, ideas, and memories. Works like remember_search_memory but searches shared spaces instead of personal memories. Can search multiple spaces in a single query.',
+  description: 'Search one or more shared spaces to discover thoughts, ideas, and memories. By default, excludes comments to keep discovery clean. Set include_comments: true to include threaded discussions. Can search multiple spaces in a single query.',
   inputSchema: {
     type: 'object',
     properties: {
@@ -65,6 +65,11 @@ export const searchSpaceTool: Tool = {
         type: 'string',
         description: 'Filter memories created before this date (ISO 8601)',
       },
+      include_comments: {
+        type: 'boolean',
+        description: 'Include comments in search results (default: false)',
+        default: false,
+      },
       limit: {
         type: 'number',
         default: 10,
@@ -89,6 +94,7 @@ interface SearchSpaceArgs {
   max_weight?: number;
   date_from?: string;
   date_to?: string;
+  include_comments?: boolean;
   limit?: number;
   offset?: number;
 }
@@ -150,6 +156,13 @@ export async function handleSearchSpace(
       filterList.push(publicCollection.filter.byProperty('type').equal(args.content_type));
     }
 
+    // Exclude comments by default (unless explicitly included)
+    if (!args.include_comments && !args.content_type) {
+      // Only exclude comments if not filtering by content_type
+      // (if content_type is set, user has explicit control)
+      filterList.push(publicCollection.filter.byProperty('type').notEqual('comment'));
+    }
+
     // Apply tags filter
     if (args.tags && args.tags.length > 0) {
       args.tags.forEach(tag => {

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,
+  });
 }
 
 /**