@mastra/agent-builder 1.0.10 → 1.0.11

package/dist/index.js

@@ -14,7 +14,9 @@ import { promisify } from 'util';
 import { ModelRouterLanguageModel } from '@mastra/core/llm';
 import { tmpdir } from 'os';
 import { openai } from '@ai-sdk/openai';
+import { toStandardSchema } from '@mastra/core/schema';
 import { createStep, createWorkflow } from '@mastra/core/workflows';
+import { standardSchemaToJSONSchema } from '@mastra/schema-compat/schema';
 import * as z4 from 'zod/v4';
 import { z as z$1 } from 'zod/v4';
 import { ZodFirstPartyTypeKind } from 'zod/v3';
@@ -36,7 +38,7 @@ var AgentBuilderInputSchema = z.object({
   ref: z.string().optional().describe("Tag/branch/commit to checkout (defaults to main/master)"),
   slug: z.string().optional().describe("Slug for branch/scripts; defaults to inferred from repo"),
   targetPath: z.string().optional().describe("Project path to merge into; defaults to current directory"),
-  variables: z.record(z.string()).optional().describe("Environment variables to set in .env file")
+  variables: z.record(z.string(), z.string()).optional().describe("Environment variables to set in .env file")
 });
 z.object({
   slug: z.string(),
@@ -67,7 +69,7 @@ var FileCopyInputSchema = z.object({
   commitSha: z.string(),
   slug: z.string(),
   targetPath: z.string().optional(),
-  variables: z.record(z.string()).optional()
+  variables: z.record(z.string(), z.string()).optional()
 });
 var FileCopyResultSchema = z.object({
   success: z.boolean(),
@@ -160,10 +162,10 @@ var PackageAnalysisSchema = z.object({
   name: z.string().optional(),
   version: z.string().optional(),
   description: z.string().optional(),
-  dependencies: z.record(z.string()).optional(),
-  devDependencies: z.record(z.string()).optional(),
-  peerDependencies: z.record(z.string()).optional(),
-  scripts: z.record(z.string()).optional(),
+  dependencies: z.record(z.string(), z.string()).optional(),
+  devDependencies: z.record(z.string(), z.string()).optional(),
+  peerDependencies: z.record(z.string(), z.string()).optional(),
+  scripts: z.record(z.string(), z.string()).optional(),
   success: z.boolean().optional(),
   error: z.string().optional()
 });
@@ -1192,7 +1194,7 @@ export const mastra = new Mastra({
           timeout: z.number().default(3e4).describe("Timeout in milliseconds"),
           captureOutput: z.boolean().default(true).describe("Capture command output"),
           shell: z.string().optional().describe("Shell to use (defaults to system shell)"),
-          env: z.record(z.string()).optional().describe("Environment variables")
+          env: z.record(z.string(), z.string()).optional().describe("Environment variables")
         }),
         outputSchema: z.object({
           success: z.boolean(),
@@ -1207,7 +1209,8 @@ export const mastra = new Mastra({
         execute: async (inputData) => {
           return await _AgentBuilderDefaults.executeCommand({
             ...inputData,
-            workingDirectory: inputData.workingDirectory || projectPath
+            workingDirectory: inputData.workingDirectory || projectPath,
+            env: inputData.env
           });
         }
       }),
@@ -1642,7 +1645,7 @@ export const mastra = new Mastra({
           method: z.enum(["GET", "POST", "PUT", "DELETE", "PATCH"]).describe("HTTP method"),
           url: z.string().describe("Full URL or path (if baseUrl provided)"),
           baseUrl: z.string().optional().describe("Base URL for the server (e.g., http://localhost:4200)"),
-          headers: z.record(z.string()).optional().describe("HTTP headers"),
+          headers: z.record(z.string(), z.string()).optional().describe("HTTP headers"),
           body: z.any().optional().describe("Request body (will be JSON stringified if object)"),
           timeout: z.number().optional().default(3e4).describe("Request timeout in milliseconds")
         }),
@@ -1650,7 +1653,7 @@ export const mastra = new Mastra({
           success: z.boolean(),
           status: z.number().optional(),
           statusText: z.string().optional(),
-          headers: z.record(z.string()).optional(),
+          headers: z.record(z.string(), z.string()).optional(),
           data: z.any().optional(),
           errorMessage: z.string().optional(),
           url: z.string(),
@@ -3497,15 +3500,22 @@ Return the actual exported names of the units, as well as the file names.`,
         networks: z.array(z.object({ name: z.string(), file: z.string() })).optional(),
         other: z.array(z.object({ name: z.string(), file: z.string() })).optional()
       });
-      const result = isSupported ? await tryGenerateWithJsonFallback(agent, prompt, {
-        structuredOutput: {
-          schema: output
-        },
-        maxSteps: 100
-      }) : await agent.generateLegacy(prompt, {
-        experimental_output: output,
-        maxSteps: 100
-      });
+      let result;
+      if (isSupported) {
+        result = await tryGenerateWithJsonFallback(agent, prompt, {
+          structuredOutput: {
+            schema: output
+          },
+          maxSteps: 100
+        });
+      } else {
+        const standardSchema = toStandardSchema(output);
+        const jsonSchema2 = standardSchemaToJSONSchema(standardSchema);
+        result = await agent.generateLegacy(prompt, {
+          experimental_output: jsonSchema2,
+          maxSteps: 100
+        });
+      }
       const template = result.object ?? {};
       const units = [];
       template.agents?.forEach((agentId) => {
@@ -9103,7 +9113,7 @@ var PlanningIterationResultSchema = z.object({
   message: z.string(),
   error: z.string().optional(),
   allPreviousQuestions: z.array(z.any()).optional(),
-  allPreviousAnswers: z.record(z.string()).optional()
+  allPreviousAnswers: z.record(z.string(), z.string()).optional()
 });
 
 // src/workflows/task-planning/prompts.ts
@@ -9214,7 +9224,7 @@ var ProjectDiscoveryResultSchema = z.object({
     existingAgents: z.array(z.string()),
     existingTools: z.array(z.string())
   }),
-  dependencies: z.record(z.string()),
+  dependencies: z.record(z.string(), z.string()),
   message: z.string(),
   error: z.string().optional()
 });
@@ -9283,7 +9293,7 @@ z.object({
   questions: QuestionSchema
 });
 z.object({
-  answers: z.record(z.string()),
+  answers: z.record(z.string(), z.string()),
   hasAnswers: z.boolean()
 });
 var WorkflowBuilderResultSchema = z.object({
@@ -9324,7 +9334,7 @@ var PlanningIterationInputSchema = z.object({
   discoveredWorkflows: z.array(DiscoveredWorkflowSchema),
   projectStructure: ProjectDiscoveryResultSchema,
   research: WorkflowResearchResultSchema,
-  userAnswers: z.record(z.string()).optional()
+  userAnswers: z.record(z.string(), z.string()).optional()
 });
 var PlanningIterationSuspendSchema = z.object({
   questions: QuestionSchema,
@@ -9335,7 +9345,7 @@ var PlanningIterationSuspendSchema = z.object({
   })
 });
 var PlanningIterationResumeSchema = z.object({
-  answers: z.record(z.string())
+  answers: z.record(z.string(), z.string())
 });
 var PlanningAgentOutputSchema = z.object({
   tasks: TaskSchema,
@@ -9384,10 +9394,11 @@ var planningIterationStep = createStep({
     const newAnswers = { ...userAnswers || {}, ...resumeData?.answers || {} };
     if (Object.keys(newAnswers).length > 0) {
       storedQAPairs = storedQAPairs.map((pair) => {
-        if (newAnswers[pair.question.id]) {
+        const answerValue = newAnswers[pair.question.id];
+        if (answerValue) {
           return {
             ...pair,
-            answer: newAnswers[pair.question.id] || null,
+            answer: String(answerValue) || null,
             answeredAt: (/* @__PURE__ */ new Date()).toISOString()
           };
         }