@prmichaelsen/remember-mcp 2.3.4 → 2.5.1

package/dist/server-factory.js

@@ -3241,33 +3241,72 @@ var ConfirmationTokenService = class {
    * @returns Request ID and token
    */
   async createRequest(userId, action, payload, targetCollection) {
-    const token = randomUUID();
-    const now = /* @__PURE__ */ new Date();
-    const expiresAt = new Date(now.getTime() + this.EXPIRY_MINUTES * 60 * 1e3);
-    const request = {
-      user_id: userId,
-      token,
-      action,
-      target_collection: targetCollection,
-      payload,
-      created_at: now.toISOString(),
-      expires_at: expiresAt.toISOString(),
-      status: "pending"
-    };
-    const collectionPath = `users/${userId}/requests`;
-    console.log("[ConfirmationTokenService] Creating request:", {
-      userId,
-      action,
-      targetCollection,
-      collectionPath
-    });
-    const docRef = await addDocument(collectionPath, request);
-    console.log("[ConfirmationTokenService] Request created:", {
-      requestId: docRef.id,
-      token,
-      expiresAt: request.expires_at
-    });
-    return { requestId: docRef.id, token };
+    try {
+      const token = randomUUID();
+      const now = /* @__PURE__ */ new Date();
+      const expiresAt = new Date(now.getTime() + this.EXPIRY_MINUTES * 60 * 1e3);
+      const request = {
+        user_id: userId,
+        token,
+        action,
+        target_collection: targetCollection,
+        payload,
+        created_at: now.toISOString(),
+        expires_at: expiresAt.toISOString(),
+        status: "pending"
+      };
+      const collectionPath = `users/${userId}/requests`;
+      console.log("[ConfirmationTokenService] Creating request:", {
+        userId,
+        action,
+        targetCollection,
+        collectionPath,
+        requestData: {
+          token,
+          action,
+          payloadKeys: Object.keys(payload)
+        }
+      });
+      console.log("[ConfirmationTokenService] Calling addDocument...");
+      const docRef = await addDocument(collectionPath, request);
+      console.log("[ConfirmationTokenService] addDocument returned:", {
+        hasDocRef: !!docRef,
+        hasId: !!docRef?.id,
+        docRefId: docRef?.id
+      });
+      if (!docRef) {
+        const error = new Error("Firestore addDocument returned null/undefined");
+        console.error("[ConfirmationTokenService] CRITICAL: addDocument returned null", {
+          userId,
+          collectionPath
+        });
+        throw error;
+      }
+      if (!docRef.id) {
+        const error = new Error("Firestore addDocument returned docRef without ID");
+        console.error("[ConfirmationTokenService] CRITICAL: docRef has no ID", {
+          userId,
+          collectionPath,
+          docRef
+        });
+        throw error;
+      }
+      console.log("[ConfirmationTokenService] Request created successfully:", {
+        requestId: docRef.id,
+        token,
+        expiresAt: request.expires_at
+      });
+      return { requestId: docRef.id, token };
+    } catch (error) {
+      console.error("[ConfirmationTokenService] FAILED to create request:", {
+        error: error instanceof Error ? error.message : String(error),
+        stack: error instanceof Error ? error.stack : void 0,
+        userId,
+        action,
+        collectionPath: `users/${userId}/requests`
+      });
+      throw error;
+    }
   }
   /**
    * Validate and retrieve a confirmation request
@@ -3404,6 +3443,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 +3451,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 +3469,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 +3611,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 +3624,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 +3632,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 +3649,7 @@ var publishTool = {
         default: []
       }
     },
-    required: ["memory_id", "target"]
+    required: ["memory_id", "spaces"]
   }
 };
 async function handlePublish(args, userId) {
@@ -3611,18 +3657,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 +3679,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 +3749,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 +3757,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 +3774,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 +3841,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,18 +3880,16 @@ 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,
       // Add space-specific fields
-      space_id: request.target_collection || "the_void",
+      spaces: request.payload.spaces || ["the_void"],
+      // ✅ Array of spaces!
       author_id: userId,
       // Track original author
       published_at: (/* @__PURE__ */ new Date()).toISOString(),
@@ -3841,14 +3903,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,
       hasUserId: !!publishedMemory.user_id,
-      hasAuthorId: !!publishedMemory.author_id,
-      hasSpaceId: !!publishedMemory.space_id
+      hasAuthorId: !!publishedMemory.author_id
     });
-    const result = await targetCollection.data.insert(publishedMemory);
+    const result = await publicCollection.data.insert(publishedMemory);
     console.log("[executePublishMemory] Insert result:", {
       success: !!result,
       spaceMemoryId: result
@@ -3856,7 +3918,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 +3985,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 +3993,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 +4043,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 +4113,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 +4122,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 +4185,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 +4248,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 +4262,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 +4270,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
     });
   }