@mastra/ai-sdk 0.0.5 → 0.1.0

package/CHANGELOG.md

@@ -1,5 +1,27 @@
 # @mastra/ai-sdk
 
+## 0.1.0
+
+### Minor Changes
+
+- Breaking change to move the agent.streamVNext/generateVNext implementation to the default stream/generate. The old stream/generate have now been moved to streamLegacy and generateLegacy ([#8097](https://github.com/mastra-ai/mastra/pull/8097))
+
+### Patch Changes
+
+- Updated dependencies [[`00cb6bd`](https://github.com/mastra-ai/mastra/commit/00cb6bdf78737c0fac14a5a0c7b532a11e38558a), [`869ba22`](https://github.com/mastra-ai/mastra/commit/869ba222e1d6b58fc1b65e7c9fd55ca4e01b8c2f), [`1b73665`](https://github.com/mastra-ai/mastra/commit/1b73665e8e23f5c09d49fcf3e7d709c75259259e), [`f7d7475`](https://github.com/mastra-ai/mastra/commit/f7d747507341aef60ed39e4b49318db1f86034a6), [`084b77b`](https://github.com/mastra-ai/mastra/commit/084b77b2955960e0190af8db3f77138aa83ed65c), [`a93ff84`](https://github.com/mastra-ai/mastra/commit/a93ff84b5e1af07ee236ac8873dac9b49aa5d501), [`bc5aacb`](https://github.com/mastra-ai/mastra/commit/bc5aacb646d468d325327e36117129f28cd13bf6), [`6b5af12`](https://github.com/mastra-ai/mastra/commit/6b5af12ce9e09066e0c32e821c203a6954498bea), [`bf60e4a`](https://github.com/mastra-ai/mastra/commit/bf60e4a89c515afd9570b7b79f33b95e7d07c397), [`d41aee5`](https://github.com/mastra-ai/mastra/commit/d41aee526d124e35f42720a08e64043229193679), [`e8fe13c`](https://github.com/mastra-ai/mastra/commit/e8fe13c4b4c255a42520127797ec394310f7c919), [`3ca833d`](https://github.com/mastra-ai/mastra/commit/3ca833dc994c38e3c9b4f9b4478a61cd8e07b32a), [`1edb8d1`](https://github.com/mastra-ai/mastra/commit/1edb8d1cfb963e72a12412990fb9170936c9904c), [`fbf6e32`](https://github.com/mastra-ai/mastra/commit/fbf6e324946332d0f5ed8930bf9d4d4479cefd7a), [`4753027`](https://github.com/mastra-ai/mastra/commit/4753027ee889288775c6958bdfeda03ff909af67)]:
+  - @mastra/core@0.20.0
+
+## 0.1.0-alpha.0
+
+### Minor Changes
+
+- Breaking change to move the agent.streamVNext/generateVNext implementation to the default stream/generate. The old stream/generate have now been moved to streamLegacy and generateLegacy ([#8097](https://github.com/mastra-ai/mastra/pull/8097))
+
+### Patch Changes
+
+- Updated dependencies [[`00cb6bd`](https://github.com/mastra-ai/mastra/commit/00cb6bdf78737c0fac14a5a0c7b532a11e38558a), [`869ba22`](https://github.com/mastra-ai/mastra/commit/869ba222e1d6b58fc1b65e7c9fd55ca4e01b8c2f), [`1b73665`](https://github.com/mastra-ai/mastra/commit/1b73665e8e23f5c09d49fcf3e7d709c75259259e), [`f7d7475`](https://github.com/mastra-ai/mastra/commit/f7d747507341aef60ed39e4b49318db1f86034a6), [`084b77b`](https://github.com/mastra-ai/mastra/commit/084b77b2955960e0190af8db3f77138aa83ed65c), [`a93ff84`](https://github.com/mastra-ai/mastra/commit/a93ff84b5e1af07ee236ac8873dac9b49aa5d501), [`bc5aacb`](https://github.com/mastra-ai/mastra/commit/bc5aacb646d468d325327e36117129f28cd13bf6), [`6b5af12`](https://github.com/mastra-ai/mastra/commit/6b5af12ce9e09066e0c32e821c203a6954498bea), [`bf60e4a`](https://github.com/mastra-ai/mastra/commit/bf60e4a89c515afd9570b7b79f33b95e7d07c397), [`d41aee5`](https://github.com/mastra-ai/mastra/commit/d41aee526d124e35f42720a08e64043229193679), [`e8fe13c`](https://github.com/mastra-ai/mastra/commit/e8fe13c4b4c255a42520127797ec394310f7c919), [`3ca833d`](https://github.com/mastra-ai/mastra/commit/3ca833dc994c38e3c9b4f9b4478a61cd8e07b32a), [`1edb8d1`](https://github.com/mastra-ai/mastra/commit/1edb8d1cfb963e72a12412990fb9170936c9904c), [`fbf6e32`](https://github.com/mastra-ai/mastra/commit/fbf6e324946332d0f5ed8930bf9d4d4479cefd7a), [`4753027`](https://github.com/mastra-ai/mastra/commit/4753027ee889288775c6958bdfeda03ff909af67)]:
+  - @mastra/core@0.20.0-alpha.0
+
 ## 0.0.5
 
 ### Patch Changes

package/dist/index.cjs

@@ -124,7 +124,7 @@ function chatRoute({
       if (!agentObj) {
         throw new Error(`Agent ${agentToUse} not found`);
       }
-      const result = await agentObj.streamVNext(messages, {
+      const result = await agentObj.stream(messages, {
         ...defaultOptions,
         ...rest,
         format: "aisdk"

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/chat-route.ts","../src/to-ai-sdk-format.ts"],"names":["registerApiRoute"],"mappings":";;;;;AAiBO,SAAS,SAAA,CAAmD;AAAA,EACjE,IAAA,GAAO,gBAAA;AAAA,EACP,KAAA;AAAA,EACA;AACF,CAAA,EAAkE;AAChE,EAAA,IAAI,CAAC,KAAA,IAAS,CAAC,IAAA,CAAK,QAAA,CAAS,WAAW,CAAA,EAAG;AACzC,IAAA,MAAM,IAAI,MAAM,uFAAuF,CAAA;AAAA,EACzG;AAEA,EAAA,OAAOA,wBAAiB,IAAA,EAAM;AAAA,IAC5B,MAAA,EAAQ,MAAA;AAAA,IACR,OAAA,EAAS;AAAA,MACP,OAAA,EAAS,oBAAA;AAAA,MACT,WAAA,EAAa,wEAAA;AAAA,MACb,IAAA,EAAM,CAAC,QAAQ,CAAA;AAAA,MACf,UAAA,EAAY;AAAA,QACV;AAAA,UACE,IAAA,EAAM,SAAA;AAAA,UACN,EAAA,EAAI,MAAA;AAAA,UACJ,QAAA,EAAU,IAAA;AAAA,UACV,WAAA,EAAa,kCAAA;AAAA,UACb,MAAA,EAAQ;AAAA,YACN,IAAA,EAAM;AAAA;AACR;AACF,OACF;AAAA,MACA,WAAA,EAAa;AAAA,QACX,QAAA,EAAU,IAAA;AAAA,QACV,OAAA,EAAS;AAAA,UACP,kBAAA,EAAoB;AAAA,YAClB,MAAA,EAAQ;AAAA,cACN,IAAA,EAAM,QAAA;AAAA,cACN,UAAA,EAAY;AAAA,gBACV,QAAA,EAAU;AAAA,kBACR,IAAA,EAAM,OAAA;AAAA,kBACN,WAAA,EAAa,uCAAA;AAAA,kBACb,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM,QAAA;AAAA,oBACN,UAAA,EAAY;AAAA,sBACV,IAAA,EAAM;AAAA,wBACJ,IAAA,EAAM,QAAA;AAAA,wBACN,IAAA,EAAM,CAAC,MAAA,EAAQ,WAAA,EAAa,QAAQ,CAAA;AAAA,wBACpC,WAAA,EAAa;AAAA,uBACf;AAAA,sBACA,OAAA,EAAS;AAAA,wBACP,IAAA,EAAM,QAAA;AAAA,wBACN,WAAA,EAAa;AAAA;AACf,qBACF;AAAA,oBACA,QAAA,EAAU,CAAC,MAAA,EAAQ,SAAS;AAAA;AAC9B;AACF,eACF;AAAA,cACA,QAAA,EAAU,CAAC,UAAU;AAAA;AACvB;AACF;AACF,OACF;AAAA,MACA,SAAA,EAAW;AAAA,QACT,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,mCAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,YAAA,EAAc;AAAA,cACZ,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,WAAA,EAAa;AAAA;AACf;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,6BAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,iBAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF;AACF;AACF,KACF;AAAA,IACA,OAAA,EAAS,OAAM,CAAA,KAAK;AAClB,MAAA,MAAM,EAAE,UAAU,GAAG,IAAA,KAAS,MAAM,CAAA,CAAE,IAAI,IAAA,EAAK;AAC/C,MAAA,MAAM,MAAA,GAAS,CAAA,CAAE,GAAA,CAAI,QAAQ,CAAA;AAE7B,MAAA,IAAI,UAAA,GAAiC,KAAA;AACrC,MAAA,IAAI,CAAC,KAAA,EAAO;AACV,QAAA,MAAM,OAAA,GAAU,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,CAAA;AACrC,QAAA,UAAA,GAAa,OAAA;AAAA,MACf;AAEA,MAAA,IAAI,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,KAAK,KAAA,EAAO;AACnC,QAAA,MAAA,CACG,WAAU,EACT,IAAA;AAAA,UACA,CAAA,qGAAA;AAAA,SACF;AAAA,MACJ;AAEA,MAAA,IAAI,CAAC,UAAA,EAAY;AACf,QAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,MACxC;AAEA,MAAA,MAAM,QAAA,GAAW,MAAA,CAAO,QAAA,CAAS,UAAU,CAAA;AAC3C,MAAA,IAAI,CAAC,QAAA,EAAU;AACb,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,MAAA,EAAS,UAAU,CAAA,UAAA,CAAY,CAAA;AAAA,MACjD;AAEA,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,WAAA,CAA6B,QAAA,EAAU;AAAA,QACnE,GAAG,cAAA;AAAA,QACH,GAAG,IAAA;AAAA,QACH,MAAA,EAAQ;AAAA,OACT,CAAA;AAED,MAAA,OAAO,OAAO,yBAAA,EAA0B;AAAA,IAC1C;AAAA,GACD,CAAA;AACH;;;AC1HO,SAAS,+BAAA,GAAkC;AAChD,EAAA,MAAM,QAAoC,EAAC;AAC3C,EAAA,OAAO,IAAI,eAAA,CAKT;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW;AAAA,SACZ;AAAA,OACF,CAAA;AAAA,IACH,CAAA;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM;AAAA,SACP;AAAA,OACF,CAAA;AACD,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM;AAAA,OACP,CAAA;AAAA,IACH,CAAA;AAAA,IACA,SAAA,CAAU,OAAO,UAAA,EAAY;AAC3B,MAAA,IAAI,YAAA,GAAe,EAAA;AACnB,MAAA,IAAI,KAAA,CAAM,SAAS,gBAAA,EAAkB;AAEnC,QAAA,YAAA,GAAe,MAAM,OAAA,CAAQ,UAAA;AAC7B,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,OAAO,EAAC;AAAA,cACR,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,qBAAA,EAAuB;AAC/C,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA;AAAA,UAExB,IAAA,EAAM,MAAM,OAAA,CAAQ,EAAA;AAAA,UACpB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,KAAA,EAAO,KAAA,CAAM,OAAA,CAAQ,OAAA,IAAW,IAAA;AAAA,UAChC,MAAA,EAAQ;AAAA,SACV;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,sBAAA,EAAwB;AAChD,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA,UACxB,GAAG,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA;AAAA,UACzB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU;AAAA,SAClC;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,iBAAA,EAAmB;AAC3C,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,KAAA;AAAA,cACA,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU,IAAA;AAAA,cAChC,MAAA,EAAQ,MAAM,OAAA,CAAQ;AAAA;AACxB,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH;AAAA,IACF;AAAA,GACD,CAAA;AACH;AAEO,SAAS,cAId,MAAA,EAAuD;AACvD,EAAA,OAAO,MAAA,CAAO,WAAA,CAAY,+BAAA,EAAiC,CAAA;AAC7D","file":"index.cjs","sourcesContent":["import type { AgentExecutionOptions } from '@mastra/core/agent';\nimport { registerApiRoute } from '@mastra/core/server';\nimport type { OutputSchema } from '@mastra/core/stream';\n\nexport type chatRouteOptions<OUTPUT extends OutputSchema = undefined> = {\n  defaultOptions?: AgentExecutionOptions<OUTPUT, 'aisdk'>;\n} & (\n  | {\n      path: `${string}:agentId${string}`;\n      agent?: never;\n    }\n  | {\n      path: string;\n      agent: string;\n    }\n);\n\nexport function chatRoute<OUTPUT extends OutputSchema = undefined>({\n  path = '/chat/:agentId',\n  agent,\n  defaultOptions,\n}: chatRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute> {\n  if (!agent && !path.includes('/:agentId')) {\n    throw new Error('Path must include :agentId to route to the correct agent or pass the agent explicitly');\n  }\n\n  return registerApiRoute(path, {\n    method: 'POST',\n    openapi: {\n      summary: 'Chat with an agent',\n      description: 'Send messages to an agent and stream the response in the AI SDK format',\n      tags: ['ai-sdk'],\n      parameters: [\n        {\n          name: 'agentId',\n          in: 'path',\n          required: true,\n          description: 'The ID of the agent to chat with',\n          schema: {\n            type: 'string',\n          },\n        },\n      ],\n      requestBody: {\n        required: true,\n        content: {\n          'application/json': {\n            schema: {\n              type: 'object',\n              properties: {\n                messages: {\n                  type: 'array',\n                  description: 'Array of messages in the conversation',\n                  items: {\n                    type: 'object',\n                    properties: {\n                      role: {\n                        type: 'string',\n                        enum: ['user', 'assistant', 'system'],\n                        description: 'The role of the message sender',\n                      },\n                      content: {\n                        type: 'string',\n                        description: 'The content of the message',\n                      },\n                    },\n                    required: ['role', 'content'],\n                  },\n                },\n              },\n              required: ['messages'],\n            },\n          },\n        },\n      },\n      responses: {\n        '200': {\n          description: 'Streaming response from the agent',\n          content: {\n            'text/plain': {\n              schema: {\n                type: 'string',\n                description: 'Server-sent events stream containing the agent response',\n              },\n            },\n          },\n        },\n        '400': {\n          description: 'Bad request - invalid input',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n        '404': {\n          description: 'Agent not found',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n      },\n    },\n    handler: async c => {\n      const { messages, ...rest } = await c.req.json();\n      const mastra = c.get('mastra');\n\n      let agentToUse: string | undefined = agent;\n      if (!agent) {\n        const agentId = c.req.param('agentId');\n        agentToUse = agentId;\n      }\n\n      if (c.req.param('agentId') && agent) {\n        mastra\n          .getLogger()\n          ?.warn(\n            `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`,\n          );\n      }\n\n      if (!agentToUse) {\n        throw new Error('Agent ID is required');\n      }\n\n      const agentObj = mastra.getAgent(agentToUse);\n      if (!agentObj) {\n        throw new Error(`Agent ${agentToUse} not found`);\n      }\n\n      const result = await agentObj.streamVNext<OUTPUT, 'aisdk'>(messages, {\n        ...defaultOptions,\n        ...rest,\n        format: 'aisdk',\n      });\n\n      return result.toUIMessageStreamResponse();\n    },\n  });\n}\n","import type {\n  ChunkType,\n  MastraWorkflowStream,\n  Step,\n  WorkflowRunStatus,\n  WorkflowStepStatus,\n} from '@mastra/core/workflows';\nimport type { ZodType } from 'zod';\n\ntype StepResult = {\n  name: string;\n  status: WorkflowStepStatus;\n  input: Record<string, unknown> | null;\n  output: Record<string, unknown> | null;\n};\n\nexport type WorkflowAiSDKType = {\n  type: 'data-workflow';\n  id: string;\n  data: {\n    name: string;\n    status: WorkflowRunStatus;\n    steps: Record<string, StepResult>;\n    output: {\n      usage: {\n        inputTokens: number;\n        outputTokens: number;\n        totalTokens: number;\n      };\n    } | null;\n  };\n};\n\nexport function WokflowStreamToAISDKTransformer() {\n  const steps: Record<string, StepResult> = {};\n  return new TransformStream<\n    ChunkType,\n    {\n      data: string;\n    }\n  >({\n    start(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'start',\n          messageId: '1',\n        }),\n      });\n    },\n    flush(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'finish',\n        }),\n      });\n      controller.enqueue({\n        data: '[DONE]',\n      });\n    },\n    transform(chunk, controller) {\n      let workflowName = '';\n      if (chunk.type === 'workflow-start') {\n        // TODO swap with name\n        workflowName = chunk.payload.workflowId;\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps: {} as Record<string, StepResult>,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-start') {\n        steps[chunk.payload.id] = {\n          // TODO swap with name\n          name: chunk.payload.id,\n          status: chunk.payload.status,\n          input: chunk.payload.payload ?? null,\n          output: null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-result') {\n        steps[chunk.payload.id] = {\n          ...steps[chunk.payload.id]!,\n          status: chunk.payload.status,\n          output: chunk.payload.output ?? null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-finish') {\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              steps,\n              output: chunk.payload.output ?? null,\n              status: chunk.payload.workflowStatus,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      }\n    },\n  });\n}\n\nexport function toAISdkFormat<\n  TInput extends ZodType<any>,\n  TOutput extends ZodType<any>,\n  TSteps extends Step<string, any, any>[],\n>(stream: MastraWorkflowStream<TInput, TOutput, TSteps>) {\n  return stream.pipeThrough(WokflowStreamToAISDKTransformer());\n}\n"]}
+{"version":3,"sources":["../src/chat-route.ts","../src/to-ai-sdk-format.ts"],"names":["registerApiRoute"],"mappings":";;;;;AAiBO,SAAS,SAAA,CAAmD;AAAA,EACjE,IAAA,GAAO,gBAAA;AAAA,EACP,KAAA;AAAA,EACA;AACF,CAAA,EAAkE;AAChE,EAAA,IAAI,CAAC,KAAA,IAAS,CAAC,IAAA,CAAK,QAAA,CAAS,WAAW,CAAA,EAAG;AACzC,IAAA,MAAM,IAAI,MAAM,uFAAuF,CAAA;AAAA,EACzG;AAEA,EAAA,OAAOA,wBAAiB,IAAA,EAAM;AAAA,IAC5B,MAAA,EAAQ,MAAA;AAAA,IACR,OAAA,EAAS;AAAA,MACP,OAAA,EAAS,oBAAA;AAAA,MACT,WAAA,EAAa,wEAAA;AAAA,MACb,IAAA,EAAM,CAAC,QAAQ,CAAA;AAAA,MACf,UAAA,EAAY;AAAA,QACV;AAAA,UACE,IAAA,EAAM,SAAA;AAAA,UACN,EAAA,EAAI,MAAA;AAAA,UACJ,QAAA,EAAU,IAAA;AAAA,UACV,WAAA,EAAa,kCAAA;AAAA,UACb,MAAA,EAAQ;AAAA,YACN,IAAA,EAAM;AAAA;AACR;AACF,OACF;AAAA,MACA,WAAA,EAAa;AAAA,QACX,QAAA,EAAU,IAAA;AAAA,QACV,OAAA,EAAS;AAAA,UACP,kBAAA,EAAoB;AAAA,YAClB,MAAA,EAAQ;AAAA,cACN,IAAA,EAAM,QAAA;AAAA,cACN,UAAA,EAAY;AAAA,gBACV,QAAA,EAAU;AAAA,kBACR,IAAA,EAAM,OAAA;AAAA,kBACN,WAAA,EAAa,uCAAA;AAAA,kBACb,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM,QAAA;AAAA,oBACN,UAAA,EAAY;AAAA,sBACV,IAAA,EAAM;AAAA,wBACJ,IAAA,EAAM,QAAA;AAAA,wBACN,IAAA,EAAM,CAAC,MAAA,EAAQ,WAAA,EAAa,QAAQ,CAAA;AAAA,wBACpC,WAAA,EAAa;AAAA,uBACf;AAAA,sBACA,OAAA,EAAS;AAAA,wBACP,IAAA,EAAM,QAAA;AAAA,wBACN,WAAA,EAAa;AAAA;AACf,qBACF;AAAA,oBACA,QAAA,EAAU,CAAC,MAAA,EAAQ,SAAS;AAAA;AAC9B;AACF,eACF;AAAA,cACA,QAAA,EAAU,CAAC,UAAU;AAAA;AACvB;AACF;AACF,OACF;AAAA,MACA,SAAA,EAAW;AAAA,QACT,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,mCAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,YAAA,EAAc;AAAA,cACZ,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,WAAA,EAAa;AAAA;AACf;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,6BAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,iBAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF;AACF;AACF,KACF;AAAA,IACA,OAAA,EAAS,OAAM,CAAA,KAAK;AAClB,MAAA,MAAM,EAAE,UAAU,GAAG,IAAA,KAAS,MAAM,CAAA,CAAE,IAAI,IAAA,EAAK;AAC/C,MAAA,MAAM,MAAA,GAAS,CAAA,CAAE,GAAA,CAAI,QAAQ,CAAA;AAE7B,MAAA,IAAI,UAAA,GAAiC,KAAA;AACrC,MAAA,IAAI,CAAC,KAAA,EAAO;AACV,QAAA,MAAM,OAAA,GAAU,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,CAAA;AACrC,QAAA,UAAA,GAAa,OAAA;AAAA,MACf;AAEA,MAAA,IAAI,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,KAAK,KAAA,EAAO;AACnC,QAAA,MAAA,CACG,WAAU,EACT,IAAA;AAAA,UACA,CAAA,qGAAA;AAAA,SACF;AAAA,MACJ;AAEA,MAAA,IAAI,CAAC,UAAA,EAAY;AACf,QAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,MACxC;AAEA,MAAA,MAAM,QAAA,GAAW,MAAA,CAAO,QAAA,CAAS,UAAU,CAAA;AAC3C,MAAA,IAAI,CAAC,QAAA,EAAU;AACb,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,MAAA,EAAS,UAAU,CAAA,UAAA,CAAY,CAAA;AAAA,MACjD;AAEA,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,MAAA,CAAwB,QAAA,EAAU;AAAA,QAC9D,GAAG,cAAA;AAAA,QACH,GAAG,IAAA;AAAA,QACH,MAAA,EAAQ;AAAA,OACT,CAAA;AAED,MAAA,OAAO,OAAO,yBAAA,EAA0B;AAAA,IAC1C;AAAA,GACD,CAAA;AACH;;;AC1HO,SAAS,+BAAA,GAAkC;AAChD,EAAA,MAAM,QAAoC,EAAC;AAC3C,EAAA,OAAO,IAAI,eAAA,CAKT;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW;AAAA,SACZ;AAAA,OACF,CAAA;AAAA,IACH,CAAA;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM;AAAA,SACP;AAAA,OACF,CAAA;AACD,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM;AAAA,OACP,CAAA;AAAA,IACH,CAAA;AAAA,IACA,SAAA,CAAU,OAAO,UAAA,EAAY;AAC3B,MAAA,IAAI,YAAA,GAAe,EAAA;AACnB,MAAA,IAAI,KAAA,CAAM,SAAS,gBAAA,EAAkB;AAEnC,QAAA,YAAA,GAAe,MAAM,OAAA,CAAQ,UAAA;AAC7B,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,OAAO,EAAC;AAAA,cACR,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,qBAAA,EAAuB;AAC/C,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA;AAAA,UAExB,IAAA,EAAM,MAAM,OAAA,CAAQ,EAAA;AAAA,UACpB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,KAAA,EAAO,KAAA,CAAM,OAAA,CAAQ,OAAA,IAAW,IAAA;AAAA,UAChC,MAAA,EAAQ;AAAA,SACV;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,sBAAA,EAAwB;AAChD,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA,UACxB,GAAG,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA;AAAA,UACzB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU;AAAA,SAClC;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,iBAAA,EAAmB;AAC3C,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,KAAA;AAAA,cACA,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU,IAAA;AAAA,cAChC,MAAA,EAAQ,MAAM,OAAA,CAAQ;AAAA;AACxB,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH;AAAA,IACF;AAAA,GACD,CAAA;AACH;AAEO,SAAS,cAId,MAAA,EAAuD;AACvD,EAAA,OAAO,MAAA,CAAO,WAAA,CAAY,+BAAA,EAAiC,CAAA;AAC7D","file":"index.cjs","sourcesContent":["import type { AgentExecutionOptions } from '@mastra/core/agent';\nimport { registerApiRoute } from '@mastra/core/server';\nimport type { OutputSchema } from '@mastra/core/stream';\n\nexport type chatRouteOptions<OUTPUT extends OutputSchema = undefined> = {\n  defaultOptions?: AgentExecutionOptions<OUTPUT, 'aisdk'>;\n} & (\n  | {\n      path: `${string}:agentId${string}`;\n      agent?: never;\n    }\n  | {\n      path: string;\n      agent: string;\n    }\n);\n\nexport function chatRoute<OUTPUT extends OutputSchema = undefined>({\n  path = '/chat/:agentId',\n  agent,\n  defaultOptions,\n}: chatRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute> {\n  if (!agent && !path.includes('/:agentId')) {\n    throw new Error('Path must include :agentId to route to the correct agent or pass the agent explicitly');\n  }\n\n  return registerApiRoute(path, {\n    method: 'POST',\n    openapi: {\n      summary: 'Chat with an agent',\n      description: 'Send messages to an agent and stream the response in the AI SDK format',\n      tags: ['ai-sdk'],\n      parameters: [\n        {\n          name: 'agentId',\n          in: 'path',\n          required: true,\n          description: 'The ID of the agent to chat with',\n          schema: {\n            type: 'string',\n          },\n        },\n      ],\n      requestBody: {\n        required: true,\n        content: {\n          'application/json': {\n            schema: {\n              type: 'object',\n              properties: {\n                messages: {\n                  type: 'array',\n                  description: 'Array of messages in the conversation',\n                  items: {\n                    type: 'object',\n                    properties: {\n                      role: {\n                        type: 'string',\n                        enum: ['user', 'assistant', 'system'],\n                        description: 'The role of the message sender',\n                      },\n                      content: {\n                        type: 'string',\n                        description: 'The content of the message',\n                      },\n                    },\n                    required: ['role', 'content'],\n                  },\n                },\n              },\n              required: ['messages'],\n            },\n          },\n        },\n      },\n      responses: {\n        '200': {\n          description: 'Streaming response from the agent',\n          content: {\n            'text/plain': {\n              schema: {\n                type: 'string',\n                description: 'Server-sent events stream containing the agent response',\n              },\n            },\n          },\n        },\n        '400': {\n          description: 'Bad request - invalid input',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n        '404': {\n          description: 'Agent not found',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n      },\n    },\n    handler: async c => {\n      const { messages, ...rest } = await c.req.json();\n      const mastra = c.get('mastra');\n\n      let agentToUse: string | undefined = agent;\n      if (!agent) {\n        const agentId = c.req.param('agentId');\n        agentToUse = agentId;\n      }\n\n      if (c.req.param('agentId') && agent) {\n        mastra\n          .getLogger()\n          ?.warn(\n            `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`,\n          );\n      }\n\n      if (!agentToUse) {\n        throw new Error('Agent ID is required');\n      }\n\n      const agentObj = mastra.getAgent(agentToUse);\n      if (!agentObj) {\n        throw new Error(`Agent ${agentToUse} not found`);\n      }\n\n      const result = await agentObj.stream<OUTPUT, 'aisdk'>(messages, {\n        ...defaultOptions,\n        ...rest,\n        format: 'aisdk',\n      });\n\n      return result.toUIMessageStreamResponse();\n    },\n  });\n}\n","import type {\n  ChunkType,\n  MastraWorkflowStream,\n  Step,\n  WorkflowRunStatus,\n  WorkflowStepStatus,\n} from '@mastra/core/workflows';\nimport type { ZodType } from 'zod';\n\ntype StepResult = {\n  name: string;\n  status: WorkflowStepStatus;\n  input: Record<string, unknown> | null;\n  output: Record<string, unknown> | null;\n};\n\nexport type WorkflowAiSDKType = {\n  type: 'data-workflow';\n  id: string;\n  data: {\n    name: string;\n    status: WorkflowRunStatus;\n    steps: Record<string, StepResult>;\n    output: {\n      usage: {\n        inputTokens: number;\n        outputTokens: number;\n        totalTokens: number;\n      };\n    } | null;\n  };\n};\n\nexport function WokflowStreamToAISDKTransformer() {\n  const steps: Record<string, StepResult> = {};\n  return new TransformStream<\n    ChunkType,\n    {\n      data: string;\n    }\n  >({\n    start(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'start',\n          messageId: '1',\n        }),\n      });\n    },\n    flush(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'finish',\n        }),\n      });\n      controller.enqueue({\n        data: '[DONE]',\n      });\n    },\n    transform(chunk, controller) {\n      let workflowName = '';\n      if (chunk.type === 'workflow-start') {\n        // TODO swap with name\n        workflowName = chunk.payload.workflowId;\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps: {} as Record<string, StepResult>,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-start') {\n        steps[chunk.payload.id] = {\n          // TODO swap with name\n          name: chunk.payload.id,\n          status: chunk.payload.status,\n          input: chunk.payload.payload ?? null,\n          output: null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-result') {\n        steps[chunk.payload.id] = {\n          ...steps[chunk.payload.id]!,\n          status: chunk.payload.status,\n          output: chunk.payload.output ?? null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-finish') {\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              steps,\n              output: chunk.payload.output ?? null,\n              status: chunk.payload.workflowStatus,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      }\n    },\n  });\n}\n\nexport function toAISdkFormat<\n  TInput extends ZodType<any>,\n  TOutput extends ZodType<any>,\n  TSteps extends Step<string, any, any>[],\n>(stream: MastraWorkflowStream<TInput, TOutput, TSteps>) {\n  return stream.pipeThrough(WokflowStreamToAISDKTransformer());\n}\n"]}

package/dist/index.js

@@ -122,7 +122,7 @@ function chatRoute({
       if (!agentObj) {
         throw new Error(`Agent ${agentToUse} not found`);
       }
-      const result = await agentObj.streamVNext(messages, {
+      const result = await agentObj.stream(messages, {
         ...defaultOptions,
         ...rest,
         format: "aisdk"

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/chat-route.ts","../src/to-ai-sdk-format.ts"],"names":[],"mappings":";;;AAiBO,SAAS,SAAA,CAAmD;AAAA,EACjE,IAAA,GAAO,gBAAA;AAAA,EACP,KAAA;AAAA,EACA;AACF,CAAA,EAAkE;AAChE,EAAA,IAAI,CAAC,KAAA,IAAS,CAAC,IAAA,CAAK,QAAA,CAAS,WAAW,CAAA,EAAG;AACzC,IAAA,MAAM,IAAI,MAAM,uFAAuF,CAAA;AAAA,EACzG;AAEA,EAAA,OAAO,iBAAiB,IAAA,EAAM;AAAA,IAC5B,MAAA,EAAQ,MAAA;AAAA,IACR,OAAA,EAAS;AAAA,MACP,OAAA,EAAS,oBAAA;AAAA,MACT,WAAA,EAAa,wEAAA;AAAA,MACb,IAAA,EAAM,CAAC,QAAQ,CAAA;AAAA,MACf,UAAA,EAAY;AAAA,QACV;AAAA,UACE,IAAA,EAAM,SAAA;AAAA,UACN,EAAA,EAAI,MAAA;AAAA,UACJ,QAAA,EAAU,IAAA;AAAA,UACV,WAAA,EAAa,kCAAA;AAAA,UACb,MAAA,EAAQ;AAAA,YACN,IAAA,EAAM;AAAA;AACR;AACF,OACF;AAAA,MACA,WAAA,EAAa;AAAA,QACX,QAAA,EAAU,IAAA;AAAA,QACV,OAAA,EAAS;AAAA,UACP,kBAAA,EAAoB;AAAA,YAClB,MAAA,EAAQ;AAAA,cACN,IAAA,EAAM,QAAA;AAAA,cACN,UAAA,EAAY;AAAA,gBACV,QAAA,EAAU;AAAA,kBACR,IAAA,EAAM,OAAA;AAAA,kBACN,WAAA,EAAa,uCAAA;AAAA,kBACb,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM,QAAA;AAAA,oBACN,UAAA,EAAY;AAAA,sBACV,IAAA,EAAM;AAAA,wBACJ,IAAA,EAAM,QAAA;AAAA,wBACN,IAAA,EAAM,CAAC,MAAA,EAAQ,WAAA,EAAa,QAAQ,CAAA;AAAA,wBACpC,WAAA,EAAa;AAAA,uBACf;AAAA,sBACA,OAAA,EAAS;AAAA,wBACP,IAAA,EAAM,QAAA;AAAA,wBACN,WAAA,EAAa;AAAA;AACf,qBACF;AAAA,oBACA,QAAA,EAAU,CAAC,MAAA,EAAQ,SAAS;AAAA;AAC9B;AACF,eACF;AAAA,cACA,QAAA,EAAU,CAAC,UAAU;AAAA;AACvB;AACF;AACF,OACF;AAAA,MACA,SAAA,EAAW;AAAA,QACT,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,mCAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,YAAA,EAAc;AAAA,cACZ,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,WAAA,EAAa;AAAA;AACf;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,6BAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,iBAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF;AACF;AACF,KACF;AAAA,IACA,OAAA,EAAS,OAAM,CAAA,KAAK;AAClB,MAAA,MAAM,EAAE,UAAU,GAAG,IAAA,KAAS,MAAM,CAAA,CAAE,IAAI,IAAA,EAAK;AAC/C,MAAA,MAAM,MAAA,GAAS,CAAA,CAAE,GAAA,CAAI,QAAQ,CAAA;AAE7B,MAAA,IAAI,UAAA,GAAiC,KAAA;AACrC,MAAA,IAAI,CAAC,KAAA,EAAO;AACV,QAAA,MAAM,OAAA,GAAU,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,CAAA;AACrC,QAAA,UAAA,GAAa,OAAA;AAAA,MACf;AAEA,MAAA,IAAI,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,KAAK,KAAA,EAAO;AACnC,QAAA,MAAA,CACG,WAAU,EACT,IAAA;AAAA,UACA,CAAA,qGAAA;AAAA,SACF;AAAA,MACJ;AAEA,MAAA,IAAI,CAAC,UAAA,EAAY;AACf,QAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,MACxC;AAEA,MAAA,MAAM,QAAA,GAAW,MAAA,CAAO,QAAA,CAAS,UAAU,CAAA;AAC3C,MAAA,IAAI,CAAC,QAAA,EAAU;AACb,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,MAAA,EAAS,UAAU,CAAA,UAAA,CAAY,CAAA;AAAA,MACjD;AAEA,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,WAAA,CAA6B,QAAA,EAAU;AAAA,QACnE,GAAG,cAAA;AAAA,QACH,GAAG,IAAA;AAAA,QACH,MAAA,EAAQ;AAAA,OACT,CAAA;AAED,MAAA,OAAO,OAAO,yBAAA,EAA0B;AAAA,IAC1C;AAAA,GACD,CAAA;AACH;;;AC1HO,SAAS,+BAAA,GAAkC;AAChD,EAAA,MAAM,QAAoC,EAAC;AAC3C,EAAA,OAAO,IAAI,eAAA,CAKT;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW;AAAA,SACZ;AAAA,OACF,CAAA;AAAA,IACH,CAAA;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM;AAAA,SACP;AAAA,OACF,CAAA;AACD,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM;AAAA,OACP,CAAA;AAAA,IACH,CAAA;AAAA,IACA,SAAA,CAAU,OAAO,UAAA,EAAY;AAC3B,MAAA,IAAI,YAAA,GAAe,EAAA;AACnB,MAAA,IAAI,KAAA,CAAM,SAAS,gBAAA,EAAkB;AAEnC,QAAA,YAAA,GAAe,MAAM,OAAA,CAAQ,UAAA;AAC7B,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,OAAO,EAAC;AAAA,cACR,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,qBAAA,EAAuB;AAC/C,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA;AAAA,UAExB,IAAA,EAAM,MAAM,OAAA,CAAQ,EAAA;AAAA,UACpB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,KAAA,EAAO,KAAA,CAAM,OAAA,CAAQ,OAAA,IAAW,IAAA;AAAA,UAChC,MAAA,EAAQ;AAAA,SACV;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,sBAAA,EAAwB;AAChD,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA,UACxB,GAAG,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA;AAAA,UACzB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU;AAAA,SAClC;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,iBAAA,EAAmB;AAC3C,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,KAAA;AAAA,cACA,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU,IAAA;AAAA,cAChC,MAAA,EAAQ,MAAM,OAAA,CAAQ;AAAA;AACxB,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH;AAAA,IACF;AAAA,GACD,CAAA;AACH;AAEO,SAAS,cAId,MAAA,EAAuD;AACvD,EAAA,OAAO,MAAA,CAAO,WAAA,CAAY,+BAAA,EAAiC,CAAA;AAC7D","file":"index.js","sourcesContent":["import type { AgentExecutionOptions } from '@mastra/core/agent';\nimport { registerApiRoute } from '@mastra/core/server';\nimport type { OutputSchema } from '@mastra/core/stream';\n\nexport type chatRouteOptions<OUTPUT extends OutputSchema = undefined> = {\n  defaultOptions?: AgentExecutionOptions<OUTPUT, 'aisdk'>;\n} & (\n  | {\n      path: `${string}:agentId${string}`;\n      agent?: never;\n    }\n  | {\n      path: string;\n      agent: string;\n    }\n);\n\nexport function chatRoute<OUTPUT extends OutputSchema = undefined>({\n  path = '/chat/:agentId',\n  agent,\n  defaultOptions,\n}: chatRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute> {\n  if (!agent && !path.includes('/:agentId')) {\n    throw new Error('Path must include :agentId to route to the correct agent or pass the agent explicitly');\n  }\n\n  return registerApiRoute(path, {\n    method: 'POST',\n    openapi: {\n      summary: 'Chat with an agent',\n      description: 'Send messages to an agent and stream the response in the AI SDK format',\n      tags: ['ai-sdk'],\n      parameters: [\n        {\n          name: 'agentId',\n          in: 'path',\n          required: true,\n          description: 'The ID of the agent to chat with',\n          schema: {\n            type: 'string',\n          },\n        },\n      ],\n      requestBody: {\n        required: true,\n        content: {\n          'application/json': {\n            schema: {\n              type: 'object',\n              properties: {\n                messages: {\n                  type: 'array',\n                  description: 'Array of messages in the conversation',\n                  items: {\n                    type: 'object',\n                    properties: {\n                      role: {\n                        type: 'string',\n                        enum: ['user', 'assistant', 'system'],\n                        description: 'The role of the message sender',\n                      },\n                      content: {\n                        type: 'string',\n                        description: 'The content of the message',\n                      },\n                    },\n                    required: ['role', 'content'],\n                  },\n                },\n              },\n              required: ['messages'],\n            },\n          },\n        },\n      },\n      responses: {\n        '200': {\n          description: 'Streaming response from the agent',\n          content: {\n            'text/plain': {\n              schema: {\n                type: 'string',\n                description: 'Server-sent events stream containing the agent response',\n              },\n            },\n          },\n        },\n        '400': {\n          description: 'Bad request - invalid input',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n        '404': {\n          description: 'Agent not found',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n      },\n    },\n    handler: async c => {\n      const { messages, ...rest } = await c.req.json();\n      const mastra = c.get('mastra');\n\n      let agentToUse: string | undefined = agent;\n      if (!agent) {\n        const agentId = c.req.param('agentId');\n        agentToUse = agentId;\n      }\n\n      if (c.req.param('agentId') && agent) {\n        mastra\n          .getLogger()\n          ?.warn(\n            `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`,\n          );\n      }\n\n      if (!agentToUse) {\n        throw new Error('Agent ID is required');\n      }\n\n      const agentObj = mastra.getAgent(agentToUse);\n      if (!agentObj) {\n        throw new Error(`Agent ${agentToUse} not found`);\n      }\n\n      const result = await agentObj.streamVNext<OUTPUT, 'aisdk'>(messages, {\n        ...defaultOptions,\n        ...rest,\n        format: 'aisdk',\n      });\n\n      return result.toUIMessageStreamResponse();\n    },\n  });\n}\n","import type {\n  ChunkType,\n  MastraWorkflowStream,\n  Step,\n  WorkflowRunStatus,\n  WorkflowStepStatus,\n} from '@mastra/core/workflows';\nimport type { ZodType } from 'zod';\n\ntype StepResult = {\n  name: string;\n  status: WorkflowStepStatus;\n  input: Record<string, unknown> | null;\n  output: Record<string, unknown> | null;\n};\n\nexport type WorkflowAiSDKType = {\n  type: 'data-workflow';\n  id: string;\n  data: {\n    name: string;\n    status: WorkflowRunStatus;\n    steps: Record<string, StepResult>;\n    output: {\n      usage: {\n        inputTokens: number;\n        outputTokens: number;\n        totalTokens: number;\n      };\n    } | null;\n  };\n};\n\nexport function WokflowStreamToAISDKTransformer() {\n  const steps: Record<string, StepResult> = {};\n  return new TransformStream<\n    ChunkType,\n    {\n      data: string;\n    }\n  >({\n    start(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'start',\n          messageId: '1',\n        }),\n      });\n    },\n    flush(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'finish',\n        }),\n      });\n      controller.enqueue({\n        data: '[DONE]',\n      });\n    },\n    transform(chunk, controller) {\n      let workflowName = '';\n      if (chunk.type === 'workflow-start') {\n        // TODO swap with name\n        workflowName = chunk.payload.workflowId;\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps: {} as Record<string, StepResult>,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-start') {\n        steps[chunk.payload.id] = {\n          // TODO swap with name\n          name: chunk.payload.id,\n          status: chunk.payload.status,\n          input: chunk.payload.payload ?? null,\n          output: null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-result') {\n        steps[chunk.payload.id] = {\n          ...steps[chunk.payload.id]!,\n          status: chunk.payload.status,\n          output: chunk.payload.output ?? null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-finish') {\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              steps,\n              output: chunk.payload.output ?? null,\n              status: chunk.payload.workflowStatus,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      }\n    },\n  });\n}\n\nexport function toAISdkFormat<\n  TInput extends ZodType<any>,\n  TOutput extends ZodType<any>,\n  TSteps extends Step<string, any, any>[],\n>(stream: MastraWorkflowStream<TInput, TOutput, TSteps>) {\n  return stream.pipeThrough(WokflowStreamToAISDKTransformer());\n}\n"]}
+{"version":3,"sources":["../src/chat-route.ts","../src/to-ai-sdk-format.ts"],"names":[],"mappings":";;;AAiBO,SAAS,SAAA,CAAmD;AAAA,EACjE,IAAA,GAAO,gBAAA;AAAA,EACP,KAAA;AAAA,EACA;AACF,CAAA,EAAkE;AAChE,EAAA,IAAI,CAAC,KAAA,IAAS,CAAC,IAAA,CAAK,QAAA,CAAS,WAAW,CAAA,EAAG;AACzC,IAAA,MAAM,IAAI,MAAM,uFAAuF,CAAA;AAAA,EACzG;AAEA,EAAA,OAAO,iBAAiB,IAAA,EAAM;AAAA,IAC5B,MAAA,EAAQ,MAAA;AAAA,IACR,OAAA,EAAS;AAAA,MACP,OAAA,EAAS,oBAAA;AAAA,MACT,WAAA,EAAa,wEAAA;AAAA,MACb,IAAA,EAAM,CAAC,QAAQ,CAAA;AAAA,MACf,UAAA,EAAY;AAAA,QACV;AAAA,UACE,IAAA,EAAM,SAAA;AAAA,UACN,EAAA,EAAI,MAAA;AAAA,UACJ,QAAA,EAAU,IAAA;AAAA,UACV,WAAA,EAAa,kCAAA;AAAA,UACb,MAAA,EAAQ;AAAA,YACN,IAAA,EAAM;AAAA;AACR;AACF,OACF;AAAA,MACA,WAAA,EAAa;AAAA,QACX,QAAA,EAAU,IAAA;AAAA,QACV,OAAA,EAAS;AAAA,UACP,kBAAA,EAAoB;AAAA,YAClB,MAAA,EAAQ;AAAA,cACN,IAAA,EAAM,QAAA;AAAA,cACN,UAAA,EAAY;AAAA,gBACV,QAAA,EAAU;AAAA,kBACR,IAAA,EAAM,OAAA;AAAA,kBACN,WAAA,EAAa,uCAAA;AAAA,kBACb,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM,QAAA;AAAA,oBACN,UAAA,EAAY;AAAA,sBACV,IAAA,EAAM;AAAA,wBACJ,IAAA,EAAM,QAAA;AAAA,wBACN,IAAA,EAAM,CAAC,MAAA,EAAQ,WAAA,EAAa,QAAQ,CAAA;AAAA,wBACpC,WAAA,EAAa;AAAA,uBACf;AAAA,sBACA,OAAA,EAAS;AAAA,wBACP,IAAA,EAAM,QAAA;AAAA,wBACN,WAAA,EAAa;AAAA;AACf,qBACF;AAAA,oBACA,QAAA,EAAU,CAAC,MAAA,EAAQ,SAAS;AAAA;AAC9B;AACF,eACF;AAAA,cACA,QAAA,EAAU,CAAC,UAAU;AAAA;AACvB;AACF;AACF,OACF;AAAA,MACA,SAAA,EAAW;AAAA,QACT,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,mCAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,YAAA,EAAc;AAAA,cACZ,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,WAAA,EAAa;AAAA;AACf;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,6BAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF,SACF;AAAA,QACA,KAAA,EAAO;AAAA,UACL,WAAA,EAAa,iBAAA;AAAA,UACb,OAAA,EAAS;AAAA,YACP,kBAAA,EAAoB;AAAA,cAClB,MAAA,EAAQ;AAAA,gBACN,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY;AAAA,kBACV,KAAA,EAAO;AAAA,oBACL,IAAA,EAAM;AAAA;AACR;AACF;AACF;AACF;AACF;AACF;AACF,KACF;AAAA,IACA,OAAA,EAAS,OAAM,CAAA,KAAK;AAClB,MAAA,MAAM,EAAE,UAAU,GAAG,IAAA,KAAS,MAAM,CAAA,CAAE,IAAI,IAAA,EAAK;AAC/C,MAAA,MAAM,MAAA,GAAS,CAAA,CAAE,GAAA,CAAI,QAAQ,CAAA;AAE7B,MAAA,IAAI,UAAA,GAAiC,KAAA;AACrC,MAAA,IAAI,CAAC,KAAA,EAAO;AACV,QAAA,MAAM,OAAA,GAAU,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,CAAA;AACrC,QAAA,UAAA,GAAa,OAAA;AAAA,MACf;AAEA,MAAA,IAAI,CAAA,CAAE,GAAA,CAAI,KAAA,CAAM,SAAS,KAAK,KAAA,EAAO;AACnC,QAAA,MAAA,CACG,WAAU,EACT,IAAA;AAAA,UACA,CAAA,qGAAA;AAAA,SACF;AAAA,MACJ;AAEA,MAAA,IAAI,CAAC,UAAA,EAAY;AACf,QAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,MACxC;AAEA,MAAA,MAAM,QAAA,GAAW,MAAA,CAAO,QAAA,CAAS,UAAU,CAAA;AAC3C,MAAA,IAAI,CAAC,QAAA,EAAU;AACb,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,MAAA,EAAS,UAAU,CAAA,UAAA,CAAY,CAAA;AAAA,MACjD;AAEA,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,MAAA,CAAwB,QAAA,EAAU;AAAA,QAC9D,GAAG,cAAA;AAAA,QACH,GAAG,IAAA;AAAA,QACH,MAAA,EAAQ;AAAA,OACT,CAAA;AAED,MAAA,OAAO,OAAO,yBAAA,EAA0B;AAAA,IAC1C;AAAA,GACD,CAAA;AACH;;;AC1HO,SAAS,+BAAA,GAAkC;AAChD,EAAA,MAAM,QAAoC,EAAC;AAC3C,EAAA,OAAO,IAAI,eAAA,CAKT;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW;AAAA,SACZ;AAAA,OACF,CAAA;AAAA,IACH,CAAA;AAAA,IACA,MAAM,UAAA,EAAY;AAChB,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,IAAA,EAAM;AAAA,SACP;AAAA,OACF,CAAA;AACD,MAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,QACjB,IAAA,EAAM;AAAA,OACP,CAAA;AAAA,IACH,CAAA;AAAA,IACA,SAAA,CAAU,OAAO,UAAA,EAAY;AAC3B,MAAA,IAAI,YAAA,GAAe,EAAA;AACnB,MAAA,IAAI,KAAA,CAAM,SAAS,gBAAA,EAAkB;AAEnC,QAAA,YAAA,GAAe,MAAM,OAAA,CAAQ,UAAA;AAC7B,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,OAAO,EAAC;AAAA,cACR,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,qBAAA,EAAuB;AAC/C,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA;AAAA,UAExB,IAAA,EAAM,MAAM,OAAA,CAAQ,EAAA;AAAA,UACpB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,KAAA,EAAO,KAAA,CAAM,OAAA,CAAQ,OAAA,IAAW,IAAA;AAAA,UAChC,MAAA,EAAQ;AAAA,SACV;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,sBAAA,EAAwB;AAChD,QAAA,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA,GAAI;AAAA,UACxB,GAAG,KAAA,CAAM,KAAA,CAAM,OAAA,CAAQ,EAAE,CAAA;AAAA,UACzB,MAAA,EAAQ,MAAM,OAAA,CAAQ,MAAA;AAAA,UACtB,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU;AAAA,SAClC;AAEA,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,MAAA,EAAQ,SAAA;AAAA,cACR,KAAA;AAAA,cACA,MAAA,EAAQ;AAAA;AACV,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH,CAAA,MAAA,IAAW,KAAA,CAAM,IAAA,KAAS,iBAAA,EAAmB;AAC3C,QAAA,UAAA,CAAW,OAAA,CAAQ;AAAA,UACjB,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,YACnB,IAAA,EAAM,eAAA;AAAA,YACN,IAAI,KAAA,CAAM,KAAA;AAAA,YACV,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,YAAA;AAAA,cACN,KAAA;AAAA,cACA,MAAA,EAAQ,KAAA,CAAM,OAAA,CAAQ,MAAA,IAAU,IAAA;AAAA,cAChC,MAAA,EAAQ,MAAM,OAAA,CAAQ;AAAA;AACxB,WAC2B;AAAA,SAC9B,CAAA;AAAA,MACH;AAAA,IACF;AAAA,GACD,CAAA;AACH;AAEO,SAAS,cAId,MAAA,EAAuD;AACvD,EAAA,OAAO,MAAA,CAAO,WAAA,CAAY,+BAAA,EAAiC,CAAA;AAC7D","file":"index.js","sourcesContent":["import type { AgentExecutionOptions } from '@mastra/core/agent';\nimport { registerApiRoute } from '@mastra/core/server';\nimport type { OutputSchema } from '@mastra/core/stream';\n\nexport type chatRouteOptions<OUTPUT extends OutputSchema = undefined> = {\n  defaultOptions?: AgentExecutionOptions<OUTPUT, 'aisdk'>;\n} & (\n  | {\n      path: `${string}:agentId${string}`;\n      agent?: never;\n    }\n  | {\n      path: string;\n      agent: string;\n    }\n);\n\nexport function chatRoute<OUTPUT extends OutputSchema = undefined>({\n  path = '/chat/:agentId',\n  agent,\n  defaultOptions,\n}: chatRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute> {\n  if (!agent && !path.includes('/:agentId')) {\n    throw new Error('Path must include :agentId to route to the correct agent or pass the agent explicitly');\n  }\n\n  return registerApiRoute(path, {\n    method: 'POST',\n    openapi: {\n      summary: 'Chat with an agent',\n      description: 'Send messages to an agent and stream the response in the AI SDK format',\n      tags: ['ai-sdk'],\n      parameters: [\n        {\n          name: 'agentId',\n          in: 'path',\n          required: true,\n          description: 'The ID of the agent to chat with',\n          schema: {\n            type: 'string',\n          },\n        },\n      ],\n      requestBody: {\n        required: true,\n        content: {\n          'application/json': {\n            schema: {\n              type: 'object',\n              properties: {\n                messages: {\n                  type: 'array',\n                  description: 'Array of messages in the conversation',\n                  items: {\n                    type: 'object',\n                    properties: {\n                      role: {\n                        type: 'string',\n                        enum: ['user', 'assistant', 'system'],\n                        description: 'The role of the message sender',\n                      },\n                      content: {\n                        type: 'string',\n                        description: 'The content of the message',\n                      },\n                    },\n                    required: ['role', 'content'],\n                  },\n                },\n              },\n              required: ['messages'],\n            },\n          },\n        },\n      },\n      responses: {\n        '200': {\n          description: 'Streaming response from the agent',\n          content: {\n            'text/plain': {\n              schema: {\n                type: 'string',\n                description: 'Server-sent events stream containing the agent response',\n              },\n            },\n          },\n        },\n        '400': {\n          description: 'Bad request - invalid input',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n        '404': {\n          description: 'Agent not found',\n          content: {\n            'application/json': {\n              schema: {\n                type: 'object',\n                properties: {\n                  error: {\n                    type: 'string',\n                  },\n                },\n              },\n            },\n          },\n        },\n      },\n    },\n    handler: async c => {\n      const { messages, ...rest } = await c.req.json();\n      const mastra = c.get('mastra');\n\n      let agentToUse: string | undefined = agent;\n      if (!agent) {\n        const agentId = c.req.param('agentId');\n        agentToUse = agentId;\n      }\n\n      if (c.req.param('agentId') && agent) {\n        mastra\n          .getLogger()\n          ?.warn(\n            `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`,\n          );\n      }\n\n      if (!agentToUse) {\n        throw new Error('Agent ID is required');\n      }\n\n      const agentObj = mastra.getAgent(agentToUse);\n      if (!agentObj) {\n        throw new Error(`Agent ${agentToUse} not found`);\n      }\n\n      const result = await agentObj.stream<OUTPUT, 'aisdk'>(messages, {\n        ...defaultOptions,\n        ...rest,\n        format: 'aisdk',\n      });\n\n      return result.toUIMessageStreamResponse();\n    },\n  });\n}\n","import type {\n  ChunkType,\n  MastraWorkflowStream,\n  Step,\n  WorkflowRunStatus,\n  WorkflowStepStatus,\n} from '@mastra/core/workflows';\nimport type { ZodType } from 'zod';\n\ntype StepResult = {\n  name: string;\n  status: WorkflowStepStatus;\n  input: Record<string, unknown> | null;\n  output: Record<string, unknown> | null;\n};\n\nexport type WorkflowAiSDKType = {\n  type: 'data-workflow';\n  id: string;\n  data: {\n    name: string;\n    status: WorkflowRunStatus;\n    steps: Record<string, StepResult>;\n    output: {\n      usage: {\n        inputTokens: number;\n        outputTokens: number;\n        totalTokens: number;\n      };\n    } | null;\n  };\n};\n\nexport function WokflowStreamToAISDKTransformer() {\n  const steps: Record<string, StepResult> = {};\n  return new TransformStream<\n    ChunkType,\n    {\n      data: string;\n    }\n  >({\n    start(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'start',\n          messageId: '1',\n        }),\n      });\n    },\n    flush(controller) {\n      controller.enqueue({\n        data: JSON.stringify({\n          type: 'finish',\n        }),\n      });\n      controller.enqueue({\n        data: '[DONE]',\n      });\n    },\n    transform(chunk, controller) {\n      let workflowName = '';\n      if (chunk.type === 'workflow-start') {\n        // TODO swap with name\n        workflowName = chunk.payload.workflowId;\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps: {} as Record<string, StepResult>,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-start') {\n        steps[chunk.payload.id] = {\n          // TODO swap with name\n          name: chunk.payload.id,\n          status: chunk.payload.status,\n          input: chunk.payload.payload ?? null,\n          output: null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-step-result') {\n        steps[chunk.payload.id] = {\n          ...steps[chunk.payload.id]!,\n          status: chunk.payload.status,\n          output: chunk.payload.output ?? null,\n        } satisfies StepResult;\n\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              status: 'running',\n              steps,\n              output: null,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      } else if (chunk.type === 'workflow-finish') {\n        controller.enqueue({\n          data: JSON.stringify({\n            type: 'data-workflow',\n            id: chunk.runId,\n            data: {\n              name: workflowName,\n              steps,\n              output: chunk.payload.output ?? null,\n              status: chunk.payload.workflowStatus,\n            },\n          } satisfies WorkflowAiSDKType),\n        });\n      }\n    },\n  });\n}\n\nexport function toAISdkFormat<\n  TInput extends ZodType<any>,\n  TOutput extends ZodType<any>,\n  TSteps extends Step<string, any, any>[],\n>(stream: MastraWorkflowStream<TInput, TOutput, TSteps>) {\n  return stream.pipeThrough(WokflowStreamToAISDKTransformer());\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/ai-sdk",
-  "version": "0.0.5",
+  "version": "0.1.0",
   "description": "Adds custom API routes to be compatible with the AI SDK UI parts",
   "type": "module",
   "main": "dist/index.js",
@@ -22,7 +22,7 @@
     "CHANGELOG.md"
   ],
   "peerDependencies": {
-    "@mastra/core": ">=0.18.1-0 <0.20.0-0",
+    "@mastra/core": ">=0.20.0-0 <0.21.0-0",
     "zod": "^3.25.0 || ^4.0.0"
   },
   "devDependencies": {
@@ -32,9 +32,9 @@
     "typescript": "^5.8.3",
     "vitest": "^3.2.4",
     "zod": "^3.25.76",
-    "@internal/types-builder": "0.0.18",
-    "@mastra/core": "0.19.0",
-    "@internal/lint": "0.0.43"
+    "@internal/lint": "0.0.45",
+    "@internal/types-builder": "0.0.20",
+    "@mastra/core": "0.20.0"
   },
   "keywords": [
     "mastra",