@skyramp/mcp 0.2.8 → 0.2.150-rc.ldw

package/build/tools/generate-tests/batchMockGenerationTool.js

@@ -0,0 +1,432 @@
+import { z } from "zod";
+import { SkyrampClient } from "@skyramp/skyramp";
+import { MockProtocol, RuntimeEnvironment, runtimeEnvironmentInputSchema } from "../../types/TestTypes.js";
+export { MockProtocol } from "../../types/TestTypes.js";
+import { AnalyticsService } from "../../services/AnalyticsService.js";
+import { logger } from "../../utils/logger.js";
+import { getEntryPoint } from "../../utils/telemetry.js";
+import fs from "fs";
+import { getRestMockMethodValidationError, normalizeRestMockMethod, } from "../../utils/httpMethodValidation.js";
+import { getGrpcMockResponseValidationError } from "../../utils/grpcMockValidation.js";
+import { validateMockCompatibility } from "../../utils/mockCompatibility.js";
+export const BATCH_MOCK_PROTOCOL_EXAMPLES = `
+<example protocol="rest">
+Batch REST item. Required fields: protocol="rest", endpointURL, method unless apiSchema is provided.
+{
+  "protocol": "rest",
+  "endpointURL": "http://identity-service:4000/api/internal/v1/users/validate",
+  "method": "POST",
+  "requestAware": true,
+  "responseData": "{\"valid\":true}"
+}
+</example>
+
+<example protocol="grpc">
+Batch gRPC item. Required fields: protocol="grpc", endpointURL as host:port, protoPath, grpcServiceName. method is the RPC name; empty means all service methods.
+{
+  "protocol": "grpc",
+  "endpointURL": "profile-service:50051",
+  "protoPath": "/repo/proto/profiles.proto",
+  "grpcServiceName": "ProfileService",
+  "method": "GetProfile",
+  "responseData": "{\"id\":\"profile_123\"}"
+}
+</example>
+
+<example protocol="grpc-status-error">
+Batch gRPC status error item. Do not provide responseData with grpcStatusCode.
+{
+  "protocol": "grpc",
+  "endpointURL": "profile-service:50051",
+  "protoPath": "/repo/proto/profiles.proto",
+  "grpcServiceName": "ProfileService",
+  "method": "GetProfile",
+  "grpcStatusCode": 5,
+  "grpcStatusMessage": "profile not found"
+}
+</example>
+
+<example protocol="kafka-producer">
+Batch Kafka producer validation item. Required fields: protocol="kafka", endpointURL as broker host:port, kafkaTopic, and requestData. Do not pass method or responseData.
+{
+  "protocol": "kafka",
+  "endpointURL": "broker:9092",
+  "kafkaTopic": "orders",
+  "requestData": "{\"order_id\":\"ord_123\"}"
+}
+</example>
+
+<example protocol="kafka-consumer">
+Batch Kafka consumer message item. Required fields: protocol="kafka", endpointURL as broker host:port, kafkaTopic, and responseData. Do not pass method or requestData.
+{
+  "protocol": "kafka",
+  "endpointURL": "broker:9092",
+  "kafkaTopic": "orders",
+  "responseData": "{\"accepted\":true}"
+}
+</example>`;
+const mockSpecSchema = z.object({
+    protocol: z
+        .nativeEnum(MockProtocol)
+        .default(MockProtocol.Rest)
+        .describe("Protocol for this mock: 'rest', 'grpc', or 'kafka'. See <example> blocks in the tool description for required fields."),
+    endpointURL: z
+        .string()
+        .describe("For REST: the base URL or endpoint URL. " +
+        "For gRPC: the target service address as host:port. " +
+        "For Kafka: the broker address as host:port."),
+    method: z
+        .string()
+        .default("")
+        .describe("For REST: HTTP method, required unless apiSchema is provided. For gRPC: RPC method name, empty generates all service methods. For Kafka: MUST be empty; use kafkaTopic."),
+    responseData: z
+        .string()
+        .optional()
+        .describe("REST/gRPC response body data; Kafka consumer messages. Inline JSON/YAML or @/path/to/file."),
+    responseStatusCode: z
+        .string()
+        .default("")
+        .refine((val) => !val || /^\d{3}$/.test(val), {
+        message: "Must be a 3-digit HTTP status code (e.g., '200', '201') or empty string",
+    })
+        .describe("HTTP status code the mock should return (e.g., '200', '201')."),
+    grpcStatusCode: z
+        .number()
+        .int()
+        .min(1)
+        .max(16)
+        .optional()
+        .describe("For protocol='grpc' error mocks only. Non-OK gRPC status code number. Do not combine with responseData."),
+    grpcStatusMessage: z
+        .string()
+        .optional()
+        .describe("For protocol='grpc' error mocks only. gRPC status message to return with grpcStatusCode."),
+    requestData: z
+        .string()
+        .default("")
+        .describe("REST request body sample for request-aware mocks; Kafka producer validation payload/schema."),
+    requestAware: z
+        .boolean()
+        .default(false)
+        .describe("When true, generates a dynamic mock that varies by request."),
+    apiSchema: z
+        .string()
+        .default("")
+        .describe("Path or URL to OpenAPI/Swagger schema file."),
+    protoPath: z
+        .string()
+        .optional()
+        .describe("Required for protocol='grpc'. Path to the .proto file."),
+    grpcServiceName: z
+        .string()
+        .optional()
+        .describe("Required for protocol='grpc'. gRPC service name."),
+    protobufExtraFiles: z
+        .array(z.string())
+        .optional()
+        .describe("Optional extra .proto files for gRPC compilation."),
+    protobufExtraImportPath: z
+        .array(z.string())
+        .optional()
+        .describe("Optional extra import directories for protobuf compilation."),
+    kafkaTopic: z
+        .string()
+        .optional()
+        .describe("Required for protocol='kafka'. Kafka topic name."),
+    output: z
+        .string()
+        .optional()
+        .describe("Optional output file name for this mock. Use when multiple mocks share the same endpoint and method so generated files do not overwrite each other."),
+    outputDir: z
+        .string()
+        .optional()
+        .describe("Optional output directory for this mock. Defaults to the batch-level outputDir."),
+});
+function validateMockSpec(spec) {
+    const protocol = spec.protocol ?? MockProtocol.Rest;
+    if (protocol === MockProtocol.Rest) {
+        const methodError = getRestMockMethodValidationError(spec.method);
+        if (methodError)
+            return methodError;
+        if (!spec.apiSchema && !spec.method) {
+            return "REST mock method is required unless apiSchema is provided for schema-wide generation";
+        }
+    }
+    if (protocol === MockProtocol.Grpc) {
+        const missing = [];
+        if (!spec.protoPath)
+            missing.push("protoPath");
+        if (!spec.grpcServiceName)
+            missing.push("grpcServiceName");
+        if (missing.length > 0) {
+            return `gRPC mock requires: ${missing.join(", ")}`;
+        }
+        if (spec.grpcStatusCode !== undefined && spec.responseData?.trim()) {
+            return "gRPC status error mocks must use grpcStatusCode/grpcStatusMessage instead of responseData";
+        }
+        if (spec.grpcStatusMessage && spec.grpcStatusCode === undefined) {
+            return "grpcStatusMessage requires grpcStatusCode";
+        }
+        const grpcResponseError = getGrpcMockResponseValidationError(spec.responseData);
+        if (grpcResponseError)
+            return grpcResponseError;
+    }
+    if (protocol === MockProtocol.Kafka) {
+        if (!spec.kafkaTopic)
+            return "Kafka mock requires: kafkaTopic";
+        if (spec.method) {
+            return "Kafka mock method must be empty; use kafkaTopic for the topic name. Produce validation uses requestData and consumer messages use responseData";
+        }
+        const hasRequestData = Boolean(spec.requestData && spec.requestData.trim());
+        const hasResponseData = Boolean(spec.responseData && spec.responseData.trim());
+        if (hasRequestData === hasResponseData) {
+            return "Kafka mock requires exactly one of requestData (producer validation) or responseData (consumer messages)";
+        }
+    }
+    return null;
+}
+function buildGenerateOptions(spec, shared) {
+    const protocol = spec.protocol ?? MockProtocol.Rest;
+    const language = shared.language ?? "python";
+    const framework = shared.framework ??
+        (language === "typescript" || language === "javascript"
+            ? "playwright"
+            : "");
+    const outputDir = spec.outputDir ?? shared.outputDir;
+    const common = {
+        protocol,
+        language,
+        framework,
+        outputDir,
+        output: spec.output,
+        force: true,
+        overwrite: true,
+        runtime: shared.runtime,
+        dockerNetwork: shared.dockerNetwork,
+        entrypoint: getEntryPoint(),
+    };
+    const responseDataOption = spec.responseData !== undefined &&
+        !(protocol === MockProtocol.Grpc && spec.grpcStatusCode !== undefined)
+        ? { responseData: spec.responseData }
+        : {};
+    if (protocol === MockProtocol.Grpc) {
+        return {
+            ...common,
+            ...responseDataOption,
+            method: spec.method ?? "",
+            url: spec.endpointURL,
+            protoPath: spec.protoPath,
+            service: spec.grpcServiceName,
+            grpcStatusCode: spec.grpcStatusCode,
+            grpcStatusMessage: spec.grpcStatusMessage,
+            extraFiles: spec.protobufExtraFiles ?? [],
+            extraImports: spec.protobufExtraImportPath ?? [],
+        };
+    }
+    if (protocol === MockProtocol.Kafka) {
+        return {
+            ...common,
+            ...responseDataOption,
+            url: spec.endpointURL,
+            topic: spec.kafkaTopic,
+            requestData: spec.requestData?.trim() ? spec.requestData : undefined,
+        };
+    }
+    // REST
+    return {
+        ...common,
+        ...responseDataOption,
+        method: normalizeRestMockMethod(spec.method),
+        uri: spec.endpointURL,
+        apiSchema: spec.apiSchema ? [spec.apiSchema] : [],
+        requestData: spec.requestData ?? "",
+        responseStatusCode: spec.responseStatusCode ?? "",
+        requestAware: spec.requestAware ?? false,
+        mockPort: 0,
+        optionalFields: false,
+    };
+}
+export async function executeBatchMockGeneration(params) {
+    const { mocks, outputDir } = params;
+    fs.mkdirSync(outputDir, { recursive: true });
+    const client = new SkyrampClient();
+    const results = [];
+    for (let i = 0; i < mocks.length; i++) {
+        const spec = mocks[i];
+        const protocol = spec.protocol ?? MockProtocol.Rest;
+        const method = spec.method ?? "";
+        // Validate spec
+        const validationError = validateMockSpec(spec);
+        if (validationError) {
+            results.push({
+                index: i,
+                protocol,
+                endpointURL: spec.endpointURL,
+                method,
+                topic: spec.kafkaTopic,
+                success: false,
+                error: validationError,
+            });
+            continue;
+        }
+        const dockerLocalDev = params.runtime === "docker" || Boolean(params.dockerNetwork?.trim());
+        const compatibility = validateMockCompatibility([{ ...spec, protocol }], {
+            dockerLocalDev,
+        });
+        const blockingWarnings = compatibility.warnings.filter((warning) => warning.blocking);
+        if (blockingWarnings.length > 0) {
+            results.push({
+                index: i,
+                protocol,
+                endpointURL: spec.endpointURL,
+                method,
+                topic: spec.kafkaTopic,
+                success: false,
+                error: blockingWarnings
+                    .map((warning) => `${warning.code}: ${warning.message} Suggestion: ${warning.suggestion}`)
+                    .join("\n"),
+            });
+            continue;
+        }
+        const resolvedOutputDir = spec.outputDir ?? outputDir;
+        fs.mkdirSync(resolvedOutputDir, { recursive: true });
+        // Build options and generate
+        const options = buildGenerateOptions(spec, {
+            outputDir,
+            runtime: params.runtime,
+            dockerNetwork: params.dockerNetwork,
+            language: params.language,
+            framework: params.framework,
+        });
+        try {
+            let result;
+            if (protocol === MockProtocol.Grpc) {
+                result = await client.generateGrpcMock(options);
+            }
+            else if (protocol === MockProtocol.Kafka) {
+                result = await client.generateKafkaMock(options);
+            }
+            else {
+                result = await client.generateRestMock(options);
+            }
+            if (!result || !result.toLowerCase().includes("success")) {
+                results.push({
+                    index: i,
+                    protocol,
+                    endpointURL: spec.endpointURL,
+                    method,
+                    topic: spec.kafkaTopic,
+                    success: false,
+                    error: `Mock generation failed: ${result || "no output from backend"}`,
+                });
+                continue;
+            }
+            results.push({
+                index: i,
+                protocol,
+                endpointURL: spec.endpointURL,
+                method,
+                topic: spec.kafkaTopic,
+                success: true,
+                outputDir: resolvedOutputDir,
+                output: spec.output,
+                detail: result,
+            });
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            logger.error("Batch mock generation failed for spec", {
+                index: i,
+                protocol,
+                endpointURL: spec.endpointURL,
+                error: errorMessage,
+            });
+            results.push({
+                index: i,
+                protocol,
+                endpointURL: spec.endpointURL,
+                method,
+                topic: spec.kafkaTopic,
+                success: false,
+                error: errorMessage,
+            });
+        }
+    }
+    const succeeded = results.filter((r) => r.success);
+    const failed = results.filter((r) => !r.success);
+    const summaryLines = [
+        `## Batch Mock Generation Summary`,
+        ``,
+        `- **Total:** ${mocks.length}`,
+        `- **Succeeded:** ${succeeded.length}`,
+        `- **Failed:** ${failed.length}`,
+        `- **Default Output Directory:** ${outputDir}`,
+        ``,
+    ];
+    if (succeeded.length > 0) {
+        summaryLines.push(`### Successful Mocks`);
+        for (const r of succeeded) {
+            summaryLines.push(`- [${r.index}] ${r.protocol.toUpperCase()} ${r.endpointURL} ${r.method || (r.protocol === MockProtocol.Kafka ? `topic:${r.topic ?? "?"}` : "(all methods)")}${r.output ? ` -> ${r.outputDir ?? outputDir}/${r.output}` : ""}`);
+        }
+        summaryLines.push(``);
+    }
+    if (failed.length > 0) {
+        summaryLines.push(`### Failed Mocks`);
+        for (const r of failed) {
+            summaryLines.push(`- [${r.index}] ${r.protocol.toUpperCase()} ${r.endpointURL} ${r.method || (r.protocol === MockProtocol.Kafka ? `topic:${r.topic ?? "?"}` : "(all methods)")}: ${r.error}`);
+        }
+        summaryLines.push(``);
+    }
+    return {
+        content: [{ type: "text", text: summaryLines.join("\n") }],
+        isError: failed.length > 0,
+    };
+}
+const TOOL_NAME = "skyramp_batch_mock_generation";
+const batchMockSchema = {
+    mocks: z
+        .array(mockSpecSchema)
+        .min(1)
+        .describe("Array of mock specifications. Each element describes one mock to generate."),
+    outputDir: z
+        .string()
+        .describe("Directory where generated mock files will be written."),
+    runtime: runtimeEnvironmentInputSchema
+        .default(RuntimeEnvironment.LOCAL)
+        .describe("Runtime environment: 'local', 'docker', or 'k8s' (also accepts 'kubernetes')."),
+    dockerNetwork: z
+        .string()
+        .default("")
+        .describe("Docker network name for containerized testing."),
+    language: z
+        .string()
+        .default("python")
+        .describe("Programming language for generated mocks (python, typescript, javascript, java)."),
+    framework: z
+        .string()
+        .default("")
+        .describe("Testing framework (e.g., pytest, playwright)."),
+};
+export function registerBatchMockTool(server) {
+    server.registerTool(TOOL_NAME, {
+        description: `Generate multiple mock server implementations in a single call using Skyramp.
+
+This tool accepts an array of mock specifications and generates all mocks in one operation,
+avoiding the need for N separate tool calls when mocking multiple downstream services.
+
+Each mock spec can target REST, gRPC, or Kafka protocols. Partial failures are handled
+gracefully — if some mocks fail, the others still succeed and are reported.
+
+Use this tool when you need to mock 2+ downstream services simultaneously (e.g., for a
+local-dev workflow with multiple dependencies).
+
+${BATCH_MOCK_PROTOCOL_EXAMPLES}`,
+        inputSchema: batchMockSchema,
+    }, async (params) => {
+        const result = await executeBatchMockGeneration(params);
+        AnalyticsService.pushTestGenerationToolEvent(TOOL_NAME, result, params).catch(() => {
+            // Silently ignore analytics errors
+        });
+        return result;
+    });
+}

