@mastra/core 0.24.2 → 0.24.3-alpha.0

package/dist/test-utils/llm-mock.cjs

@@ -3,56 +3,94 @@
 var chunkYSQE5IHK_cjs = require('../chunk-YSQE5IHK.cjs');
 var ai = require('ai');
 var test = require('ai/test');
+var test$1 = require('ai-v5/test');
 
 function createMockModel({
   objectGenerationMode,
   mockText,
   spyGenerate,
-  spyStream
+  spyStream,
+  version = "v2"
 }) {
-  const mockModel = new test.MockLanguageModelV1({
-    defaultObjectGenerationMode: objectGenerationMode,
-    doGenerate: async (props) => {
-      if (spyGenerate) {
-        spyGenerate(props);
-      }
-      if (objectGenerationMode === "json") {
+  const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
+  const finalText = objectGenerationMode === "json" ? JSON.stringify(mockText) : text;
+  if (version === "v1") {
+    const mockModel2 = new test.MockLanguageModelV1({
+      defaultObjectGenerationMode: objectGenerationMode,
+      doGenerate: async (props) => {
+        if (spyGenerate) {
+          spyGenerate(props);
+        }
         return {
           rawCall: { rawPrompt: null, rawSettings: {} },
           finishReason: "stop",
           usage: { promptTokens: 10, completionTokens: 20 },
-          text: JSON.stringify(mockText)
+          text: finalText
+        };
+      },
+      doStream: async (props) => {
+        if (spyStream) {
+          spyStream(props);
+        }
+        const chunks = finalText.split(" ").map((word) => ({
+          type: "text-delta",
+          textDelta: word + " "
+        }));
+        return {
+          stream: ai.simulateReadableStream({
+            chunks: [
+              ...chunks,
+              {
+                type: "finish",
+                finishReason: "stop",
+                logprobs: void 0,
+                usage: { completionTokens: 10, promptTokens: 3 }
+              }
+            ]
+          }),
+          rawCall: { rawPrompt: null, rawSettings: {} }
         };
       }
+    });
+    return mockModel2;
+  }
+  const mockModel = new test$1.MockLanguageModelV2({
+    doGenerate: async (props) => {
+      if (spyGenerate) {
+        spyGenerate(props);
+      }
       return {
         rawCall: { rawPrompt: null, rawSettings: {} },
         finishReason: "stop",
-        usage: { promptTokens: 10, completionTokens: 20 },
-        text: typeof mockText === "string" ? mockText : JSON.stringify(mockText)
+        usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+        content: [
+          {
+            type: "text",
+            text: finalText
+          }
+        ],
+        warnings: []
       };
     },
     doStream: async (props) => {
       if (spyStream) {
         spyStream(props);
       }
-      const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
-      const chunks = text.split(" ").map((word) => ({
-        type: "text-delta",
-        textDelta: word + " "
-      }));
       return {
-        stream: ai.simulateReadableStream({
-          chunks: [
-            ...chunks,
-            {
-              type: "finish",
-              finishReason: "stop",
-              logprobs: void 0,
-              usage: { completionTokens: 10, promptTokens: 3 }
-            }
-          ]
-        }),
-        rawCall: { rawPrompt: null, rawSettings: {} }
+        rawCall: { rawPrompt: null, rawSettings: {} },
+        warnings: [],
+        stream: test$1.convertArrayToReadableStream([
+          { type: "stream-start", warnings: [] },
+          { type: "response-metadata", id: "id-0", modelId: "mock-model-id", timestamp: /* @__PURE__ */ new Date(0) },
+          { type: "text-start", id: "1" },
+          { type: "text-delta", id: "1", delta: finalText },
+          { type: "text-end", id: "1" },
+          {
+            type: "finish",
+            finishReason: "stop",
+            usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 }
+          }
+        ])
       };
     }
   });

package/dist/test-utils/llm-mock.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/test-utils/llm-mock.ts"],"names":["MockLanguageModelV1","simulateReadableStream","MastraLLMV1"],"mappings":";;;;;;AAKO,SAAS,eAAA,CAAgB;AAAA,EAC9B,oBAAA;AAAA,EACA,QAAA;AAAA,EACA,WAAA;AAAA,EACA;AACF,CAAA,EAKG;AACD,EAAA,MAAM,SAAA,GAAY,IAAIA,wBAAA,CAAoB;AAAA,IACxC,2BAAA,EAA6B,oBAAA;AAAA,IAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,MAAA,IAAI,WAAA,EAAa;AACf,QAAA,WAAA,CAAY,KAAK,CAAA;AAAA,MACnB;AAEA,MAAA,IAAI,yBAAyB,MAAA,EAAQ;AACnC,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,QAAQ;AAAA,SAC/B;AAAA,MACF;AAEA,MAAA,OAAO;AAAA,QACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,QAC5C,YAAA,EAAc,MAAA;AAAA,QACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,QAChD,MAAM,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ;AAAA,OACzE;AAAA,IACF,CAAA;AAAA,IACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,SAAA,CAAU,KAAK,CAAA;AAAA,MACjB;AAEA,MAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAE9E,MAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,QAC1C,IAAA,EAAM,YAAA;AAAA,QACN,WAAW,IAAA,GAAO;AAAA,OACpB,CAAE,CAAA;AAEF,MAAA,OAAO;AAAA,QACL,QAAQC,yBAAA,CAAuB;AAAA,UAC7B,MAAA,EAAQ;AAAA,YACN,GAAG,MAAA;AAAA,YACH;AAAA,cACE,IAAA,EAAM,QAAA;AAAA,cACN,YAAA,EAAc,MAAA;AAAA,cACd,QAAA,EAAU,MAAA;AAAA,cACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,SACD,CAAA;AAAA,QACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,OAC9C;AAAA,IACF;AAAA,GACD,CAAA;AAED,EAAA,OAAO,SAAA;AACT;AAEO,IAAM,YAAA,GAAN,cAA2BC,6BAAA,CAAY;AAAA,EAC5C,WAAA,CAAY;AAAA,IACV,WAAA;AAAA,IACA,SAAA;AAAA,IACA,oBAAA;AAAA,IACA,QAAA,GAAW;AAAA,GACb,EAKG;AACD,IAAA,MAAM,SAAA,GAAY,IAAIF,wBAAA,CAAoB;AAAA,MACxC,2BAAA,EAA6B,oBAAA;AAAA,MAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,QAAA,IAAI,WAAA,EAAa;AACf,UAAA,WAAA,CAAY,KAAK,CAAA;AAAA,QACnB;AAEA,QAAA,IAAI,yBAAyB,MAAA,EAAQ;AACnC,UAAA,OAAO;AAAA,YACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,YAC5C,YAAA,EAAc,MAAA;AAAA,YACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,YAChD,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,QAAQ;AAAA,WAC/B;AAAA,QACF;AAEA,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,MAAM,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ;AAAA,SACzE;AAAA,MACF,CAAA;AAAA,MACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,SAAA,CAAU,KAAK,CAAA;AAAA,QACjB;AAEA,QAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAE9E,QAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,UAC1C,IAAA,EAAM,YAAA;AAAA,UACN,WAAW,IAAA,GAAO;AAAA,SACpB,CAAE,CAAA;AAEF,QAAA,OAAO;AAAA,UACL,QAAQC,yBAAA,CAAuB;AAAA,YAC7B,MAAA,EAAQ;AAAA,cACN,GAAG,MAAA;AAAA,cACH;AAAA,gBACE,IAAA,EAAM,QAAA;AAAA,gBACN,YAAA,EAAc,MAAA;AAAA,gBACd,QAAA,EAAU,MAAA;AAAA,gBACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,WACD,CAAA;AAAA,UACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,SAC9C;AAAA,MACF;AAAA,KACD,CAAA;AAED,IAAA,KAAA,CAAM,EAAE,KAAA,EAAO,SAAA,EAAW,CAAA;AAAA,EAC5B;AAAA;AAAA,EAGA,UAAU,IAAA,EAAqD;AAE7D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,MAAA,CAAO,GAAG,IAAI,CAAA;AAEnC,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AAAA;AAAA,EAGA,kBAAkB,IAAA,EAA4C;AAE5D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,cAAA,CAAe,GAAG,IAAI,CAAA;AAE3C,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AACF","file":"llm-mock.cjs","sourcesContent":["import { simulateReadableStream } from 'ai';\nimport { MockLanguageModelV1 } from 'ai/test';\n\nimport { MastraLLMV1 } from '../llm/model/model';\n\nexport function createMockModel({\n  objectGenerationMode,\n  mockText,\n  spyGenerate,\n  spyStream,\n}: {\n  objectGenerationMode?: 'json';\n  mockText: string | Record<string, any>;\n  spyGenerate?: (props: any) => void;\n  spyStream?: (props: any) => void;\n}) {\n  const mockModel = new MockLanguageModelV1({\n    defaultObjectGenerationMode: objectGenerationMode,\n    doGenerate: async props => {\n      if (spyGenerate) {\n        spyGenerate(props);\n      }\n\n      if (objectGenerationMode === 'json') {\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: JSON.stringify(mockText),\n        };\n      }\n\n      return {\n        rawCall: { rawPrompt: null, rawSettings: {} },\n        finishReason: 'stop',\n        usage: { promptTokens: 10, completionTokens: 20 },\n        text: typeof mockText === 'string' ? mockText : JSON.stringify(mockText),\n      };\n    },\n    doStream: async props => {\n      if (spyStream) {\n        spyStream(props);\n      }\n\n      const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n      // Split the mock text into chunks for streaming\n      const chunks = text.split(' ').map(word => ({\n        type: 'text-delta' as const,\n        textDelta: word + ' ',\n      }));\n\n      return {\n        stream: simulateReadableStream({\n          chunks: [\n            ...chunks,\n            {\n              type: 'finish',\n              finishReason: 'stop',\n              logprobs: undefined,\n              usage: { completionTokens: 10, promptTokens: 3 },\n            },\n          ],\n        }),\n        rawCall: { rawPrompt: null, rawSettings: {} },\n      };\n    },\n  });\n\n  return mockModel;\n}\n\nexport class MockProvider extends MastraLLMV1 {\n  constructor({\n    spyGenerate,\n    spyStream,\n    objectGenerationMode,\n    mockText = 'Hello, world!',\n  }: {\n    spyGenerate?: (props: any) => void;\n    spyStream?: (props: any) => void;\n    objectGenerationMode?: 'json';\n    mockText?: string | Record<string, any>;\n  }) {\n    const mockModel = new MockLanguageModelV1({\n      defaultObjectGenerationMode: objectGenerationMode,\n      doGenerate: async props => {\n        if (spyGenerate) {\n          spyGenerate(props);\n        }\n\n        if (objectGenerationMode === 'json') {\n          return {\n            rawCall: { rawPrompt: null, rawSettings: {} },\n            finishReason: 'stop',\n            usage: { promptTokens: 10, completionTokens: 20 },\n            text: JSON.stringify(mockText),\n          };\n        }\n\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: typeof mockText === 'string' ? mockText : JSON.stringify(mockText),\n        };\n      },\n      doStream: async props => {\n        if (spyStream) {\n          spyStream(props);\n        }\n\n        const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n        // Split the mock text into chunks for streaming\n        const chunks = text.split(' ').map(word => ({\n          type: 'text-delta' as const,\n          textDelta: word + ' ',\n        }));\n\n        return {\n          stream: simulateReadableStream({\n            chunks: [\n              ...chunks,\n              {\n                type: 'finish',\n                finishReason: 'stop',\n                logprobs: undefined,\n                usage: { completionTokens: 10, promptTokens: 3 },\n              },\n            ],\n          }),\n          rawCall: { rawPrompt: null, rawSettings: {} },\n        };\n      },\n    });\n\n    super({ model: mockModel });\n  }\n\n  // @ts-ignore\n  stream(...args: any): PromiseLike<StreamReturn<any, any, any>> {\n    // @ts-ignore\n    const result = super.stream(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n\n  // @ts-ignore\n  __streamObject(...args): PromiseLike<StreamObjectResult<any>> {\n    // @ts-ignore\n    const result = super.__streamObject(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/test-utils/llm-mock.ts"],"names":["mockModel","MockLanguageModelV1","simulateReadableStream","MockLanguageModelV2","convertArrayToReadableStream","MastraLLMV1"],"mappings":";;;;;;;AAMO,SAAS,eAAA,CAAgB;AAAA,EAC9B,oBAAA;AAAA,EACA,QAAA;AAAA,EACA,WAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA,GAAU;AACZ,CAAA,EAMG;AACD,EAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAC9E,EAAA,MAAM,YAAY,oBAAA,KAAyB,MAAA,GAAS,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAA,GAAI,IAAA;AAE/E,EAAA,IAAI,YAAY,IAAA,EAAM;AAEpB,IAAA,MAAMA,UAAAA,GAAY,IAAIC,wBAAA,CAAoB;AAAA,MACxC,2BAAA,EAA6B,oBAAA;AAAA,MAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,QAAA,IAAI,WAAA,EAAa;AACf,UAAA,WAAA,CAAY,KAAK,CAAA;AAAA,QACnB;AAEA,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,IAAA,EAAM;AAAA,SACR;AAAA,MACF,CAAA;AAAA,MACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,SAAA,CAAU,KAAK,CAAA;AAAA,QACjB;AAGA,QAAA,MAAM,SAAS,SAAA,CAAU,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,UAC/C,IAAA,EAAM,YAAA;AAAA,UACN,WAAW,IAAA,GAAO;AAAA,SACpB,CAAE,CAAA;AAEF,QAAA,OAAO;AAAA,UACL,QAAQC,yBAAA,CAAuB;AAAA,YAC7B,MAAA,EAAQ;AAAA,cACN,GAAG,MAAA;AAAA,cACH;AAAA,gBACE,IAAA,EAAM,QAAA;AAAA,gBACN,YAAA,EAAc,MAAA;AAAA,gBACd,QAAA,EAAU,MAAA;AAAA,gBACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,WACD,CAAA;AAAA,UACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,SAC9C;AAAA,MACF;AAAA,KACD,CAAA;AAED,IAAA,OAAOF,UAAAA;AAAA,EACT;AAGA,EAAA,MAAM,SAAA,GAAY,IAAIG,0BAAA,CAAoB;AAAA,IACxC,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,MAAA,IAAI,WAAA,EAAa;AACf,QAAA,WAAA,CAAY,KAAK,CAAA;AAAA,MACnB;AAEA,MAAA,OAAO;AAAA,QACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,QAC5C,YAAA,EAAc,MAAA;AAAA,QACd,OAAO,EAAE,WAAA,EAAa,IAAI,YAAA,EAAc,EAAA,EAAI,aAAa,EAAA,EAAG;AAAA,QAC5D,OAAA,EAAS;AAAA,UACP;AAAA,YACE,IAAA,EAAM,MAAA;AAAA,YACN,IAAA,EAAM;AAAA;AACR,SACF;AAAA,QACA,UAAU;AAAC,OACb;AAAA,IACF,CAAA;AAAA,IACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,SAAA,CAAU,KAAK,CAAA;AAAA,MACjB;AAEA,MAAA,OAAO;AAAA,QACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,QAC5C,UAAU,EAAC;AAAA,QACX,QAAQC,mCAAA,CAA6B;AAAA,UACnC,EAAE,IAAA,EAAM,cAAA,EAAgB,QAAA,EAAU,EAAC,EAAE;AAAA,UACrC,EAAE,IAAA,EAAM,mBAAA,EAAqB,EAAA,EAAI,MAAA,EAAQ,OAAA,EAAS,eAAA,EAAiB,SAAA,kBAAW,IAAI,IAAA,CAAK,CAAC,CAAA,EAAE;AAAA,UAC1F,EAAE,IAAA,EAAM,YAAA,EAAc,EAAA,EAAI,GAAA,EAAI;AAAA,UAC9B,EAAE,IAAA,EAAM,YAAA,EAAc,EAAA,EAAI,GAAA,EAAK,OAAO,SAAA,EAAU;AAAA,UAChD,EAAE,IAAA,EAAM,UAAA,EAAY,EAAA,EAAI,GAAA,EAAI;AAAA,UAC5B;AAAA,YACE,IAAA,EAAM,QAAA;AAAA,YACN,YAAA,EAAc,MAAA;AAAA,YACd,OAAO,EAAE,WAAA,EAAa,IAAI,YAAA,EAAc,EAAA,EAAI,aAAa,EAAA;AAAG;AAC9D,SACD;AAAA,OACH;AAAA,IACF;AAAA,GACD,CAAA;AAED,EAAA,OAAO,SAAA;AACT;AAEO,IAAM,YAAA,GAAN,cAA2BC,6BAAA,CAAY;AAAA,EAC5C,WAAA,CAAY;AAAA,IACV,WAAA;AAAA,IACA,SAAA;AAAA,IACA,oBAAA;AAAA,IACA,QAAA,GAAW;AAAA,GACb,EAKG;AACD,IAAA,MAAM,SAAA,GAAY,IAAIJ,wBAAA,CAAoB;AAAA,MACxC,2BAAA,EAA6B,oBAAA;AAAA,MAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,QAAA,IAAI,WAAA,EAAa;AACf,UAAA,WAAA,CAAY,KAAK,CAAA;AAAA,QACnB;AAEA,QAAA,IAAI,yBAAyB,MAAA,EAAQ;AACnC,UAAA,OAAO;AAAA,YACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,YAC5C,YAAA,EAAc,MAAA;AAAA,YACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,YAChD,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,QAAQ;AAAA,WAC/B;AAAA,QACF;AAEA,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,MAAM,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ;AAAA,SACzE;AAAA,MACF,CAAA;AAAA,MACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,SAAA,CAAU,KAAK,CAAA;AAAA,QACjB;AAEA,QAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAE9E,QAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,UAC1C,IAAA,EAAM,YAAA;AAAA,UACN,WAAW,IAAA,GAAO;AAAA,SACpB,CAAE,CAAA;AAEF,QAAA,OAAO;AAAA,UACL,QAAQC,yBAAA,CAAuB;AAAA,YAC7B,MAAA,EAAQ;AAAA,cACN,GAAG,MAAA;AAAA,cACH;AAAA,gBACE,IAAA,EAAM,QAAA;AAAA,gBACN,YAAA,EAAc,MAAA;AAAA,gBACd,QAAA,EAAU,MAAA;AAAA,gBACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,WACD,CAAA;AAAA,UACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,SAC9C;AAAA,MACF;AAAA,KACD,CAAA;AAED,IAAA,KAAA,CAAM,EAAE,KAAA,EAAO,SAAA,EAAW,CAAA;AAAA,EAC5B;AAAA;AAAA,EAGA,UAAU,IAAA,EAAqD;AAE7D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,MAAA,CAAO,GAAG,IAAI,CAAA;AAEnC,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AAAA;AAAA,EAGA,kBAAkB,IAAA,EAA4C;AAE5D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,cAAA,CAAe,GAAG,IAAI,CAAA;AAE3C,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AACF","file":"llm-mock.cjs","sourcesContent":["import { simulateReadableStream } from 'ai';\nimport { MockLanguageModelV1 } from 'ai/test';\nimport { convertArrayToReadableStream, MockLanguageModelV2 } from 'ai-v5/test';\n\nimport { MastraLLMV1 } from '../llm/model/model';\n\nexport function createMockModel({\n  objectGenerationMode,\n  mockText,\n  spyGenerate,\n  spyStream,\n  version = 'v2',\n}: {\n  objectGenerationMode?: 'json';\n  mockText: string | Record<string, any>;\n  spyGenerate?: (props: any) => void;\n  spyStream?: (props: any) => void;\n  version?: 'v1' | 'v2';\n}) {\n  const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n  const finalText = objectGenerationMode === 'json' ? JSON.stringify(mockText) : text;\n\n  if (version === 'v1') {\n    // Return a v1 model\n    const mockModel = new MockLanguageModelV1({\n      defaultObjectGenerationMode: objectGenerationMode,\n      doGenerate: async props => {\n        if (spyGenerate) {\n          spyGenerate(props);\n        }\n\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: finalText,\n        };\n      },\n      doStream: async props => {\n        if (spyStream) {\n          spyStream(props);\n        }\n\n        // Split the mock text into chunks for streaming\n        const chunks = finalText.split(' ').map(word => ({\n          type: 'text-delta' as const,\n          textDelta: word + ' ',\n        }));\n\n        return {\n          stream: simulateReadableStream({\n            chunks: [\n              ...chunks,\n              {\n                type: 'finish',\n                finishReason: 'stop',\n                logprobs: undefined,\n                usage: { completionTokens: 10, promptTokens: 3 },\n              },\n            ],\n          }),\n          rawCall: { rawPrompt: null, rawSettings: {} },\n        };\n      },\n    });\n\n    return mockModel;\n  }\n\n  // Return a v2 model (default)\n  const mockModel = new MockLanguageModelV2({\n    doGenerate: async props => {\n      if (spyGenerate) {\n        spyGenerate(props);\n      }\n\n      return {\n        rawCall: { rawPrompt: null, rawSettings: {} },\n        finishReason: 'stop',\n        usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },\n        content: [\n          {\n            type: 'text',\n            text: finalText,\n          },\n        ],\n        warnings: [],\n      };\n    },\n    doStream: async props => {\n      if (spyStream) {\n        spyStream(props);\n      }\n\n      return {\n        rawCall: { rawPrompt: null, rawSettings: {} },\n        warnings: [],\n        stream: convertArrayToReadableStream([\n          { type: 'stream-start', warnings: [] },\n          { type: 'response-metadata', id: 'id-0', modelId: 'mock-model-id', timestamp: new Date(0) },\n          { type: 'text-start', id: '1' },\n          { type: 'text-delta', id: '1', delta: finalText },\n          { type: 'text-end', id: '1' },\n          {\n            type: 'finish',\n            finishReason: 'stop',\n            usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },\n          },\n        ]),\n      };\n    },\n  });\n\n  return mockModel;\n}\n\nexport class MockProvider extends MastraLLMV1 {\n  constructor({\n    spyGenerate,\n    spyStream,\n    objectGenerationMode,\n    mockText = 'Hello, world!',\n  }: {\n    spyGenerate?: (props: any) => void;\n    spyStream?: (props: any) => void;\n    objectGenerationMode?: 'json';\n    mockText?: string | Record<string, any>;\n  }) {\n    const mockModel = new MockLanguageModelV1({\n      defaultObjectGenerationMode: objectGenerationMode,\n      doGenerate: async props => {\n        if (spyGenerate) {\n          spyGenerate(props);\n        }\n\n        if (objectGenerationMode === 'json') {\n          return {\n            rawCall: { rawPrompt: null, rawSettings: {} },\n            finishReason: 'stop',\n            usage: { promptTokens: 10, completionTokens: 20 },\n            text: JSON.stringify(mockText),\n          };\n        }\n\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: typeof mockText === 'string' ? mockText : JSON.stringify(mockText),\n        };\n      },\n      doStream: async props => {\n        if (spyStream) {\n          spyStream(props);\n        }\n\n        const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n        // Split the mock text into chunks for streaming\n        const chunks = text.split(' ').map(word => ({\n          type: 'text-delta' as const,\n          textDelta: word + ' ',\n        }));\n\n        return {\n          stream: simulateReadableStream({\n            chunks: [\n              ...chunks,\n              {\n                type: 'finish',\n                finishReason: 'stop',\n                logprobs: undefined,\n                usage: { completionTokens: 10, promptTokens: 3 },\n              },\n            ],\n          }),\n          rawCall: { rawPrompt: null, rawSettings: {} },\n        };\n      },\n    });\n\n    super({ model: mockModel });\n  }\n\n  // @ts-ignore\n  stream(...args: any): PromiseLike<StreamReturn<any, any, any>> {\n    // @ts-ignore\n    const result = super.stream(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n\n  // @ts-ignore\n  __streamObject(...args): PromiseLike<StreamObjectResult<any>> {\n    // @ts-ignore\n    const result = super.__streamObject(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n}\n"]}

package/dist/test-utils/llm-mock.d.ts

@@ -1,11 +1,13 @@
 import { MockLanguageModelV1 } from 'ai/test';
+import { MockLanguageModelV2 } from 'ai-v5/test';
 import { MastraLLMV1 } from '../llm/model/model.js';
-export declare function createMockModel({ objectGenerationMode, mockText, spyGenerate, spyStream, }: {
+export declare function createMockModel({ objectGenerationMode, mockText, spyGenerate, spyStream, version, }: {
     objectGenerationMode?: 'json';
     mockText: string | Record<string, any>;
     spyGenerate?: (props: any) => void;
     spyStream?: (props: any) => void;
-}): MockLanguageModelV1;
+    version?: 'v1' | 'v2';
+}): MockLanguageModelV1 | MockLanguageModelV2;
 export declare class MockProvider extends MastraLLMV1 {
     constructor({ spyGenerate, spyStream, objectGenerationMode, mockText, }: {
         spyGenerate?: (props: any) => void;

package/dist/test-utils/llm-mock.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm-mock.d.ts","sourceRoot":"","sources":["../../src/test-utils/llm-mock.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAE9C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,wBAAgB,eAAe,CAAC,EAC9B,oBAAoB,EACpB,QAAQ,EACR,WAAW,EACX,SAAS,GACV,EAAE;IACD,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACvC,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IACnC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;CAClC,uBAsDA;AAED,qBAAa,YAAa,SAAQ,WAAW;gBAC/B,EACV,WAAW,EACX,SAAS,EACT,oBAAoB,EACpB,QAA0B,GAC3B,EAAE;QACD,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;QACnC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;QACjC,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACzC;IAyDD,MAAM,CAAC,GAAG,IAAI,EAAE,GAAG,GAAG,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;IAe9D,cAAc,CAAC,GAAG,IAAI,OAAA,GAAG,WAAW,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;CAa9D"}
+{"version":3,"file":"llm-mock.d.ts","sourceRoot":"","sources":["../../src/test-utils/llm-mock.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAgC,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAE/E,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,wBAAgB,eAAe,CAAC,EAC9B,oBAAoB,EACpB,QAAQ,EACR,WAAW,EACX,SAAS,EACT,OAAc,GACf,EAAE;IACD,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACvC,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IACnC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IACjC,OAAO,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;CACvB,6CAgGA;AAED,qBAAa,YAAa,SAAQ,WAAW;gBAC/B,EACV,WAAW,EACX,SAAS,EACT,oBAAoB,EACpB,QAA0B,GAC3B,EAAE;QACD,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;QACnC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;QACjC,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACzC;IAyDD,MAAM,CAAC,GAAG,IAAI,EAAE,GAAG,GAAG,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;IAe9D,cAAc,CAAC,GAAG,IAAI,OAAA,GAAG,WAAW,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;CAa9D"}

package/dist/test-utils/llm-mock.js

@@ -1,56 +1,94 @@
 import { MastraLLMV1 } from '../chunk-OLERJ2OU.js';
 import { simulateReadableStream } from 'ai';
 import { MockLanguageModelV1 } from 'ai/test';
+import { MockLanguageModelV2, convertArrayToReadableStream } from 'ai-v5/test';
 
 function createMockModel({
   objectGenerationMode,
   mockText,
   spyGenerate,
-  spyStream
+  spyStream,
+  version = "v2"
 }) {
-  const mockModel = new MockLanguageModelV1({
-    defaultObjectGenerationMode: objectGenerationMode,
-    doGenerate: async (props) => {
-      if (spyGenerate) {
-        spyGenerate(props);
-      }
-      if (objectGenerationMode === "json") {
+  const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
+  const finalText = objectGenerationMode === "json" ? JSON.stringify(mockText) : text;
+  if (version === "v1") {
+    const mockModel2 = new MockLanguageModelV1({
+      defaultObjectGenerationMode: objectGenerationMode,
+      doGenerate: async (props) => {
+        if (spyGenerate) {
+          spyGenerate(props);
+        }
         return {
           rawCall: { rawPrompt: null, rawSettings: {} },
           finishReason: "stop",
           usage: { promptTokens: 10, completionTokens: 20 },
-          text: JSON.stringify(mockText)
+          text: finalText
+        };
+      },
+      doStream: async (props) => {
+        if (spyStream) {
+          spyStream(props);
+        }
+        const chunks = finalText.split(" ").map((word) => ({
+          type: "text-delta",
+          textDelta: word + " "
+        }));
+        return {
+          stream: simulateReadableStream({
+            chunks: [
+              ...chunks,
+              {
+                type: "finish",
+                finishReason: "stop",
+                logprobs: void 0,
+                usage: { completionTokens: 10, promptTokens: 3 }
+              }
+            ]
+          }),
+          rawCall: { rawPrompt: null, rawSettings: {} }
         };
       }
+    });
+    return mockModel2;
+  }
+  const mockModel = new MockLanguageModelV2({
+    doGenerate: async (props) => {
+      if (spyGenerate) {
+        spyGenerate(props);
+      }
       return {
         rawCall: { rawPrompt: null, rawSettings: {} },
         finishReason: "stop",
-        usage: { promptTokens: 10, completionTokens: 20 },
-        text: typeof mockText === "string" ? mockText : JSON.stringify(mockText)
+        usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+        content: [
+          {
+            type: "text",
+            text: finalText
+          }
+        ],
+        warnings: []
       };
     },
     doStream: async (props) => {
       if (spyStream) {
         spyStream(props);
       }
-      const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
-      const chunks = text.split(" ").map((word) => ({
-        type: "text-delta",
-        textDelta: word + " "
-      }));
       return {
-        stream: simulateReadableStream({
-          chunks: [
-            ...chunks,
-            {
-              type: "finish",
-              finishReason: "stop",
-              logprobs: void 0,
-              usage: { completionTokens: 10, promptTokens: 3 }
-            }
-          ]
-        }),
-        rawCall: { rawPrompt: null, rawSettings: {} }
+        rawCall: { rawPrompt: null, rawSettings: {} },
+        warnings: [],
+        stream: convertArrayToReadableStream([
+          { type: "stream-start", warnings: [] },
+          { type: "response-metadata", id: "id-0", modelId: "mock-model-id", timestamp: /* @__PURE__ */ new Date(0) },
+          { type: "text-start", id: "1" },
+          { type: "text-delta", id: "1", delta: finalText },
+          { type: "text-end", id: "1" },
+          {
+            type: "finish",
+            finishReason: "stop",
+            usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 }
+          }
+        ])
       };
     }
   });

package/dist/test-utils/llm-mock.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/test-utils/llm-mock.ts"],"names":[],"mappings":";;;;AAKO,SAAS,eAAA,CAAgB;AAAA,EAC9B,oBAAA;AAAA,EACA,QAAA;AAAA,EACA,WAAA;AAAA,EACA;AACF,CAAA,EAKG;AACD,EAAA,MAAM,SAAA,GAAY,IAAI,mBAAA,CAAoB;AAAA,IACxC,2BAAA,EAA6B,oBAAA;AAAA,IAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,MAAA,IAAI,WAAA,EAAa;AACf,QAAA,WAAA,CAAY,KAAK,CAAA;AAAA,MACnB;AAEA,MAAA,IAAI,yBAAyB,MAAA,EAAQ;AACnC,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,QAAQ;AAAA,SAC/B;AAAA,MACF;AAEA,MAAA,OAAO;AAAA,QACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,QAC5C,YAAA,EAAc,MAAA;AAAA,QACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,QAChD,MAAM,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ;AAAA,OACzE;AAAA,IACF,CAAA;AAAA,IACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,SAAA,CAAU,KAAK,CAAA;AAAA,MACjB;AAEA,MAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAE9E,MAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,QAC1C,IAAA,EAAM,YAAA;AAAA,QACN,WAAW,IAAA,GAAO;AAAA,OACpB,CAAE,CAAA;AAEF,MAAA,OAAO;AAAA,QACL,QAAQ,sBAAA,CAAuB;AAAA,UAC7B,MAAA,EAAQ;AAAA,YACN,GAAG,MAAA;AAAA,YACH;AAAA,cACE,IAAA,EAAM,QAAA;AAAA,cACN,YAAA,EAAc,MAAA;AAAA,cACd,QAAA,EAAU,MAAA;AAAA,cACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,SACD,CAAA;AAAA,QACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,OAC9C;AAAA,IACF;AAAA,GACD,CAAA;AAED,EAAA,OAAO,SAAA;AACT;AAEO,IAAM,YAAA,GAAN,cAA2B,WAAA,CAAY;AAAA,EAC5C,WAAA,CAAY;AAAA,IACV,WAAA;AAAA,IACA,SAAA;AAAA,IACA,oBAAA;AAAA,IACA,QAAA,GAAW;AAAA,GACb,EAKG;AACD,IAAA,MAAM,SAAA,GAAY,IAAI,mBAAA,CAAoB;AAAA,MACxC,2BAAA,EAA6B,oBAAA;AAAA,MAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,QAAA,IAAI,WAAA,EAAa;AACf,UAAA,WAAA,CAAY,KAAK,CAAA;AAAA,QACnB;AAEA,QAAA,IAAI,yBAAyB,MAAA,EAAQ;AACnC,UAAA,OAAO;AAAA,YACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,YAC5C,YAAA,EAAc,MAAA;AAAA,YACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,YAChD,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,QAAQ;AAAA,WAC/B;AAAA,QACF;AAEA,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,MAAM,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ;AAAA,SACzE;AAAA,MACF,CAAA;AAAA,MACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,SAAA,CAAU,KAAK,CAAA;AAAA,QACjB;AAEA,QAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAE9E,QAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,UAC1C,IAAA,EAAM,YAAA;AAAA,UACN,WAAW,IAAA,GAAO;AAAA,SACpB,CAAE,CAAA;AAEF,QAAA,OAAO;AAAA,UACL,QAAQ,sBAAA,CAAuB;AAAA,YAC7B,MAAA,EAAQ;AAAA,cACN,GAAG,MAAA;AAAA,cACH;AAAA,gBACE,IAAA,EAAM,QAAA;AAAA,gBACN,YAAA,EAAc,MAAA;AAAA,gBACd,QAAA,EAAU,MAAA;AAAA,gBACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,WACD,CAAA;AAAA,UACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,SAC9C;AAAA,MACF;AAAA,KACD,CAAA;AAED,IAAA,KAAA,CAAM,EAAE,KAAA,EAAO,SAAA,EAAW,CAAA;AAAA,EAC5B;AAAA;AAAA,EAGA,UAAU,IAAA,EAAqD;AAE7D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,MAAA,CAAO,GAAG,IAAI,CAAA;AAEnC,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AAAA;AAAA,EAGA,kBAAkB,IAAA,EAA4C;AAE5D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,cAAA,CAAe,GAAG,IAAI,CAAA;AAE3C,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AACF","file":"llm-mock.js","sourcesContent":["import { simulateReadableStream } from 'ai';\nimport { MockLanguageModelV1 } from 'ai/test';\n\nimport { MastraLLMV1 } from '../llm/model/model';\n\nexport function createMockModel({\n  objectGenerationMode,\n  mockText,\n  spyGenerate,\n  spyStream,\n}: {\n  objectGenerationMode?: 'json';\n  mockText: string | Record<string, any>;\n  spyGenerate?: (props: any) => void;\n  spyStream?: (props: any) => void;\n}) {\n  const mockModel = new MockLanguageModelV1({\n    defaultObjectGenerationMode: objectGenerationMode,\n    doGenerate: async props => {\n      if (spyGenerate) {\n        spyGenerate(props);\n      }\n\n      if (objectGenerationMode === 'json') {\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: JSON.stringify(mockText),\n        };\n      }\n\n      return {\n        rawCall: { rawPrompt: null, rawSettings: {} },\n        finishReason: 'stop',\n        usage: { promptTokens: 10, completionTokens: 20 },\n        text: typeof mockText === 'string' ? mockText : JSON.stringify(mockText),\n      };\n    },\n    doStream: async props => {\n      if (spyStream) {\n        spyStream(props);\n      }\n\n      const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n      // Split the mock text into chunks for streaming\n      const chunks = text.split(' ').map(word => ({\n        type: 'text-delta' as const,\n        textDelta: word + ' ',\n      }));\n\n      return {\n        stream: simulateReadableStream({\n          chunks: [\n            ...chunks,\n            {\n              type: 'finish',\n              finishReason: 'stop',\n              logprobs: undefined,\n              usage: { completionTokens: 10, promptTokens: 3 },\n            },\n          ],\n        }),\n        rawCall: { rawPrompt: null, rawSettings: {} },\n      };\n    },\n  });\n\n  return mockModel;\n}\n\nexport class MockProvider extends MastraLLMV1 {\n  constructor({\n    spyGenerate,\n    spyStream,\n    objectGenerationMode,\n    mockText = 'Hello, world!',\n  }: {\n    spyGenerate?: (props: any) => void;\n    spyStream?: (props: any) => void;\n    objectGenerationMode?: 'json';\n    mockText?: string | Record<string, any>;\n  }) {\n    const mockModel = new MockLanguageModelV1({\n      defaultObjectGenerationMode: objectGenerationMode,\n      doGenerate: async props => {\n        if (spyGenerate) {\n          spyGenerate(props);\n        }\n\n        if (objectGenerationMode === 'json') {\n          return {\n            rawCall: { rawPrompt: null, rawSettings: {} },\n            finishReason: 'stop',\n            usage: { promptTokens: 10, completionTokens: 20 },\n            text: JSON.stringify(mockText),\n          };\n        }\n\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: typeof mockText === 'string' ? mockText : JSON.stringify(mockText),\n        };\n      },\n      doStream: async props => {\n        if (spyStream) {\n          spyStream(props);\n        }\n\n        const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n        // Split the mock text into chunks for streaming\n        const chunks = text.split(' ').map(word => ({\n          type: 'text-delta' as const,\n          textDelta: word + ' ',\n        }));\n\n        return {\n          stream: simulateReadableStream({\n            chunks: [\n              ...chunks,\n              {\n                type: 'finish',\n                finishReason: 'stop',\n                logprobs: undefined,\n                usage: { completionTokens: 10, promptTokens: 3 },\n              },\n            ],\n          }),\n          rawCall: { rawPrompt: null, rawSettings: {} },\n        };\n      },\n    });\n\n    super({ model: mockModel });\n  }\n\n  // @ts-ignore\n  stream(...args: any): PromiseLike<StreamReturn<any, any, any>> {\n    // @ts-ignore\n    const result = super.stream(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n\n  // @ts-ignore\n  __streamObject(...args): PromiseLike<StreamObjectResult<any>> {\n    // @ts-ignore\n    const result = super.__streamObject(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/test-utils/llm-mock.ts"],"names":["mockModel"],"mappings":";;;;;AAMO,SAAS,eAAA,CAAgB;AAAA,EAC9B,oBAAA;AAAA,EACA,QAAA;AAAA,EACA,WAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA,GAAU;AACZ,CAAA,EAMG;AACD,EAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAC9E,EAAA,MAAM,YAAY,oBAAA,KAAyB,MAAA,GAAS,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAA,GAAI,IAAA;AAE/E,EAAA,IAAI,YAAY,IAAA,EAAM;AAEpB,IAAA,MAAMA,UAAAA,GAAY,IAAI,mBAAA,CAAoB;AAAA,MACxC,2BAAA,EAA6B,oBAAA;AAAA,MAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,QAAA,IAAI,WAAA,EAAa;AACf,UAAA,WAAA,CAAY,KAAK,CAAA;AAAA,QACnB;AAEA,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,IAAA,EAAM;AAAA,SACR;AAAA,MACF,CAAA;AAAA,MACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,SAAA,CAAU,KAAK,CAAA;AAAA,QACjB;AAGA,QAAA,MAAM,SAAS,SAAA,CAAU,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,UAC/C,IAAA,EAAM,YAAA;AAAA,UACN,WAAW,IAAA,GAAO;AAAA,SACpB,CAAE,CAAA;AAEF,QAAA,OAAO;AAAA,UACL,QAAQ,sBAAA,CAAuB;AAAA,YAC7B,MAAA,EAAQ;AAAA,cACN,GAAG,MAAA;AAAA,cACH;AAAA,gBACE,IAAA,EAAM,QAAA;AAAA,gBACN,YAAA,EAAc,MAAA;AAAA,gBACd,QAAA,EAAU,MAAA;AAAA,gBACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,WACD,CAAA;AAAA,UACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,SAC9C;AAAA,MACF;AAAA,KACD,CAAA;AAED,IAAA,OAAOA,UAAAA;AAAA,EACT;AAGA,EAAA,MAAM,SAAA,GAAY,IAAI,mBAAA,CAAoB;AAAA,IACxC,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,MAAA,IAAI,WAAA,EAAa;AACf,QAAA,WAAA,CAAY,KAAK,CAAA;AAAA,MACnB;AAEA,MAAA,OAAO;AAAA,QACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,QAC5C,YAAA,EAAc,MAAA;AAAA,QACd,OAAO,EAAE,WAAA,EAAa,IAAI,YAAA,EAAc,EAAA,EAAI,aAAa,EAAA,EAAG;AAAA,QAC5D,OAAA,EAAS;AAAA,UACP;AAAA,YACE,IAAA,EAAM,MAAA;AAAA,YACN,IAAA,EAAM;AAAA;AACR,SACF;AAAA,QACA,UAAU;AAAC,OACb;AAAA,IACF,CAAA;AAAA,IACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,SAAA,CAAU,KAAK,CAAA;AAAA,MACjB;AAEA,MAAA,OAAO;AAAA,QACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,QAC5C,UAAU,EAAC;AAAA,QACX,QAAQ,4BAAA,CAA6B;AAAA,UACnC,EAAE,IAAA,EAAM,cAAA,EAAgB,QAAA,EAAU,EAAC,EAAE;AAAA,UACrC,EAAE,IAAA,EAAM,mBAAA,EAAqB,EAAA,EAAI,MAAA,EAAQ,OAAA,EAAS,eAAA,EAAiB,SAAA,kBAAW,IAAI,IAAA,CAAK,CAAC,CAAA,EAAE;AAAA,UAC1F,EAAE,IAAA,EAAM,YAAA,EAAc,EAAA,EAAI,GAAA,EAAI;AAAA,UAC9B,EAAE,IAAA,EAAM,YAAA,EAAc,EAAA,EAAI,GAAA,EAAK,OAAO,SAAA,EAAU;AAAA,UAChD,EAAE,IAAA,EAAM,UAAA,EAAY,EAAA,EAAI,GAAA,EAAI;AAAA,UAC5B;AAAA,YACE,IAAA,EAAM,QAAA;AAAA,YACN,YAAA,EAAc,MAAA;AAAA,YACd,OAAO,EAAE,WAAA,EAAa,IAAI,YAAA,EAAc,EAAA,EAAI,aAAa,EAAA;AAAG;AAC9D,SACD;AAAA,OACH;AAAA,IACF;AAAA,GACD,CAAA;AAED,EAAA,OAAO,SAAA;AACT;AAEO,IAAM,YAAA,GAAN,cAA2B,WAAA,CAAY;AAAA,EAC5C,WAAA,CAAY;AAAA,IACV,WAAA;AAAA,IACA,SAAA;AAAA,IACA,oBAAA;AAAA,IACA,QAAA,GAAW;AAAA,GACb,EAKG;AACD,IAAA,MAAM,SAAA,GAAY,IAAI,mBAAA,CAAoB;AAAA,MACxC,2BAAA,EAA6B,oBAAA;AAAA,MAC7B,UAAA,EAAY,OAAM,KAAA,KAAS;AACzB,QAAA,IAAI,WAAA,EAAa;AACf,UAAA,WAAA,CAAY,KAAK,CAAA;AAAA,QACnB;AAEA,QAAA,IAAI,yBAAyB,MAAA,EAAQ;AACnC,UAAA,OAAO;AAAA,YACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,YAC5C,YAAA,EAAc,MAAA;AAAA,YACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,YAChD,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,QAAQ;AAAA,WAC/B;AAAA,QACF;AAEA,QAAA,OAAO;AAAA,UACL,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC,EAAE;AAAA,UAC5C,YAAA,EAAc,MAAA;AAAA,UACd,KAAA,EAAO,EAAE,YAAA,EAAc,EAAA,EAAI,kBAAkB,EAAA,EAAG;AAAA,UAChD,MAAM,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ;AAAA,SACzE;AAAA,MACF,CAAA;AAAA,MACA,QAAA,EAAU,OAAM,KAAA,KAAS;AACvB,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,SAAA,CAAU,KAAK,CAAA;AAAA,QACjB;AAEA,QAAA,MAAM,OAAO,OAAO,QAAA,KAAa,WAAW,QAAA,GAAW,IAAA,CAAK,UAAU,QAAQ,CAAA;AAE9E,QAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,UAC1C,IAAA,EAAM,YAAA;AAAA,UACN,WAAW,IAAA,GAAO;AAAA,SACpB,CAAE,CAAA;AAEF,QAAA,OAAO;AAAA,UACL,QAAQ,sBAAA,CAAuB;AAAA,YAC7B,MAAA,EAAQ;AAAA,cACN,GAAG,MAAA;AAAA,cACH;AAAA,gBACE,IAAA,EAAM,QAAA;AAAA,gBACN,YAAA,EAAc,MAAA;AAAA,gBACd,QAAA,EAAU,MAAA;AAAA,gBACV,KAAA,EAAO,EAAE,gBAAA,EAAkB,EAAA,EAAI,cAAc,CAAA;AAAE;AACjD;AACF,WACD,CAAA;AAAA,UACD,SAAS,EAAE,SAAA,EAAW,IAAA,EAAM,WAAA,EAAa,EAAC;AAAE,SAC9C;AAAA,MACF;AAAA,KACD,CAAA;AAED,IAAA,KAAA,CAAM,EAAE,KAAA,EAAO,SAAA,EAAW,CAAA;AAAA,EAC5B;AAAA;AAAA,EAGA,UAAU,IAAA,EAAqD;AAE7D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,MAAA,CAAO,GAAG,IAAI,CAAA;AAEnC,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AAAA;AAAA,EAGA,kBAAkB,IAAA,EAA4C;AAE5D,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,cAAA,CAAe,GAAG,IAAI,CAAA;AAE3C,IAAA,OAAO;AAAA,MACL,GAAG,MAAA;AAAA;AAAA,MAEH,IAAA,EAAM,CAAC,WAAA,EAAa,UAAA,KAAe;AAEjC,QAAA,OAAO,MAAA,CAAO,WAAW,MAAA,CAAO,IAAI,gBAAgB,CAAA,CAAE,IAAA,CAAK,WAAA,EAAa,UAAU,CAAA;AAAA,MACpF;AAAA,KACF;AAAA,EACF;AACF","file":"llm-mock.js","sourcesContent":["import { simulateReadableStream } from 'ai';\nimport { MockLanguageModelV1 } from 'ai/test';\nimport { convertArrayToReadableStream, MockLanguageModelV2 } from 'ai-v5/test';\n\nimport { MastraLLMV1 } from '../llm/model/model';\n\nexport function createMockModel({\n  objectGenerationMode,\n  mockText,\n  spyGenerate,\n  spyStream,\n  version = 'v2',\n}: {\n  objectGenerationMode?: 'json';\n  mockText: string | Record<string, any>;\n  spyGenerate?: (props: any) => void;\n  spyStream?: (props: any) => void;\n  version?: 'v1' | 'v2';\n}) {\n  const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n  const finalText = objectGenerationMode === 'json' ? JSON.stringify(mockText) : text;\n\n  if (version === 'v1') {\n    // Return a v1 model\n    const mockModel = new MockLanguageModelV1({\n      defaultObjectGenerationMode: objectGenerationMode,\n      doGenerate: async props => {\n        if (spyGenerate) {\n          spyGenerate(props);\n        }\n\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: finalText,\n        };\n      },\n      doStream: async props => {\n        if (spyStream) {\n          spyStream(props);\n        }\n\n        // Split the mock text into chunks for streaming\n        const chunks = finalText.split(' ').map(word => ({\n          type: 'text-delta' as const,\n          textDelta: word + ' ',\n        }));\n\n        return {\n          stream: simulateReadableStream({\n            chunks: [\n              ...chunks,\n              {\n                type: 'finish',\n                finishReason: 'stop',\n                logprobs: undefined,\n                usage: { completionTokens: 10, promptTokens: 3 },\n              },\n            ],\n          }),\n          rawCall: { rawPrompt: null, rawSettings: {} },\n        };\n      },\n    });\n\n    return mockModel;\n  }\n\n  // Return a v2 model (default)\n  const mockModel = new MockLanguageModelV2({\n    doGenerate: async props => {\n      if (spyGenerate) {\n        spyGenerate(props);\n      }\n\n      return {\n        rawCall: { rawPrompt: null, rawSettings: {} },\n        finishReason: 'stop',\n        usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },\n        content: [\n          {\n            type: 'text',\n            text: finalText,\n          },\n        ],\n        warnings: [],\n      };\n    },\n    doStream: async props => {\n      if (spyStream) {\n        spyStream(props);\n      }\n\n      return {\n        rawCall: { rawPrompt: null, rawSettings: {} },\n        warnings: [],\n        stream: convertArrayToReadableStream([\n          { type: 'stream-start', warnings: [] },\n          { type: 'response-metadata', id: 'id-0', modelId: 'mock-model-id', timestamp: new Date(0) },\n          { type: 'text-start', id: '1' },\n          { type: 'text-delta', id: '1', delta: finalText },\n          { type: 'text-end', id: '1' },\n          {\n            type: 'finish',\n            finishReason: 'stop',\n            usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },\n          },\n        ]),\n      };\n    },\n  });\n\n  return mockModel;\n}\n\nexport class MockProvider extends MastraLLMV1 {\n  constructor({\n    spyGenerate,\n    spyStream,\n    objectGenerationMode,\n    mockText = 'Hello, world!',\n  }: {\n    spyGenerate?: (props: any) => void;\n    spyStream?: (props: any) => void;\n    objectGenerationMode?: 'json';\n    mockText?: string | Record<string, any>;\n  }) {\n    const mockModel = new MockLanguageModelV1({\n      defaultObjectGenerationMode: objectGenerationMode,\n      doGenerate: async props => {\n        if (spyGenerate) {\n          spyGenerate(props);\n        }\n\n        if (objectGenerationMode === 'json') {\n          return {\n            rawCall: { rawPrompt: null, rawSettings: {} },\n            finishReason: 'stop',\n            usage: { promptTokens: 10, completionTokens: 20 },\n            text: JSON.stringify(mockText),\n          };\n        }\n\n        return {\n          rawCall: { rawPrompt: null, rawSettings: {} },\n          finishReason: 'stop',\n          usage: { promptTokens: 10, completionTokens: 20 },\n          text: typeof mockText === 'string' ? mockText : JSON.stringify(mockText),\n        };\n      },\n      doStream: async props => {\n        if (spyStream) {\n          spyStream(props);\n        }\n\n        const text = typeof mockText === 'string' ? mockText : JSON.stringify(mockText);\n        // Split the mock text into chunks for streaming\n        const chunks = text.split(' ').map(word => ({\n          type: 'text-delta' as const,\n          textDelta: word + ' ',\n        }));\n\n        return {\n          stream: simulateReadableStream({\n            chunks: [\n              ...chunks,\n              {\n                type: 'finish',\n                finishReason: 'stop',\n                logprobs: undefined,\n                usage: { completionTokens: 10, promptTokens: 3 },\n              },\n            ],\n          }),\n          rawCall: { rawPrompt: null, rawSettings: {} },\n        };\n      },\n    });\n\n    super({ model: mockModel });\n  }\n\n  // @ts-ignore\n  stream(...args: any): PromiseLike<StreamReturn<any, any, any>> {\n    // @ts-ignore\n    const result = super.stream(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n\n  // @ts-ignore\n  __streamObject(...args): PromiseLike<StreamObjectResult<any>> {\n    // @ts-ignore\n    const result = super.__streamObject(...args);\n\n    return {\n      ...result,\n      // @ts-ignore on await read the stream\n      then: (onfulfilled, onrejected) => {\n        // @ts-ignore\n        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);\n      },\n    };\n  }\n}\n"]}

package/dist/workflows/evented/index.cjs

@@ -1,44 +1,44 @@
 'use strict';
 
-var chunkNJKNZXAS_cjs = require('../../chunk-NJKNZXAS.cjs');
+var chunkXMV6JYFI_cjs = require('../../chunk-XMV6JYFI.cjs');
 
 
 
 Object.defineProperty(exports, "EventedExecutionEngine", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.EventedExecutionEngine; }
+  get: function () { return chunkXMV6JYFI_cjs.EventedExecutionEngine; }
 });
 Object.defineProperty(exports, "EventedRun", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.EventedRun; }
+  get: function () { return chunkXMV6JYFI_cjs.EventedRun; }
 });
 Object.defineProperty(exports, "EventedWorkflow", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.EventedWorkflow; }
+  get: function () { return chunkXMV6JYFI_cjs.EventedWorkflow; }
 });
 Object.defineProperty(exports, "StepExecutor", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.StepExecutor; }
+  get: function () { return chunkXMV6JYFI_cjs.StepExecutor; }
 });
 Object.defineProperty(exports, "WorkflowEventProcessor", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.WorkflowEventProcessor; }
+  get: function () { return chunkXMV6JYFI_cjs.WorkflowEventProcessor; }
 });
 Object.defineProperty(exports, "cloneStep", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.cloneStep; }
+  get: function () { return chunkXMV6JYFI_cjs.cloneStep; }
 });
 Object.defineProperty(exports, "cloneWorkflow", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.cloneWorkflow; }
+  get: function () { return chunkXMV6JYFI_cjs.cloneWorkflow; }
 });
 Object.defineProperty(exports, "createStep", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.createStep; }
+  get: function () { return chunkXMV6JYFI_cjs.createStep; }
 });
 Object.defineProperty(exports, "createWorkflow", {
   enumerable: true,
-  get: function () { return chunkNJKNZXAS_cjs.createWorkflow; }
+  get: function () { return chunkXMV6JYFI_cjs.createWorkflow; }
 });
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/workflows/evented/index.js

@@ -1,3 +1,3 @@
-export { EventedExecutionEngine, EventedRun, EventedWorkflow, StepExecutor, WorkflowEventProcessor, cloneStep, cloneWorkflow, createStep, createWorkflow } from '../../chunk-FKV5666D.js';
+export { EventedExecutionEngine, EventedRun, EventedWorkflow, StepExecutor, WorkflowEventProcessor, cloneStep, cloneWorkflow, createStep, createWorkflow } from '../../chunk-QJSHWTRV.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/workflows/index.cjs

@@ -1,60 +1,60 @@
 'use strict';
 
-var chunkDE7YOOVP_cjs = require('../chunk-DE7YOOVP.cjs');
+var chunkNKENPWUR_cjs = require('../chunk-NKENPWUR.cjs');
 
 
 
 Object.defineProperty(exports, "DefaultExecutionEngine", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.DefaultExecutionEngine; }
+  get: function () { return chunkNKENPWUR_cjs.DefaultExecutionEngine; }
 });
 Object.defineProperty(exports, "ExecutionEngine", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.ExecutionEngine; }
