meMCP - Memory-Enhanced Model Context Protocol

Implementation Reference

• src/tools/modules/MemoryStreamingTools.js:256-285 (handler)

  Executes the memory_stream_cancel tool: calls streamingManager.cancelStream(streamId) and returns formatted JSON response with success/error status.

  async handleStreamCancel(args) {
    try {
      const result = await this.streamingManager.cancelStream(args.streamId);
      
      return {
        content: [
          {
            type: 'text',
            text: JSON.stringify({
              success: true,
              ...result,
            }),
          },
        ],
      };
    } catch (error) {
      return {
        content: [
          {
            type: 'text',
            text: JSON.stringify({
              success: false,
              error: error.message,
            }),
          },
        ],
        isError: true,
      };
    }
  }

• src/tools/modules/MemoryStreamingTools.js:97-114 (registration)

  Registers the memory_stream_cancel tool on the MCP server, defining its name, description, input schema (streamId required), and linking to handleStreamCancel handler.

  registerStreamCancelTool(server) {
    server.registerTool(
      'memory_stream_cancel',
      'Cancel an active streaming query',
      {
        type: 'object',
        properties: {
          streamId: {
            type: 'string',
            description: 'The stream ID to cancel',
          },
        },
        required: ['streamId'],
      },
      async (args) => {
        return await this.handleStreamCancel(args);
      }
    );

• src/tools/modules/MemoryStreamingTools.js:102-110 (schema)

  Input schema for memory_stream_cancel tool: requires a streamId string.

    type: 'object',
    properties: {
      streamId: {
        type: 'string',
        description: 'The stream ID to cancel',
      },
    },
    required: ['streamId'],
  },

• src/streaming/StreamingManager.js:129-144 (helper)

  Core implementation of stream cancellation: marks the stream as 'cancelled', sets end time, and returns status summary. Invoked by the tool handler.

  async cancelStream(streamId) {
    const stream = this.activeStreams.get(streamId);
    if (!stream) {
      throw new Error(`Stream ${streamId} not found`);
    }
  
    stream.status = 'cancelled';
    stream.endTime = Date.now();
  
    return {
      streamId,
      status: 'cancelled',
      processed: stream.currentIndex,
      total: stream.totalFacts,
    };
  }

• src/tools/MemoryTools.js:23-28 (registration)

  MemoryTools.registerTools calls streamingTools.registerTools(server), which registers the memory_stream_cancel tool among others.

  async registerTools(server) {
    // Register tools from modular components
    this.operations.registerTools(server);
    this.queryHandler.registerTools(server);
    this.streamingTools.registerTools(server);
    this.management.registerTools(server);