@mablhq/mabl-cli 2.70.18 → 2.70.20

package/api/mablApiClient.js

@@ -12,12 +12,22 @@ const cliConfigProvider_1 = require("../providers/cliConfigProvider");
 const basicApiClient_1 = require("./basicApiClient");
 const query_string_1 = __importDefault(require("query-string"));
 const featureSet_1 = require("./featureSet");
+const AGENT_API_URL_DEV = 'https://agent-api-172599862372.us-central1.run.app';
+const AGENT_API_URL_PROD = 'https://agent-api-259397947092.us-central1.run.app';
 class MablApiClient extends basicApiClient_1.BasicApiClient {
     constructor(options) {
         super(options);
         this.baseApiUrl =
             options.apiUrl ?? process.env.MABL_API_URL ?? env_1.BASE_API_URL;
-        this.baseApiAgentUrl = process.env.MABL_API_AGENT_URL ?? this.baseApiUrl;
+        if (this.baseApiUrl.includes('api-dev.internal')) {
+            this.baseApiAgentUrl = AGENT_API_URL_DEV;
+        }
+        else if (this.baseApiUrl.includes('api.internal')) {
+            this.baseApiAgentUrl = AGENT_API_URL_PROD;
+        }
+        else {
+            this.baseApiAgentUrl = this.baseApiUrl;
+        }
     }
     async getPlan(planId) {
         try {

package/commands/mcp/mcp_cmds/tools/getPlanResults.js

@@ -14,12 +14,14 @@ const getPlanResults = (0, mcpMablTool_1.defineTool)({
                 .describe('The id of the mabl plan to get recent results of'),
         }),
         type: 'readOnly',
-        outputSchema: zod_1.z.array(zod_1.z.object({
-            planRunId: zod_1.z.string().nullable(),
-            status: zod_1.z.string().nullable(),
-            completedTimeMs: zod_1.z.number().nullable(),
-            errorMessage: zod_1.z.string(),
-        })),
+        outputSchema: zod_1.z.object({
+            planResults: zod_1.z.array(zod_1.z.object({
+                planRunId: zod_1.z.string().nullable(),
+                status: zod_1.z.string().nullable(),
+                completedTimeMs: zod_1.z.number().nullable(),
+                errorMessage: zod_1.z.string(),
+            })),
+        }),
     },
     handle: async (params, mablApiClient, _, workspaceId) => {
         const planId = params.planId;
@@ -62,10 +64,10 @@ const getPlanResults = (0, mcpMablTool_1.defineTool)({
             content: [
                 {
                     type: 'text',
-                    text: JSON.stringify(processedResults),
+                    text: JSON.stringify({ planResults: processedResults }),
                 },
             ],
-            structuredContent: processedResults,
+            structuredContent: { planResults: processedResults },
         };
     },
 });

package/commands/mcp/mcp_cmds/tools/getTestResults.js

@@ -14,13 +14,15 @@ const getTestResults = (0, mcpMablTool_1.defineTool)({
                 .describe('The id of the mabl test to get recent results of'),
         }),
         type: 'readOnly',
-        outputSchema: zod_1.z.array(zod_1.z.object({
-            testRunId: zod_1.z.string().nullable(),
-            environmentId: zod_1.z.string().nullable(),
-            status: zod_1.z.string().nullable(),
-            completedTimeMs: zod_1.z.number().nullable(),
-            errorMessage: zod_1.z.string(),
-        })),
+        outputSchema: zod_1.z.object({
+            testResults: zod_1.z.array(zod_1.z.object({
+                testRunId: zod_1.z.string().nullable(),
+                environmentId: zod_1.z.string().nullable(),
+                status: zod_1.z.string().nullable(),
+                completedTimeMs: zod_1.z.number().nullable(),
+                errorMessage: zod_1.z.string(),
+            })),
+        }),
     },
     handle: async (params, mablApiClient, _, workspaceId) => {
         const testId = params.testId;
@@ -64,10 +66,10 @@ const getTestResults = (0, mcpMablTool_1.defineTool)({
             content: [
                 {
                     type: 'text',
-                    text: JSON.stringify(processedResults),
+                    text: JSON.stringify({ testResults: processedResults }),
                 },
             ],
-            structuredContent: processedResults,
+            structuredContent: { testResults: processedResults },
         };
     },
 });