+  get: function () { return chunkNKENPWUR_cjs.ExecutionEngine; }
 });
 Object.defineProperty(exports, "Run", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.Run; }
+  get: function () { return chunkNKENPWUR_cjs.Run; }
 });
 Object.defineProperty(exports, "Workflow", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.Workflow; }
+  get: function () { return chunkNKENPWUR_cjs.Workflow; }
 });
 Object.defineProperty(exports, "cloneStep", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.cloneStep; }
+  get: function () { return chunkNKENPWUR_cjs.cloneStep; }
 });
 Object.defineProperty(exports, "cloneWorkflow", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.cloneWorkflow; }
+  get: function () { return chunkNKENPWUR_cjs.cloneWorkflow; }
 });
 Object.defineProperty(exports, "createStep", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.createStep; }
+  get: function () { return chunkNKENPWUR_cjs.createStep; }
 });
 Object.defineProperty(exports, "createWorkflow", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.createWorkflow; }
+  get: function () { return chunkNKENPWUR_cjs.createWorkflow; }
 });
 Object.defineProperty(exports, "getResumeLabelsByStepId", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getResumeLabelsByStepId; }
+  get: function () { return chunkNKENPWUR_cjs.getResumeLabelsByStepId; }
 });
 Object.defineProperty(exports, "getStepResult", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getStepResult; }
