remove toolkits (move to tools index), pass apify token as arg, clean up old tools.ts, separate Actor readme

Author: MQ37

.actor/ACTOR.md

@@ -5,5 +5,6 @@
   "description": "Implementation of a Model Context Protocol (MCP) Server for Apify Actors that enables AI applications (and AI agents) to interact with Apify Actors",
   "version": "0.1",
   "input": "./input_schema.json",
+  "readme": "./ACTOR.md",
   "dockerfile": "./Dockerfile"
 }

.actor/actor.json

@@ -1,9 +1,10 @@
 import { Ajv } from 'ajv';
+import type { ActorStoreList } from 'apify-client';
 import { ApifyClient } from 'apify-client';
 
 import { ACTOR_ADDITIONAL_INSTRUCTIONS, defaults, MAX_DESCRIPTION_LENGTH, ACTOR_README_MAX_LENGTH, ACTOR_ENUM_MAX_LENGTH } from './const.js';
 import { log } from './logger.js';
-import type { ActorDefinitionPruned, ActorDefinitionWithDesc, IActorInputSchema, ISchemaProperties, ToolWrap } from './types.js';
+import type { ActorStorePruned, PricingInfo, ActorDefinitionPruned, ActorDefinitionWithDesc, IActorInputSchema, ISchemaProperties, ToolWrap } from './types.js';
 
 export function actorNameToToolName(actorName: string): string {
     return actorName
@@ -349,3 +350,39 @@ export async function getActorsAsTools(actors: string[]): Promise<ToolWrap[]> {
     }
     return tools;
 }
+
+function pruneActorStoreInfo(response: ActorStoreList): ActorStorePruned {
+    const stats = response.stats || {};
+    const pricingInfo = (response.currentPricingInfo || {}) as PricingInfo;
+    return {
+        id: response.id,
+        name: response.name?.toString() || '',
+        username: response.username?.toString() || '',
+        actorFullName: `${response.username}/${response.name}`,
+        title: response.title?.toString() || '',
+        description: response.description?.toString() || '',
+        stats: {
+            totalRuns: stats.totalRuns,
+            totalUsers30Days: stats.totalUsers30Days,
+            publicActorRunStats30Days: 'publicActorRunStats30Days' in stats
+                ? stats.publicActorRunStats30Days : {},
+        },
+        currentPricingInfo: {
+            pricingModel: pricingInfo.pricingModel?.toString() || '',
+            pricePerUnitUsd: pricingInfo?.pricePerUnitUsd ?? 0,
+            trialMinutes: pricingInfo?.trialMinutes ?? 0,
+        },
+        url: response.url?.toString() || '',
+        totalStars: 'totalStars' in response ? (response.totalStars as number) : null,
+    };
+}
+
+export async function searchActorsByKeywords(
+    search: string,
+    limit: number | undefined = undefined,
+    offset: number | undefined = undefined,
+): Promise<ActorStorePruned[] | null> {
+    const client = new ApifyClient({ token: process.env.APIFY_TOKEN });
+    const results = await client.store().list({ search, limit, offset });
+    return results.items.map((x) => pruneActorStoreInfo(x));
+}

src/actors.ts

@@ -4,5 +4,5 @@
 
 export { createExpressApp } from './server.js';
 export { ApifyMcpServer } from './mcp-server.js';
-export { getActorAutoLoadingTools, getActorDiscoveryTools } from './toolkits/index.js';
+export { getActorAutoLoadingTools, getActorDiscoveryTools } from './tools/index.js';
 export { addActorToTools, discoverActorsTool, getActorsDetailsTool, removeActorFromTools } from './tools/index.js';

src/index.ts

@@ -9,8 +9,7 @@ import { processInput } from './input.js';
 import { log } from './logger.js';
 import { ApifyMcpServer } from './mcp-server.js';
 import { createExpressApp } from './server.js';
