@scoutqa/mastra-core 0.24.6-fork.1

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

@@ -0,0 +1,181 @@
+'use strict';
+
+var chunkCYEQK4PM_cjs = require('../chunk-CYEQK4PM.cjs');
+var ai = require('ai');
+var test = require('ai/test');
+var test$1 = require('ai-v5/test');
+
+function createMockModel({
+  objectGenerationMode,
+  mockText,
+  spyGenerate,
+  spyStream,
+  version = "v2"
+}) {
+  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: 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: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+        content: [
+          {
+            type: "text",
+            text: finalText
+          }
+        ],
+        warnings: []
+      };
+    },
+    doStream: async (props) => {
+      if (spyStream) {
+        spyStream(props);
+      }
+      return {
+        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 }
+          }
+        ])
+      };
+    }
+  });
+  return mockModel;
+}
+var MockProvider = class extends chunkCYEQK4PM_cjs.MastraLLMV1 {
+  constructor({
+    spyGenerate,
+    spyStream,
+    objectGenerationMode,
+    mockText = "Hello, world!"
+  }) {
+    const mockModel = new test.MockLanguageModelV1({
+      defaultObjectGenerationMode: objectGenerationMode,
+      doGenerate: async (props) => {
+        if (spyGenerate) {
+          spyGenerate(props);
+        }
+        if (objectGenerationMode === "json") {
+          return {
+            rawCall: { rawPrompt: null, rawSettings: {} },
+            finishReason: "stop",
+            usage: { promptTokens: 10, completionTokens: 20 },
+            text: JSON.stringify(mockText)
+          };
+        }
+        return {
+          rawCall: { rawPrompt: null, rawSettings: {} },
+          finishReason: "stop",
+          usage: { promptTokens: 10, completionTokens: 20 },
+          text: typeof mockText === "string" ? mockText : JSON.stringify(mockText)
+        };
+      },
+      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: {} }
+        };
+      }
+    });
+    super({ model: mockModel });
+  }
+  // @ts-ignore
+  stream(...args) {
+    const result = super.stream(...args);
+    return {
+      ...result,
+      // @ts-ignore on await read the stream
+      then: (onfulfilled, onrejected) => {
+        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);
+      }
+    };
+  }
+  // @ts-ignore
+  __streamObject(...args) {
+    const result = super.__streamObject(...args);
+    return {
+      ...result,
+      // @ts-ignore on await read the stream
+      then: (onfulfilled, onrejected) => {
+        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);
+      }
+    };
+  }
+};
+
+exports.MockProvider = MockProvider;
+exports.createMockModel = createMockModel;
+//# sourceMappingURL=llm-mock.cjs.map
+//# sourceMappingURL=llm-mock.cjs.map

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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,21 @@
+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, version, }: {
+    objectGenerationMode?: 'json';
+    mockText: string | Record<string, any>;
+    spyGenerate?: (props: any) => void;
+    spyStream?: (props: any) => void;
+    version?: 'v1' | 'v2';
+}): MockLanguageModelV1 | MockLanguageModelV2;
+export declare class MockProvider extends MastraLLMV1 {
+    constructor({ spyGenerate, spyStream, objectGenerationMode, mockText, }: {
+        spyGenerate?: (props: any) => void;
+        spyStream?: (props: any) => void;
+        objectGenerationMode?: 'json';
+        mockText?: string | Record<string, any>;
+    });
+    stream(...args: any): PromiseLike<StreamReturn<any, any, any>>;
+    __streamObject(...args: any[]): PromiseLike<StreamObjectResult<any>>;
+}
+//# sourceMappingURL=llm-mock.d.ts.map

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

@@ -0,0 +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;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