package/build/tools/generate-tests/generateContractRestTool.js

@@ -357,8 +357,8 @@ const TOOL_NAME = "skyramp_contract_test_generation";
 const MODES_SECTION = CONSUMER_MODE_ENABLED
     ? `**Modes:**
 - Default (no mode set): both \`providerMode\` and \`consumerMode\` default to false. This generates both provider and consumer contract tests — equivalent to setting both modes to true.
-- \`providerMode\`: set to true ONLY if the user explicitly requests a provider-side contract test. Optionally specify \`providerOutput\` for the output file path.
-- \`consumerMode\`: set to true ONLY if the user explicitly requests a consumer-side contract test. Optionally specify \`consumerOutput\` for the output file path.
+- \`providerMode\`: set to true ONLY if the user explicitly requests a provider-side contract test. Optionally specify \`providerOutput\` for the output file path. Provider tests hit the real endpoint — no mocks.
+- \`consumerMode\`: set to true ONLY if the user explicitly requests a consumer-side contract test. Optionally specify \`consumerOutput\` for the output file path. Consumer tests include mocks wired inline (\`client.apply_mock(mock)\` in the test function) — no separate \`skyramp_enrich_test_with_mocks\` call needed.
 - Both \`providerMode\` and \`consumerMode\` can be enabled simultaneously to generate both sides.`
     : `**Mode:**
 - Only provider-side contract tests are generated. Pass \`providerMode: true\` (or omit it — the tool always behaves as provider-only). Optionally specify \`providerOutput\` for the output file path.`;