+  get: function () { return chunkNKENPWUR_cjs.getStepResult; }
 });
 Object.defineProperty(exports, "getZodErrors", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getZodErrors; }
+  get: function () { return chunkNKENPWUR_cjs.getZodErrors; }
 });
 Object.defineProperty(exports, "mapVariable", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.mapVariable; }
+  get: function () { return chunkNKENPWUR_cjs.mapVariable; }
 });
 Object.defineProperty(exports, "validateStepInput", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.validateStepInput; }
+  get: function () { return chunkNKENPWUR_cjs.validateStepInput; }
 });
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/workflows/index.js

@@ -1,3 +1,3 @@
-export { DefaultExecutionEngine, ExecutionEngine, Run, Workflow, cloneStep, cloneWorkflow, createStep, createWorkflow, getResumeLabelsByStepId, getStepResult, getZodErrors, mapVariable, validateStepInput } from '../chunk-SHPLHQFH.js';
+export { DefaultExecutionEngine, ExecutionEngine, Run, Workflow, cloneStep, cloneWorkflow, createStep, createWorkflow, getResumeLabelsByStepId, getStepResult, getZodErrors, mapVariable, validateStepInput } from '../chunk-2SILT3DE.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/workflows/legacy/index.cjs

@@ -1,92 +1,92 @@
 'use strict';
 