@@ -0,0 +1,178 @@
+import { MastraLLMV1 } from '../chunk-6D2K2CAA.js';
+import { simulateReadableStream } from 'ai';
+import { MockLanguageModelV1 } from 'ai/test';
+import { MockLanguageModelV2, convertArrayToReadableStream } from 'ai-v5/test';
+
+function createMockModel({
+  objectGenerationMode,
+  mockText,
+  spyGenerate,
+  spyStream,
+  version = "v2"
+}) {
+  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: 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: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+        content: [
+          {
+            type: "text",
+            text: finalText
+          }
+        ],
+        warnings: []
+      };
+    },
+    doStream: async (props) => {
+      if (spyStream) {
+        spyStream(props);
+      }
+      return {
+        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 }
+          }
+        ])
+      };
+    }
+  });
+  return mockModel;
+}
+var MockProvider = class extends MastraLLMV1 {
+  constructor({
+    spyGenerate,
+    spyStream,
+    objectGenerationMode,
+    mockText = "Hello, world!"
+  }) {
+    const mockModel = new MockLanguageModelV1({
+      defaultObjectGenerationMode: objectGenerationMode,
+      doGenerate: async (props) => {
+        if (spyGenerate) {
+          spyGenerate(props);
+        }
+        if (objectGenerationMode === "json") {
+          return {
+            rawCall: { rawPrompt: null, rawSettings: {} },
+            finishReason: "stop",
+            usage: { promptTokens: 10, completionTokens: 20 },
+            text: JSON.stringify(mockText)
+          };
+        }
+        return {
+          rawCall: { rawPrompt: null, rawSettings: {} },
+          finishReason: "stop",
+          usage: { promptTokens: 10, completionTokens: 20 },
+          text: typeof mockText === "string" ? mockText : JSON.stringify(mockText)
+        };
+      },
+      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: {} }
+        };
+      }
+    });
+    super({ model: mockModel });
+  }
+  // @ts-ignore
+  stream(...args) {
+    const result = super.stream(...args);
+    return {
+      ...result,
+      // @ts-ignore on await read the stream
+      then: (onfulfilled, onrejected) => {
+        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);
+      }
+    };
+  }
+  // @ts-ignore
+  __streamObject(...args) {
+    const result = super.__streamObject(...args);
+    return {
+      ...result,
+      // @ts-ignore on await read the stream
+      then: (onfulfilled, onrejected) => {
+        return result.baseStream.pipeTo(new WritableStream()).then(onfulfilled, onrejected);
+      }
+    };
+  }
+};
+
+export { MockProvider, createMockModel };
+//# sourceMappingURL=llm-mock.js.map
+//# sourceMappingURL=llm-mock.js.map

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

