@apify/actors-mcp-server 0.1.22-beta.14 → 0.1.22-beta.15

package/dist/server.js

@@ -0,0 +1,178 @@
+#!/usr/bin/env node
+/**
+ * Model Context Protocol (MCP) server for Apify Actors
+ */
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { Actor } from 'apify';
+import { ApifyClient } from 'apify-client';
+import { filterSchemaProperties, getActorDefinition, getActorsAsTools, shortenProperties, } from './actors.js';
+import { ACTOR_OUTPUT_MAX_CHARS_PER_ITEM, ACTOR_OUTPUT_TRUNCATED_MESSAGE, defaults, InternalTools, SERVER_NAME, SERVER_VERSION, USER_AGENT_ORIGIN, } from './const.js';
+import { log } from './logger.js';
+import { RemoveActorToolArgsSchema, AddActorToToolsArgsSchema, DiscoverActorsArgsSchema, searchActorsByKeywords, GetActorDefinition, } from './tools.js';
+/**
+ * Create Apify MCP server
+ */
+export class ApifyMcpServer {
+    server;
+    tools;
+    constructor() {
+        this.server = new Server({
+            name: SERVER_NAME,
+            version: SERVER_VERSION,
+        }, {
+            capabilities: {
+                tools: { listChanged: true },
+            },
+        });
+        this.tools = new Map();
+        this.setupErrorHandling();
+        this.setupToolHandlers();
+    }
+    /**
+     * Adds a User-Agent header to the request config.
+     * @param config
+     * @private
+     */
+    addUserAgent(config) {
+        const updatedConfig = { ...config };
+        updatedConfig.headers = updatedConfig.headers ?? {};
+        updatedConfig.headers['User-Agent'] = `${updatedConfig.headers['User-Agent'] ?? ''}; ${USER_AGENT_ORIGIN}`;
+        return updatedConfig;
+    }
+    /**
+     * Calls an Apify actor and retrieves the dataset items.
+     *
+     * It requires the `APIFY_TOKEN` environment variable to be set.
+     * If the `APIFY_IS_AT_HOME` the dataset items are pushed to the Apify dataset.
+     *
+     * @param {string} actorName - The name of the actor to call.
+     * @param {ActorCallOptions} callOptions - The options to pass to the actor.
+     * @param {unknown} input - The input to pass to the actor.
+     * @returns {Promise<object[]>} - A promise that resolves to an array of dataset items.
+     * @throws {Error} - Throws an error if the `APIFY_TOKEN` is not set
+     */
+    async callActorGetDataset(actorName, input, callOptions = undefined) {
+        const name = actorName;
+        try {
+            log.info(`Calling actor ${name} with input: ${JSON.stringify(input)}`);
+            const options = { requestInterceptors: [this.addUserAgent] };
+            const client = new ApifyClient({ ...options, token: process.env.APIFY_TOKEN });
+            const actorClient = client.actor(name);
+            const results = await actorClient.call(input, callOptions);
+            const dataset = await client.dataset(results.defaultDatasetId).listItems();
+            log.info(`Actor ${name} finished with ${dataset.items.length} items`);
+            if (process.env.APIFY_IS_AT_HOME) {
+                await Actor.pushData(dataset.items);
+                log.info(`Pushed ${dataset.items.length} items to the dataset`);
+            }
+            return dataset.items;
+        }
+        catch (error) {
+            log.error(`Error calling actor: ${error}. Actor: ${name}, input: ${JSON.stringify(input)}`);
+            throw new Error(`Error calling actor: ${error}`);
+        }
+    }
+    async addToolsFromActors(actors) {
+        const tools = await getActorsAsTools(actors);
+        this.updateTools(tools);
+        return tools;
+    }
+    async addToolsFromDefaultActors() {
+        await this.addToolsFromActors(defaults.actors);
+    }
+    updateTools(tools) {
+        for (const tool of tools) {
+            this.tools.set(tool.name, tool);
+            log.info(`Added/Updated tool: ${tool.actorFullName} (tool: ${tool.name})`);
+        }
+    }
+    /**
+     * Returns an array of tool names.
+     * @returns {string[]} - An array of tool names.
+     */
+    getToolNames() {
+        return Array.from(this.tools.keys());
+    }
+    setupErrorHandling() {
+        this.server.onerror = (error) => {
+            console.error('[MCP Error]', error); // eslint-disable-line no-console
+        };
+        process.on('SIGINT', async () => {
+            await this.server.close();
+            process.exit(0);
+        });
+    }
+    setupToolHandlers() {
+        this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+            return { tools: Array.from(this.tools.values()) };
+        });
+        /**
+         * Handles the request to call a tool.
+         * @param {object} request - The request object containing tool name and arguments.
+         * @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 tool = Array.from(this.tools.values()).find((t) => t.name === name || t.actorFullName === name);
+            if (!tool) {
+                throw new Error(`Unknown tool: ${name}`);
+            }
+            if (!args) {
+                throw new Error(`Missing arguments for tool: ${name}`);
+            }
+            log.info(`Validate arguments for tool: ${tool.name} with arguments: ${JSON.stringify(args)}`);
+            if (!tool.ajvValidate(args)) {
+                throw new Error(`Invalid arguments for tool ${tool.name}: args: ${JSON.stringify(args)} error: ${JSON.stringify(tool?.ajvValidate.errors)}`);
+            }
+            try {
+                switch (name) {
+                    case InternalTools.ADD_ACTOR_TO_TOOLS: {
+                        const parsed = AddActorToToolsArgsSchema.parse(args);
+                        const toolsAdded = await this.addToolsFromActors([parsed.actorName]);
+                        await this.server.notification({ method: 'notifications/tools/list_changed' });
+                        return { content: [{ type: 'text', text: `Actor added: ${toolsAdded.map((t) => `${t.actorFullName} (tool name: ${t.name})`).join(', ')}` }] };
+                    }
+                    case InternalTools.REMOVE_ACTOR_FROM_TOOLS: {
+                        const parsed = RemoveActorToolArgsSchema.parse(args);
+                        this.tools.delete(parsed.toolName);
+                        await this.server.notification({ method: 'notifications/tools/list_changed' });
+                        return { content: [{ type: 'text', text: `Tool ${parsed.toolName} was removed` }] };
+                    }
+                    case InternalTools.DISCOVER_ACTORS: {
+                        const parsed = DiscoverActorsArgsSchema.parse(args);
+                        const actors = await searchActorsByKeywords(parsed.search, parsed.limit, parsed.offset);
+                        return { content: actors?.map((item) => ({ type: 'text', text: JSON.stringify(item) })) };
+                    }
+                    case InternalTools.GET_ACTOR_DETAILS: {
+                        const parsed = GetActorDefinition.parse(args);
+                        const v = await getActorDefinition(parsed.actorName, parsed.limit);
+                        if (v && v.input && 'properties' in v.input && v.input) {
+                            const properties = filterSchemaProperties(v.input.properties);
+                            v.input.properties = shortenProperties(properties);
+                        }
+                        return { content: [{ type: 'text', text: JSON.stringify(v) }] };
+                    }
+                    default: {
+                        const items = await this.callActorGetDataset(tool.actorFullName, args, { memory: tool.memoryMbytes });
+                        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
+                                ? { type: 'text', text: `${text} ... ${ACTOR_OUTPUT_TRUNCATED_MESSAGE}` }
+                                : { type: 'text', text };
+                        });
+                        return { content };
+                    }
+                }
+            }
+            catch (error) {
+                log.error(`Error calling tool: ${error}`);
+                throw new Error(`Error calling tool: ${error}`);
+            }
+        });
+    }
+    async connect(transport) {
+        await this.server.connect(transport);
+    }
+}
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":";AACA;;GAEG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAEnE,OAAO,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AAEnG,OAAO,EAAE,KAAK,EAAE,MAAM,OAAO,CAAC;AAE9B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAG3C,OAAO,EACH,sBAAsB,EACtB,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,GACpB,MAAM,aAAa,CAAC;AACrB,OAAO,EACH,+BAA+B,EAC/B,8BAA8B,EAC9B,QAAQ,EACR,aAAa,EACb,WAAW,EACX,cAAc,EACd,iBAAiB,GACpB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,EACH,yBAAyB,EACzB,yBAAyB,EACzB,wBAAwB,EACxB,sBAAsB,EACtB,kBAAkB,GACrB,MAAM,YAAY,CAAC;AAGpB;;GAEG;AACH,MAAM,OAAO,cAAc;IACf,MAAM,CAAS;IACf,KAAK,CAAoB;IAEjC;QACI,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CACpB;YACI,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,cAAc;SAC1B,EACD;YACI,YAAY,EAAE;gBACV,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;aAC/B;SACJ,CACJ,CAAC;QACF,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACK,YAAY,CAAC,MAA0B;QAC3C,MAAM,aAAa,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;QACpC,aAAa,CAAC,OAAO,GAAG,aAAa,CAAC,OAAO,IAAI,EAAE,CAAC;QACpD,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,GAAG,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,iBAAiB,EAAE,CAAC;QAC3G,OAAO,aAAa,CAAC;IACzB,CAAC;IAED;;;;;;;;;;;OAWG;IACI,KAAK,CAAC,mBAAmB,CAC5B,SAAiB,EACjB,KAAc,EACd,cAA4C,SAAS;QAErD,MAAM,IAAI,GAAG,SAAS,CAAC;QACvB,IAAI,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,iBAAiB,IAAI,gBAAgB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAEvE,MAAM,OAAO,GAAuB,EAAE,mBAAmB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACjF,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;YAC/E,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEvC,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAC3D,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,SAAS,EAAE,CAAC;YAC3E,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,kBAAkB,OAAO,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,CAAC;YAEtE,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;gBAC/B,MAAM,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACpC,GAAG,CAAC,IAAI,CAAC,UAAU,OAAO,CAAC,KAAK,CAAC,MAAM,uBAAuB,CAAC,CAAC;YACpE,CAAC;YACD,OAAO,OAAO,CAAC,KAAK,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,wBAAwB,KAAK,YAAY,IAAI,YAAY,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC5F,MAAM,IAAI,KAAK,CAAC,wBAAwB,KAAK,EAAE,CAAC,CAAC;QACrD,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,MAAgB;QAC5C,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACxB,OAAO,KAAK,CAAC;IACjB,CAAC;IAEM,KAAK,CAAC,yBAAyB;QAClC,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC;IAEM,WAAW,CAAC,KAAa;QAC5B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,aAAa,WAAW,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;QAC/E,CAAC;IACL,CAAC;IAED;;;OAGG;IACI,YAAY;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAEO,kBAAkB;QACtB,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,CAAC,KAAK,EAAE,EAAE;YAC5B,OAAO,CAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC,CAAC,iCAAiC;QAC1E,CAAC,CAAC;QACF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YAC5B,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,iBAAiB;QACrB,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YAC7D,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACnE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;YAEjD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,aAAa,KAAK,IAAI,CAAC,CAAC;YACtG,IAAI,CAAC,IAAI,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;YAC7C,CAAC;YACD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC;YAC3D,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,gCAAgC,IAAI,CAAC,IAAI,oBAAoB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9F,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,8BAA8B,IAAI,CAAC,IAAI,WAAW,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACjJ,CAAC;YAED,IAAI,CAAC;gBACD,QAAQ,IAAI,EAAE,CAAC;oBACX,KAAK,aAAa,CAAC,kBAAkB,CAAC,CAAC,CAAC;wBACpC,MAAM,MAAM,GAAG,yBAAyB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBACrD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;wBACrE,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,kCAAkC,EAAE,CAAC,CAAC;wBAC/E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,aAAa,gBAAgB,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;oBAClJ,CAAC;oBACD,KAAK,aAAa,CAAC,uBAAuB,CAAC,CAAC,CAAC;wBACzC,MAAM,MAAM,GAAG,yBAAyB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBACrD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACnC,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,kCAAkC,EAAE,CAAC,CAAC;wBAC/E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,MAAM,CAAC,QAAQ,cAAc,EAAE,CAAC,EAAE,CAAC;oBACxF,CAAC;oBACD,KAAK,aAAa,CAAC,eAAe,CAAC,CAAC,CAAC;wBACjC,MAAM,MAAM,GAAG,wBAAwB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBACpD,MAAM,MAAM,GAAG,MAAM,sBAAsB,CACvC,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,MAAM,CAChB,CAAC;wBACF,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;oBAC9F,CAAC;oBACD,KAAK,aAAa,CAAC,iBAAiB,CAAC,CAAC,CAAC;wBACnC,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC9C,MAAM,CAAC,GAAG,MAAM,kBAAkB,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;wBACnE,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,YAAY,IAAI,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;4BACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,CAAC,CAAC,KAAK,CAAC,UAAkD,CAAC,CAAC;4BACtG,CAAC,CAAC,KAAK,CAAC,UAAU,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;wBACvD,CAAC;wBACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;oBACpE,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACN,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,YAAY,EAAsB,CAAC,CAAC;wBAC1H,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;4BAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,+BAA+B,CAAC,CAAC;4BAC5E,OAAO,IAAI,CAAC,MAAM,KAAK,+BAA+B;gCAClD,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,QAAQ,8BAA8B,EAAE,EAAE;gCACzE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;wBACjC,CAAC,CAAC,CAAC;wBACH,OAAO,EAAE,OAAO,EAAE,CAAC;oBACvB,CAAC;gBACL,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,GAAG,CAAC,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;gBAC1C,MAAM,IAAI,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;YACpD,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAoB;QAC9B,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;CACJ"}

package/dist/tools.d.ts

@@ -0,0 +1,46 @@
+import { z } from 'zod';
+import type { ActorStorePruned, Tool } from './types.js';
+export declare const DiscoverActorsArgsSchema: z.ZodObject<{
+    limit: z.ZodDefault<z.ZodNumber>;
+    offset: z.ZodDefault<z.ZodNumber>;
+    search: z.ZodDefault<z.ZodString>;
+    category: z.ZodDefault<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    search: string;
+    limit: number;
+    offset: number;
+    category: string;
+}, {
+    search?: string | undefined;
+    limit?: number | undefined;
+    offset?: number | undefined;
+    category?: string | undefined;
+}>;
+export declare const RemoveActorToolArgsSchema: z.ZodObject<{
+    toolName: z.ZodEffects<z.ZodString, string, string>;
+}, "strip", z.ZodTypeAny, {
+    toolName: string;
+}, {
+    toolName: string;
+}>;
+export declare const AddActorToToolsArgsSchema: z.ZodObject<{
+    actorName: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    actorName: string;
+}, {
+    actorName: string;
+}>;
+export declare const GetActorDefinition: z.ZodObject<{
+    actorName: z.ZodString;
+    limit: z.ZodDefault<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    limit: number;
+    actorName: string;
+}, {
+    actorName: string;
+    limit?: number | undefined;
+}>;
+export declare function getActorAutoLoadingTools(): Tool[];
+export declare function getActorDiscoveryTools(): Tool[];
+export declare function searchActorsByKeywords(search: string, limit?: number | undefined, offset?: number | undefined): Promise<ActorStorePruned[] | null>;
+//# sourceMappingURL=tools.d.ts.map

package/dist/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,OAAO,KAAK,EAAE,gBAAgB,EAAe,IAAI,EAAE,MAAM,YAAY,CAAC;AAEtE,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;EAqBnC,CAAC;AAEH,eAAO,MAAM,yBAAyB;;;;;;EAIpC,CAAC;AAEH,eAAO,MAAM,yBAAyB;;;;;;EAIpC,CAAC;AAEH,eAAO,MAAM,kBAAkB;;;;;;;;;EAQ7B,CAAC;AAEH,wBAAgB,wBAAwB,IAAI,IAAI,EAAE,CAsBjD;AAED,wBAAgB,sBAAsB,IAAI,IAAI,EAAE,CA0B/C;AA4BD,wBAAsB,sBAAsB,CACxC,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAAM,GAAG,SAAqB,EACrC,MAAM,GAAE,MAAM,GAAG,SAAqB,GACvC,OAAO,CAAC,gBAAgB,EAAE,GAAG,IAAI,CAAC,CAIpC"}

package/dist/tools.js

@@ -0,0 +1,128 @@
+import { Ajv } from 'ajv';
+import { ApifyClient } from 'apify-client';
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { actorNameToToolName } from './actors.js';
+import { ACTOR_README_MAX_LENGTH, InternalTools } from './const.js';
+export const DiscoverActorsArgsSchema = z.object({
+    limit: z.number()
+        .int()
+        .min(1)
+        .max(100)
+        .default(10)
+        .describe('The maximum number of Actors to return. Default value is 10.'),
+    offset: z.number()
+        .int()
+        .min(0)
+        .default(0)
+        .describe('The number of elements that should be skipped at the start. Default value is 0.'),
+    search: z.string()
+        .default('')
+        .describe('String of key words to search by. '
+        + 'Searches the title, name, description, username, and readme of an Actor.'
+        + 'Only key word search is supported, no advanced search.'
+        + 'Always prefer simple keywords over complex queries.'),
+    category: z.string()
+        .default('')
+        .describe('Filters the results by the specified category.'),
+});
+export const RemoveActorToolArgsSchema = z.object({
+    toolName: z.string()
+        .describe('Tool name to remove from available tools.')
+        .transform((val) => actorNameToToolName(val)),
+});
+export const AddActorToToolsArgsSchema = z.object({
+    actorName: z.string()
+        .describe('Add an Actor to available tools by Actor ID or Actor full name.'
+        + 'Actor name is always composed from `username/name`'),
+});
+export const GetActorDefinition = z.object({
+    actorName: z.string()
+        .describe('Retrieve input, readme, and other details for Actor ID or Actor full name. '
+        + 'Actor name is always composed from `username/name`'),
+    limit: z.number()
+        .int()
+        .default(ACTOR_README_MAX_LENGTH)
+        .describe(`Truncate the README to this limit. Default value is ${ACTOR_README_MAX_LENGTH}.`),
+});
+export function getActorAutoLoadingTools() {
+    const ajv = new Ajv({ coerceTypes: 'array', strict: false });
+    return [
+        {
+            name: InternalTools.ADD_ACTOR_TO_TOOLS,
+            actorFullName: InternalTools.ADD_ACTOR_TO_TOOLS,
+            description: 'Add an Actor to available tools by Actor ID or Actor name. '
+                + 'Do not execute the Actor, only add it and list it in available tools. '
+                + 'Never run the tool without user consent! '
+                + 'For example, add a tool with username/name when user wants to scrape data from a website.',
+            inputSchema: zodToJsonSchema(AddActorToToolsArgsSchema),
+            ajvValidate: ajv.compile(zodToJsonSchema(AddActorToToolsArgsSchema)),
+        },
+        {
+            name: InternalTools.REMOVE_ACTOR_FROM_TOOLS,
+            actorFullName: InternalTools.REMOVE_ACTOR_FROM_TOOLS,
+            description: 'Remove tool by name from available tools. '
+                + 'For example, when user says, I do not need a tool username/name anymore',
+            inputSchema: zodToJsonSchema(RemoveActorToolArgsSchema),
+            ajvValidate: ajv.compile(zodToJsonSchema(RemoveActorToolArgsSchema)),
+        },
+    ];
+}
+export function getActorDiscoveryTools() {
+    const ajv = new Ajv({ coerceTypes: 'array', strict: false });
+    return [
+        {
+            name: InternalTools.DISCOVER_ACTORS,
+            actorFullName: InternalTools.DISCOVER_ACTORS,
+            description: `Discover available Actors using full text search using keywords.`
+                + `Users try to discover Actors using free form query in this case search query needs to be converted to full text search. `
+                + `Prefer Actors from Apify as they are generally more reliable and have better support. `
+                + `Returns a list of Actors with name, description, run statistics, pricing, starts, and URL. `
+                + `You perhaps need to use this tool several times to find the right Actor. `
+                + `Limit number of results returned but ensure that relevant results are returned. `,
+            inputSchema: zodToJsonSchema(DiscoverActorsArgsSchema),
+            ajvValidate: ajv.compile(zodToJsonSchema(DiscoverActorsArgsSchema)),
+        },
+        {
+            name: InternalTools.GET_ACTOR_DETAILS,
+            actorFullName: InternalTools.GET_ACTOR_DETAILS,
+            description: 'Get documentation, readme, input schema and other details about an Actor. '
+                + 'For example, when user says, I need to know more about web crawler Actor.'
+                + 'Get details for an Actor with with Actor ID or Actor full name, i.e. username/name.'
+                + `Limit the length of the README if needed.`,
+            inputSchema: zodToJsonSchema(GetActorDefinition),
+            ajvValidate: ajv.compile(zodToJsonSchema(GetActorDefinition)),
+        },
+    ];
+}
+function pruneActorStoreInfo(response) {
+    const stats = response.stats || {};
+    const pricingInfo = (response.currentPricingInfo || {});
+    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 : null,
+    };
+}
+export async function searchActorsByKeywords(search, limit = undefined, offset = undefined) {
+    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));
+}
+//# sourceMappingURL=tools.js.map

package/dist/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,KAAK,CAAC;AAE1B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,uBAAuB,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAGpE,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SACZ,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,GAAG,CAAC;SACR,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,CAAC,8DAA8D,CAAC;IAC7E,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;SACb,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,iFAAiF,CAAC;IAChG,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;SACb,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,CAAC,oCAAoC;UACxC,0EAA0E;UAC1E,wDAAwD;UACxD,qDAAqD,CAAC;IAChE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;SACf,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,CAAC,gDAAgD,CAAC;CAClE,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;SACf,QAAQ,CAAC,2CAA2C,CAAC;SACrD,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;CACpD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;SAChB,QAAQ,CAAC,iEAAiE;UACrE,oDAAoD,CAAC;CAClE,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;SAChB,QAAQ,CAAC,6EAA6E;UACjF,oDAAoD,CAAC;IAC/D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SACZ,GAAG,EAAE;SACL,OAAO,CAAC,uBAAuB,CAAC;SAChC,QAAQ,CAAC,uDAAuD,uBAAuB,GAAG,CAAC;CACnG,CAAC,CAAC;AAEH,MAAM,UAAU,wBAAwB;IACpC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,OAAO;QACH;YACI,IAAI,EAAE,aAAa,CAAC,kBAAkB;YACtC,aAAa,EAAE,aAAa,CAAC,kBAAkB;YAC/C,WAAW,EAAE,6DAA6D;kBACpE,wEAAwE;kBACxE,2CAA2C;kBAC3C,2FAA2F;YACjG,WAAW,EAAE,eAAe,CAAC,yBAAyB,CAAC;YACvD,WAAW,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,yBAAyB,CAAC,CAAC;SACvE;QACD;YACI,IAAI,EAAE,aAAa,CAAC,uBAAuB;YAC3C,aAAa,EAAE,aAAa,CAAC,uBAAuB;YACpD,WAAW,EAAE,4CAA4C;kBACnD,yEAAyE;YAC/E,WAAW,EAAE,eAAe,CAAC,yBAAyB,CAAC;YACvD,WAAW,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,yBAAyB,CAAC,CAAC;SACvE;KACJ,CAAC;AACN,CAAC;AAED,MAAM,UAAU,sBAAsB;IAClC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,OAAO;QACH;YACI,IAAI,EAAE,aAAa,CAAC,eAAe;YACnC,aAAa,EAAE,aAAa,CAAC,eAAe;YAC5C,WAAW,EAAE,kEAAkE;kBACzE,0HAA0H;kBAC1H,wFAAwF;kBACxF,6FAA6F;kBAC7F,2EAA2E;kBAC3E,kFAAkF;YACxF,WAAW,EAAE,eAAe,CAAC,wBAAwB,CAAC;YACtD,WAAW,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,wBAAwB,CAAC,CAAC;SACtE;QACD;YACI,IAAI,EAAE,aAAa,CAAC,iBAAiB;YACrC,aAAa,EAAE,aAAa,CAAC,iBAAiB;YAC9C,WAAW,EAAE,4EAA4E;kBACnF,2EAA2E;kBAC3E,qFAAqF;kBACrF,2CAA2C;YACjD,WAAW,EAAE,eAAe,CAAC,kBAAkB,CAAC;YAChD,WAAW,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;SAChE;KACJ,CAAC;AACN,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAwB;IACjD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;IACnC,MAAM,WAAW,GAAG,CAAC,QAAQ,CAAC,kBAAkB,IAAI,EAAE,CAAgB,CAAC;IACvE,OAAO;QACH,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE;QACrC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QAC7C,aAAa,EAAE,GAAG,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE;QACtD,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE;QACvC,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE;QACnD,KAAK,EAAE;YACH,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,yBAAyB,EAAE,2BAA2B,IAAI,KAAK;gBAC3D,CAAC,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAC,EAAE;SAC7C;QACD,kBAAkB,EAAE;YAChB,YAAY,EAAE,WAAW,CAAC,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE;YACxD,eAAe,EAAE,WAAW,EAAE,eAAe,IAAI,CAAC;YAClD,YAAY,EAAE,WAAW,EAAE,YAAY,IAAI,CAAC;SAC/C;QACD,GAAG,EAAE,QAAQ,CAAC,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE;QACnC,UAAU,EAAE,YAAY,IAAI,QAAQ,CAAC,CAAC,CAAE,QAAQ,CAAC,UAAqB,CAAC,CAAC,CAAC,IAAI;KAChF,CAAC;AACN,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CACxC,MAAc,EACd,QAA4B,SAAS,EACrC,SAA6B,SAAS;IAEtC,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;IACnE,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IACrE,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;AAC5D,CAAC"}