fix: stdio integration tests

Author: MichalKalita

src/input.ts

@@ -59,7 +59,7 @@ const mcpOptionsSchema = z.preprocess((originalInput) => {
         }
         return items.filter((s) => toolCategoriesArray.includes(s as ToolCategory));
     }, z.array(z.enum(toolCategoriesArray)).default([])),
-    fullActorSchema: z.boolean().default(false),
+    fullActorSchema: z.preprocess(parseBoolean, z.boolean().default(false)),
 }));
 
 export type McpOptions = z.infer<typeof mcpOptionsSchema>;

src/stdio.ts

@@ -22,10 +22,10 @@ import { hideBin } from 'yargs/helpers';
 
 import log from '@apify/log';
 
+import type { McpOptions } from './input.js';
 import { ActorsMcpServer } from './mcp/server.js';
 import { toolCategories } from './tools/index.js';
 import type { ToolCategory } from './types.js';
-import type { McpOptions } from './input.js';
 import { loadToolsFromInput } from './utils/tools-loader.js';
 
 // Keeping this interface here and not types.ts since
@@ -40,6 +40,7 @@ interface CliArgs {
     enableActorAutoLoading: boolean;
     /** Tool categories to include */
     tools?: string;
+    fullActorSchema: boolean;
 }
 
 // Configure logging, set to ERROR