@@ -0,0 +1 @@
+{"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/tools/index.cjs

@@ -0,0 +1,25 @@
+'use strict';
+
+var chunkX7F4CSGR_cjs = require('../chunk-X7F4CSGR.cjs');
+var chunkVF676YCO_cjs = require('../chunk-VF676YCO.cjs');
+
+
+
+Object.defineProperty(exports, "ToolStream", {
+  enumerable: true,
+  get: function () { return chunkX7F4CSGR_cjs.ToolStream; }
+});
+Object.defineProperty(exports, "Tool", {
+  enumerable: true,
+  get: function () { return chunkVF676YCO_cjs.Tool; }
+});
+Object.defineProperty(exports, "createTool", {
+  enumerable: true,
+  get: function () { return chunkVF676YCO_cjs.createTool; }
+});
+Object.defineProperty(exports, "isVercelTool", {
+  enumerable: true,
+  get: function () { return chunkVF676YCO_cjs.isVercelTool; }
+});
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/tools/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":"","file":"index.cjs"}

package/dist/tools/index.d.ts

@@ -0,0 +1,6 @@
+export * from './tool.js';
+export * from './types.js';
+export * from './ui-types.js';
+export { isVercelTool } from './toolchecks.js';
+export { ToolStream } from './stream.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/tools/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC"}

package/dist/tools/index.js

@@ -0,0 +1,4 @@
+export { ToolStream } from '../chunk-5O52O25J.js';
+export { Tool, createTool, isVercelTool } from '../chunk-WM4RO23J.js';
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":"","file":"index.js"}

package/dist/tools/index.warning.d.ts

@@ -0,0 +1,8 @@
+import type { z } from 'zod';
+import { Tool as BaseTool } from './tool.js';
+import type { ToolAction, ToolExecutionContext } from './types.js';
+export * from './tool.js';
+export declare class Tool<TSchemaIn extends z.ZodSchema | undefined = undefined, TSchemaOut extends z.ZodSchema | undefined = undefined, TSuspendSchema extends z.ZodSchema = any, TResumeSchema extends z.ZodSchema = any, TContext extends ToolExecutionContext<TSchemaIn, TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSchemaIn, TSuspendSchema, TResumeSchema>> extends BaseTool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> {
+    constructor(opts: ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext>);
+}
+//# sourceMappingURL=index.warning.d.ts.map

package/dist/tools/index.warning.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.warning.d.ts","sourceRoot":"","sources":["../../src/tools/index.warning.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAE7B,OAAO,EAAE,IAAI,IAAI,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAC1C,OAAO,KAAK,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAEhE,cAAc,QAAQ,CAAC;AAEvB,qBAAa,IAAI,CACf,SAAS,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,GAAG,SAAS,EACrD,UAAU,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,GAAG,SAAS,EACtD,cAAc,SAAS,CAAC,CAAC,SAAS,GAAG,GAAG,EACxC,aAAa,SAAS,CAAC,CAAC,SAAS,GAAG,GAAG,EACvC,QAAQ,SAAS,oBAAoB,CAAC,SAAS,EAAE,cAAc,EAAE,aAAa,CAAC,GAAG,oBAAoB,CACpG,SAAS,EACT,cAAc,EACd,aAAa,CACd,CACD,SAAQ,QAAQ,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC;gBACpE,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC;CAK7F"}

package/dist/tools/is-vercel-tool.cjs

@@ -0,0 +1,12 @@
+'use strict';
+
+var chunkVF676YCO_cjs = require('../chunk-VF676YCO.cjs');
+
+
+
+Object.defineProperty(exports, "isVercelTool", {
+  enumerable: true,
+  get: function () { return chunkVF676YCO_cjs.isVercelTool; }
+});
+//# sourceMappingURL=is-vercel-tool.cjs.map
+//# sourceMappingURL=is-vercel-tool.cjs.map

package/dist/tools/is-vercel-tool.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":"","file":"is-vercel-tool.cjs"}

package/dist/tools/is-vercel-tool.d.ts

@@ -0,0 +1,2 @@
+export { isVercelTool } from './toolchecks.js';
+//# sourceMappingURL=is-vercel-tool.d.ts.map

package/dist/tools/is-vercel-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"is-vercel-tool.d.ts","sourceRoot":"","sources":["../../src/tools/is-vercel-tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC"}

package/dist/tools/is-vercel-tool.js

@@ -0,0 +1,3 @@
+export { isVercelTool } from '../chunk-WM4RO23J.js';
+//# sourceMappingURL=is-vercel-tool.js.map
+//# sourceMappingURL=is-vercel-tool.js.map

package/dist/tools/is-vercel-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":"","file":"is-vercel-tool.js"}

package/dist/tools/stream.d.ts

@@ -0,0 +1,19 @@
+import { WritableStream } from 'stream/web';
+import type { DataChunkType } from '../stream/types.js';
+export declare class ToolStream<T> extends WritableStream<T> {
+    originalStream?: WritableStream;
+    private writeQueue;
+    constructor({ prefix, callId, name, runId, }: {
+        prefix: string;
+        callId: string;
+        name: string;
+        runId: string;
+    }, originalStream?: WritableStream);
+    write(data: any): Promise<void>;
+    custom<T extends {
+        type: string;
+    }>(data: T extends {
+        type: `data-${string}`;
+    } ? DataChunkType : T): Promise<void>;
+}
+//# sourceMappingURL=stream.d.ts.map

package/dist/tools/stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/tools/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAErD,qBAAa,UAAU,CAAC,CAAC,CAAE,SAAQ,cAAc,CAAC,CAAC,CAAC;IAClD,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,OAAO,CAAC,UAAU,CAAoC;gBAGpD,EACE,MAAM,EACN,MAAM,EACN,IAAI,EACJ,KAAK,GACN,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;KACf,EACD,cAAc,CAAC,EAAE,cAAc;IAgC3B,KAAK,CAAC,IAAI,EAAE,GAAG;IAUf,MAAM,CAAC,CAAC,SAAS;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,EAAE,CAAC,SAAS;QAAE,IAAI,EAAE,QAAQ,MAAM,EAAE,CAAA;KAAE,GAAG,aAAa,GAAG,CAAC;CAiBxG"}

package/dist/tools/tool-builder/builder.d.ts

@@ -0,0 +1,28 @@
+import type { ProviderDefinedTool } from '@internal/external-types';
+import { MastraBase } from '../../base.js';
+import type { ToolOptions } from '../../utils.js';
+import type { CoreTool, ToolAction, VercelTool, VercelToolV5 } from '../types.js';
+/**
+ * Types that can be converted to Mastra tools.
+ * Includes provider-defined tools from external packages via ProviderDefinedTool.
+ */
+export type ToolToConvert = VercelTool | ToolAction<any, any, any> | VercelToolV5 | ProviderDefinedTool;
+export type LogType = 'tool' | 'toolset' | 'client-tool';
+export declare class CoreToolBuilder extends MastraBase {
+    private originalTool;
+    private options;
+    private logType?;
+    constructor(input: {
+        originalTool: ToolToConvert;
+        options: ToolOptions;
+        logType?: LogType;
+    });
+    private getParameters;
+    private getOutputSchema;
+    private buildProviderTool;
+    private createLogMessageOptions;
+    private createExecute;
+    buildV5(): VercelToolV5;
+    build(): CoreTool;
+}
+//# sourceMappingURL=builder.d.ts.map

package/dist/tools/tool-builder/builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"builder.d.ts","sourceRoot":"","sources":["../../../src/tools/tool-builder/builder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAmB,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAcrF,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAIxC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE/C,OAAO,KAAK,EAAE,QAAQ,EAA+B,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAG5G;;;GAGG;AACH,MAAM,MAAM,aAAa,GAAG,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,YAAY,GAAG,mBAAmB,CAAC;AACxG,MAAM,MAAM,OAAO,GAAG,MAAM,GAAG,SAAS,GAAG,aAAa,CAAC;AAazD,qBAAa,eAAgB,SAAQ,UAAU;IAC7C,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,OAAO,CAAC,CAAU;gBAEd,KAAK,EAAE;QAAE,YAAY,EAAE,aAAa,CAAC;QAAC,OAAO,EAAE,WAAW,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE;IAQ3F,OAAO,CAAC,aAAa,CA0BnB;IAEF,OAAO,CAAC,eAAe,CAYrB;IAGF,OAAO,CAAC,iBAAiB;IAiCzB,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,aAAa;IAsJrB,OAAO;IA+BP,KAAK,IAAI,QAAQ;CA6FlB"}