@prmichaelsen/remember-mcp 2.3.1 → 2.5.0

package/dist/server-factory.js

@@ -3404,6 +3404,7 @@ var confirmationTokenService = new ConfirmationTokenService();
 // src/weaviate/space-schema.ts
 init_space_memory();
 import weaviate3 from "weaviate-client";
+var PUBLIC_COLLECTION_NAME = "Memory_public";
 function getSpaceCollectionName(spaceId) {
   return `Memory_${spaceId}`;
 }
@@ -3411,7 +3412,7 @@ function isValidSpaceId(spaceId) {
   return SUPPORTED_SPACES.includes(spaceId);
 }
 async function createSpaceCollection(client2, spaceId) {
-  const collectionName = getSpaceCollectionName(spaceId);
+  const collectionName = spaceId === "public" ? PUBLIC_COLLECTION_NAME : getSpaceCollectionName(spaceId);
   console.log(`[Weaviate] Creating space collection ${collectionName}...`);
   await client2.collections.create({
     name: collectionName,
@@ -3429,10 +3430,15 @@ async function createSpaceCollection(client2, spaceId) {
         description: 'Document type: "space_memory"'
       },
       // Space identity
+      {
+        name: "spaces",
+        dataType: "text[]",
+        description: 'Spaces this memory is published to (e.g., ["the_void", "dogs"])'
+      },
       {
         name: "space_id",
         dataType: "text",
-        description: 'Space identifier (e.g., "the_void")'
+        description: "DEPRECATED: Use spaces array instead. Will be removed in v3.0.0."
       },
       {
         name: "author_id",
@@ -3566,14 +3572,11 @@ async function createSpaceCollection(client2, spaceId) {
   });
   console.log(`[Weaviate] Space collection ${collectionName} created successfully`);
 }
-async function ensureSpaceCollection(client2, spaceId) {
-  if (!isValidSpaceId(spaceId)) {
-    throw new Error(`Invalid space ID: ${spaceId}. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`);
-  }
-  const collectionName = getSpaceCollectionName(spaceId);
+async function ensurePublicCollection(client2) {
+  const collectionName = PUBLIC_COLLECTION_NAME;
   const exists = await client2.collections.exists(collectionName);
   if (!exists) {
-    await createSpaceCollection(client2, spaceId);
+    await createSpaceCollection(client2, "public");
   }
   return client2.collections.get(collectionName);
 }
@@ -3582,7 +3585,7 @@ async function ensureSpaceCollection(client2, spaceId) {
 init_space_memory();
 var publishTool = {
   name: "remember_publish",
-  description: 'Publish a memory to a shared space (like "The Void"). The memory will be COPIED (not moved) from your personal collection. Generates a confirmation token. Use remember_confirm to execute.',
+  description: 'Publish a memory to one or more shared spaces (like "The Void"). The memory will be COPIED (not moved) from your personal collection. Generates a confirmation token. Use remember_confirm to execute.',
   inputSchema: {
     type: "object",
     properties: {
@@ -3590,11 +3593,15 @@ var publishTool = {
         type: "string",
         description: "ID of the memory from your personal collection to publish"
       },
-      target: {
-        type: "string",
-        description: "Target space to publish to (snake_case ID)",
-        enum: SUPPORTED_SPACES,
-        default: "the_void"
+      spaces: {
+        type: "array",
+        items: {
+          type: "string",
+          enum: SUPPORTED_SPACES
+        },
+        description: 'Spaces to publish to (e.g., ["the_void", "dogs"]). Can publish to multiple spaces at once.',
+        minItems: 1,
+        default: ["the_void"]
       },
       additional_tags: {
         type: "array",
@@ -3603,7 +3610,7 @@ var publishTool = {
         default: []
       }
     },
-    required: ["memory_id", "target"]
+    required: ["memory_id", "spaces"]
   }
 };
 async function handlePublish(args, userId) {
@@ -3611,18 +3618,21 @@ async function handlePublish(args, userId) {
     console.log("[remember_publish] Starting publish request:", {
       userId,
       memoryId: args.memory_id,
-      target: args.target,
+      spaces: args.spaces,
+      spaceCount: args.spaces.length,
       additionalTags: args.additional_tags?.length || 0
     });
-    if (!isValidSpaceId(args.target)) {
-      console.log("[remember_publish] Invalid space ID:", args.target);
+    const invalidSpaces = args.spaces.filter((s) => !isValidSpaceId(s));
+    if (invalidSpaces.length > 0) {
+      console.log("[remember_publish] Invalid space IDs:", invalidSpaces);
       return JSON.stringify(
         {
           success: false,
-          error: "Invalid space ID",
-          message: `Space "${args.target}" is not supported. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`,
+          error: "Invalid space IDs",
+          message: `Invalid spaces: ${invalidSpaces.join(", ")}. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`,
           context: {
-            provided_space: args.target,
+            invalid_spaces: invalidSpaces,
+            provided_spaces: args.spaces,
             supported_spaces: SUPPORTED_SPACES
           }
         },
@@ -3630,6 +3640,18 @@ async function handlePublish(args, userId) {
         2
       );
     }
+    if (args.spaces.length === 0) {
+      console.log("[remember_publish] Empty spaces array");
+      return JSON.stringify(
+        {
+          success: false,
+          error: "Empty spaces array",
+          message: "Must specify at least one space to publish to"
+        },
+        null,
+        2
+      );
+    }
     const weaviateClient = getWeaviateClient();
     const collectionName = getMemoryCollectionName(userId);
     console.log("[remember_publish] Fetching memory from collection:", collectionName);
@@ -3688,6 +3710,7 @@ async function handlePublish(args, userId) {
     }
     const payload = {
       memory_id: args.memory_id,
+      spaces: args.spaces,
       additional_tags: args.additional_tags || []
     };
     console.log("[remember_publish] Generating confirmation token");
@@ -3695,7 +3718,8 @@ async function handlePublish(args, userId) {
       userId,
       "publish_memory",
       payload,
-      args.target
+      void 0
+      // No single target_collection anymore
     );
     console.log("[remember_publish] Token generated:", {
       requestId,
@@ -3711,12 +3735,12 @@ async function handlePublish(args, userId) {
       2
     );
   } catch (error) {
-    handleToolError(error, {
+    return handleToolError(error, {
       toolName: "remember_publish",
       userId,
       operation: "publish memory",
       memory_id: args.memory_id,
-      target: args.target
+      spaces: args.spaces
     });
   }
 }
@@ -3778,7 +3802,8 @@ async function executePublishMemory(request, userId) {
     console.log("[executePublishMemory] Starting execution:", {
       userId,
       memoryId: request.payload.memory_id,
-      targetSpace: request.target_collection
+      spaces: request.payload.spaces,
+      spaceCount: request.payload.spaces?.length || 0
     });
     const weaviateClient = getWeaviateClient();
     const userCollection = weaviateClient.collections.get(
@@ -3816,20 +3841,18 @@ async function executePublishMemory(request, userId) {
         2
       );
     }
-    console.log("[executePublishMemory] Ensuring space collection:", request.target_collection || "the_void");
-    const targetCollection = await ensureSpaceCollection(
-      weaviateClient,
-      request.target_collection || "the_void"
-    );
-    console.log("[executePublishMemory] Space collection ready");
+    console.log("[executePublishMemory] Ensuring public collection");
+    const publicCollection = await ensurePublicCollection(weaviateClient);
+    console.log("[executePublishMemory] Public collection ready");
     const originalTags = Array.isArray(originalMemory.properties.tags) ? originalMemory.properties.tags : [];
     const additionalTags = Array.isArray(request.payload.additional_tags) ? request.payload.additional_tags : [];
     const publishedMemory = {
       ...originalMemory.properties,
-      // Override specific fields
-      space_id: request.target_collection || "the_void",
+      // Add space-specific fields
+      spaces: request.payload.spaces || ["the_void"],
+      // ✅ Array of spaces!
       author_id: userId,
-      // Always attributed
+      // Track original author
       published_at: (/* @__PURE__ */ new Date()).toISOString(),
       discovery_count: 0,
       doc_type: "space_memory",
@@ -3841,14 +3864,14 @@ async function executePublishMemory(request, userId) {
       updated_at: (/* @__PURE__ */ new Date()).toISOString(),
       version: 1
     };
-    console.log("[executePublishMemory] Inserting into space collection:", {
-      spaceId: request.target_collection || "the_void",
+    console.log("[executePublishMemory] Inserting into Memory_public:", {
+      spaces: request.payload.spaces,
+      spaceCount: request.payload.spaces?.length || 0,
       memoryId: request.payload.memory_id,
-      hasProperties: !!publishedMemory
-    });
-    const result = await targetCollection.data.insert({
-      properties: publishedMemory
+      hasUserId: !!publishedMemory.user_id,
+      hasAuthorId: !!publishedMemory.author_id
     });
+    const result = await publicCollection.data.insert(publishedMemory);
     console.log("[executePublishMemory] Insert result:", {
       success: !!result,
       spaceMemoryId: result
@@ -3856,7 +3879,8 @@ async function executePublishMemory(request, userId) {
     return JSON.stringify(
       {
         success: true,
-        space_memory_id: result
+        space_memory_id: result,
+        spaces: request.payload.spaces || ["the_void"]
       },
       null,
       2
@@ -3922,7 +3946,7 @@ import { Filters as Filters3 } from "weaviate-client";
 init_space_memory();
 var searchSpaceTool = {
   name: "remember_search_space",
-  description: "Search shared spaces to discover thoughts, ideas, and memories. Works like remember_search_memory but searches shared spaces instead of personal memories.",
+  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.",
   inputSchema: {
     type: "object",
     properties: {
@@ -3930,11 +3954,15 @@ var searchSpaceTool = {
         type: "string",
         description: "Search query (semantic + keyword hybrid)"
       },
-      space: {
-        type: "string",
-        description: "Which space to search",
-        enum: SUPPORTED_SPACES,
-        default: "the_void"
+      spaces: {
+        type: "array",
+        items: {
+          type: "string",
+          enum: SUPPORTED_SPACES
+        },
+        description: 'Spaces to search (e.g., ["the_void", "dogs"]). Can search multiple spaces at once.',
+        minItems: 1,
+        default: ["the_void"]
       },
       content_type: {
         type: "string",
@@ -3976,49 +4004,66 @@ var searchSpaceTool = {
         description: "Offset for pagination"
       }
     },
-    required: ["query", "space"]
+    required: ["query", "spaces"]
   }
 };
 async function handleSearchSpace(args, userId) {
   try {
-    if (!isValidSpaceId(args.space)) {
+    const invalidSpaces = args.spaces.filter((s) => !isValidSpaceId(s));
+    if (invalidSpaces.length > 0) {
       return JSON.stringify(
         {
           success: false,
-          error: "Invalid space ID",
-          message: `Space "${args.space}" is not supported. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`
+          error: "Invalid space IDs",
+          message: `Invalid spaces: ${invalidSpaces.join(", ")}. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`,
+          context: {
+            invalid_spaces: invalidSpaces,
+            provided_spaces: args.spaces,
+            supported_spaces: SUPPORTED_SPACES
+          }
+        },
+        null,
+        2
+      );
+    }
+    if (args.spaces.length === 0) {
+      return JSON.stringify(
+        {
+          success: false,
+          error: "Empty spaces array",
+          message: "Must specify at least one space to search"
         },
         null,
         2
       );
     }
     const weaviateClient = getWeaviateClient();
-    const spaceCollection = await ensureSpaceCollection(weaviateClient, args.space);
+    const publicCollection = await ensurePublicCollection(weaviateClient);
     const filterList = [];
-    filterList.push(spaceCollection.filter.byProperty("space_id").equal(args.space));
-    filterList.push(spaceCollection.filter.byProperty("doc_type").equal("space_memory"));
+    filterList.push(publicCollection.filter.byProperty("spaces").containsAny(args.spaces));
+    filterList.push(publicCollection.filter.byProperty("doc_type").equal("space_memory"));
     if (args.content_type) {
-      filterList.push(spaceCollection.filter.byProperty("type").equal(args.content_type));
+      filterList.push(publicCollection.filter.byProperty("type").equal(args.content_type));
     }
     if (args.tags && args.tags.length > 0) {
       args.tags.forEach((tag) => {
-        filterList.push(spaceCollection.filter.byProperty("tags").containsAny([tag]));
+        filterList.push(publicCollection.filter.byProperty("tags").containsAny([tag]));
       });
     }
     if (args.min_weight !== void 0) {
-      filterList.push(spaceCollection.filter.byProperty("weight").greaterOrEqual(args.min_weight));
+      filterList.push(publicCollection.filter.byProperty("weight").greaterOrEqual(args.min_weight));
     }
     if (args.max_weight !== void 0) {
-      filterList.push(spaceCollection.filter.byProperty("weight").lessOrEqual(args.max_weight));
+      filterList.push(publicCollection.filter.byProperty("weight").lessOrEqual(args.max_weight));
     }
     if (args.date_from) {
-      filterList.push(spaceCollection.filter.byProperty("created_at").greaterOrEqual(new Date(args.date_from)));
+      filterList.push(publicCollection.filter.byProperty("created_at").greaterOrEqual(new Date(args.date_from)));
     }
     if (args.date_to) {
-      filterList.push(spaceCollection.filter.byProperty("created_at").lessOrEqual(new Date(args.date_to)));
+      filterList.push(publicCollection.filter.byProperty("created_at").lessOrEqual(new Date(args.date_to)));
     }
     const whereFilter = filterList.length > 0 ? Filters3.and(...filterList) : void 0;
-    const searchResults = await spaceCollection.query.hybrid(args.query, {
+    const searchResults = await publicCollection.query.hybrid(args.query, {
       limit: args.limit || 10,
       offset: args.offset || 0,
       ...whereFilter && { where: whereFilter }
@@ -4029,7 +4074,7 @@ async function handleSearchSpace(args, userId) {
       _score: obj.metadata?.score
     }));
     const result = {
-      space: args.space,
+      spaces_searched: args.spaces,
       query: args.query,
       memories,
       total: memories.length,
@@ -4038,10 +4083,10 @@ async function handleSearchSpace(args, userId) {
     };
     return JSON.stringify(result, null, 2);
   } catch (error) {
-    handleToolError(error, {
+    return handleToolError(error, {
       toolName: "remember_search_space",
-      operation: "search space",
-      space: args.space,
+      operation: "search spaces",
+      spaces: args.spaces,
       query: args.query
     });
   }
@@ -4101,46 +4146,58 @@ var querySpaceTool = {
         description: "Output format: detailed (full objects) or compact (text summary)"
       }
     },
-    required: ["question", "space"]
+    required: ["question", "spaces"]
   }
 };
 async function handleQuerySpace(args, userId) {
   try {
-    if (!isValidSpaceId(args.space)) {
+    const invalidSpaces = args.spaces.filter((s) => !isValidSpaceId(s));
+    if (invalidSpaces.length > 0) {
       return JSON.stringify(
         {
           success: false,
-          error: "Invalid space ID",
-          message: `Space "${args.space}" is not supported. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`
+          error: "Invalid space IDs",
+          message: `Invalid spaces: ${invalidSpaces.join(", ")}. Supported spaces: ${SUPPORTED_SPACES.join(", ")}`
+        },
+        null,
+        2
+      );
+    }
+    if (args.spaces.length === 0) {
+      return JSON.stringify(
+        {
+          success: false,
+          error: "Empty spaces array",
+          message: "Must specify at least one space to query"
         },
         null,
         2
       );
     }
     const weaviateClient = getWeaviateClient();
-    const spaceCollection = await ensureSpaceCollection(weaviateClient, args.space);
+    const publicCollection = await ensurePublicCollection(weaviateClient);
     const filterList = [];
-    filterList.push(spaceCollection.filter.byProperty("space_id").equal(args.space));
-    filterList.push(spaceCollection.filter.byProperty("doc_type").equal("space_memory"));
+    filterList.push(publicCollection.filter.byProperty("spaces").containsAny(args.spaces));
+    filterList.push(publicCollection.filter.byProperty("doc_type").equal("space_memory"));
     if (args.content_type) {
-      filterList.push(spaceCollection.filter.byProperty("type").equal(args.content_type));
+      filterList.push(publicCollection.filter.byProperty("type").equal(args.content_type));
     }
     if (args.tags && args.tags.length > 0) {
       args.tags.forEach((tag) => {
-        filterList.push(spaceCollection.filter.byProperty("tags").containsAny([tag]));
+        filterList.push(publicCollection.filter.byProperty("tags").containsAny([tag]));
       });
     }
     if (args.min_weight !== void 0) {
-      filterList.push(spaceCollection.filter.byProperty("weight").greaterOrEqual(args.min_weight));
+      filterList.push(publicCollection.filter.byProperty("weight").greaterOrEqual(args.min_weight));
     }
     if (args.date_from) {
-      filterList.push(spaceCollection.filter.byProperty("created_at").greaterOrEqual(new Date(args.date_from)));
+      filterList.push(publicCollection.filter.byProperty("created_at").greaterOrEqual(new Date(args.date_from)));
     }
     if (args.date_to) {
-      filterList.push(spaceCollection.filter.byProperty("created_at").lessOrEqual(new Date(args.date_to)));
+      filterList.push(publicCollection.filter.byProperty("created_at").lessOrEqual(new Date(args.date_to)));
     }
     const whereFilter = filterList.length > 0 ? Filters4.and(...filterList) : void 0;
-    const searchResults = await spaceCollection.query.nearText(args.question, {
+    const searchResults = await publicCollection.query.nearText(args.question, {
       limit: args.limit || 10,
       ...whereFilter && { where: whereFilter }
     });
@@ -4152,7 +4209,7 @@ async function handleQuerySpace(args, userId) {
       });
       const result = {
         question: args.question,
-        space: args.space,
+        spaces_queried: args.spaces,
         format: "compact",
         summary: summaries.join("\n"),
         count: searchResults.objects.length
@@ -4166,7 +4223,7 @@ async function handleQuerySpace(args, userId) {
       }));
       const result = {
         question: args.question,
-        space: args.space,
+        spaces_queried: args.spaces,
         format: "detailed",
         memories,
         total: memories.length
@@ -4174,10 +4231,10 @@ async function handleQuerySpace(args, userId) {
       return JSON.stringify(result, null, 2);
     }
   } catch (error) {
-    handleToolError(error, {
+    return handleToolError(error, {
       toolName: "remember_query_space",
-      operation: "query space",
-      space: args.space,
+      operation: "query spaces",
+      spaces: args.spaces,
       question: args.question
     });
   }