@@ -80,6 +81,11 @@ Note: Tools that enable you to search Actors from the Apify Store and get their
 `,
         example: 'docs,runs,storage',
     })
+    .option('full-actor-schema', {
+        type: 'boolean',
+        default: false,
+        describe: 'Enable full Actor schema for all Actors.',
+    })
     .help('help')
     .alias('h', 'help')
     .version(false)
@@ -116,7 +122,8 @@ async function main() {
         actors: actorList.length ? actorList : [],
         enableAddingActors,
         tools: toolCategoryKeys as ToolCategory[],
-        fullActorSchema: false,
+        enableDefaultActors: false,
+        fullActorSchema: argv.fullActorSchema,
     };
 
     const mcpServer = new ActorsMcpServer(input);

tests/helpers.ts

@@ -92,12 +92,12 @@ export async function createMcpStreamableClient(
 }
 
 export async function createMcpStdioClient(
-    options?: McpClientOptions,
+    options?: Partial<McpOptions>,
 ): Promise<Client> {
     if (!process.env.APIFY_TOKEN) {
         throw new Error('APIFY_TOKEN environment variable is not set.');
     }
-    const { actors, enableAddingActors, tools } = options || {};
+    const { actors, enableAddingActors, tools, fullActorSchema } = options || {};
     const args = ['dist/stdio.js'];
     if (actors) {
         args.push('--actors', actors.join(','));
@@ -108,6 +108,9 @@ export async function createMcpStdioClient(
     if (tools && tools.length > 0) {
         args.push('--tools', tools.join(','));
     }
+    if (fullActorSchema !== undefined) {
+        args.push('--full-actor-schema', fullActorSchema.toString());
+    }
 
     const transport = new StdioClientTransport({
         command: 'node',

tests/integration/suite.ts

@@ -3,18 +3,19 @@ import type { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/cl
 import { ToolListChangedNotificationSchema } from '@modelcontextprotocol/sdk/types.js';
 import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it } from 'vitest';
 
-import { defaults, HelperTools } from '../../src/const.js';
+import { ADVANCED_INPUT_KEY, defaults, HelperTools } from '../../src/const.js';
+import type { McpOptions } from '../../src/input.js';
 import { latestNewsOnTopicPrompt } from '../../src/prompts/latest-news-on-topic.js';
 import { addRemoveTools, defaultTools, toolCategories, toolCategoriesEnabledByDefault } from '../../src/tools/index.js';
 import { actorNameToToolName } from '../../src/tools/utils.js';
 import type { ToolCategory } from '../../src/types.js';
 import { ACTOR_MCP_SERVER_ACTOR_NAME, ACTOR_PYTHON_EXAMPLE, DEFAULT_ACTOR_NAMES, DEFAULT_TOOL_NAMES } from '../const.js';
-import { addActor, type McpClientOptions } from '../helpers.js';
+import { addActor } from '../helpers.js';
 
 interface IntegrationTestsSuiteOptions {
     suiteName: string;
     transport: 'sse' | 'streamable-http' | 'stdio';
-    createClientFn: (options?: McpClientOptions) => Promise<Client>;
+    createClientFn: (options?: Partial<McpOptions>) => Promise<Client>;
     beforeAllFn?: () => Promise<void>;
     afterAllFn?: () => Promise<void>;
     beforeEachFn?: () => Promise<void>;
@@ -273,7 +274,7 @@ export function createIntegrationTestsSuite(
                     limit: 5,
                 },
             });
-            const content = result.content as {text: string}[];
+            const content = result.content as { text: string }[];
             expect(content.some((item) => item.text.includes(ACTOR_PYTHON_EXAMPLE))).toBe(true);
 
             await client.close();
@@ -292,7 +293,7 @@ export function createIntegrationTestsSuite(
                     limit: 100,
                 },
             });
-            const content = result.content as {text: string}[];
+            const content = result.content as { text: string }[];
             expect(content.length).toBe(1);
             const outputText = content[0].text;
 
@@ -485,6 +486,37 @@ export function createIntegrationTestsSuite(
             await client.close();
         });
 
+        it(`should make ${ADVANCED_INPUT_KEY} in Actor properties available`, async () => {
+            const client = await createClientFn({ enableAddingActors: true, fullActorSchema: true });
+            await client.callTool({ name: HelperTools.ACTOR_ADD, arguments: { actor: 'compass/crawler-google-places' } });
+            // Get input type for actor 'compass-slash-crawler-google-places'
+            const tools = await client.listTools();
+            const googlePlacesTool = tools.tools.find((tool) => tool.name === 'compass-slash-crawler-google-places');
+            const properties = googlePlacesTool!.inputSchema.properties as Record<string, unknown>;
+            expect(Object.keys(properties)).toMatchInlineSnapshot(`
+              [
+                "searchStringsArray",
+                "locationQuery",
+                "maxCrawledPlacesPerSearch",
+                "language",
+                "advancedInput",
+              ]
+            `);
+            expect(Object.keys((properties[ADVANCED_INPUT_KEY] as { properties: Record<string, unknown> }).properties).length).toBeGreaterThan(0);
+
+            await client.close();
+        });
+
+        it(`should not create ${ADVANCED_INPUT_KEY} if is disabled`, async () => {
+            const client = await createClientFn({ enableAddingActors: true, fullActorSchema: false });
+            await client.callTool({ name: HelperTools.ACTOR_ADD, arguments: { actor: 'compass/crawler-google-places' } });
+            const tools = await client.listTools();
+            const googlePlacesTool = tools.tools.find((tool) => tool.name === 'compass-slash-crawler-google-places');
+            const properties = googlePlacesTool!.inputSchema.properties as Record<string, unknown>;
+            expect(Object.keys(properties)).not.toContain(ADVANCED_INPUT_KEY);
+            await client.close();
+        });
+
         // Session termination is only possible for streamable HTTP transport.
         it.runIf(options.transport === 'streamable-http')('should successfully terminate streamable session', async () => {
             const client = await createClientFn();

tests/unit/mcp.utils.test.ts

@@ -1,5 +1,6 @@
 import { describe, expect, it } from 'vitest';
 
+import { defaults } from '../../src/const.js';
 import { parseInputParamsFromUrl } from '../../src/mcp/utils.js';
 
 describe('parseInputParamsFromUrl', () => {
@@ -18,7 +19,7 @@ describe('parseInputParamsFromUrl', () => {
     it('should handle URL without query params', () => {
         const url = 'https://actors-mcp-server.apify.actor';
         const result = parseInputParamsFromUrl(url);
-        expect(result.actors).toEqual([]);
+        expect(result.actors).toEqual(defaults.actors);
     });
 
     it('should parse enableActorAutoLoading flag', () => {
@@ -44,4 +45,10 @@ describe('parseInputParamsFromUrl', () => {
         const result = parseInputParamsFromUrl(url);
         expect(result.actors).toEqual(['apify/rag-web-browser']);
     });
+
+    it('should parse fullActorSchema flag', () => {
+        const url = 'https://actors-mcp-server.apify.actor?fullActorSchema=true';
+        const result = parseInputParamsFromUrl(url);
+        expect(result.fullActorSchema).toBe(true);
+    });
 });