-var chunkDE7YOOVP_cjs = require('../../chunk-DE7YOOVP.cjs');
+var chunkNKENPWUR_cjs = require('../../chunk-NKENPWUR.cjs');
 
 
 
 Object.defineProperty(exports, "LegacyStep", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.LegacyStep; }
+  get: function () { return chunkNKENPWUR_cjs.LegacyStep; }
 });
 Object.defineProperty(exports, "LegacyWorkflow", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.LegacyWorkflow; }
+  get: function () { return chunkNKENPWUR_cjs.LegacyWorkflow; }
 });
 Object.defineProperty(exports, "WhenConditionReturnValue", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.WhenConditionReturnValue; }
+  get: function () { return chunkNKENPWUR_cjs.WhenConditionReturnValue; }
 });
 Object.defineProperty(exports, "agentToStep", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.agentToStep; }
+  get: function () { return chunkNKENPWUR_cjs.agentToStep; }
 });
 Object.defineProperty(exports, "getActivePathsAndStatus", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getActivePathsAndStatus; }
+  get: function () { return chunkNKENPWUR_cjs.getActivePathsAndStatus; }
 });
 Object.defineProperty(exports, "getResultActivePaths", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getResultActivePaths; }
+  get: function () { return chunkNKENPWUR_cjs.getResultActivePaths; }
 });
 Object.defineProperty(exports, "getStepResult", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getStepResult2; }