-import { getActorAutoLoadingTools } from './toolkits/actor-auto-loading-tools.js';
-import { getActorDiscoveryTools } from './toolkits/discovery-tools.js';
+import { getActorAutoLoadingTools, getActorDiscoveryTools } from './tools/index.js';
 import type { Input } from './types.js';
 import { isActorStandby } from './utils.js';
 
@@ -70,6 +69,6 @@ if (isActorStandby()) {
         await Actor.fail('If you need to debug a specific actor, please provide the debugActor and debugActorInput fields in the input');
     }
     const options = { memory: input.maxActorMemoryBytes } as ActorCallOptions;
-    await mcpServer.callActorGetDataset(input.debugActor!, input.debugActorInput!, options);
+    await mcpServer.callActorGetDataset(input.debugActor!, input.debugActorInput!, process.env.APIFY_TOKEN, options);
     await Actor.exit();
 }

src/main.ts

@@ -26,7 +26,7 @@ import {
 } from './const.js';
 import { processInput } from './input.js';
 import { log } from './logger.js';
-import { getActorAutoLoadingTools } from './toolkits/actor-auto-loading-tools.js';
+import { getActorAutoLoadingTools } from './tools/index.js';
 import type { Input, ActorTool, ToolWrap, InternalTool } from './types.js';
 
 /**
@@ -80,14 +80,15 @@ export class ApifyMcpServer {
     public async callActorGetDataset(
         actorName: string,
         input: unknown,
+        apifyToken: string,
         callOptions: ActorCallOptions | undefined = undefined,
     ): Promise<object[]> {
         const name = actorName;
         try {
             log.info(`Calling actor ${name} with input: ${JSON.stringify(input)}`);
 
             const options: ApifyClientOptions = { requestInterceptors: [this.addUserAgent] };
-            const client = new ApifyClient({ ...options, token: process.env.APIFY_TOKEN });
+            const client = new ApifyClient({ ...options, token: apifyToken });
             const actorClient = client.actor(name);
 
             const results = await actorClient.call(input, callOptions);
@@ -172,7 +173,7 @@ export class ApifyMcpServer {
          * @throws {Error} - Throws an error if the tool is unknown or arguments are invalid.
          */
         this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-            const { name, arguments: args } = request.params;
+            const { name, arguments: args, apifyToken } = request.params;
 
             const tool = Array.from(this.tools.values())
                 .find((t) => t.tool.name === name || (t.type === 'actor' && (t.tool as ActorTool).actorFullName === name));
@@ -203,7 +204,9 @@ export class ApifyMcpServer {
                 if (tool.type === 'actor') {
                     const actorTool = tool.tool as ActorTool;
 
-                    const items = await this.callActorGetDataset(actorTool.actorFullName, args, { memory: actorTool.memoryMbytes } as ActorCallOptions);
+                    const items = await this.callActorGetDataset(actorTool.actorFullName, args, apifyToken as string, {
+                        memory: actorTool.memoryMbytes,
+                    } as ActorCallOptions);
                     const content = items.map((item) => {
                         const text = JSON.stringify(item).slice(0, ACTOR_OUTPUT_MAX_CHARS_PER_ITEM);
                         return text.length === ACTOR_OUTPUT_MAX_CHARS_PER_ITEM

src/mcp-server.ts

@@ -1,3 +1,7 @@
+/*
+ * This file contains the express server implementation used for standby Actor mode.
+ */
+
 import { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';
 import type { Request, Response } from 'express';
 import express from 'express';

src/server.ts

@@ -17,8 +17,7 @@ import minimist from 'minimist';
 
 import { log } from './logger.js';
 import { ApifyMcpServer } from './mcp-server.js';
-import { getActorAutoLoadingTools } from './toolkits/actor-auto-loading-tools.js';
-import { getActorDiscoveryTools } from './toolkits/discovery-tools.js';
+import { getActorAutoLoadingTools, getActorDiscoveryTools } from './tools/index.js';
 
 log.setLevel(log.LEVELS.ERROR);