package/build/tools/generate-tests/generateMockRestTool.d.ts

@@ -1,13 +1,23 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { TestType } from "../../types/TestTypes.js";
+import { MockProtocol } from "../../types/TestTypes.js";
+export { MockProtocol } from "../../types/TestTypes.js";
 import { TestGenerationService } from "../../services/TestGenerationService.js";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+export declare const MOCK_PROTOCOL_EXAMPLES = "\n<example protocol=\"rest\">\nGenerate a REST mock for one HTTP endpoint. Required fields: protocol=\"rest\", endpointURL, method. Use apiSchema with an empty method only when intentionally generating all schema methods.\n{\n  \"protocol\": \"rest\",\n  \"endpointURL\": \"http://payment-service:5001/orders/:order_id/payment_authorizations\",\n  \"method\": \"PUT\",\n  \"requestAware\": true,\n  \"requestData\": \"{\"payment_method_id\":\"pm_123\"}\",\n  \"responseData\": \"{\"accepted\":true}\",\n  \"responseStatusCode\": \"200\"\n}\n</example>\n\n<example protocol=\"grpc\">\nGenerate a gRPC mock. Required fields: protocol=\"grpc\", endpointURL as host:port, protoPath, grpcServiceName. method is the RPC method name; omit or leave empty to generate all service methods. use responseData for successful protobuf-shaped responses, or grpcStatusCode/grpcStatusMessage for gRPC status errors.\n{\n  \"protocol\": \"grpc\",\n  \"endpointURL\": \"profile-service:50051\",\n  \"protoPath\": \"/repo/proto/profiles.proto\",\n  \"grpcServiceName\": \"ProfileService\",\n  \"method\": \"GetProfile\",\n  \"responseData\": \"{\"eligible\":true}\"\n}\n</example>\n\n<example protocol=\"grpc-status-error\">\nGenerate a gRPC status error mock. Do not provide responseData with grpcStatusCode.\n{\n  \"protocol\": \"grpc\",\n  \"endpointURL\": \"profile-service:50051\",\n  \"protoPath\": \"/repo/proto/profiles.proto\",\n  \"grpcServiceName\": \"ProfileService\",\n  \"method\": \"GetProfile\",\n  \"grpcStatusCode\": 5,\n  \"grpcStatusMessage\": \"profile not found\"\n}\n</example>\n\n<example protocol=\"kafka-producer\">\nGenerate a Kafka producer validation mock. Required fields: protocol=\"kafka\", endpointURL as broker host:port, kafkaTopic, and requestData. Do not pass method or responseData.\n{\n  \"protocol\": \"kafka\",\n  \"endpointURL\": \"broker:9092\",\n  \"kafkaTopic\": \"events\",\n  \"requestData\": \"{\"name\":\"order.created\"}\"\n}\n</example>\n\n<example protocol=\"kafka-consumer\">\nGenerate a Kafka consumer message mock. Required fields: protocol=\"kafka\", endpointURL as broker host:port, kafkaTopic, and responseData. Do not pass method or requestData.\n{\n  \"protocol\": \"kafka\",\n  \"endpointURL\": \"broker:9092\",\n  \"kafkaTopic\": \"events\",\n  \"responseData\": \"{\"accepted\":true}\"\n}\n</example>";
 type MockParams = {
+    protocol?: MockProtocol;
     endpointURL?: string;
+    protoPath?: string;
+    grpcServiceName?: string;
+    protobufExtraFiles?: string[];
+    protobufExtraImportPath?: string[];
+    kafkaTopic?: string;
     method?: string;
     apiSchema?: string;
     responseData?: string;
     responseStatusCode?: string;
+    grpcStatusCode?: number;
+    grpcStatusMessage?: string;
     requestData?: string;
     formParams?: string;
     requestAware?: boolean;
@@ -29,13 +39,21 @@ type MockParams = {
     prompt?: string;
 };
 export declare class MockGenerationService extends TestGenerationService {
-    protected getTestType(): TestType;
+    protected getTestType(): "mock";
     generateTest(params: MockParams & Record<string, any>): Promise<CallToolResult>;
     private buildReorganizationInstructions;
+    protected validateInputs(params: MockParams & Record<string, any>): CallToolResult;
     protected buildGenerationOptions(params: MockParams): {
-        uri: string | undefined;
         method: string | undefined;
-        apiSchema: string[];
+        url: string | undefined;
+        protoPath: string | undefined;
+        service: string | undefined;
+        grpcStatusCode: number | undefined;
+        grpcStatusMessage: string | undefined;
+        extraFiles: string[];
+        extraImports: string[];
+        responseData: string;
+        protocol: MockProtocol;
         language: string;
         framework: string;
         output: string | undefined;
@@ -48,17 +66,122 @@ export declare class MockGenerationService extends TestGenerationService {
         k8sNamespace: string | undefined;
         k8sConfig: string | undefined;
         k8sContext: string | undefined;
+        entrypoint: string;
+    } | {
+        method: string | undefined;
+        url: string | undefined;
+        protoPath: string | undefined;
+        service: string | undefined;
+        grpcStatusCode: number | undefined;
+        grpcStatusMessage: string | undefined;
+        extraFiles: string[];
+        extraImports: string[];
+        responseData?: undefined;
+        protocol: MockProtocol;
+        language: string;
+        framework: string;
+        output: string | undefined;
+        outputDir: string;
+        force: boolean;
+        deployDashboard: boolean | undefined;
+        runtime: string | undefined;
+        dockerNetwork: string | undefined;
+        dockerWorkerPort: string;
+        k8sNamespace: string | undefined;
+        k8sConfig: string | undefined;
+        k8sContext: string | undefined;
+        entrypoint: string;
+    } | {
+        url: string | undefined;
+        topic: string | undefined;
+        requestData: string | undefined;
+        responseData: string;
+        protocol: MockProtocol;
+        language: string;
+        framework: string;
+        output: string | undefined;
+        outputDir: string;
+        force: boolean;
+        deployDashboard: boolean | undefined;
+        runtime: string | undefined;
+        dockerNetwork: string | undefined;
+        dockerWorkerPort: string;
+        k8sNamespace: string | undefined;
+        k8sConfig: string | undefined;
+        k8sContext: string | undefined;
+        entrypoint: string;
+    } | {
+        url: string | undefined;
+        topic: string | undefined;
+        requestData: string | undefined;
+        responseData?: undefined;
+        protocol: MockProtocol;
+        language: string;
+        framework: string;
+        output: string | undefined;
+        outputDir: string;
+        force: boolean;
+        deployDashboard: boolean | undefined;
+        runtime: string | undefined;
+        dockerNetwork: string | undefined;
+        dockerWorkerPort: string;
+        k8sNamespace: string | undefined;
+        k8sConfig: string | undefined;
+        k8sContext: string | undefined;
+        entrypoint: string;
+    } | {
+        method: string;
+        uri: string | undefined;
+        apiSchema: string[];
         requestData: string | undefined;
-        responseData: string | undefined;
         responseStatusCode: string | undefined;
         formParams: string | undefined;
         requestAware: boolean | undefined;
         traceFilePath: string | undefined;
+        mockPort: number;
+        optionalFields: boolean;
+        responseData: string;
+        protocol: MockProtocol;
+        language: string;
+        framework: string;
+        output: string | undefined;
+        outputDir: string;
+        force: boolean;
+        deployDashboard: boolean | undefined;
+        runtime: string | undefined;
+        dockerNetwork: string | undefined;
+        dockerWorkerPort: string;
+        k8sNamespace: string | undefined;
+        k8sConfig: string | undefined;
+        k8sContext: string | undefined;
         entrypoint: string;
+    } | {
+        method: string;
+        uri: string | undefined;
+        apiSchema: string[];
+        requestData: string | undefined;
+        responseStatusCode: string | undefined;
+        formParams: string | undefined;
+        requestAware: boolean | undefined;
+        traceFilePath: string | undefined;
         mockPort: number;
         optionalFields: boolean;
+        responseData?: undefined;
+        protocol: MockProtocol;
+        language: string;
+        framework: string;
+        output: string | undefined;
+        outputDir: string;
+        force: boolean;
+        deployDashboard: boolean | undefined;
+        runtime: string | undefined;
+        dockerNetwork: string | undefined;
+        dockerWorkerPort: string;
+        k8sNamespace: string | undefined;
+        k8sConfig: string | undefined;
+        k8sContext: string | undefined;
+        entrypoint: string;
     };
     protected executeGeneration(generateOptions: any): Promise<string>;
 }
 export declare function registerMockTool(server: McpServer): void;
-export {};