+  get: function () { return chunkNKENPWUR_cjs.getStepResult2; }
 });
 Object.defineProperty(exports, "getSuspendedPaths", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.getSuspendedPaths; }
+  get: function () { return chunkNKENPWUR_cjs.getSuspendedPaths; }
 });
 Object.defineProperty(exports, "isAgent", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isAgent; }
+  get: function () { return chunkNKENPWUR_cjs.isAgent; }
 });
 Object.defineProperty(exports, "isConditionalKey", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isConditionalKey; }
+  get: function () { return chunkNKENPWUR_cjs.isConditionalKey; }
 });
 Object.defineProperty(exports, "isErrorEvent", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isErrorEvent; }
+  get: function () { return chunkNKENPWUR_cjs.isErrorEvent; }
 });
 Object.defineProperty(exports, "isFinalState", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isFinalState; }
+  get: function () { return chunkNKENPWUR_cjs.isFinalState; }
 });
 Object.defineProperty(exports, "isLimboState", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isLimboState; }
+  get: function () { return chunkNKENPWUR_cjs.isLimboState; }
 });
 Object.defineProperty(exports, "isTransitionEvent", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isTransitionEvent; }
+  get: function () { return chunkNKENPWUR_cjs.isTransitionEvent; }
 });
 Object.defineProperty(exports, "isVariableReference", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isVariableReference; }
