@mastra/core 1.0.0-beta.2 → 1.0.0-beta.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +206 -0
- package/dist/agent/agent-legacy.d.ts +2 -2
- package/dist/agent/agent-legacy.d.ts.map +1 -1
- package/dist/agent/agent.d.ts +1 -1
- package/dist/agent/agent.d.ts.map +1 -1
- package/dist/agent/agent.types.d.ts +3 -2
- package/dist/agent/agent.types.d.ts.map +1 -1
- package/dist/agent/index.cjs +9 -9
- package/dist/agent/index.js +2 -2
- package/dist/agent/message-list/index.cjs +3 -3
- package/dist/agent/message-list/index.d.ts +5 -0
- package/dist/agent/message-list/index.d.ts.map +1 -1
- package/dist/agent/message-list/index.js +1 -1
- package/dist/agent/message-list/prompt/convert-file.d.ts +1 -1
- package/dist/agent/message-list/prompt/convert-file.d.ts.map +1 -1
- package/dist/agent/message-list/prompt/download-assets.d.ts.map +1 -1
- package/dist/agent/types.d.ts +1 -0
- package/dist/agent/types.d.ts.map +1 -1
- package/dist/agent/utils.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/index.d.ts +2 -1
- package/dist/agent/workflows/prepare-stream/index.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/map-results-step.d.ts +3 -1
- package/dist/agent/workflows/prepare-stream/map-results-step.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/prepare-memory-step.d.ts +2 -1
- package/dist/agent/workflows/prepare-stream/prepare-memory-step.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/prepare-tools-step.d.ts +2 -1
- package/dist/agent/workflows/prepare-stream/prepare-tools-step.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/stream-step.d.ts +3 -1
- package/dist/agent/workflows/prepare-stream/stream-step.d.ts.map +1 -1
- package/dist/{chunk-YCVEJ3UN.cjs → chunk-2NVBZKZI.cjs} +988 -233
- package/dist/chunk-2NVBZKZI.cjs.map +1 -0
- package/dist/{chunk-SNPVZPLB.js → chunk-2OTDXX73.js} +6 -5
- package/dist/chunk-2OTDXX73.js.map +1 -0
- package/dist/chunk-3PSWNGBF.js +3 -0
- package/dist/{chunk-ZV5CC35D.js.map → chunk-3PSWNGBF.js.map} +1 -1
- package/dist/{chunk-ZGHTOYHW.js → chunk-3RW5EMSB.js} +155 -24
- package/dist/chunk-3RW5EMSB.js.map +1 -0
- package/dist/chunk-3W5RQCCY.cjs +440 -0
- package/dist/chunk-3W5RQCCY.cjs.map +1 -0
- package/dist/{chunk-ET6UOTTU.cjs → chunk-4IKJAKCD.cjs} +40 -4
- package/dist/chunk-4IKJAKCD.cjs.map +1 -0
- package/dist/{chunk-2ZVKF4HP.cjs → chunk-4RXG622P.cjs} +184 -285
- package/dist/chunk-4RXG622P.cjs.map +1 -0
- package/dist/{chunk-22443P6A.cjs → chunk-5WXEYDFI.cjs} +173 -42
- package/dist/chunk-5WXEYDFI.cjs.map +1 -0
- package/dist/{chunk-ZWNI5IWX.cjs → chunk-CYVNOIXS.cjs} +13 -12
- package/dist/chunk-CYVNOIXS.cjs.map +1 -0
- package/dist/{chunk-4CDL2QJT.js → chunk-D6EDHNGV.js} +53 -16
- package/dist/chunk-D6EDHNGV.js.map +1 -0
- package/dist/chunk-FVNT7VTO.js +436 -0
- package/dist/chunk-FVNT7VTO.js.map +1 -0
- package/dist/{chunk-WM6CK2F3.cjs → chunk-HBJPYQRN.cjs} +57 -19
- package/dist/chunk-HBJPYQRN.cjs.map +1 -0
- package/dist/{chunk-HDJFSJCK.js → chunk-IHJDOC3A.js} +35 -587
- package/dist/chunk-IHJDOC3A.js.map +1 -0
- package/dist/{chunk-QUKUN6NR.cjs → chunk-ISMGVGUM.cjs} +105 -5
- package/dist/chunk-ISMGVGUM.cjs.map +1 -0
- package/dist/{chunk-JYYQQEBH.cjs → chunk-IWB65P37.cjs} +241 -4
- package/dist/chunk-IWB65P37.cjs.map +1 -0
- package/dist/{chunk-MV7KHWUT.js → chunk-IWQDBVJK.js} +25 -4
- package/dist/chunk-IWQDBVJK.js.map +1 -0
- package/dist/{chunk-XEVG546F.js → chunk-JXESKY4A.js} +3 -3
- package/dist/{chunk-XEVG546F.js.map → chunk-JXESKY4A.js.map} +1 -1
- package/dist/{chunk-7AHYOMHJ.js → chunk-KEURQGCQ.js} +40 -5
- package/dist/chunk-KEURQGCQ.js.map +1 -0
- package/dist/{chunk-JPGVRWWL.js → chunk-MDKPL2R2.js} +470 -124
- package/dist/chunk-MDKPL2R2.js.map +1 -0
- package/dist/{chunk-UIZSWUKP.js → chunk-NZAXAFI3.js} +104 -6
- package/dist/chunk-NZAXAFI3.js.map +1 -0
- package/dist/{chunk-VOY2RXOC.cjs → chunk-O6NA3Z43.cjs} +6 -6
- package/dist/{chunk-VOY2RXOC.cjs.map → chunk-O6NA3Z43.cjs.map} +1 -1
- package/dist/chunk-PE3V7GUL.cjs +4 -0
- package/dist/{chunk-LJFJTTZQ.cjs.map → chunk-PE3V7GUL.cjs.map} +1 -1
- package/dist/{chunk-I4CXL4SR.js → chunk-RXDJL5QT.js} +5 -4
- package/dist/chunk-RXDJL5QT.js.map +1 -0
- package/dist/{chunk-LWBQ4P4N.cjs → chunk-S6OEQHEI.cjs} +62 -614
- package/dist/chunk-S6OEQHEI.cjs.map +1 -0
- package/dist/{chunk-7PO6SEJF.js → chunk-U7VECK2G.js} +240 -3
- package/dist/chunk-U7VECK2G.js.map +1 -0
- package/dist/{chunk-ECFXGXWO.cjs → chunk-VSM3NLUX.cjs} +312 -210
- package/dist/chunk-VSM3NLUX.cjs.map +1 -0
- package/dist/{chunk-IQO7ANVS.cjs → chunk-VZC4BWWH.cjs} +10 -9
- package/dist/chunk-VZC4BWWH.cjs.map +1 -0
- package/dist/{chunk-JV2KH24V.js → chunk-W3DD3XP5.js} +312 -210
- package/dist/chunk-W3DD3XP5.js.map +1 -0
- package/dist/{chunk-GGYKYORQ.cjs → chunk-WQSGX6XA.cjs} +27 -6
- package/dist/chunk-WQSGX6XA.cjs.map +1 -0
- package/dist/{chunk-W7UH2PWL.js → chunk-WTYNK7Q4.js} +179 -282
- package/dist/chunk-WTYNK7Q4.js.map +1 -0
- package/dist/{chunk-7CBEP2ZQ.js → chunk-XXBWX7DT.js} +949 -197
- package/dist/chunk-XXBWX7DT.js.map +1 -0
- package/dist/{chunk-CB575O6L.cjs → chunk-ZCVTH3CH.cjs} +470 -130
- package/dist/chunk-ZCVTH3CH.cjs.map +1 -0
- package/dist/evals/base.d.ts.map +1 -1
- package/dist/evals/index.cjs +4 -4
- package/dist/evals/index.js +1 -1
- package/dist/evals/scoreTraces/index.cjs +5 -4
- package/dist/evals/scoreTraces/index.cjs.map +1 -1
- package/dist/evals/scoreTraces/index.js +3 -2
- package/dist/evals/scoreTraces/index.js.map +1 -1
- package/dist/evals/scoreTraces/scoreTracesWorkflow.d.ts.map +1 -1
- package/dist/index.cjs +2 -2
- package/dist/index.js +1 -1
- package/dist/integration/index.cjs +2 -2
- package/dist/integration/index.js +1 -1
- package/dist/llm/index.cjs +26 -10
- package/dist/llm/index.d.ts +1 -0
- package/dist/llm/index.d.ts.map +1 -1
- package/dist/llm/index.js +5 -1
- package/dist/llm/model/aisdk/v5/model.d.ts +47 -0
- package/dist/llm/model/aisdk/v5/model.d.ts.map +1 -0
- package/dist/llm/model/gateways/base.d.ts +8 -0
- package/dist/llm/model/gateways/base.d.ts.map +1 -1
- package/dist/llm/model/gateways/models-dev.d.ts +1 -0
- package/dist/llm/model/gateways/models-dev.d.ts.map +1 -1
- package/dist/llm/model/gateways/netlify.d.ts +2 -1
- package/dist/llm/model/gateways/netlify.d.ts.map +1 -1
- package/dist/llm/model/is-v2-model.d.ts +3 -0
- package/dist/llm/model/is-v2-model.d.ts.map +1 -0
- package/dist/llm/model/model-method-from-agent.d.ts +4 -0
- package/dist/llm/model/model-method-from-agent.d.ts.map +1 -0
- package/dist/llm/model/model.loop.d.ts +2 -2
- package/dist/llm/model/model.loop.d.ts.map +1 -1
- package/dist/llm/model/model.loop.types.d.ts +2 -0
- package/dist/llm/model/model.loop.types.d.ts.map +1 -1
- package/dist/llm/model/provider-registry.d.ts +11 -1
- package/dist/llm/model/provider-registry.d.ts.map +1 -1
- package/dist/llm/model/provider-types.generated.d.ts +56 -7
- package/dist/llm/model/registry-generator.d.ts.map +1 -1
- package/dist/llm/model/resolve-model.d.ts +1 -1
- package/dist/llm/model/resolve-model.d.ts.map +1 -1
- package/dist/llm/model/router.d.ts +12 -5
- package/dist/llm/model/router.d.ts.map +1 -1
- package/dist/llm/model/shared.types.d.ts +10 -4
- package/dist/llm/model/shared.types.d.ts.map +1 -1
- package/dist/loop/index.cjs +2 -2
- package/dist/loop/index.js +1 -1
- package/dist/loop/network/index.d.ts.map +1 -1
- package/dist/loop/test-utils/MastraLanguageModelV2Mock.d.ts +37 -0
- package/dist/loop/test-utils/MastraLanguageModelV2Mock.d.ts.map +1 -0
- package/dist/loop/test-utils/fullStream.d.ts.map +1 -1
- package/dist/loop/test-utils/generateText.d.ts.map +1 -1
- package/dist/loop/test-utils/options.d.ts.map +1 -1
- package/dist/loop/test-utils/resultObject.d.ts.map +1 -1
- package/dist/loop/test-utils/streamObject.d.ts.map +1 -1
- package/dist/loop/test-utils/textStream.d.ts.map +1 -1
- package/dist/loop/test-utils/toUIMessageStream.d.ts.map +1 -1
- package/dist/loop/test-utils/tools.d.ts.map +1 -1
- package/dist/loop/test-utils/utils.d.ts +1 -1
- package/dist/loop/test-utils/utils.d.ts.map +1 -1
- package/dist/loop/types.d.ts +8 -3
- package/dist/loop/types.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-execution/index.d.ts +12 -12
- package/dist/loop/workflows/agentic-execution/index.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-execution/llm-execution-step.d.ts +9 -9
- package/dist/loop/workflows/agentic-execution/llm-execution-step.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-execution/llm-mapping-step.d.ts +4 -4
- package/dist/loop/workflows/agentic-execution/llm-mapping-step.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-loop/index.d.ts +13 -12
- package/dist/loop/workflows/agentic-loop/index.d.ts.map +1 -1
- package/dist/loop/workflows/run-state.d.ts +2 -2
- package/dist/loop/workflows/run-state.d.ts.map +1 -1
- package/dist/loop/workflows/schema.d.ts +4 -4
- package/dist/loop/workflows/stream.d.ts.map +1 -1
- package/dist/mastra/index.cjs +2 -2
- package/dist/mastra/index.d.ts +127 -3
- package/dist/mastra/index.d.ts.map +1 -1
- package/dist/mastra/index.js +1 -1
- package/dist/memory/index.cjs +8 -8
- package/dist/memory/index.js +4 -4
- package/dist/models-dev-6PRLJKVZ.js +3 -0
- package/dist/{models-dev-DNBKXHT4.js.map → models-dev-6PRLJKVZ.js.map} +1 -1
- package/dist/models-dev-WHMI5G6Y.cjs +12 -0
- package/dist/{models-dev-YBEEQIX6.cjs.map → models-dev-WHMI5G6Y.cjs.map} +1 -1
- package/dist/netlify-46I3SNNV.cjs +12 -0
- package/dist/{netlify-GWNGSIRZ.cjs.map → netlify-46I3SNNV.cjs.map} +1 -1
- package/dist/netlify-EBQ6YUC6.js +3 -0
- package/dist/{netlify-7G2L5VSH.js.map → netlify-EBQ6YUC6.js.map} +1 -1
- package/dist/processors/index.cjs +11 -11
- package/dist/processors/index.js +1 -1
- package/dist/provider-registry-HDG6UMUC.js +3 -0
- package/dist/provider-registry-HDG6UMUC.js.map +1 -0
- package/dist/provider-registry-RP2W4B24.cjs +40 -0
- package/dist/provider-registry-RP2W4B24.cjs.map +1 -0
- package/dist/provider-registry.json +140 -18
- package/dist/{registry-generator-MK63POJO.cjs → registry-generator-JPCV47SC.cjs} +6 -4
- package/dist/registry-generator-JPCV47SC.cjs.map +1 -0
- package/dist/{registry-generator-H4YNODDH.js → registry-generator-XD4FPZTU.js} +6 -4
- package/dist/registry-generator-XD4FPZTU.js.map +1 -0
- package/dist/relevance/index.cjs +2 -2
- package/dist/relevance/index.js +1 -1
- package/dist/server/auth.d.ts +11 -0
- package/dist/server/auth.d.ts.map +1 -1
- package/dist/server/index.cjs +12 -1
- package/dist/server/index.cjs.map +1 -1
- package/dist/server/index.d.ts +4 -0
- package/dist/server/index.d.ts.map +1 -1
- package/dist/server/index.js +12 -1
- package/dist/server/index.js.map +1 -1
- package/dist/storage/domains/workflows/inmemory.d.ts +1 -1
- package/dist/storage/domains/workflows/inmemory.d.ts.map +1 -1
- package/dist/storage/index.cjs +29 -29
- package/dist/storage/index.js +1 -1
- package/dist/storage/types.d.ts +2 -1
- package/dist/storage/types.d.ts.map +1 -1
- package/dist/stream/RunOutput.d.ts +1 -1
- package/dist/stream/RunOutput.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/compat/prepare-tools.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/execute.d.ts +6 -3
- package/dist/stream/aisdk/v5/execute.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/output.d.ts.map +1 -1
- package/dist/stream/base/input.d.ts +1 -1
- package/dist/stream/base/output.d.ts.map +1 -1
- package/dist/stream/index.cjs +11 -11
- package/dist/stream/index.js +2 -2
- package/dist/stream/types.d.ts +4 -3
- package/dist/stream/types.d.ts.map +1 -1
- package/dist/test-utils/llm-mock.cjs +68 -31
- package/dist/test-utils/llm-mock.cjs.map +1 -1
- package/dist/test-utils/llm-mock.d.ts +4 -2
- package/dist/test-utils/llm-mock.d.ts.map +1 -1
- package/dist/test-utils/llm-mock.js +67 -30
- package/dist/test-utils/llm-mock.js.map +1 -1
- package/dist/tools/index.cjs +4 -4
- package/dist/tools/index.js +1 -1
- package/dist/tools/is-vercel-tool.cjs +2 -2
- package/dist/tools/is-vercel-tool.js +1 -1
- package/dist/tools/tool-builder/builder.d.ts.map +1 -1
- package/dist/tools/tool.d.ts.map +1 -1
- package/dist/tools/types.d.ts +1 -0
- package/dist/tools/types.d.ts.map +1 -1
- package/dist/tools/validation.d.ts +12 -0
- package/dist/tools/validation.d.ts.map +1 -1
- package/dist/utils.cjs +25 -21
- package/dist/utils.d.ts +4 -1
- package/dist/utils.d.ts.map +1 -1
- package/dist/utils.js +1 -1
- package/dist/voice/aisdk/index.d.ts +3 -0
- package/dist/voice/aisdk/index.d.ts.map +1 -0
- package/dist/voice/aisdk/speech.d.ts +23 -0
- package/dist/voice/aisdk/speech.d.ts.map +1 -0
- package/dist/voice/aisdk/transcription.d.ts +22 -0
- package/dist/voice/aisdk/transcription.d.ts.map +1 -0
- package/dist/voice/composite-voice.d.ts +4 -3
- package/dist/voice/composite-voice.d.ts.map +1 -1
- package/dist/voice/index.cjs +12 -4
- package/dist/voice/index.d.ts +1 -0
- package/dist/voice/index.d.ts.map +1 -1
- package/dist/voice/index.js +1 -1
- package/dist/workflows/default.d.ts +24 -8
- package/dist/workflows/default.d.ts.map +1 -1
- package/dist/workflows/evented/execution-engine.d.ts +3 -1
- package/dist/workflows/evented/execution-engine.d.ts.map +1 -1
- package/dist/workflows/evented/index.cjs +10 -10
- package/dist/workflows/evented/index.js +1 -1
- package/dist/workflows/evented/workflow-event-processor/index.d.ts +5 -4
- package/dist/workflows/evented/workflow-event-processor/index.d.ts.map +1 -1
- package/dist/workflows/evented/workflow-event-processor/loop.d.ts +1 -1
- package/dist/workflows/evented/workflow-event-processor/loop.d.ts.map +1 -1
- package/dist/workflows/evented/workflow-event-processor/parallel.d.ts +2 -2
- package/dist/workflows/evented/workflow-event-processor/parallel.d.ts.map +1 -1
- package/dist/workflows/evented/workflow-event-processor/sleep.d.ts +2 -2
- package/dist/workflows/evented/workflow-event-processor/sleep.d.ts.map +1 -1
- package/dist/workflows/evented/workflow.d.ts +2 -1
- package/dist/workflows/evented/workflow.d.ts.map +1 -1
- package/dist/workflows/execution-engine.d.ts +4 -2
- package/dist/workflows/execution-engine.d.ts.map +1 -1
- package/dist/workflows/index.cjs +28 -16
- package/dist/workflows/index.js +1 -1
- package/dist/workflows/step.d.ts +1 -0
- package/dist/workflows/step.d.ts.map +1 -1
- package/dist/workflows/types.d.ts +54 -2
- package/dist/workflows/types.d.ts.map +1 -1
- package/dist/workflows/utils.d.ts +20 -0
- package/dist/workflows/utils.d.ts.map +1 -1
- package/dist/workflows/workflow.d.ts +108 -40
- package/dist/workflows/workflow.d.ts.map +1 -1
- package/package.json +19 -27
- package/src/llm/model/provider-types.generated.d.ts +56 -7
- package/dist/chunk-22443P6A.cjs.map +0 -1
- package/dist/chunk-2ZVKF4HP.cjs.map +0 -1
- package/dist/chunk-4CDL2QJT.js.map +0 -1
- package/dist/chunk-7AHYOMHJ.js.map +0 -1
- package/dist/chunk-7CBEP2ZQ.js.map +0 -1
- package/dist/chunk-7PO6SEJF.js.map +0 -1
- package/dist/chunk-CB575O6L.cjs.map +0 -1
- package/dist/chunk-ECFXGXWO.cjs.map +0 -1
- package/dist/chunk-ET6UOTTU.cjs.map +0 -1
- package/dist/chunk-GGYKYORQ.cjs.map +0 -1
- package/dist/chunk-HDJFSJCK.js.map +0 -1
- package/dist/chunk-I4CXL4SR.js.map +0 -1
- package/dist/chunk-IQO7ANVS.cjs.map +0 -1
- package/dist/chunk-JPGVRWWL.js.map +0 -1
- package/dist/chunk-JV2KH24V.js.map +0 -1
- package/dist/chunk-JYYQQEBH.cjs.map +0 -1
- package/dist/chunk-LJFJTTZQ.cjs +0 -4
- package/dist/chunk-LWBQ4P4N.cjs.map +0 -1
- package/dist/chunk-MV7KHWUT.js.map +0 -1
- package/dist/chunk-QUKUN6NR.cjs.map +0 -1
- package/dist/chunk-SNPVZPLB.js.map +0 -1
- package/dist/chunk-UIZSWUKP.js.map +0 -1
- package/dist/chunk-W7UH2PWL.js.map +0 -1
- package/dist/chunk-WM6CK2F3.cjs.map +0 -1
- package/dist/chunk-YCVEJ3UN.cjs.map +0 -1
- package/dist/chunk-ZGHTOYHW.js.map +0 -1
- package/dist/chunk-ZV5CC35D.js +0 -3
- package/dist/chunk-ZWNI5IWX.cjs.map +0 -1
- package/dist/models-dev-DNBKXHT4.js +0 -3
- package/dist/models-dev-YBEEQIX6.cjs +0 -12
- package/dist/netlify-7G2L5VSH.js +0 -3
- package/dist/netlify-GWNGSIRZ.cjs +0 -12
- package/dist/registry-generator-H4YNODDH.js.map +0 -1
- package/dist/registry-generator-MK63POJO.cjs.map +0 -1
|
@@ -1,64 +1,101 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunkO6NA3Z43_cjs = require('../chunk-O6NA3Z43.cjs');
|
|
4
4
|
var chunkSVLMF4UZ_cjs = require('../chunk-SVLMF4UZ.cjs');
|
|
5
|
+
var test = require('ai-v5/test');
|
|
5
6
|
|
|
6
|
-
// src/test-utils/llm-mock.ts
|
|
7
7
|
function createMockModel({
|
|
8
8
|
objectGenerationMode,
|
|
9
9
|
mockText,
|
|
10
10
|
spyGenerate,
|
|
11
|
-
spyStream
|
|
11
|
+
spyStream,
|
|
12
|
+
version = "v2"
|
|
12
13
|
}) {
|
|
13
|
-
const
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
14
|
+
const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
|
|
15
|
+
const finalText = objectGenerationMode === "json" ? JSON.stringify(mockText) : text;
|
|
16
|
+
if (version === "v1") {
|
|
17
|
+
const mockModel2 = new chunkSVLMF4UZ_cjs.MockLanguageModelV1({
|
|
18
|
+
defaultObjectGenerationMode: objectGenerationMode,
|
|
19
|
+
doGenerate: async (props) => {
|
|
20
|
+
if (spyGenerate) {
|
|
21
|
+
spyGenerate(props);
|
|
22
|
+
}
|
|
20
23
|
return {
|
|
21
24
|
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
22
25
|
finishReason: "stop",
|
|
23
26
|
usage: { promptTokens: 10, completionTokens: 20 },
|
|
24
|
-
text:
|
|
27
|
+
text: finalText
|
|
28
|
+
};
|
|
29
|
+
},
|
|
30
|
+
doStream: async (props) => {
|
|
31
|
+
if (spyStream) {
|
|
32
|
+
spyStream(props);
|
|
33
|
+
}
|
|
34
|
+
const chunks = finalText.split(" ").map((word) => ({
|
|
35
|
+
type: "text-delta",
|
|
36
|
+
textDelta: word + " "
|
|
37
|
+
}));
|
|
38
|
+
return {
|
|
39
|
+
stream: chunkSVLMF4UZ_cjs.simulateReadableStream({
|
|
40
|
+
chunks: [
|
|
41
|
+
...chunks,
|
|
42
|
+
{
|
|
43
|
+
type: "finish",
|
|
44
|
+
finishReason: "stop",
|
|
45
|
+
logprobs: void 0,
|
|
46
|
+
usage: { completionTokens: 10, promptTokens: 3 }
|
|
47
|
+
}
|
|
48
|
+
]
|
|
49
|
+
}),
|
|
50
|
+
rawCall: { rawPrompt: null, rawSettings: {} }
|
|
25
51
|
};
|
|
26
52
|
}
|
|
53
|
+
});
|
|
54
|
+
return mockModel2;
|
|
55
|
+
}
|
|
56
|
+
const mockModel = new test.MockLanguageModelV2({
|
|
57
|
+
doGenerate: async (props) => {
|
|
58
|
+
if (spyGenerate) {
|
|
59
|
+
spyGenerate(props);
|
|
60
|
+
}
|
|
27
61
|
return {
|
|
28
62
|
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
29
63
|
finishReason: "stop",
|
|
30
|
-
usage: {
|
|
31
|
-
|
|
64
|
+
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
|
|
65
|
+
content: [
|
|
66
|
+
{
|
|
67
|
+
type: "text",
|
|
68
|
+
text: finalText
|
|
69
|
+
}
|
|
70
|
+
],
|
|
71
|
+
warnings: []
|
|
32
72
|
};
|
|
33
73
|
},
|
|
34
74
|
doStream: async (props) => {
|
|
35
75
|
if (spyStream) {
|
|
36
76
|
spyStream(props);
|
|
37
77
|
}
|
|
38
|
-
const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
|
|
39
|
-
const chunks = text.split(" ").map((word) => ({
|
|
40
|
-
type: "text-delta",
|
|
41
|
-
textDelta: word + " "
|
|
42
|
-
}));
|
|
43
78
|
return {
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
79
|
+
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
80
|
+
warnings: [],
|
|
81
|
+
stream: test.convertArrayToReadableStream([
|
|
82
|
+
{ type: "stream-start", warnings: [] },
|
|
83
|
+
{ type: "response-metadata", id: "id-0", modelId: "mock-model-id", timestamp: /* @__PURE__ */ new Date(0) },
|
|
84
|
+
{ type: "text-start", id: "1" },
|
|
85
|
+
{ type: "text-delta", id: "1", delta: finalText },
|
|
86
|
+
{ type: "text-end", id: "1" },
|
|
87
|
+
{
|
|
88
|
+
type: "finish",
|
|
89
|
+
finishReason: "stop",
|
|
90
|
+
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 }
|
|
91
|
+
}
|
|
92
|
+
])
|
|
56
93
|
};
|
|
57
94
|
}
|
|
58
95
|
});
|
|
59
96
|
return mockModel;
|
|
60
97
|
}
|
|
61
|
-
var MockProvider = class extends
|
|
98
|
+
var MockProvider = class extends chunkO6NA3Z43_cjs.MastraLLMV1 {
|
|
62
99
|
constructor({
|
|
63
100
|
spyGenerate,
|
|
64
101
|
spyStream,
|
|
@@ -1 +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,qCAAA,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,wCAAA,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,qCAAA,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,wCAAA,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 '@internal/ai-sdk-v4';\nimport { MockLanguageModelV1 } from '@internal/ai-sdk-v4';\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"]}
|
|
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,qCAAA,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,wCAAA,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,wBAAA,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,iCAAA,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,qCAAA,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,wCAAA,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 '@internal/ai-sdk-v4';\nimport { MockLanguageModelV1 } from '@internal/ai-sdk-v4';\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"]}
|
|
@@ -1,11 +1,13 @@
|
|
|
1
1
|
import { MockLanguageModelV1 } from '../ai-sdk.types.d.ts';
|
|
2
|
+
import { MockLanguageModelV2 } from 'ai-v5/test';
|
|
2
3
|
import { MastraLLMV1 } from '../llm/model/model.js';
|
|
3
|
-
export declare function createMockModel({ objectGenerationMode, mockText, spyGenerate, spyStream, }: {
|
|
4
|
+
export declare function createMockModel({ objectGenerationMode, mockText, spyGenerate, spyStream, version, }: {
|
|
4
5
|
objectGenerationMode?: 'json';
|
|
5
6
|
mockText: string | Record<string, any>;
|
|
6
7
|
spyGenerate?: (props: any) => void;
|
|
7
8
|
spyStream?: (props: any) => void;
|
|
8
|
-
|
|
9
|
+
version?: 'v1' | 'v2';
|
|
10
|
+
}): MockLanguageModelV1 | MockLanguageModelV2;
|
|
9
11
|
export declare class MockProvider extends MastraLLMV1 {
|
|
10
12
|
constructor({ spyGenerate, spyStream, objectGenerationMode, mockText, }: {
|
|
11
13
|
spyGenerate?: (props: any) => void;
|
|
@@ -1 +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,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"llm-mock.d.ts","sourceRoot":"","sources":["../../src/test-utils/llm-mock.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,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"}
|
|
@@ -1,56 +1,93 @@
|
|
|
1
|
-
import { MastraLLMV1 } from '../chunk-
|
|
1
|
+
import { MastraLLMV1 } from '../chunk-JXESKY4A.js';
|
|
2
2
|
import { MockLanguageModelV1, simulateReadableStream } from '../chunk-QM5SRDJX.js';
|
|
3
|
+
import { MockLanguageModelV2, convertArrayToReadableStream } from 'ai-v5/test';
|
|
3
4
|
|
|
4
|
-
// src/test-utils/llm-mock.ts
|
|
5
5
|
function createMockModel({
|
|
6
6
|
objectGenerationMode,
|
|
7
7
|
mockText,
|
|
8
8
|
spyGenerate,
|
|
9
|
-
spyStream
|
|
9
|
+
spyStream,
|
|
10
|
+
version = "v2"
|
|
10
11
|
}) {
|
|
11
|
-
const
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
12
|
+
const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
|
|
13
|
+
const finalText = objectGenerationMode === "json" ? JSON.stringify(mockText) : text;
|
|
14
|
+
if (version === "v1") {
|
|
15
|
+
const mockModel2 = new MockLanguageModelV1({
|
|
16
|
+
defaultObjectGenerationMode: objectGenerationMode,
|
|
17
|
+
doGenerate: async (props) => {
|
|
18
|
+
if (spyGenerate) {
|
|
19
|
+
spyGenerate(props);
|
|
20
|
+
}
|
|
18
21
|
return {
|
|
19
22
|
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
20
23
|
finishReason: "stop",
|
|
21
24
|
usage: { promptTokens: 10, completionTokens: 20 },
|
|
22
|
-
text:
|
|
25
|
+
text: finalText
|
|
26
|
+
};
|
|
27
|
+
},
|
|
28
|
+
doStream: async (props) => {
|
|
29
|
+
if (spyStream) {
|
|
30
|
+
spyStream(props);
|
|
31
|
+
}
|
|
32
|
+
const chunks = finalText.split(" ").map((word) => ({
|
|
33
|
+
type: "text-delta",
|
|
34
|
+
textDelta: word + " "
|
|
35
|
+
}));
|
|
36
|
+
return {
|
|
37
|
+
stream: simulateReadableStream({
|
|
38
|
+
chunks: [
|
|
39
|
+
...chunks,
|
|
40
|
+
{
|
|
41
|
+
type: "finish",
|
|
42
|
+
finishReason: "stop",
|
|
43
|
+
logprobs: void 0,
|
|
44
|
+
usage: { completionTokens: 10, promptTokens: 3 }
|
|
45
|
+
}
|
|
46
|
+
]
|
|
47
|
+
}),
|
|
48
|
+
rawCall: { rawPrompt: null, rawSettings: {} }
|
|
23
49
|
};
|
|
24
50
|
}
|
|
51
|
+
});
|
|
52
|
+
return mockModel2;
|
|
53
|
+
}
|
|
54
|
+
const mockModel = new MockLanguageModelV2({
|
|
55
|
+
doGenerate: async (props) => {
|
|
56
|
+
if (spyGenerate) {
|
|
57
|
+
spyGenerate(props);
|
|
58
|
+
}
|
|
25
59
|
return {
|
|
26
60
|
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
27
61
|
finishReason: "stop",
|
|
28
|
-
usage: {
|
|
29
|
-
|
|
62
|
+
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
|
|
63
|
+
content: [
|
|
64
|
+
{
|
|
65
|
+
type: "text",
|
|
66
|
+
text: finalText
|
|
67
|
+
}
|
|
68
|
+
],
|
|
69
|
+
warnings: []
|
|
30
70
|
};
|
|
31
71
|
},
|
|
32
72
|
doStream: async (props) => {
|
|
33
73
|
if (spyStream) {
|
|
34
74
|
spyStream(props);
|
|
35
75
|
}
|
|
36
|
-
const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
|
|
37
|
-
const chunks = text.split(" ").map((word) => ({
|
|
38
|
-
type: "text-delta",
|
|
39
|
-
textDelta: word + " "
|
|
40
|
-
}));
|
|
41
76
|
return {
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
77
|
+
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
78
|
+
warnings: [],
|
|
79
|
+
stream: convertArrayToReadableStream([
|
|
80
|
+
{ type: "stream-start", warnings: [] },
|
|
81
|
+
{ type: "response-metadata", id: "id-0", modelId: "mock-model-id", timestamp: /* @__PURE__ */ new Date(0) },
|
|
82
|
+
{ type: "text-start", id: "1" },
|
|
83
|
+
{ type: "text-delta", id: "1", delta: finalText },
|
|
84
|
+
{ type: "text-end", id: "1" },
|
|
85
|
+
{
|
|
86
|
+
type: "finish",
|
|
87
|
+
finishReason: "stop",
|
|
88
|
+
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 }
|
|
89
|
+
}
|
|
90
|
+
])
|
|
54
91
|
};
|
|
55
92
|
}
|
|
56
93
|
});
|
|
@@ -1 +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 '@internal/ai-sdk-v4';\nimport { MockLanguageModelV1 } from '@internal/ai-sdk-v4';\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"]}
|
|
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 '@internal/ai-sdk-v4';\nimport { MockLanguageModelV1 } from '@internal/ai-sdk-v4';\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
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
3
|
var chunkLRSB62Z6_cjs = require('../chunk-LRSB62Z6.cjs');
|
|
4
|
-
var
|
|
4
|
+
var chunk4IKJAKCD_cjs = require('../chunk-4IKJAKCD.cjs');
|
|
5
5
|
|
|
6
6
|
|
|
7
7
|
|
|
@@ -11,15 +11,15 @@ Object.defineProperty(exports, "ToolStream", {
|
|
|
11
11
|
});
|
|
12
12
|
Object.defineProperty(exports, "Tool", {
|
|
13
13
|
enumerable: true,
|
|
14
|
-
get: function () { return
|
|
14
|
+
get: function () { return chunk4IKJAKCD_cjs.Tool; }
|
|
15
15
|
});
|
|
16
16
|
Object.defineProperty(exports, "createTool", {
|
|
17
17
|
enumerable: true,
|
|
18
|
-
get: function () { return
|
|
18
|
+
get: function () { return chunk4IKJAKCD_cjs.createTool; }
|
|
19
19
|
});
|
|
20
20
|
Object.defineProperty(exports, "isVercelTool", {
|
|
21
21
|
enumerable: true,
|
|
22
|
-
get: function () { return
|
|
22
|
+
get: function () { return chunk4IKJAKCD_cjs.isVercelTool; }
|
|
23
23
|
});
|
|
24
24
|
//# sourceMappingURL=index.cjs.map
|
|
25
25
|
//# sourceMappingURL=index.cjs.map
|
package/dist/tools/index.js
CHANGED
|
@@ -1,12 +1,12 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunk4IKJAKCD_cjs = require('../chunk-4IKJAKCD.cjs');
|
|
4
4
|
|
|
5
5
|
|
|
6
6
|
|
|
7
7
|
Object.defineProperty(exports, "isVercelTool", {
|
|
8
8
|
enumerable: true,
|
|
9
|
-
get: function () { return
|
|
9
|
+
get: function () { return chunk4IKJAKCD_cjs.isVercelTool; }
|
|
10
10
|
});
|
|
11
11
|
//# sourceMappingURL=is-vercel-tool.cjs.map
|
|
12
12
|
//# sourceMappingURL=is-vercel-tool.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"builder.d.ts","sourceRoot":"","sources":["../../../src/tools/tool-builder/builder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAwB,MAAM,0BAA0B,CAAC;AAY1F,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAKxC,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;
|
|
1
|
+
{"version":3,"file":"builder.d.ts","sourceRoot":"","sources":["../../../src/tools/tool-builder/builder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAwB,MAAM,0BAA0B,CAAC;AAY1F,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAKxC,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;IAqEzB,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,aAAa;IA6NrB,OAAO;IA+BP,KAAK,IAAI,QAAQ;CA+FlB"}
|
package/dist/tools/tool.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/tools/tool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAExC,OAAO,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAC7E,OAAO,KAAK,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAGhE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,qBAAa,IAAI,CACf,SAAS,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACvD,UAAU,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACxD,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG,EACzC,QAAQ,SAAS,oBAAoB,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,oBAAoB,CACzF,cAAc,EACd,aAAa,CACd,CACD,YAAW,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC;IAErF,qCAAqC;IACrC,EAAE,EAAE,MAAM,CAAC;IAEX,wCAAwC;IACxC,WAAW,EAAE,MAAM,CAAC;IAEpB,6CAA6C;IAC7C,WAAW,CAAC,EAAE,SAAS,CAAC;IAExB,6CAA6C;IAC7C,YAAY,CAAC,EAAE,UAAU,CAAC;IAE1B,wCAAwC;IACxC,aAAa,CAAC,EAAE,cAAc,CAAC;IAE/B,uCAAuC;IACvC,YAAY,CAAC,EAAE,aAAa,CAAC;IAE7B;;;;;OAKG;IACH,OAAO,CAAC,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAC;IAEhG,4DAA4D;IAC5D,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE,OAAO,CAAC;IAE1B;;;;;;;;;;;;;OAaG;gBACS,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC;
|
|
1
|
+
{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/tools/tool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAExC,OAAO,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAC7E,OAAO,KAAK,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAGhE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,qBAAa,IAAI,CACf,SAAS,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACvD,UAAU,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACxD,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG,EACzC,QAAQ,SAAS,oBAAoB,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,oBAAoB,CACzF,cAAc,EACd,aAAa,CACd,CACD,YAAW,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC;IAErF,qCAAqC;IACrC,EAAE,EAAE,MAAM,CAAC;IAEX,wCAAwC;IACxC,WAAW,EAAE,MAAM,CAAC;IAEpB,6CAA6C;IAC7C,WAAW,CAAC,EAAE,SAAS,CAAC;IAExB,6CAA6C;IAC7C,YAAY,CAAC,EAAE,UAAU,CAAC;IAE1B,wCAAwC;IACxC,aAAa,CAAC,EAAE,cAAc,CAAC;IAE/B,uCAAuC;IACvC,YAAY,CAAC,EAAE,aAAa,CAAC;IAE7B;;;;;OAKG;IACH,OAAO,CAAC,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAC;IAEhG,4DAA4D;IAC5D,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE,OAAO,CAAC;IAE1B;;;;;;;;;;;;;OAaG;gBACS,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC;CA8F7F;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2EG;AACH,wBAAgB,UAAU,CACxB,SAAS,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACvD,UAAU,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACxD,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG,EACzC,QAAQ,SAAS,oBAAoB,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,oBAAoB,CACzF,cAAc,EACd,aAAa,CACd,EACD,QAAQ,SAAS,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC,SAAS,CAAC,GAAG,UAAU,CACjH,SAAS,EACT,UAAU,EACV,cAAc,EACd,aAAa,EACb,QAAQ,CACT,CAAC,SAAS,CAAC,EAEZ,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC,GAAG;IACjF,OAAO,CAAC,EAAE,QAAQ,CAAC;CACpB,GACA,CAAC,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,aAAa,EAAE,QAAQ,CAAC,GACjF,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC,GAAG;IACrE,WAAW,EAAE,SAAS,CAAC;IACvB,YAAY,EAAE,UAAU,CAAC;IACzB,OAAO,EAAE,CACP,SAAS,EAAE,SAAS,SAAS,aAAa,GAAG,kBAAkB,CAAC,SAAS,CAAC,GAAG,OAAO,EACpF,OAAO,CAAC,EAAE,QAAQ,KACf,OAAO,CAAC,UAAU,SAAS,aAAa,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,CAAC;CAC3F,GACD,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,CAAC,CAEvE"}
|
package/dist/tools/types.d.ts
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import type { WritableStream } from 'stream/web';
|
|
1
2
|
import type { Tool, ToolV5, FlexibleSchema, ToolCallOptions, ToolExecutionOptions, Schema } from '@internal/external-types';
|
|
2
3
|
import type { RequestHandlerExtra } from '@modelcontextprotocol/sdk/shared/protocol.js';
|
|
3
4
|
import type { ElicitRequest, ElicitResult } from '@modelcontextprotocol/sdk/types.js';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/tools/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,IAAI,EACJ,MAAM,EACN,cAAc,EACd,eAAe,EACf,oBAAoB,EACpB,MAAM,EACP,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;AACxF,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AAEtF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAC7E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,MAAM,MAAM,UAAU,GAAG,IAAI,CAAC;AAC9B,MAAM,MAAM,YAAY,GAAG,MAAM,CAAC;AAElC,MAAM,MAAM,qBAAqB,GAAG,oBAAoB,GAAG,eAAe,CAAC;AAE3E;;GAEG;AAEH,MAAM,WAAW,yBAAyB,CACxC,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG;IAGzC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,GAAG,EAAE,CAAC;IAChB,OAAO,EAAE,CAAC,cAAc,EAAE,kBAAkB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAG9E,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,UAAU,CAAC,EAAE,kBAAkB,CAAC,aAAa,CAAC,CAAC;IAG/C,cAAc,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC;CACtC;AAGD,MAAM,WAAW,4BAA4B,CAC3C,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG;IAGzC,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,GAAG,CAAC;IACX,QAAQ,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAC/B,OAAO,EAAE,CAAC,cAAc,EAAE,kBAAkB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAG9E,UAAU,CAAC,EAAE,kBAAkB,CAAC,aAAa,CAAC,CAAC;CAChD;AAGD,MAAM,WAAW,uBAAuB;IACtC,gDAAgD;IAChD,KAAK,EAAE,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACrC,2EAA2E;IAC3E,WAAW,EAAE;QACX,WAAW,EAAE,CAAC,OAAO,EAAE,aAAa,CAAC,QAAQ,CAAC,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;KAC1E,CAAC;CACH;AAED;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,2BAA2B,GAAG,qBAAqB,GAAG;IAChE,OAAO,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAChD,UAAU,CAAC,EAAE,GAAG,CAAC;IACjB,cAAc,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IACvD,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;;OAGG;IACH,GAAG,CAAC,EAAE,uBAAuB,CAAC;CAC/B,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,WAAW,GAAG,OAAO,GAAG,UAAU,CAAC;AAG/C,MAAM,WAAW,iBAAiB;IAChC;;;;OAIG;IACH,QAAQ,CAAC,EAAE,WAAW,CAAC;CACxB;AAED;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,QAAQ,GAAG;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;IACzC,YAAY,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;IAC5C,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC9E;;;OAGG;IACH,GAAG,CAAC,EAAE,iBAAiB,CAAC;CACzB,GAAG,CACA;IACE,IAAI,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;IAC9B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,EAAE,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CACJ,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC9E;;;OAGG;IACH,GAAG,CAAC,EAAE,iBAAiB,CAAC;CACzB,GAAG,CACA;IACE,IAAI,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;IAC9B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,EAAE,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CACJ,CAAC;AAGF,MAAM,WAAW,oBAAoB,CACnC,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG;IAGzC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,WAAW,CAAC,EAAE,WAAW,CAAC;IAI1B,MAAM,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAKzB,KAAK,CAAC,EAAE,yBAAyB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IAGjE,QAAQ,CAAC,EAAE,4BAA4B,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IAGvE,GAAG,CAAC,EAAE,uBAAuB,CAAC;CAC/B;AAED,MAAM,WAAW,UAAU,CACzB,SAAS,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACvD,UAAU,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACxD,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG,EACzC,QAAQ,SAAS,oBAAoB,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,oBAAoB,CACzF,cAAc,EACd,aAAa,CACd;IAED,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,SAAS,CAAC;IACxB,YAAY,CAAC,EAAE,UAAU,CAAC;IAC1B,aAAa,CAAC,EAAE,cAAc,CAAC;IAC/B,YAAY,CAAC,EAAE,aAAa,CAAC;IAM7B,OAAO,CAAC,EAAE,CACR,SAAS,EAAE,SAAS,SAAS,aAAa,GAAG,kBAAkB,CAAC,SAAS,CAAC,GAAG,OAAO,EACpF,OAAO,CAAC,EAAE,QAAQ,KACf,OAAO,CAAC,CAAC,UAAU,SAAS,aAAa,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;IAC1G,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,eAAe,KAAK,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACtE,YAAY,CAAC,EAAE,CACb,OAAO,EAAE;QACP,cAAc,EAAE,MAAM,CAAC;KACxB,GAAG,eAAe,KAChB,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC9B,gBAAgB,CAAC,EAAE,CACjB,OAAO,EAAE;QACP,KAAK,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;KACtC,GAAG,eAAe,KAChB,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;CAC/B"}
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/tools/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EACV,IAAI,EACJ,MAAM,EACN,cAAc,EACd,eAAe,EACf,oBAAoB,EACpB,MAAM,EACP,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;AACxF,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AAEtF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAC7E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,MAAM,MAAM,UAAU,GAAG,IAAI,CAAC;AAC9B,MAAM,MAAM,YAAY,GAAG,MAAM,CAAC;AAElC,MAAM,MAAM,qBAAqB,GAAG,oBAAoB,GAAG,eAAe,CAAC;AAE3E;;GAEG;AAEH,MAAM,WAAW,yBAAyB,CACxC,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG;IAGzC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,GAAG,EAAE,CAAC;IAChB,OAAO,EAAE,CAAC,cAAc,EAAE,kBAAkB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAG9E,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,UAAU,CAAC,EAAE,kBAAkB,CAAC,aAAa,CAAC,CAAC;IAG/C,cAAc,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC;CACtC;AAGD,MAAM,WAAW,4BAA4B,CAC3C,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG;IAGzC,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,GAAG,CAAC;IACX,QAAQ,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAC/B,OAAO,EAAE,CAAC,cAAc,EAAE,kBAAkB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAG9E,UAAU,CAAC,EAAE,kBAAkB,CAAC,aAAa,CAAC,CAAC;CAChD;AAGD,MAAM,WAAW,uBAAuB;IACtC,gDAAgD;IAChD,KAAK,EAAE,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACrC,2EAA2E;IAC3E,WAAW,EAAE;QACX,WAAW,EAAE,CAAC,OAAO,EAAE,aAAa,CAAC,QAAQ,CAAC,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;KAC1E,CAAC;CACH;AAED;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,2BAA2B,GAAG,qBAAqB,GAAG;IAChE,OAAO,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAChD,UAAU,CAAC,EAAE,GAAG,CAAC;IACjB,cAAc,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IACvD,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;;OAGG;IACH,GAAG,CAAC,EAAE,uBAAuB,CAAC;CAC/B,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,WAAW,GAAG,OAAO,GAAG,UAAU,CAAC;AAG/C,MAAM,WAAW,iBAAiB;IAChC;;;;OAIG;IACH,QAAQ,CAAC,EAAE,WAAW,CAAC;CACxB;AAED;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,QAAQ,GAAG;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;IACzC,YAAY,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;IAC5C,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC9E;;;OAGG;IACH,GAAG,CAAC,EAAE,iBAAiB,CAAC;CACzB,GAAG,CACA;IACE,IAAI,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;IAC9B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,EAAE,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CACJ,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,2BAA2B,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC9E;;;OAGG;IACH,GAAG,CAAC,EAAE,iBAAiB,CAAC;CACzB,GAAG,CACA;IACE,IAAI,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;IAC9B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,EAAE,EAAE,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CACJ,CAAC;AAGF,MAAM,WAAW,oBAAoB,CACnC,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG;IAGzC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,WAAW,CAAC,EAAE,WAAW,CAAC;IAI1B,MAAM,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAKzB,KAAK,CAAC,EAAE,yBAAyB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IAGjE,QAAQ,CAAC,EAAE,4BAA4B,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IAGvE,GAAG,CAAC,EAAE,uBAAuB,CAAC;CAC/B;AAED,MAAM,WAAW,UAAU,CACzB,SAAS,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACvD,UAAU,SAAS,aAAa,GAAG,SAAS,GAAG,SAAS,EACxD,cAAc,SAAS,aAAa,GAAG,GAAG,EAC1C,aAAa,SAAS,aAAa,GAAG,GAAG,EACzC,QAAQ,SAAS,oBAAoB,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,oBAAoB,CACzF,cAAc,EACd,aAAa,CACd;IAED,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,SAAS,CAAC;IACxB,YAAY,CAAC,EAAE,UAAU,CAAC;IAC1B,aAAa,CAAC,EAAE,cAAc,CAAC;IAC/B,YAAY,CAAC,EAAE,aAAa,CAAC;IAM7B,OAAO,CAAC,EAAE,CACR,SAAS,EAAE,SAAS,SAAS,aAAa,GAAG,kBAAkB,CAAC,SAAS,CAAC,GAAG,OAAO,EACpF,OAAO,CAAC,EAAE,QAAQ,KACf,OAAO,CAAC,CAAC,UAAU,SAAS,aAAa,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;IAC1G,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,eAAe,KAAK,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACtE,YAAY,CAAC,EAAE,CACb,OAAO,EAAE;QACP,cAAc,EAAE,MAAM,CAAC;KACxB,GAAG,eAAe,KAChB,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC9B,gBAAgB,CAAC,EAAE,CACjB,OAAO,EAAE;QACP,KAAK,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;KACtC,GAAG,eAAe,KAChB,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;CAC/B"}
|
|
@@ -17,4 +17,16 @@ export declare function validateToolInput<T = any>(schema: ZodLikeSchema | undef
|
|
|
17
17
|
data: T | unknown;
|
|
18
18
|
error?: ValidationError<T>;
|
|
19
19
|
};
|
|
20
|
+
/**
|
|
21
|
+
* Validates tool output data against a Zod schema.
|
|
22
|
+
*
|
|
23
|
+
* @param schema The Zod schema to validate against
|
|
24
|
+
* @param output The output data to validate
|
|
25
|
+
* @param toolId Optional tool ID for better error messages
|
|
26
|
+
* @returns The validated data or a validation error
|
|
27
|
+
*/
|
|
28
|
+
export declare function validateToolOutput<T = any>(schema: ZodLikeSchema | undefined, output: unknown, toolId?: string): {
|
|
29
|
+
data: T | unknown;
|
|
30
|
+
error?: ValidationError<T>;
|
|
31
|
+
};
|
|
20
32
|
//# sourceMappingURL=validation.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"validation.d.ts","sourceRoot":"","sources":["../../src/tools/validation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEzD,MAAM,WAAW,eAAe,CAAC,CAAC,GAAG,GAAG;IACtC,KAAK,EAAE,IAAI,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;CAC1C;
|
|
1
|
+
{"version":3,"file":"validation.d.ts","sourceRoot":"","sources":["../../src/tools/validation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEzD,MAAM,WAAW,eAAe,CAAC,CAAC,GAAG,GAAG;IACtC,KAAK,EAAE,IAAI,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;CAC1C;AAoBD;;;;;;;GAOG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,GAAG,GAAG,EACvC,MAAM,EAAE,aAAa,GAAG,SAAS,EACjC,KAAK,EAAE,OAAO,EACd,MAAM,CAAC,EAAE,MAAM,GACd;IAAE,IAAI,EAAE,CAAC,GAAG,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,CAAA;CAAE,CAyBnD;AAED;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAAC,CAAC,GAAG,GAAG,EACxC,MAAM,EAAE,aAAa,GAAG,SAAS,EACjC,MAAM,EAAE,OAAO,EACf,MAAM,CAAC,EAAE,MAAM,GACd;IAAE,IAAI,EAAE,CAAC,GAAG,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,CAAA;CAAE,CAyBnD"}
|
package/dist/utils.cjs
CHANGED
|
@@ -1,88 +1,92 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunkHBJPYQRN_cjs = require('./chunk-HBJPYQRN.cjs');
|
|
4
4
|
|
|
5
5
|
|
|
6
6
|
|
|
7
7
|
Object.defineProperty(exports, "checkEvalStorageFields", {
|
|
8
8
|
enumerable: true,
|
|
9
|
-
get: function () { return
|
|
9
|
+
get: function () { return chunkHBJPYQRN_cjs.checkEvalStorageFields; }
|
|
10
10
|
});
|
|
11
11
|
Object.defineProperty(exports, "createMastraProxy", {
|
|
12
12
|
enumerable: true,
|
|
13
|
-
get: function () { return
|
|
13
|
+
get: function () { return chunkHBJPYQRN_cjs.createMastraProxy; }
|
|
14
14
|
});
|
|
15
15
|
Object.defineProperty(exports, "deepMerge", {
|
|
16
16
|
enumerable: true,
|
|
17
|
-
get: function () { return
|
|
17
|
+
get: function () { return chunkHBJPYQRN_cjs.deepMerge; }
|
|
18
18
|
});
|
|
19
19
|
Object.defineProperty(exports, "delay", {
|
|
20
20
|
enumerable: true,
|
|
21
|
-
get: function () { return
|
|
21
|
+
get: function () { return chunkHBJPYQRN_cjs.delay; }
|
|
22
22
|
});
|
|
23
23
|
Object.defineProperty(exports, "ensureToolProperties", {
|
|
24
24
|
enumerable: true,
|
|
25
|
-
get: function () { return
|
|
25
|
+
get: function () { return chunkHBJPYQRN_cjs.ensureToolProperties; }
|
|
26
26
|
});
|
|
27
27
|
Object.defineProperty(exports, "fetchWithRetry", {
|
|
28
28
|
enumerable: true,
|
|
29
|
-
get: function () { return
|
|
29
|
+
get: function () { return chunkHBJPYQRN_cjs.fetchWithRetry; }
|
|
30
30
|
});
|
|
31
31
|
Object.defineProperty(exports, "generateEmptyFromSchema", {
|
|
32
32
|
enumerable: true,
|
|
33
|
-
get: function () { return
|
|
33
|
+
get: function () { return chunkHBJPYQRN_cjs.generateEmptyFromSchema; }
|
|
34
34
|
});
|
|
35
35
|
Object.defineProperty(exports, "getNestedValue", {
|
|
36
36
|
enumerable: true,
|
|
37
|
-
get: function () { return
|
|
37
|
+
get: function () { return chunkHBJPYQRN_cjs.getNestedValue; }
|
|
38
38
|
});
|
|
39
39
|
Object.defineProperty(exports, "isCoreMessage", {
|
|
40
40
|
enumerable: true,
|
|
41
|
-
get: function () { return
|
|
41
|
+
get: function () { return chunkHBJPYQRN_cjs.isCoreMessage; }
|
|
42
42
|
});
|
|
43
43
|
Object.defineProperty(exports, "isUiMessage", {
|
|
44
44
|
enumerable: true,
|
|
45
|
-
get: function () { return
|
|
45
|
+
get: function () { return chunkHBJPYQRN_cjs.isUiMessage; }
|
|
46
46
|
});
|
|
47
47
|
Object.defineProperty(exports, "isZodType", {
|
|
48
48
|
enumerable: true,
|
|
49
|
-
get: function () { return
|
|
49
|
+
get: function () { return chunkHBJPYQRN_cjs.isZodType; }
|
|
50
50
|
});
|
|
51
51
|
Object.defineProperty(exports, "makeCoreTool", {
|
|
52
52
|
enumerable: true,
|
|
53
|
-
get: function () { return
|
|
53
|
+
get: function () { return chunkHBJPYQRN_cjs.makeCoreTool; }
|
|
54
54
|
});
|
|
55
55
|
Object.defineProperty(exports, "makeCoreToolV5", {
|
|
56
56
|
enumerable: true,
|
|
57
|
-
get: function () { return
|
|
57
|
+
get: function () { return chunkHBJPYQRN_cjs.makeCoreToolV5; }
|
|
58
58
|
});
|
|
59
59
|
Object.defineProperty(exports, "maskStreamTags", {
|
|
60
60
|
enumerable: true,
|
|
61
|
-
get: function () { return
|
|
61
|
+
get: function () { return chunkHBJPYQRN_cjs.maskStreamTags; }
|
|
62
62
|
});
|
|
63
63
|
Object.defineProperty(exports, "omitKeys", {
|
|
64
64
|
enumerable: true,
|
|
65
|
-
get: function () { return
|
|
65
|
+
get: function () { return chunkHBJPYQRN_cjs.omitKeys; }
|
|
66
66
|
});
|
|
67
67
|
Object.defineProperty(exports, "parseFieldKey", {
|
|
68
68
|
enumerable: true,
|
|
69
|
-
get: function () { return
|
|
69
|
+
get: function () { return chunkHBJPYQRN_cjs.parseFieldKey; }
|
|
70
70
|
});
|
|
71
71
|
Object.defineProperty(exports, "parseSqlIdentifier", {
|
|
72
72
|
enumerable: true,
|
|
73
|
-
get: function () { return
|
|
73
|
+
get: function () { return chunkHBJPYQRN_cjs.parseSqlIdentifier; }
|
|
74
|
+
});
|
|
75
|
+
Object.defineProperty(exports, "removeUndefinedValues", {
|
|
76
|
+
enumerable: true,
|
|
77
|
+
get: function () { return chunkHBJPYQRN_cjs.removeUndefinedValues; }
|
|
74
78
|
});
|
|
75
79
|
Object.defineProperty(exports, "resolveSerializedZodOutput", {
|
|
76
80
|
enumerable: true,
|
|
77
|
-
get: function () { return
|
|
81
|
+
get: function () { return chunkHBJPYQRN_cjs.resolveSerializedZodOutput; }
|
|
78
82
|
});
|
|
79
83
|
Object.defineProperty(exports, "selectFields", {
|
|
80
84
|
enumerable: true,
|
|
81
|
-
get: function () { return
|
|
85
|
+
get: function () { return chunkHBJPYQRN_cjs.selectFields; }
|
|
82
86
|
});
|
|
83
87
|
Object.defineProperty(exports, "setNestedValue", {
|
|
84
88
|
enumerable: true,
|
|
85
|
-
get: function () { return
|
|
89
|
+
get: function () { return chunkHBJPYQRN_cjs.setNestedValue; }
|
|
86
90
|
});
|
|
87
91
|
//# sourceMappingURL=utils.cjs.map
|
|
88
92
|
//# sourceMappingURL=utils.cjs.map
|