@prmichaelsen/remember-mcp 3.12.0 → 3.14.0

package/src/tools/revise.ts

@@ -9,18 +9,10 @@
  */
 
 import type { Tool } from '@modelcontextprotocol/sdk/types.js';
-import {
-  getWeaviateClient,
-  getMemoryCollectionName,
-  fetchMemoryWithAllProperties,
-} from '../weaviate/client.js';
-import { confirmationTokenService } from '../services/confirmation-token.service.js';
 import { handleToolError } from '../utils/error-handler.js';
-import { logger } from '../utils/logger.js';
 import { createDebugLogger } from '../utils/debug.js';
-import { CollectionType, getCollectionName } from '../collections/dot-notation.js';
-import { generateCompositeId } from '../collections/composite-ids.js';
 import type { AuthContext } from '../types/auth.js';
+import { createCoreServices } from '../core-services.js';
 
 /** Maximum number of revision history entries to retain */
 const MAX_REVISION_HISTORY = 10;
@@ -128,141 +120,17 @@ export async function handleRevise(
     debug.info('Tool invoked');
     debug.trace('Arguments', { args });
 
-    logger.info('Starting revise request', {
-      tool: 'remember_revise',
-      userId,
-      memoryId: args.memory_id,
-    });
-
-    // Load source memory from user's personal collection
-    const weaviateClient = getWeaviateClient();
-    const userCollectionName = getMemoryCollectionName(userId);
-    const userCollection = weaviateClient.collections.get(userCollectionName);
-
-    const sourceMemory = await fetchMemoryWithAllProperties(
-      userCollection,
-      args.memory_id
-    );
-
-    if (!sourceMemory) {
-      logger.info('Source memory not found', {
-        tool: 'remember_revise',
-        memoryId: args.memory_id,
-      });
-      return JSON.stringify(
-        {
-          success: false,
-          error: 'Memory not found',
-          message: `Memory ${args.memory_id} does not exist`,
-        },
-        null,
-        2
-      );
-    }
-
-    // Verify ownership
-    if (sourceMemory.properties.user_id !== userId) {
-      logger.warn('Permission denied', {
-        tool: 'remember_revise',
-        memoryId: args.memory_id,
-        memoryOwner: sourceMemory.properties.user_id,
-        requestingUser: userId,
-      });
-      return JSON.stringify(
-        {
-          success: false,
-          error: 'Permission denied',
-          message: 'You can only revise your own memories',
-        },
-        null,
-        2
-      );
-    }
-
-    // Get tracking arrays
-    const spaceIds: string[] = Array.isArray(sourceMemory.properties.space_ids)
-      ? sourceMemory.properties.space_ids
-      : [];
-    const groupIds: string[] = Array.isArray(sourceMemory.properties.group_ids)
-      ? sourceMemory.properties.group_ids
-      : [];
-
-    // Validate the memory is published somewhere
-    if (spaceIds.length === 0 && groupIds.length === 0) {
-      logger.info('Memory has no published copies', {
-        tool: 'remember_revise',
-        memoryId: args.memory_id,
-      });
-      return JSON.stringify(
-        {
-          success: false,
-          error: 'Not published',
-          message:
-            'Memory has no published copies to revise. Publish first with remember_publish.',
-          context: {
-            memory_id: args.memory_id,
-            space_ids: [],
-            group_ids: [],
-          },
-        },
-        null,
-        2
-      );
-    }
-
-    // Create payload for confirmation token
-    const payload = {
+    const { space } = createCoreServices(userId);
+    const result = await space.revise({
       memory_id: args.memory_id,
-      space_ids: spaceIds,
-      group_ids: groupIds,
-    };
-
-    logger.info('Generating confirmation token for revise', {
-      tool: 'remember_revise',
-      userId,
-      memoryId: args.memory_id,
-      spaceIds,
-      groupIds,
-    });
-
-    // Generate confirmation token
-    const { requestId, token } = await confirmationTokenService.createRequest(
-      userId,
-      'revise_memory',
-      payload
-    );
-
-    logger.info('Confirmation token generated for revise', {
-      tool: 'remember_revise',
-      requestId,
-      token,
-      action: 'revise_memory',
-      spaceIds,
-      groupIds,
     });
 
-    // Build destination summary for user
-    const destinations: string[] = [];
-    if (spaceIds.length > 0) {
-      destinations.push(`spaces: ${spaceIds.join(', ')}`);
-    }
-    if (groupIds.length > 0) {
-      destinations.push(`groups: ${groupIds.join(', ')}`);
-    }
-
     return JSON.stringify(
       {
         success: true,
-        token,
-        message: 'Revision request created. Please confirm to sync content to all published copies.',
+        token: result.token,
         action: 'revise_memory',
         memory_id: args.memory_id,
-        destinations: destinations.join('; '),
-        revision_details: {
-          space_ids: spaceIds,
-          group_ids: groupIds,
-          total_locations: (spaceIds.length > 0 ? 1 : 0) + groupIds.length,
-        },
         confirmation_required: true,
       },
       null,

package/src/tools/search-relationship.ts

@@ -3,14 +3,11 @@
  * Search relationships by observation text or type
  */
 
-import { Filters } from 'weaviate-client';
 import type { Relationship, DeletedFilter } from '../types/memory.js';
-import { getMemoryCollection } from '../weaviate/schema.js';
-import { logger } from '../utils/logger.js';
 import { handleToolError } from '../utils/error-handler.js';
 import { createDebugLogger } from '../utils/debug.js';
-import { buildDeletedFilter, combineFiltersWithAnd } from '../utils/weaviate-filters.js';
 import type { AuthContext } from '../types/auth.js';
+import { createCoreServices } from '../core-services.js';
 
 /**
  * Tool definition for remember_search_relationship
@@ -118,123 +115,27 @@ export async function handleSearchRelationship(
     debug.info('Tool invoked');
     debug.trace('Arguments', { args });
 
-    logger.info('Searching relationships', {
-      userId,
+    const { relationship } = createCoreServices(userId);
+    const result = await relationship.search({
       query: args.query,
-      types: args.relationship_types
-    });
-
-    const collection = getMemoryCollection(userId);
-    const limit = args.limit ?? 10;
-    const offset = args.offset ?? 0;
-
-    // Build deleted filter
-    const deletedFilter = buildDeletedFilter(collection, args.deleted_filter || 'exclude');
-
-    // Build filters using Weaviate v3 API
-    const filterList: any[] = [];
-
-    // Add deleted filter if present
-    if (deletedFilter) {
-      filterList.push(deletedFilter);
-    }
-
-    // Always filter by doc_type = 'relationship'
-    filterList.push(
-      collection.filter.byProperty('doc_type').equal('relationship')
-    );
-
-    // Add relationship type filter
-    if (args.relationship_types && args.relationship_types.length > 0) {
-      if (args.relationship_types.length === 1) {
-        filterList.push(
-          collection.filter.byProperty('relationship_type').equal(args.relationship_types[0])
-        );
-      } else {
-        // Multiple types: use OR logic
-        const typeFilters = args.relationship_types.map(type =>
-          collection.filter.byProperty('relationship_type').equal(type)
-        );
-        filterList.push(Filters.or(...typeFilters));
-      }
-    }
-
-    // Add strength filter
-    if (args.strength_min !== undefined) {
-      filterList.push(
-        collection.filter.byProperty('strength').greaterOrEqual(args.strength_min)
-      );
-    }
-
-    // Add confidence filter
-    if (args.confidence_min !== undefined) {
-      filterList.push(
-        collection.filter.byProperty('confidence').greaterOrEqual(args.confidence_min)
-      );
-    }
-
-    // Add tags filter
-    if (args.tags && args.tags.length > 0) {
-      filterList.push(
-        collection.filter.byProperty('tags').containsAny(args.tags)
-      );
-    }
-
-    // Combine all filters with AND logic using the helper
-    const combinedFilters = combineFiltersWithAnd(filterList);
-
-    // Build search options
-    const searchOptions: any = {
-      alpha: 1.0, // Pure semantic search for relationships
-      limit: limit + offset, // Get extra for offset
-    };
-
-    // Add filters
-    if (combinedFilters) {
-      searchOptions.filters = combinedFilters;
-    }
-
-    // Perform hybrid search (semantic search on observation field)
-    const results = await collection.query.hybrid(args.query, searchOptions);
-
-    // Apply offset manually (Weaviate v4 doesn't have built-in offset for nearText)
-    const paginatedResults = results.objects.slice(offset, offset + limit);
-
-    // Map results to Relationship type
-    const relationships: Relationship[] = paginatedResults.map((obj: any) => ({
-      id: obj.uuid,
-      user_id: obj.properties.user_id,
-      doc_type: 'relationship',
-      memory_ids: obj.properties.related_memory_ids || [],
-      relationship_type: obj.properties.relationship_type,
-      observation: obj.properties.observation,
-      strength: obj.properties.strength,
-      confidence: obj.properties.confidence,
-      context: obj.properties.context || {
-        timestamp: obj.properties.created_at,
-        source: { type: 'api', platform: 'mcp' },
-      },
-      created_at: obj.properties.created_at,
-      updated_at: obj.properties.updated_at,
-      version: obj.properties.version,
-      tags: obj.properties.tags || [],
-    }));
-
-    logger.info('Relationship search completed', { 
-      userId, 
-      found: relationships.length,
-      total: results.objects.length
+      relationship_types: args.relationship_types,
+      strength_min: args.strength_min,
+      confidence_min: args.confidence_min,
+      tags: args.tags,
+      limit: args.limit,
+      offset: args.offset,
+      deleted_filter: args.deleted_filter,
     });
 
-    const result: SearchRelationshipResult = {
-      relationships,
-      total: results.objects.length,
-      offset,
-      limit,
-      message: `Found ${relationships.length} relationship(s) matching query "${args.query}"`,
+    const response: SearchRelationshipResult = {
+      relationships: result.relationships as unknown as Relationship[],
+      total: result.total,
+      offset: result.offset,
+      limit: result.limit,
+      message: `Found ${result.relationships.length} relationship(s) matching query "${args.query}"`,
     };
 
-    return JSON.stringify(result, null, 2);
+    return JSON.stringify(response, null, 2);
   } catch (error) {
     debug.error('Tool failed', { error: error instanceof Error ? error.message : String(error) });
     handleToolError(error, {