+  get: function () { return chunkNKENPWUR_cjs.isVariableReference; }
 });
 Object.defineProperty(exports, "isWorkflow", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.isWorkflow; }
+  get: function () { return chunkNKENPWUR_cjs.isWorkflow; }
 });
 Object.defineProperty(exports, "mergeChildValue", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.mergeChildValue; }
+  get: function () { return chunkNKENPWUR_cjs.mergeChildValue; }
 });
 Object.defineProperty(exports, "recursivelyCheckForFinalState", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.recursivelyCheckForFinalState; }
+  get: function () { return chunkNKENPWUR_cjs.recursivelyCheckForFinalState; }
 });
 Object.defineProperty(exports, "resolveVariables", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.resolveVariables; }
+  get: function () { return chunkNKENPWUR_cjs.resolveVariables; }
 });
 Object.defineProperty(exports, "updateStepInHierarchy", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.updateStepInHierarchy; }
+  get: function () { return chunkNKENPWUR_cjs.updateStepInHierarchy; }
 });
 Object.defineProperty(exports, "workflowToStep", {
   enumerable: true,
-  get: function () { return chunkDE7YOOVP_cjs.workflowToStep; }
+  get: function () { return chunkNKENPWUR_cjs.workflowToStep; }
 });
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/workflows/legacy/index.js

@@ -1,3 +1,3 @@
-export { LegacyStep, LegacyWorkflow, WhenConditionReturnValue, agentToStep, getActivePathsAndStatus, getResultActivePaths, getStepResult2 as getStepResult, getSuspendedPaths, isAgent, isConditionalKey, isErrorEvent, isFinalState, isLimboState, isTransitionEvent, isVariableReference, isWorkflow, mergeChildValue, recursivelyCheckForFinalState, resolveVariables, updateStepInHierarchy, workflowToStep } from '../../chunk-SHPLHQFH.js';
+export { LegacyStep, LegacyWorkflow, WhenConditionReturnValue, agentToStep, getActivePathsAndStatus, getResultActivePaths, getStepResult2 as getStepResult, getSuspendedPaths, isAgent, isConditionalKey, isErrorEvent, isFinalState, isLimboState, isTransitionEvent, isVariableReference, isWorkflow, mergeChildValue, recursivelyCheckForFinalState, resolveVariables, updateStepInHierarchy, workflowToStep } from '../../chunk-2SILT3DE.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map