fix: load default Actors for the /mcp route (#86)

fix: load default Actors for /mcp route

Author: jirispilka

package-lock.json

@@ -31,7 +31,7 @@
   ],
   "dependencies": {
     "@apify/log": "^2.5.16",
-    "@modelcontextprotocol/sdk": "^1.10.0",
+    "@modelcontextprotocol/sdk": "^1.10.1",
     "ajv": "^8.17.1",
     "apify": "^3.4.0",
     "apify-client": "^2.12.1",

package.json

@@ -121,6 +121,14 @@ export function createExpressApp(
                     sessionIdGenerator: () => randomUUID(),
                     enableJsonResponse: true, // Enable JSON response mode
                 });
+                // Load MCP server tools
+                // TODO using query parameters in POST request is not standard
+                const urlSearchParams = new URLSearchParams(req.url.split('?')[1]);
+                if (urlSearchParams.get('actors')) {
+                    await mcpServer.loadToolsFromUrl(req.url, process.env.APIFY_TOKEN as string);
+                } else {
+                    await mcpServer.loadDefaultTools(process.env.APIFY_TOKEN as string);
+                }
 
                 // Connect the transport to the MCP server BEFORE handling the request
                 await mcpServer.connect(transport);

src/actor/server.ts

@@ -57,7 +57,10 @@ export class ActorsMcpServer {
     public async loadDefaultTools(apifyToken: string) {
         const missingDefaultTools = defaults.actors.filter((name) => !this.tools.has(actorNameToToolName(name)));
         const tools = await getActorsAsTools(missingDefaultTools, apifyToken);
-        if (tools.length > 0) this.updateTools(tools);
+        if (tools.length > 0) {
+            log.info('Loading default tools...');
+            this.updateTools(tools);
+        }
     }
 
     /**
@@ -69,7 +72,10 @@ export class ActorsMcpServer {
      */
     public async loadToolsFromUrl(url: string, apifyToken: string) {
         const tools = await processParamsGetTools(url, apifyToken);
-        if (tools.length > 0) this.updateTools(tools);
+        if (tools.length > 0) {
+            log.info('Loading tools from query parameters...');
+            this.updateTools(tools);
+        }
     }
 
     /**