feat: add client for /mcp endpoint

Author: jirispilka

src/examples/clientStreamableHttp.ts

@@ -0,0 +1,109 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import type { CallToolRequest, ListToolsRequest } from '@modelcontextprotocol/sdk/types.js';
+import {
+    CallToolResultSchema,
+    ListToolsResultSchema,
+    LoggingMessageNotificationSchema,
+} from '@modelcontextprotocol/sdk/types.js';
+
+import log from '@apify/log';
+
+log.setLevel(log.LEVELS.DEBUG);
+
+async function main(): Promise<void> {
+    // Create a new client with streamable HTTP transport
+    const client = new Client({
+        name: 'example-client',
+        version: '1.0.0',
+    });
+
+    const transport = new StreamableHTTPClientTransport(
+        new URL('http://localhost:3000/mcp'),
+    );
+
+    // Connect the client using the transport and initialize the server
+    await client.connect(transport);
+    log.debug('Connected to MCP server');
+
+    // Set up notification handlers for server-initiated messages
+    client.setNotificationHandler(LoggingMessageNotificationSchema, (notification) => {
+        log.debug(`Notification received: ${notification.params.level} - ${notification.params.data}`);
+    });
+
+    // List and call tools
+    await listTools(client);
+
+    await callSearchTool(client);
+    await callActor(client);
+
+    // Keep the connection open to receive notifications
+    log.debug('\nKeeping connection open to receive notifications. Press Ctrl+C to exit.');
+}
+
+async function listTools(client: Client): Promise<void> {
+    try {
+        const toolsRequest: ListToolsRequest = {
+            method: 'tools/list',
+            params: {},
+        };
+        const toolsResult = await client.request(toolsRequest, ListToolsResultSchema);
+        log.debug(`Tools available, count: ${toolsResult.tools.length}`);
+        for (const tool of toolsResult.tools) {
+            log.debug(`Tool: ${tool.name}, Description: ${tool.description}`);
+        }
+        if (toolsResult.tools.length === 0) {
+            log.debug('No tools available from the server');
+        }
+    } catch (error) {
+        log.error(`Tools not supported by this server (${error})`);
+    }
+}
+
+async function callSearchTool(client: Client): Promise<void> {
+    try {
+        const searchRequest: CallToolRequest = {
+            method: 'tools/call',
+            params: {
+                name: 'search',
+                arguments: { search: 'rag web browser', limit: 1 },
+            },
+        };
+        const searchResult = await client.request(searchRequest, CallToolResultSchema);
+        log.debug('Search result:');
+        searchResult.content.forEach((item) => {
+            if (item.type === 'text') {
+                log.debug(`\t${item.text}`);
+            }
+        });
+    } catch (error) {
+        log.error(`Error calling greet tool: ${error}`);
+    }
+}
+
+async function callActor(client: Client): Promise<void> {
+    try {
+        log.debug('\nCalling Actor...');
+        const actorRequest: CallToolRequest = {
+            method: 'tools/call',
+            params: {
+                name: 'apify/rag-web-browser',
+                arguments: { query: 'apify mcp server' },
+            },
+        };
+        const actorResult = await client.request(actorRequest, CallToolResultSchema);
+        log.debug('Actor results:');
+        actorResult.content.forEach((item) => {
+            if (item.type === 'text') {
+                log.debug(`- ${item.text}`);
+            }
+        });
+    } catch (error) {
+        log.error(`Error calling Actor: ${error}`);
+    }
+}
+
+main().catch((error: unknown) => {
+    log.error('Error running MCP client:', error as Error);
+    process.exit(1);
+});

src/index.ts

@@ -5,5 +5,4 @@
 
 import { ActorsMcpServer } from './mcp-server.js';
 
-// eslint-disable-next-line import/no-default-export
-export default ActorsMcpServer;
+export { ActorsMcpServer };

src/main.ts

@@ -12,14 +12,20 @@ import { processInput } from './actor/input.js';
 import { createExpressApp } from './actor/server.js';
 import type { Input } from './actor/types';
 import { ActorsMcpServer } from './mcp-server.js';
-import { actorDefinitionTool, addTool, callActorGetDataset, removeTool, searchTool } from './tools/index.js';
+import {
+    actorDefinitionTool,
+    addTool,
+    callActorGetDataset,
+    removeTool,
+    searchTool,
+} from './tools/index.js';
 
 const STANDBY_MODE = Actor.getEnv().metaOrigin === 'STANDBY';
 
 await Actor.init();
 
 const HOST = Actor.isAtHome() ? process.env.ACTOR_STANDBY_URL as string : 'http://localhost';
-const PORT = Actor.isAtHome() ? Number(process.env.ACTOR_STANDBY_PORT) : 3001;
+const PORT = Actor.isAtHome() ? Number(process.env.ACTOR_STANDBY_PORT) : 3000;
 
 if (!process.env.APIFY_TOKEN) {
     log.error('APIFY_TOKEN is required but not set in the environment variables.');