@mastra/core 1.0.0-beta.1 → 1.0.0-beta.11
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 +1218 -0
- package/dist/agent/agent-legacy.d.ts +35 -30
- package/dist/agent/agent-legacy.d.ts.map +1 -1
- package/dist/agent/agent-types.test-d.d.ts +2 -0
- package/dist/agent/agent-types.test-d.d.ts.map +1 -0
- package/dist/agent/agent.d.ts +24 -9
- package/dist/agent/agent.d.ts.map +1 -1
- package/dist/agent/agent.types.d.ts +17 -9
- 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 +61 -3
- 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/attachments-to-parts.d.ts.map +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/save-queue/index.d.ts +3 -1
- package/dist/agent/save-queue/index.d.ts.map +1 -1
- package/dist/agent/test-utils.d.ts +37 -0
- package/dist/agent/test-utils.d.ts.map +1 -1
- package/dist/agent/trip-wire.d.ts +38 -4
- package/dist/agent/trip-wire.d.ts.map +1 -1
- package/dist/agent/types.d.ts +46 -14
- package/dist/agent/types.d.ts.map +1 -1
- package/dist/agent/utils.d.ts +4 -5
- package/dist/agent/utils.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/index.d.ts +37 -13
- package/dist/agent/workflows/prepare-stream/index.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/map-results-step.d.ts +3 -3
- 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 +34 -11
- 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/schema.d.ts +38 -15
- package/dist/agent/workflows/prepare-stream/schema.d.ts.map +1 -1
- package/dist/agent/workflows/prepare-stream/stream-step.d.ts +11 -1
- package/dist/agent/workflows/prepare-stream/stream-step.d.ts.map +1 -1
- package/dist/base.cjs +2 -2
- package/dist/base.js +1 -1
- package/dist/bundler/index.cjs +2 -2
- package/dist/bundler/index.js +1 -1
- package/dist/cache/index.cjs +3 -3
- package/dist/cache/index.js +1 -1
- package/dist/{chunk-JUBFO5J3.js → chunk-2ULLRN4Y.js} +16333 -11839
- package/dist/chunk-2ULLRN4Y.js.map +1 -0
- package/dist/{chunk-OJNJA5ZI.js → chunk-3CKZSDTQ.js} +3 -3
- package/dist/{chunk-OJNJA5ZI.js.map → chunk-3CKZSDTQ.js.map} +1 -1
- package/dist/chunk-3E3ILV6T.cjs +518 -0
- package/dist/chunk-3E3ILV6T.cjs.map +1 -0
- package/dist/{chunk-TQTAMPSC.js → chunk-3N3QE4ZK.js} +3 -3
- package/dist/chunk-3N3QE4ZK.js.map +1 -0
- package/dist/{chunk-QUKUN6NR.cjs → chunk-4JKEUSCC.cjs} +107 -7
- package/dist/chunk-4JKEUSCC.cjs.map +1 -0
- package/dist/{chunk-MSWTA73A.cjs → chunk-52RSUALV.cjs} +16490 -11957
- package/dist/chunk-52RSUALV.cjs.map +1 -0
- package/dist/chunk-5PAEYE3Q.js +513 -0
- package/dist/chunk-5PAEYE3Q.js.map +1 -0
- package/dist/{chunk-ECFXGXWO.cjs → chunk-5Q6WAYEY.cjs} +695 -375
- package/dist/chunk-5Q6WAYEY.cjs.map +1 -0
- package/dist/{chunk-CBAB7GOD.cjs → chunk-6BJ7XHRH.cjs} +2 -2
- package/dist/{chunk-CBAB7GOD.cjs.map → chunk-6BJ7XHRH.cjs.map} +1 -1
- package/dist/{chunk-GRBGQ2GE.js → chunk-6TBWJV35.js} +5 -3
- package/dist/chunk-6TBWJV35.js.map +1 -0
- package/dist/chunk-6XCINXZ7.cjs +194 -0
- package/dist/chunk-6XCINXZ7.cjs.map +1 -0
- package/dist/{chunk-E7K4FTLN.cjs → chunk-72YCRTEX.cjs} +18 -2
- package/dist/chunk-72YCRTEX.cjs.map +1 -0
- package/dist/{chunk-W7UH2PWL.js → chunk-7P6BNIJH.js} +1142 -331
- package/dist/chunk-7P6BNIJH.js.map +1 -0
- package/dist/{chunk-76K3IYWM.js → chunk-ADADLFDR.js} +3 -3
- package/dist/{chunk-76K3IYWM.js.map → chunk-ADADLFDR.js.map} +1 -1
- package/dist/{chunk-DSNPWVIG.cjs → chunk-AF74UXR5.cjs} +4 -3
- package/dist/chunk-AF74UXR5.cjs.map +1 -0
- package/dist/{chunk-GGYKYORQ.cjs → chunk-BJXKH4LG.cjs} +407 -22
- package/dist/chunk-BJXKH4LG.cjs.map +1 -0
- package/dist/{chunk-2ZVKF4HP.cjs → chunk-BUKY6CTR.cjs} +1147 -334
- package/dist/chunk-BUKY6CTR.cjs.map +1 -0
- package/dist/chunk-BWYU7D33.js +192 -0
- package/dist/chunk-BWYU7D33.js.map +1 -0
- package/dist/{chunk-XEVG546F.js → chunk-C36YRTZ6.js} +178 -12
- package/dist/chunk-C36YRTZ6.js.map +1 -0
- package/dist/{chunk-CB575O6L.cjs → chunk-CZEJQSWB.cjs} +473 -124
- package/dist/chunk-CZEJQSWB.cjs.map +1 -0
- package/dist/chunk-DD2VNRQM.js +62 -0
- package/dist/chunk-DD2VNRQM.js.map +1 -0
- package/dist/{chunk-KEXGB7FK.cjs → chunk-DGV2FWB4.cjs} +7 -7
- package/dist/{chunk-KEXGB7FK.cjs.map → chunk-DGV2FWB4.cjs.map} +1 -1
- package/dist/{chunk-Y63IFHEZ.cjs → chunk-EOPEDKLG.cjs} +4 -4
- package/dist/chunk-EOPEDKLG.cjs.map +1 -0
- package/dist/{chunk-UIZSWUKP.js → chunk-F2GAJSBI.js} +105 -7
- package/dist/chunk-F2GAJSBI.js.map +1 -0
- package/dist/{chunk-4CDL2QJT.js → chunk-IVV5TOMD.js} +122 -37
- package/dist/chunk-IVV5TOMD.js.map +1 -0
- package/dist/{chunk-MV7KHWUT.js → chunk-JIGDJK2O.js} +391 -15
- package/dist/chunk-JIGDJK2O.js.map +1 -0
- package/dist/chunk-KJBMTK5B.cjs +64 -0
- package/dist/chunk-KJBMTK5B.cjs.map +1 -0
- package/dist/{chunk-S6URFGCZ.js → chunk-LSHPJWM5.js} +3 -3
- package/dist/{chunk-S6URFGCZ.js.map → chunk-LSHPJWM5.js.map} +1 -1
- package/dist/chunk-MGCGWPQJ.cjs +275 -0
- package/dist/chunk-MGCGWPQJ.cjs.map +1 -0
- package/dist/chunk-MRFUISXC.cjs +4981 -0
- package/dist/chunk-MRFUISXC.cjs.map +1 -0
- package/dist/{chunk-KJ2SW6VA.js → chunk-NRUZYMHE.js} +4 -3
- package/dist/chunk-NRUZYMHE.js.map +1 -0
- package/dist/{chunk-CINNK34N.js → chunk-O2BJW7YA.js} +26 -11
- package/dist/chunk-O2BJW7YA.js.map +1 -0
- package/dist/chunk-OEIVMCWX.js +4959 -0
- package/dist/chunk-OEIVMCWX.js.map +1 -0
- package/dist/{chunk-WM6CK2F3.cjs → chunk-PG5H6QIO.cjs} +131 -49
- package/dist/chunk-PG5H6QIO.cjs.map +1 -0
- package/dist/{chunk-JPGVRWWL.js → chunk-PK2A5WBG.js} +472 -125
- package/dist/chunk-PK2A5WBG.js.map +1 -0
- package/dist/{chunk-X7JMA3IY.js → chunk-PSCMWPLC.js} +2 -2
- package/dist/{chunk-X7JMA3IY.js.map → chunk-PSCMWPLC.js.map} +1 -1
- package/dist/{chunk-VOY2RXOC.cjs → chunk-S73Z3PBJ.cjs} +182 -15
- package/dist/chunk-S73Z3PBJ.cjs.map +1 -0
- package/dist/{chunk-MA7TEM62.cjs → chunk-SCUWP4II.cjs} +769 -27
- package/dist/chunk-SCUWP4II.cjs.map +1 -0
- package/dist/{chunk-JV2KH24V.js → chunk-SXNQRJQD.js} +695 -375
- package/dist/chunk-SXNQRJQD.js.map +1 -0
- package/dist/{chunk-BMAFVZ2D.cjs → chunk-T2VLUFGG.cjs} +4 -4
- package/dist/{chunk-BMAFVZ2D.cjs.map → chunk-T2VLUFGG.cjs.map} +1 -1
- package/dist/chunk-THZTRBFS.js +268 -0
- package/dist/chunk-THZTRBFS.js.map +1 -0
- package/dist/{chunk-VOQ3ULMT.js → chunk-U3XOLEPX.js} +759 -17
- package/dist/chunk-U3XOLEPX.js.map +1 -0
- package/dist/{chunk-Y6ROD72V.cjs → chunk-UIGRFDO6.cjs} +4 -4
- package/dist/{chunk-Y6ROD72V.cjs.map → chunk-UIGRFDO6.cjs.map} +1 -1
- package/dist/{chunk-H6CZGPZD.js → chunk-US2U7ECW.js} +179 -38
- package/dist/chunk-US2U7ECW.js.map +1 -0
- package/dist/{chunk-J7O6WENZ.cjs → chunk-UVHSM2GU.cjs} +6 -2
- package/dist/chunk-UVHSM2GU.cjs.map +1 -0
- package/dist/{chunk-CD56CXVE.cjs → chunk-WTSZBHIZ.cjs} +32 -17
- package/dist/chunk-WTSZBHIZ.cjs.map +1 -0
- package/dist/{chunk-KIZIOFZC.js → chunk-Y36Y5MTD.js} +17 -3
- package/dist/chunk-Y36Y5MTD.js.map +1 -0
- package/dist/{chunk-DNEURYF3.cjs → chunk-YC6PJEPH.cjs} +206 -65
- package/dist/chunk-YC6PJEPH.cjs.map +1 -0
- package/dist/{chunk-IQO7ANVS.cjs → chunk-YWMMBIOM.cjs} +21 -20
- package/dist/chunk-YWMMBIOM.cjs.map +1 -0
- package/dist/{chunk-I4CXL4SR.js → chunk-Z57R5WS4.js} +16 -15
- package/dist/chunk-Z57R5WS4.js.map +1 -0
- package/dist/deployer/index.cjs +2 -2
- package/dist/deployer/index.js +1 -1
- package/dist/di/index.cjs +10 -2
- package/dist/di/index.d.ts +1 -1
- package/dist/di/index.d.ts.map +1 -1
- package/dist/di/index.js +1 -1
- package/dist/evals/base.d.ts +10 -2
- package/dist/evals/base.d.ts.map +1 -1
- package/dist/evals/base.test-utils.d.ts +25 -25
- package/dist/evals/index.cjs +9 -9
- package/dist/evals/index.js +2 -2
- package/dist/evals/scoreTraces/index.cjs +7 -6
- package/dist/evals/scoreTraces/index.cjs.map +1 -1
- package/dist/evals/scoreTraces/index.js +4 -3
- package/dist/evals/scoreTraces/index.js.map +1 -1
- package/dist/evals/scoreTraces/scoreTracesWorkflow.d.ts.map +1 -1
- package/dist/evals/types.d.ts +25 -24
- package/dist/evals/types.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 +30 -10
- package/dist/llm/index.d.ts +2 -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/base.types.d.ts +2 -2
- package/dist/llm/model/base.types.d.ts.map +1 -1
- package/dist/llm/model/gateway-resolver.d.ts.map +1 -1
- package/dist/llm/model/gateways/azure.d.ts +36 -0
- package/dist/llm/model/gateways/azure.d.ts.map +1 -0
- package/dist/llm/model/gateways/base.d.ts +9 -4
- package/dist/llm/model/gateways/base.d.ts.map +1 -1
- package/dist/llm/model/gateways/constants.d.ts.map +1 -1
- package/dist/llm/model/gateways/index.d.ts +4 -1
- package/dist/llm/model/gateways/index.d.ts.map +1 -1
- package/dist/llm/model/gateways/models-dev.d.ts +3 -2
- package/dist/llm/model/gateways/models-dev.d.ts.map +1 -1
- package/dist/llm/model/gateways/netlify.d.ts +4 -3
- 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.d.ts.map +1 -1
- 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 +4 -2
- package/dist/llm/model/model.loop.types.d.ts.map +1 -1
- package/dist/llm/model/provider-options.d.ts +4 -1
- package/dist/llm/model/provider-options.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 +489 -27
- package/dist/llm/model/registry-generator.d.ts +12 -0
- 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 +17 -6
- package/dist/llm/model/shared.types.d.ts.map +1 -1
- package/dist/logger/constants.d.ts +1 -0
- package/dist/logger/constants.d.ts.map +1 -1
- package/dist/logger/index.cjs +11 -11
- package/dist/logger/index.js +2 -2
- package/dist/logger/transport.d.ts +1 -1
- package/dist/logger/transport.d.ts.map +1 -1
- package/dist/loop/index.cjs +2 -2
- package/dist/loop/index.js +1 -1
- package/dist/loop/loop.d.ts.map +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 +42 -20
- package/dist/loop/types.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-execution/index.d.ts +60 -42
- package/dist/loop/workflows/agentic-execution/index.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-execution/llm-execution-step.d.ts +41 -29
- 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 +20 -14
- package/dist/loop/workflows/agentic-execution/llm-mapping-step.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-execution/tool-call-step.d.ts +1 -1
- package/dist/loop/workflows/agentic-execution/tool-call-step.d.ts.map +1 -1
- package/dist/loop/workflows/agentic-loop/index.d.ts +62 -43
- 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 +32 -15
- package/dist/loop/workflows/schema.d.ts.map +1 -1
- package/dist/loop/workflows/stream.d.ts +1 -1
- package/dist/loop/workflows/stream.d.ts.map +1 -1
- package/dist/mastra/index.cjs +2 -2
- package/dist/mastra/index.d.ts +377 -12
- package/dist/mastra/index.d.ts.map +1 -1
- package/dist/mastra/index.js +1 -1
- package/dist/mcp/index.cjs +4 -4
- package/dist/mcp/index.cjs.map +1 -1
- package/dist/mcp/index.d.ts +4 -4
- package/dist/mcp/index.d.ts.map +1 -1
- package/dist/mcp/index.js +2 -2
- package/dist/mcp/index.js.map +1 -1
- package/dist/mcp/types.d.ts +2 -2
- package/dist/mcp/types.d.ts.map +1 -1
- package/dist/memory/index.cjs +21 -327
- package/dist/memory/index.cjs.map +1 -1
- package/dist/memory/index.js +1 -327
- package/dist/memory/index.js.map +1 -1
- package/dist/memory/memory.d.ts +28 -18
- package/dist/memory/memory.d.ts.map +1 -1
- package/dist/memory/mock.d.ts +15 -14
- package/dist/memory/mock.d.ts.map +1 -1
- package/dist/memory/types.d.ts +69 -11
- package/dist/memory/types.d.ts.map +1 -1
- package/dist/models-dev-23RN2WHG.js +3 -0
- package/dist/{models-dev-DNBKXHT4.js.map → models-dev-23RN2WHG.js.map} +1 -1
- package/dist/models-dev-EO3SUIY2.cjs +12 -0
- package/dist/{models-dev-YBEEQIX6.cjs.map → models-dev-EO3SUIY2.cjs.map} +1 -1
- package/dist/netlify-GXJ5D5DD.js +3 -0
- package/dist/{netlify-7G2L5VSH.js.map → netlify-GXJ5D5DD.js.map} +1 -1
- package/dist/netlify-KJLY3GFS.cjs +12 -0
- package/dist/{netlify-GWNGSIRZ.cjs.map → netlify-KJLY3GFS.cjs.map} +1 -1
- package/dist/observability/index.cjs +16 -8
- package/dist/observability/index.js +1 -1
- package/dist/observability/types/tracing.d.ts +236 -13
- package/dist/observability/types/tracing.d.ts.map +1 -1
- package/dist/observability/utils.d.ts +47 -1
- package/dist/observability/utils.d.ts.map +1 -1
- package/dist/processors/index.cjs +115 -11
- package/dist/processors/index.d.ts +246 -26
- package/dist/processors/index.d.ts.map +1 -1
- package/dist/processors/index.js +1 -1
- package/dist/processors/memory/index.d.ts +7 -0
- package/dist/processors/memory/index.d.ts.map +1 -0
- package/dist/processors/memory/message-history.d.ts +43 -0
- package/dist/processors/memory/message-history.d.ts.map +1 -0
- package/dist/processors/memory/semantic-recall.d.ts +152 -0
- package/dist/processors/memory/semantic-recall.d.ts.map +1 -0
- package/dist/processors/memory/working-memory.d.ts +65 -0
- package/dist/processors/memory/working-memory.d.ts.map +1 -0
- package/dist/processors/processors/batch-parts.d.ts +1 -1
- package/dist/processors/processors/batch-parts.d.ts.map +1 -1
- package/dist/processors/processors/index.d.ts +2 -1
- package/dist/processors/processors/index.d.ts.map +1 -1
- package/dist/processors/processors/language-detector.d.ts +15 -1
- package/dist/processors/processors/language-detector.d.ts.map +1 -1
- package/dist/processors/processors/moderation.d.ts +15 -1
- package/dist/processors/processors/moderation.d.ts.map +1 -1
- package/dist/processors/processors/pii-detector.d.ts +15 -1
- package/dist/processors/processors/pii-detector.d.ts.map +1 -1
- package/dist/processors/processors/prepare-step.d.ts +12 -0
- package/dist/processors/processors/prepare-step.d.ts.map +1 -0
- package/dist/processors/processors/prompt-injection-detector.d.ts +15 -1
- package/dist/processors/processors/prompt-injection-detector.d.ts.map +1 -1
- package/dist/processors/processors/structured-output.d.ts +7 -5
- package/dist/processors/processors/structured-output.d.ts.map +1 -1
- package/dist/processors/processors/system-prompt-scrubber.d.ts +1 -1
- package/dist/processors/processors/system-prompt-scrubber.d.ts.map +1 -1
- package/dist/processors/processors/token-limiter.d.ts +22 -12
- package/dist/processors/processors/token-limiter.d.ts.map +1 -1
- package/dist/processors/processors/tool-call-filter.d.ts +28 -0
- package/dist/processors/processors/tool-call-filter.d.ts.map +1 -0
- package/dist/processors/processors/unicode-normalizer.d.ts +1 -1
- package/dist/processors/processors/unicode-normalizer.d.ts.map +1 -1
- package/dist/processors/runner.d.ts +83 -9
- package/dist/processors/runner.d.ts.map +1 -1
- package/dist/processors/step-schema.d.ts +49731 -0
- package/dist/processors/step-schema.d.ts.map +1 -0
- package/dist/provider-registry-3TG2KUD2.cjs +40 -0
- package/dist/provider-registry-3TG2KUD2.cjs.map +1 -0
- package/dist/provider-registry-F67Y6OF2.js +3 -0
- package/dist/provider-registry-F67Y6OF2.js.map +1 -0
- package/dist/provider-registry.json +1056 -61
- package/dist/{registry-generator-MK63POJO.cjs → registry-generator-34SC4TAU.cjs} +23 -6
- package/dist/registry-generator-34SC4TAU.cjs.map +1 -0
- package/dist/{registry-generator-H4YNODDH.js → registry-generator-UMTNPBJX.js} +23 -7
- package/dist/registry-generator-UMTNPBJX.js.map +1 -0
- package/dist/relevance/index.cjs +2 -2
- package/dist/relevance/index.js +1 -1
- package/dist/request-context/index.cjs +10 -2
- package/dist/request-context/index.d.ts +26 -0
- package/dist/request-context/index.d.ts.map +1 -1
- package/dist/request-context/index.js +1 -1
- package/dist/server/auth.d.ts +11 -0
- package/dist/server/auth.d.ts.map +1 -1
- package/dist/server/base.d.ts +51 -0
- package/dist/server/base.d.ts.map +1 -0
- package/dist/server/composite-auth.d.ts +9 -0
- package/dist/server/composite-auth.d.ts.map +1 -0
- package/dist/server/index.cjs +124 -3
- package/dist/server/index.cjs.map +1 -1
- package/dist/server/index.d.ts +8 -0
- package/dist/server/index.d.ts.map +1 -1
- package/dist/server/index.js +121 -3
- package/dist/server/index.js.map +1 -1
- package/dist/server/simple-auth.d.ts +27 -0
- package/dist/server/simple-auth.d.ts.map +1 -0
- package/dist/server/types.d.ts +6 -0
- package/dist/server/types.d.ts.map +1 -1
- package/dist/storage/base.d.ts +76 -3
- package/dist/storage/base.d.ts.map +1 -1
- package/dist/storage/constants.d.ts +3 -1
- package/dist/storage/constants.d.ts.map +1 -1
- package/dist/storage/domains/agents/base.d.ts +49 -0
- package/dist/storage/domains/agents/base.d.ts.map +1 -0
- package/dist/storage/domains/agents/index.d.ts +3 -0
- package/dist/storage/domains/agents/index.d.ts.map +1 -0
- package/dist/storage/domains/agents/inmemory.d.ts +22 -0
- package/dist/storage/domains/agents/inmemory.d.ts.map +1 -0
- package/dist/storage/domains/index.d.ts +1 -0
- package/dist/storage/domains/index.d.ts.map +1 -1
- package/dist/storage/domains/memory/inmemory.d.ts.map +1 -1
- package/dist/storage/domains/operations/inmemory.d.ts.map +1 -1
- package/dist/storage/domains/scores/base.d.ts +2 -2
- package/dist/storage/domains/scores/base.d.ts.map +1 -1
- package/dist/storage/domains/scores/inmemory.d.ts +2 -2
- package/dist/storage/domains/scores/inmemory.d.ts.map +1 -1
- package/dist/storage/domains/workflows/base.d.ts +4 -0
- package/dist/storage/domains/workflows/base.d.ts.map +1 -1
- package/dist/storage/domains/workflows/inmemory.d.ts +5 -1
- package/dist/storage/domains/workflows/inmemory.d.ts.map +1 -1
- package/dist/storage/index.cjs +65 -29
- package/dist/storage/index.js +1 -1
- package/dist/storage/mock.d.ts +7 -2
- package/dist/storage/mock.d.ts.map +1 -1
- package/dist/storage/storageWithInit.d.ts.map +1 -1
- package/dist/storage/types.d.ts +88 -5
- package/dist/storage/types.d.ts.map +1 -1
- package/dist/storage/utils.d.ts +82 -0
- package/dist/storage/utils.d.ts.map +1 -1
- package/dist/stream/MastraAgentNetworkStream.d.ts +2 -2
- package/dist/stream/MastraAgentNetworkStream.d.ts.map +1 -1
- package/dist/stream/MastraWorkflowStream.d.ts +2 -2
- package/dist/stream/MastraWorkflowStream.d.ts.map +1 -1
- package/dist/stream/RunOutput.d.ts +2 -2
- package/dist/stream/RunOutput.d.ts.map +1 -1
- package/dist/stream/aisdk/v4/transform.d.ts.map +1 -1
- package/dist/stream/aisdk/v4/usage.d.ts +19 -0
- package/dist/stream/aisdk/v4/usage.d.ts.map +1 -0
- package/dist/stream/aisdk/v5/compat/prepare-tools.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/execute.d.ts +9 -5
- package/dist/stream/aisdk/v5/execute.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/input.d.ts +4 -1
- package/dist/stream/aisdk/v5/input.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/output-helpers.d.ts +5 -1
- package/dist/stream/aisdk/v5/output-helpers.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/output.d.ts +11 -13
- package/dist/stream/aisdk/v5/output.d.ts.map +1 -1
- package/dist/stream/aisdk/v5/transform.d.ts +2 -1
- package/dist/stream/aisdk/v5/transform.d.ts.map +1 -1
- package/dist/stream/base/base.d.ts +1 -1
- package/dist/stream/base/base.d.ts.map +1 -1
- package/dist/stream/base/input.d.ts +1 -1
- package/dist/stream/base/output-format-handlers.d.ts +14 -1
- package/dist/stream/base/output-format-handlers.d.ts.map +1 -1
- package/dist/stream/base/output.d.ts +9 -14
- package/dist/stream/base/output.d.ts.map +1 -1
- package/dist/stream/index.cjs +15 -11
- package/dist/stream/index.d.ts +2 -2
- package/dist/stream/index.d.ts.map +1 -1
- package/dist/stream/index.js +2 -2
- package/dist/stream/types.d.ts +97 -32
- package/dist/stream/types.d.ts.map +1 -1
- package/dist/test-utils/llm-mock.cjs +101 -30
- package/dist/test-utils/llm-mock.cjs.map +1 -1
- package/dist/test-utils/llm-mock.d.ts +7 -2
- package/dist/test-utils/llm-mock.d.ts.map +1 -1
- package/dist/test-utils/llm-mock.js +97 -30
- package/dist/test-utils/llm-mock.js.map +1 -1
- package/dist/tools/index.cjs +6 -6
- package/dist/tools/index.js +2 -2
- package/dist/tools/is-vercel-tool.cjs +2 -2
- package/dist/tools/is-vercel-tool.js +1 -1
- package/dist/tools/stream.d.ts +10 -4
- package/dist/tools/stream.d.ts.map +1 -1
- package/dist/tools/tool-builder/builder.d.ts +2 -0
- package/dist/tools/tool-builder/builder.d.ts.map +1 -1
- package/dist/tools/tool-stream-types.test-d.d.ts +2 -0
- package/dist/tools/tool-stream-types.test-d.d.ts.map +1 -0
- package/dist/tools/tool.d.ts +21 -8
- package/dist/tools/tool.d.ts.map +1 -1
- package/dist/tools/types.d.ts +33 -6
- package/dist/tools/types.d.ts.map +1 -1
- package/dist/tools/validation.d.ts +25 -1
- package/dist/tools/validation.d.ts.map +1 -1
- package/dist/tts/index.cjs +2 -2
- package/dist/tts/index.js +1 -1
- package/dist/types/zod-compat.d.ts +14 -1
- package/dist/types/zod-compat.d.ts.map +1 -1
- package/dist/utils.cjs +25 -21
- package/dist/utils.d.ts +9 -4
- package/dist/utils.d.ts.map +1 -1
- package/dist/utils.js +1 -1
- package/dist/vector/filter/index.cjs +7 -189
- package/dist/vector/filter/index.cjs.map +1 -1
- package/dist/vector/filter/index.js +1 -190
- package/dist/vector/filter/index.js.map +1 -1
- package/dist/vector/index.cjs +7 -2
- package/dist/vector/index.cjs.map +1 -1
- package/dist/vector/index.d.ts +1 -0
- package/dist/vector/index.d.ts.map +1 -1
- package/dist/vector/index.js +2 -1
- package/dist/vector/index.js.map +1 -1
- package/dist/vector/types.d.ts +86 -3
- package/dist/vector/types.d.ts.map +1 -1
- package/dist/vector/vector.d.ts +39 -2
- package/dist/vector/vector.d.ts.map +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 +180 -270
- 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/step-executor.d.ts +1 -1
- package/dist/workflows/evented/step-executor.d.ts.map +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 +8 -6
- package/dist/workflows/execution-engine.d.ts.map +1 -1
- package/dist/workflows/handlers/control-flow.d.ts +133 -0
- package/dist/workflows/handlers/control-flow.d.ts.map +1 -0
- package/dist/workflows/handlers/entry.d.ts +43 -0
- package/dist/workflows/handlers/entry.d.ts.map +1 -0
- package/dist/workflows/handlers/sleep.d.ts +60 -0
- package/dist/workflows/handlers/sleep.d.ts.map +1 -0
- package/dist/workflows/handlers/step.d.ts +58 -0
- package/dist/workflows/handlers/step.d.ts.map +1 -0
- package/dist/workflows/index.cjs +40 -16
- package/dist/workflows/index.js +1 -1
- package/dist/workflows/step.d.ts +8 -6
- package/dist/workflows/step.d.ts.map +1 -1
- package/dist/workflows/types.d.ts +257 -10
- package/dist/workflows/types.d.ts.map +1 -1
- package/dist/workflows/utils.d.ts +36 -0
- package/dist/workflows/utils.d.ts.map +1 -1
- package/dist/workflows/workflow.d.ts +146 -55
- package/dist/workflows/workflow.d.ts.map +1 -1
- package/package.json +26 -30
- package/src/llm/model/provider-types.generated.d.ts +489 -27
- package/dist/chunk-2ZVKF4HP.cjs.map +0 -1
- package/dist/chunk-436FFEF6.js +0 -34
- package/dist/chunk-436FFEF6.js.map +0 -1
- package/dist/chunk-4CDL2QJT.js.map +0 -1
- package/dist/chunk-7AHYOMHJ.js +0 -149
- package/dist/chunk-7AHYOMHJ.js.map +0 -1
- package/dist/chunk-CB575O6L.cjs.map +0 -1
- package/dist/chunk-CD56CXVE.cjs.map +0 -1
- package/dist/chunk-CINNK34N.js.map +0 -1
- package/dist/chunk-DNEURYF3.cjs.map +0 -1
- package/dist/chunk-DSNPWVIG.cjs.map +0 -1
- package/dist/chunk-E7K4FTLN.cjs.map +0 -1
- package/dist/chunk-ECFXGXWO.cjs.map +0 -1
- package/dist/chunk-ET6UOTTU.cjs +0 -154
- package/dist/chunk-ET6UOTTU.cjs.map +0 -1
- package/dist/chunk-GGYKYORQ.cjs.map +0 -1
- package/dist/chunk-GRBGQ2GE.js.map +0 -1
- package/dist/chunk-H6CZGPZD.js.map +0 -1
- package/dist/chunk-HDJFSJCK.js +0 -2237
- package/dist/chunk-HDJFSJCK.js.map +0 -1
- package/dist/chunk-HGNRQ3OG.js +0 -58
- package/dist/chunk-HGNRQ3OG.js.map +0 -1
- package/dist/chunk-I4CXL4SR.js.map +0 -1
- package/dist/chunk-IQO7ANVS.cjs.map +0 -1
- package/dist/chunk-J7O6WENZ.cjs.map +0 -1
- package/dist/chunk-JPGVRWWL.js.map +0 -1
- package/dist/chunk-JUBFO5J3.js.map +0 -1
- package/dist/chunk-JV2KH24V.js.map +0 -1
- package/dist/chunk-KIZIOFZC.js.map +0 -1
- package/dist/chunk-KJ2SW6VA.js.map +0 -1
- package/dist/chunk-LJFJTTZQ.cjs +0 -4
- package/dist/chunk-LJFJTTZQ.cjs.map +0 -1
- package/dist/chunk-LRSB62Z6.cjs +0 -60
- package/dist/chunk-LRSB62Z6.cjs.map +0 -1
- package/dist/chunk-LWBQ4P4N.cjs +0 -2240
- package/dist/chunk-LWBQ4P4N.cjs.map +0 -1
- package/dist/chunk-MA7TEM62.cjs.map +0 -1
- package/dist/chunk-MSWTA73A.cjs.map +0 -1
- package/dist/chunk-MV7KHWUT.js.map +0 -1
- package/dist/chunk-QUKUN6NR.cjs.map +0 -1
- package/dist/chunk-TQTAMPSC.js.map +0 -1
- package/dist/chunk-UIZSWUKP.js.map +0 -1
- package/dist/chunk-V3VLOOSW.cjs +0 -36
- package/dist/chunk-V3VLOOSW.cjs.map +0 -1
- package/dist/chunk-VOQ3ULMT.js.map +0 -1
- package/dist/chunk-VOY2RXOC.cjs.map +0 -1
- package/dist/chunk-W7UH2PWL.js.map +0 -1
- package/dist/chunk-WM6CK2F3.cjs.map +0 -1
- package/dist/chunk-XEVG546F.js.map +0 -1
- package/dist/chunk-Y63IFHEZ.cjs.map +0 -1
- package/dist/chunk-ZV5CC35D.js +0 -3
- package/dist/chunk-ZV5CC35D.js.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,11 +1,16 @@
|
|
|
1
|
+
import { simulateReadableStream } from '../ai-sdk.types.d.ts';
|
|
1
2
|
import { MockLanguageModelV1 } from '../ai-sdk.types.d.ts';
|
|
3
|
+
import { MockLanguageModelV2 } from 'ai-v5/test';
|
|
2
4
|
import { MastraLLMV1 } from '../llm/model/model.js';
|
|
3
|
-
|
|
5
|
+
import { MastraLanguageModelV2Mock } from '../loop/test-utils/MastraLanguageModelV2Mock.js';
|
|
6
|
+
export { simulateReadableStream, MastraLanguageModelV2Mock };
|
|
7
|
+
export declare function createMockModel({ objectGenerationMode, mockText, spyGenerate, spyStream, version, }: {
|
|
4
8
|
objectGenerationMode?: 'json';
|
|
5
9
|
mockText: string | Record<string, any>;
|
|
6
10
|
spyGenerate?: (props: any) => void;
|
|
7
11
|
spyStream?: (props: any) => void;
|
|
8
|
-
|
|
12
|
+
version?: 'v1' | 'v2';
|
|
13
|
+
}): MockLanguageModelV1 | MockLanguageModelV2;
|
|
9
14
|
export declare class MockProvider extends MastraLLMV1 {
|
|
10
15
|
constructor({ spyGenerate, spyStream, objectGenerationMode, mockText, }: {
|
|
11
16
|
spyGenerate?: (props: any) => void;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"llm-mock.d.ts","sourceRoot":"","sources":["../../src/test-utils/llm-mock.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"llm-mock.d.ts","sourceRoot":"","sources":["../../src/test-utils/llm-mock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAgC,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAE/E,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,yBAAyB,EAAE,MAAM,8CAA8C,CAAC;AAGzF,OAAO,EAAE,sBAAsB,EAAE,yBAAyB,EAAE,CAAC;AAE7D,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,123 @@
|
|
|
1
|
-
import { MastraLLMV1 } from '../chunk-
|
|
1
|
+
import { AISDKV5LanguageModel, MastraLLMV1 } from '../chunk-C36YRTZ6.js';
|
|
2
2
|
import { MockLanguageModelV1, simulateReadableStream } from '../chunk-QM5SRDJX.js';
|
|
3
|
+
export { simulateReadableStream } from '../chunk-QM5SRDJX.js';
|
|
4
|
+
import { MockLanguageModelV2, convertArrayToReadableStream } from 'ai-v5/test';
|
|
5
|
+
|
|
6
|
+
var MastraLanguageModelV2Mock = class {
|
|
7
|
+
specificationVersion = "v2";
|
|
8
|
+
provider;
|
|
9
|
+
modelId;
|
|
10
|
+
supportedUrls;
|
|
11
|
+
#model;
|
|
12
|
+
constructor(config) {
|
|
13
|
+
this.#model = new MockLanguageModelV2(config);
|
|
14
|
+
this.provider = this.#model.provider;
|
|
15
|
+
this.modelId = this.#model.modelId;
|
|
16
|
+
this.supportedUrls = this.#model.supportedUrls;
|
|
17
|
+
}
|
|
18
|
+
doGenerate(options) {
|
|
19
|
+
const aiSDKModel = new AISDKV5LanguageModel(this.#model);
|
|
20
|
+
return aiSDKModel.doGenerate(options);
|
|
21
|
+
}
|
|
22
|
+
doStream(options) {
|
|
23
|
+
const aiSDKModel = new AISDKV5LanguageModel(this.#model);
|
|
24
|
+
return aiSDKModel.doStream(options);
|
|
25
|
+
}
|
|
26
|
+
get doGenerateCalls() {
|
|
27
|
+
return this.#model.doGenerateCalls;
|
|
28
|
+
}
|
|
29
|
+
get doStreamCalls() {
|
|
30
|
+
return this.#model.doStreamCalls;
|
|
31
|
+
}
|
|
32
|
+
};
|
|
3
33
|
|
|
4
34
|
// src/test-utils/llm-mock.ts
|
|
5
35
|
function createMockModel({
|
|
6
36
|
objectGenerationMode,
|
|
7
37
|
mockText,
|
|
8
38
|
spyGenerate,
|
|
9
|
-
spyStream
|
|
39
|
+
spyStream,
|
|
40
|
+
version = "v2"
|
|
10
41
|
}) {
|
|
11
|
-
const
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
42
|
+
const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
|
|
43
|
+
const finalText = objectGenerationMode === "json" ? JSON.stringify(mockText) : text;
|
|
44
|
+
if (version === "v1") {
|
|
45
|
+
const mockModel2 = new MockLanguageModelV1({
|
|
46
|
+
defaultObjectGenerationMode: objectGenerationMode,
|
|
47
|
+
doGenerate: async (props) => {
|
|
48
|
+
if (spyGenerate) {
|
|
49
|
+
spyGenerate(props);
|
|
50
|
+
}
|
|
18
51
|
return {
|
|
19
52
|
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
20
53
|
finishReason: "stop",
|
|
21
54
|
usage: { promptTokens: 10, completionTokens: 20 },
|
|
22
|
-
text:
|
|
55
|
+
text: finalText
|
|
56
|
+
};
|
|
57
|
+
},
|
|
58
|
+
doStream: async (props) => {
|
|
59
|
+
if (spyStream) {
|
|
60
|
+
spyStream(props);
|
|
61
|
+
}
|
|
62
|
+
const chunks = finalText.split(" ").map((word) => ({
|
|
63
|
+
type: "text-delta",
|
|
64
|
+
textDelta: word + " "
|
|
65
|
+
}));
|
|
66
|
+
return {
|
|
67
|
+
stream: simulateReadableStream({
|
|
68
|
+
chunks: [
|
|
69
|
+
...chunks,
|
|
70
|
+
{
|
|
71
|
+
type: "finish",
|
|
72
|
+
finishReason: "stop",
|
|
73
|
+
logprobs: void 0,
|
|
74
|
+
usage: { completionTokens: 10, promptTokens: 3 }
|
|
75
|
+
}
|
|
76
|
+
]
|
|
77
|
+
}),
|
|
78
|
+
rawCall: { rawPrompt: null, rawSettings: {} }
|
|
23
79
|
};
|
|
24
80
|
}
|
|
81
|
+
});
|
|
82
|
+
return mockModel2;
|
|
83
|
+
}
|
|
84
|
+
const mockModel = new MockLanguageModelV2({
|
|
85
|
+
doGenerate: async (props) => {
|
|
86
|
+
if (spyGenerate) {
|
|
87
|
+
spyGenerate(props);
|
|
88
|
+
}
|
|
25
89
|
return {
|
|
26
90
|
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
27
91
|
finishReason: "stop",
|
|
28
|
-
usage: {
|
|
29
|
-
|
|
92
|
+
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
|
|
93
|
+
content: [
|
|
94
|
+
{
|
|
95
|
+
type: "text",
|
|
96
|
+
text: finalText
|
|
97
|
+
}
|
|
98
|
+
],
|
|
99
|
+
warnings: []
|
|
30
100
|
};
|
|
31
101
|
},
|
|
32
102
|
doStream: async (props) => {
|
|
33
103
|
if (spyStream) {
|
|
34
104
|
spyStream(props);
|
|
35
105
|
}
|
|
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
106
|
return {
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
107
|
+
rawCall: { rawPrompt: null, rawSettings: {} },
|
|
108
|
+
warnings: [],
|
|
109
|
+
stream: convertArrayToReadableStream([
|
|
110
|
+
{ type: "stream-start", warnings: [] },
|
|
111
|
+
{ type: "response-metadata", id: "id-0", modelId: "mock-model-id", timestamp: /* @__PURE__ */ new Date(0) },
|
|
112
|
+
{ type: "text-start", id: "text-1" },
|
|
113
|
+
{ type: "text-delta", id: "text-1", delta: finalText },
|
|
114
|
+
{ type: "text-end", id: "text-1" },
|
|
115
|
+
{
|
|
116
|
+
type: "finish",
|
|
117
|
+
finishReason: "stop",
|
|
118
|
+
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 }
|
|
119
|
+
}
|
|
120
|
+
])
|
|
54
121
|
};
|
|
55
122
|
}
|
|
56
123
|
});
|
|
@@ -135,6 +202,6 @@ var MockProvider = class extends MastraLLMV1 {
|
|
|
135
202
|
}
|
|
136
203
|
};
|
|
137
204
|
|
|
138
|
-
export { MockProvider, createMockModel };
|
|
205
|
+
export { MastraLanguageModelV2Mock, MockProvider, createMockModel };
|
|
139
206
|
//# sourceMappingURL=llm-mock.js.map
|
|
140
207
|
//# sourceMappingURL=llm-mock.js.map
|
|
@@ -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/loop/test-utils/MastraLanguageModelV2Mock.ts","../../src/test-utils/llm-mock.ts"],"names":["mockModel","MockLanguageModelV2"],"mappings":";;;;;AAMO,IAAM,4BAAN,MAAiE;AAAA,EAC7D,oBAAA,GAAuB,IAAA;AAAA,EACvB,QAAA;AAAA,EACA,OAAA;AAAA,EACA,aAAA;AAAA,EACT,MAAA;AAAA,EAEA,YAAY,MAAA,EAYT;AACD,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,mBAAA,CAAoB,MAAM,CAAA;AAC5C,IAAA,IAAA,CAAK,QAAA,GAAW,KAAK,MAAA,CAAO,QAAA;AAC5B,IAAA,IAAA,CAAK,OAAA,GAAU,KAAK,MAAA,CAAO,OAAA;AAC3B,IAAA,IAAA,CAAK,aAAA,GAAgB,KAAK,MAAA,CAAO,aAAA;AAAA,EACnC;AAAA,EAEA,WAAW,OAAA,EAAqC;AAC9C,IAAA,MAAM,UAAA,GAAa,IAAI,oBAAA,CAAqB,IAAA,CAAK,MAAM,CAAA;AAEvD,IAAA,OAAO,UAAA,CAAW,WAAW,OAAO,CAAA;AAAA,EACtC;AAAA,EAEA,SAAS,OAAA,EAAqC;AAC5C,IAAA,MAAM,UAAA,GAAa,IAAI,oBAAA,CAAqB,IAAA,CAAK,MAAM,CAAA;AAEvD,IAAA,OAAO,UAAA,CAAW,SAAS,OAAO,CAAA;AAAA,EACpC;AAAA,EAEA,IAAI,eAAA,GAAkB;AACpB,IAAA,OAAO,KAAK,MAAA,CAAO,eAAA;AAAA,EACrB;AAAA,EACA,IAAI,aAAA,GAAgB;AAClB,IAAA,OAAO,KAAK,MAAA,CAAO,aAAA;AAAA,EACrB;AACF;;;ACxCO,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,IAAIC,mBAAAA,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,QAAA,EAAS;AAAA,UACnC,EAAE,IAAA,EAAM,YAAA,EAAc,EAAA,EAAI,QAAA,EAAU,OAAO,SAAA,EAAU;AAAA,UACrD,EAAE,IAAA,EAAM,UAAA,EAAY,EAAA,EAAI,QAAA,EAAS;AAAA,UACjC;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 type { LanguageModelV2, LanguageModelV2CallOptions } from '@ai-sdk/provider-v5';\nimport { MockLanguageModelV2 } from 'ai-v5/test';\nimport type { MastraLanguageModelV2 } from '../../llm/model/shared.types';\nimport type { LanguageModelV2StreamResult } from '../../stream/types';\nimport { AISDKV5LanguageModel } from '../../llm/model/aisdk/v5/model';\n\nexport class MastraLanguageModelV2Mock implements MastraLanguageModelV2 {\n readonly specificationVersion = 'v2';\n readonly provider: LanguageModelV2['provider'];\n readonly modelId: LanguageModelV2['modelId'];\n readonly supportedUrls: LanguageModelV2['supportedUrls'];\n #model: MockLanguageModelV2;\n\n constructor(config: {\n provider?: LanguageModelV2['provider'];\n modelId?: LanguageModelV2['modelId'];\n supportedUrls?: LanguageModelV2['supportedUrls'] | (() => LanguageModelV2['supportedUrls']);\n doGenerate?:\n | LanguageModelV2['doGenerate']\n | Awaited<ReturnType<LanguageModelV2['doGenerate']>>\n | Awaited<ReturnType<LanguageModelV2['doGenerate']>>[];\n doStream?:\n | LanguageModelV2['doStream']\n | Awaited<ReturnType<LanguageModelV2['doStream']>>\n | Awaited<ReturnType<LanguageModelV2['doStream']>>[];\n }) {\n this.#model = new MockLanguageModelV2(config);\n this.provider = this.#model.provider;\n this.modelId = this.#model.modelId;\n this.supportedUrls = this.#model.supportedUrls;\n }\n\n doGenerate(options: LanguageModelV2CallOptions) {\n const aiSDKModel = new AISDKV5LanguageModel(this.#model);\n\n return aiSDKModel.doGenerate(options);\n }\n\n doStream(options: LanguageModelV2CallOptions) {\n const aiSDKModel = new AISDKV5LanguageModel(this.#model);\n\n return aiSDKModel.doStream(options);\n }\n\n get doGenerateCalls() {\n return this.#model.doGenerateCalls;\n }\n get doStreamCalls() {\n return this.#model.doStreamCalls;\n }\n}\n","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';\nimport { MastraLanguageModelV2Mock } from '../loop/test-utils/MastraLanguageModelV2Mock';\n\n// Re-export for external use\nexport { simulateReadableStream, MastraLanguageModelV2Mock };\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: 'text-1' },\n { type: 'text-delta', id: 'text-1', delta: finalText },\n { type: 'text-end', id: 'text-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,25 +1,25 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
4
|
-
var
|
|
3
|
+
var chunkKJBMTK5B_cjs = require('../chunk-KJBMTK5B.cjs');
|
|
4
|
+
var chunkMGCGWPQJ_cjs = require('../chunk-MGCGWPQJ.cjs');
|
|
5
5
|
|
|
6
6
|
|
|
7
7
|
|
|
8
8
|
Object.defineProperty(exports, "ToolStream", {
|
|
9
9
|
enumerable: true,
|
|
10
|
-
get: function () { return
|
|
10
|
+
get: function () { return chunkKJBMTK5B_cjs.ToolStream; }
|
|
11
11
|
});
|
|
12
12
|
Object.defineProperty(exports, "Tool", {
|
|
13
13
|
enumerable: true,
|
|
14
|
-
get: function () { return
|
|
14
|
+
get: function () { return chunkMGCGWPQJ_cjs.Tool; }
|
|
15
15
|
});
|
|
16
16
|
Object.defineProperty(exports, "createTool", {
|
|
17
17
|
enumerable: true,
|
|
18
|
-
get: function () { return
|
|
18
|
+
get: function () { return chunkMGCGWPQJ_cjs.createTool; }
|
|
19
19
|
});
|
|
20
20
|
Object.defineProperty(exports, "isVercelTool", {
|
|
21
21
|
enumerable: true,
|
|
22
|
-
get: function () { return
|
|
22
|
+
get: function () { return chunkMGCGWPQJ_cjs.isVercelTool; }
|
|
23
23
|
});
|
|
24
24
|
//# sourceMappingURL=index.cjs.map
|
|
25
25
|
//# sourceMappingURL=index.cjs.map
|
package/dist/tools/index.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
export { ToolStream } from '../chunk-
|
|
2
|
-
export { Tool, createTool, isVercelTool } from '../chunk-
|
|
1
|
+
export { ToolStream } from '../chunk-DD2VNRQM.js';
|
|
2
|
+
export { Tool, createTool, isVercelTool } from '../chunk-THZTRBFS.js';
|
|
3
3
|
//# sourceMappingURL=index.js.map
|
|
4
4
|
//# sourceMappingURL=index.js.map
|
|
@@ -1,12 +1,12 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunkMGCGWPQJ_cjs = require('../chunk-MGCGWPQJ.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 chunkMGCGWPQJ_cjs.isVercelTool; }
|
|
10
10
|
});
|
|
11
11
|
//# sourceMappingURL=is-vercel-tool.cjs.map
|
|
12
12
|
//# sourceMappingURL=is-vercel-tool.cjs.map
|
package/dist/tools/stream.d.ts
CHANGED
|
@@ -1,13 +1,19 @@
|
|
|
1
|
-
import { WritableStream } from 'stream/web';
|
|
1
|
+
import { WritableStream } from 'node:stream/web';
|
|
2
2
|
import type { DataChunkType } from '../stream/types.js';
|
|
3
|
-
|
|
4
|
-
|
|
3
|
+
import type { OutputWriter } from '../workflows/index.js';
|
|
4
|
+
export declare class ToolStream extends WritableStream<unknown> {
|
|
5
|
+
private prefix;
|
|
6
|
+
private callId;
|
|
7
|
+
private name;
|
|
8
|
+
private runId;
|
|
9
|
+
private writeFn?;
|
|
5
10
|
constructor({ prefix, callId, name, runId, }: {
|
|
6
11
|
prefix: string;
|
|
7
12
|
callId: string;
|
|
8
13
|
name: string;
|
|
9
14
|
runId: string;
|
|
10
|
-
},
|
|
15
|
+
}, writeFn?: OutputWriter);
|
|
16
|
+
private _write;
|
|
11
17
|
write(data: any): Promise<void>;
|
|
12
18
|
custom<T extends {
|
|
13
19
|
type: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/tools/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/tools/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAEjD,qBAAa,UAAW,SAAQ,cAAc,CAAC,OAAO,CAAC;IACrD,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAC,CAAe;gBAG7B,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,OAAO,CAAC,EAAE,YAAY;YAoBV,MAAM;IAsBd,KAAK,CAAC,IAAI,EAAE,GAAG;IAIf,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;CAKxG"}
|
|
@@ -19,6 +19,8 @@ export declare class CoreToolBuilder extends MastraBase {
|
|
|
19
19
|
});
|
|
20
20
|
private getParameters;
|
|
21
21
|
private getOutputSchema;
|
|
22
|
+
private getResumeSchema;
|
|
23
|
+
private getSuspendSchema;
|
|
22
24
|
private buildProviderTool;
|
|
23
25
|
private createLogMessageOptions;
|
|
24
26
|
private createExecute;
|
|
@@ -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;IAEF,OAAO,CAAC,eAAe,CAYrB;IAEF,OAAO,CAAC,gBAAgB,CAYtB;IAGF,OAAO,CAAC,iBAAiB;IAqEzB,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,aAAa;IA+PrB,OAAO;IAgCP,KAAK,IAAI,QAAQ;CAiGlB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-stream-types.test-d.d.ts","sourceRoot":"","sources":["../../src/tools/tool-stream-types.test-d.ts"],"names":[],"mappings":""}
|
package/dist/tools/tool.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { Mastra } from '../mastra/index.js';
|
|
2
|
-
import type { ZodLikeSchema, InferZodLikeSchema } from '../types/zod-compat.js';
|
|
2
|
+
import type { ZodLikeSchema, InferZodLikeSchema, InferZodLikeSchemaInput } from '../types/zod-compat.js';
|
|
3
3
|
import type { ToolAction, ToolExecutionContext } from './types.js';
|
|
4
4
|
/**
|
|
5
5
|
* A type-safe tool that agents and workflows can call to perform specific actions.
|
|
@@ -53,9 +53,9 @@ import type { ToolAction, ToolExecutionContext } from './types.js';
|
|
|
53
53
|
* });
|
|
54
54
|
* ```
|
|
55
55
|
*/
|
|
56
|
-
export declare class Tool<TSchemaIn extends ZodLikeSchema | undefined = undefined, TSchemaOut extends ZodLikeSchema | undefined = undefined, TSuspendSchema extends ZodLikeSchema = any, TResumeSchema extends ZodLikeSchema = any, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSuspendSchema, TResumeSchema
|
|
56
|
+
export declare class Tool<TSchemaIn extends ZodLikeSchema | undefined = undefined, TSchemaOut extends ZodLikeSchema | undefined = undefined, TSuspendSchema extends ZodLikeSchema = any, TResumeSchema extends ZodLikeSchema = any, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSuspendSchema, TResumeSchema>, TId extends string = string> implements ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId> {
|
|
57
57
|
/** Unique identifier for the tool */
|
|
58
|
-
id:
|
|
58
|
+
id: TId;
|
|
59
59
|
/** Description of what the tool does */
|
|
60
60
|
description: string;
|
|
61
61
|
/** Schema for validating input parameters */
|
|
@@ -84,6 +84,19 @@ export declare class Tool<TSchemaIn extends ZodLikeSchema | undefined = undefine
|
|
|
84
84
|
* ```
|
|
85
85
|
*/
|
|
86
86
|
requireApproval?: boolean;
|
|
87
|
+
/**
|
|
88
|
+
* Provider-specific options passed to the model when this tool is used.
|
|
89
|
+
* Keys are provider names (e.g., 'anthropic', 'openai'), values are provider-specific configs.
|
|
90
|
+
* @example
|
|
91
|
+
* ```typescript
|
|
92
|
+
* providerOptions: {
|
|
93
|
+
* anthropic: {
|
|
94
|
+
* cacheControl: { type: 'ephemeral' }
|
|
95
|
+
* }
|
|
96
|
+
* }
|
|
97
|
+
* ```
|
|
98
|
+
*/
|
|
99
|
+
providerOptions?: Record<string, Record<string, unknown>>;
|
|
87
100
|
/**
|
|
88
101
|
* Creates a new Tool instance with input validation wrapper.
|
|
89
102
|
*
|
|
@@ -98,7 +111,7 @@ export declare class Tool<TSchemaIn extends ZodLikeSchema | undefined = undefine
|
|
|
98
111
|
* });
|
|
99
112
|
* ```
|
|
100
113
|
*/
|
|
101
|
-
constructor(opts: ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext>);
|
|
114
|
+
constructor(opts: ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId>);
|
|
102
115
|
}
|
|
103
116
|
/**
|
|
104
117
|
* Creates a type-safe tool with automatic input validation.
|
|
@@ -176,11 +189,11 @@ export declare class Tool<TSchemaIn extends ZodLikeSchema | undefined = undefine
|
|
|
176
189
|
* });
|
|
177
190
|
* ```
|
|
178
191
|
*/
|
|
179
|
-
export declare function createTool<TSchemaIn extends ZodLikeSchema | undefined = undefined, TSchemaOut extends ZodLikeSchema | undefined = undefined, TSuspendSchema extends ZodLikeSchema = any, TResumeSchema extends ZodLikeSchema = any, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSuspendSchema, TResumeSchema>, TExecute extends ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext>['execute'] = ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext>['execute']>(opts: ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> & {
|
|
192
|
+
export declare function createTool<TId extends string = string, TSchemaIn extends ZodLikeSchema | undefined = undefined, TSchemaOut extends ZodLikeSchema | undefined = undefined, TSuspendSchema extends ZodLikeSchema = any, TResumeSchema extends ZodLikeSchema = any, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSuspendSchema, TResumeSchema>, TExecute extends ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId>['execute'] = ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId>['execute']>(opts: ToolAction<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId> & {
|
|
180
193
|
execute?: TExecute;
|
|
181
|
-
}): [TSchemaIn, TSchemaOut, TExecute] extends [ZodLikeSchema, ZodLikeSchema, Function] ? Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> & {
|
|
194
|
+
}): [TSchemaIn, TSchemaOut, TExecute] extends [ZodLikeSchema, ZodLikeSchema, Function] ? Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId> & {
|
|
182
195
|
inputSchema: TSchemaIn;
|
|
183
196
|
outputSchema: TSchemaOut;
|
|
184
|
-
execute: (inputData: TSchemaIn extends ZodLikeSchema ? InferZodLikeSchema<TSchemaIn> : unknown, context?: TContext) => Promise<TSchemaOut extends ZodLikeSchema ?
|
|
185
|
-
} : Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext>;
|
|
197
|
+
execute: (inputData: TSchemaIn extends ZodLikeSchema ? InferZodLikeSchema<TSchemaIn> : unknown, context?: TContext) => Promise<TSchemaOut extends ZodLikeSchema ? InferZodLikeSchemaInput<TSchemaOut> : unknown>;
|
|
198
|
+
} : Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext, TId>;
|
|
186
199
|
//# sourceMappingURL=tool.d.ts.map
|
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;
|
|
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,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AACtG,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,EACD,GAAG,SAAS,MAAM,GAAG,MAAM,CAC3B,YAAW,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,CAAC;IAE1F,qCAAqC;IACrC,EAAE,EAAE,GAAG,CAAC;IAER,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;;;;;;;;;;;OAWG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAE1D;;;;;;;;;;;;;OAaG;gBACS,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,CAAC;CAoJlG;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2EG;AACH,wBAAgB,UAAU,CACxB,GAAG,SAAS,MAAM,GAAG,MAAM,EAC3B,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,CACzB,SAAS,EACT,UAAU,EACV,cAAc,EACd,aAAa,EACb,QAAQ,EACR,GAAG,CACJ,CAAC,SAAS,CAAC,GAAG,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,EAEzG,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,CAAC,GAAG;IACtF,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,EAAE,GAAG,CAAC,GAAG;IAC1E,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,uBAAuB,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,CAAC;CAChG,GACD,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,CAAC,CAE5E"}
|
package/dist/tools/types.d.ts
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import type { WritableStream } from 'node: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';
|
|
@@ -5,7 +6,8 @@ import type { MastraUnion } from '../action/index.js';
|
|
|
5
6
|
import type { Mastra } from '../mastra/index.js';
|
|
6
7
|
import type { TracingContext } from '../observability/index.js';
|
|
7
8
|
import type { RequestContext } from '../request-context/index.js';
|
|
8
|
-
import type { ZodLikeSchema, InferZodLikeSchema } from '../types/zod-compat.js';
|
|
9
|
+
import type { ZodLikeSchema, InferZodLikeSchema, InferZodLikeSchemaInput } from '../types/zod-compat.js';
|
|
10
|
+
import type { OutputWriter } from '../workflows/types.js';
|
|
9
11
|
import type { ToolStream } from './stream.js';
|
|
10
12
|
import type { ValidationError } from './validation.js';
|
|
11
13
|
export type VercelTool = Tool;
|
|
@@ -54,7 +56,7 @@ export interface MCPToolExecutionContext {
|
|
|
54
56
|
export type MastraToolInvocationOptions = ToolInvocationOptions & {
|
|
55
57
|
suspend?: (suspendPayload: any) => Promise<any>;
|
|
56
58
|
resumeData?: any;
|
|
57
|
-
|
|
59
|
+
outputWriter?: OutputWriter;
|
|
58
60
|
tracingContext?: TracingContext;
|
|
59
61
|
/**
|
|
60
62
|
* Optional MCP-specific context passed when tool is executed in MCP server.
|
|
@@ -93,6 +95,10 @@ export type CoreTool = {
|
|
|
93
95
|
parameters: FlexibleSchema<any> | Schema;
|
|
94
96
|
outputSchema?: FlexibleSchema<any> | Schema;
|
|
95
97
|
execute?: (params: any, options: MastraToolInvocationOptions) => Promise<any>;
|
|
98
|
+
/**
|
|
99
|
+
* Provider-specific options passed to the model when this tool is used.
|
|
100
|
+
*/
|
|
101
|
+
providerOptions?: Record<string, Record<string, unknown>>;
|
|
96
102
|
/**
|
|
97
103
|
* Optional MCP-specific properties.
|
|
98
104
|
* Only populated when the tool is being used in an MCP context.
|
|
@@ -117,6 +123,10 @@ export type InternalCoreTool = {
|
|
|
117
123
|
parameters: Schema;
|
|
118
124
|
outputSchema?: Schema;
|
|
119
125
|
execute?: (params: any, options: MastraToolInvocationOptions) => Promise<any>;
|
|
126
|
+
/**
|
|
127
|
+
* Provider-specific options passed to the model when this tool is used.
|
|
128
|
+
*/
|
|
129
|
+
providerOptions?: Record<string, Record<string, unknown>>;
|
|
120
130
|
/**
|
|
121
131
|
* Optional MCP-specific properties.
|
|
122
132
|
* Only populated when the tool is being used in an MCP context.
|
|
@@ -135,21 +145,34 @@ export interface ToolExecutionContext<TSuspendSchema extends ZodLikeSchema = any
|
|
|
135
145
|
requestContext?: RequestContext;
|
|
136
146
|
tracingContext?: TracingContext;
|
|
137
147
|
abortSignal?: AbortSignal;
|
|
138
|
-
writer?: ToolStream
|
|
148
|
+
writer?: ToolStream;
|
|
139
149
|
agent?: AgentToolExecutionContext<TSuspendSchema, TResumeSchema>;
|
|
140
150
|
workflow?: WorkflowToolExecutionContext<TSuspendSchema, TResumeSchema>;
|
|
141
151
|
mcp?: MCPToolExecutionContext;
|
|
142
152
|
}
|
|
143
|
-
export interface ToolAction<TSchemaIn extends ZodLikeSchema | undefined = undefined, TSchemaOut extends ZodLikeSchema | undefined = undefined, TSuspendSchema extends ZodLikeSchema = any, TResumeSchema extends ZodLikeSchema = any, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSuspendSchema, TResumeSchema
|
|
144
|
-
id:
|
|
153
|
+
export interface ToolAction<TSchemaIn extends ZodLikeSchema | undefined = undefined, TSchemaOut extends ZodLikeSchema | undefined = undefined, TSuspendSchema extends ZodLikeSchema = any, TResumeSchema extends ZodLikeSchema = any, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema> = ToolExecutionContext<TSuspendSchema, TResumeSchema>, TId extends string = string> {
|
|
154
|
+
id: TId;
|
|
145
155
|
description: string;
|
|
146
156
|
inputSchema?: TSchemaIn;
|
|
147
157
|
outputSchema?: TSchemaOut;
|
|
148
158
|
suspendSchema?: TSuspendSchema;
|
|
149
159
|
resumeSchema?: TResumeSchema;
|
|
150
|
-
execute?: (inputData: TSchemaIn extends ZodLikeSchema ? InferZodLikeSchema<TSchemaIn> : unknown, context?: TContext) => Promise<(TSchemaOut extends ZodLikeSchema ?
|
|
160
|
+
execute?: (inputData: TSchemaIn extends ZodLikeSchema ? InferZodLikeSchema<TSchemaIn> : unknown, context?: TContext) => Promise<(TSchemaOut extends ZodLikeSchema ? InferZodLikeSchemaInput<TSchemaOut> : any) | ValidationError>;
|
|
151
161
|
mastra?: Mastra;
|
|
152
162
|
requireApproval?: boolean;
|
|
163
|
+
/**
|
|
164
|
+
* Provider-specific options passed to the model when this tool is used.
|
|
165
|
+
* Keys are provider names (e.g., 'anthropic', 'openai'), values are provider-specific configs.
|
|
166
|
+
* @example
|
|
167
|
+
* ```typescript
|
|
168
|
+
* providerOptions: {
|
|
169
|
+
* anthropic: {
|
|
170
|
+
* cacheControl: { type: 'ephemeral' }
|
|
171
|
+
* }
|
|
172
|
+
* }
|
|
173
|
+
* ```
|
|
174
|
+
*/
|
|
175
|
+
providerOptions?: Record<string, Record<string, unknown>>;
|
|
153
176
|
onInputStart?: (options: ToolCallOptions) => void | PromiseLike<void>;
|
|
154
177
|
onInputDelta?: (options: {
|
|
155
178
|
inputTextDelta: string;
|
|
@@ -157,5 +180,9 @@ export interface ToolAction<TSchemaIn extends ZodLikeSchema | undefined = undefi
|
|
|
157
180
|
onInputAvailable?: (options: {
|
|
158
181
|
input: InferZodLikeSchema<TSchemaIn>;
|
|
159
182
|
} & ToolCallOptions) => void | PromiseLike<void>;
|
|
183
|
+
onOutput?: (options: {
|
|
184
|
+
output: TSchemaOut extends ZodLikeSchema ? InferZodLikeSchema<TSchemaOut> : any;
|
|
185
|
+
toolName: string;
|
|
186
|
+
} & Omit<ToolCallOptions, 'messages'>) => void | PromiseLike<void>;
|
|
160
187
|
}
|
|
161
188
|
//# sourceMappingURL=types.d.ts.map
|
|
@@ -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;
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/tools/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,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,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AACtG,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,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,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,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;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC1D;;;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;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC1D;;;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;IAKpB,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,EACD,GAAG,SAAS,MAAM,GAAG,MAAM;IAE3B,EAAE,EAAE,GAAG,CAAC;IACR,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;IAO7B,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,uBAAuB,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;IAC/G,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B;;;;;;;;;;;OAWG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC1D,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;IAC9B,QAAQ,CAAC,EAAE,CACT,OAAO,EAAE;QACP,MAAM,EAAE,UAAU,SAAS,aAAa,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC;QAChF,QAAQ,EAAE,MAAM,CAAC;KAClB,GAAG,IAAI,CAAC,eAAe,EAAE,UAAU,CAAC,KAClC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;CAC/B"}
|
|
@@ -1,10 +1,22 @@
|
|
|
1
|
-
import
|
|
1
|
+
import { z } from 'zod';
|
|
2
2
|
import type { ZodLikeSchema } from '../types/zod-compat.js';
|
|
3
3
|
export interface ValidationError<T = any> {
|
|
4
4
|
error: true;
|
|
5
5
|
message: string;
|
|
6
6
|
validationErrors: z.ZodFormattedError<T>;
|
|
7
7
|
}
|
|
8
|
+
/**
|
|
9
|
+
* Validates raw suspend data against a Zod schema.
|
|
10
|
+
*
|
|
11
|
+
* @param schema The Zod schema to validate against
|
|
12
|
+
* @param suspendData The raw suspend data to validate
|
|
13
|
+
* @param toolId Optional tool ID for better error messages
|
|
14
|
+
* @returns The validated data or a validation error
|
|
15
|
+
*/
|
|
16
|
+
export declare function validateToolSuspendData<T = any>(schema: ZodLikeSchema | undefined, suspendData: unknown, toolId?: string): {
|
|
17
|
+
data: T | unknown;
|
|
18
|
+
error?: ValidationError<T>;
|
|
19
|
+
};
|
|
8
20
|
/**
|
|
9
21
|
* Validates raw input data against a Zod schema.
|
|
10
22
|
*
|
|
@@ -17,4 +29,16 @@ export declare function validateToolInput<T = any>(schema: ZodLikeSchema | undef
|
|
|
17
29
|
data: T | unknown;
|
|
18
30
|
error?: ValidationError<T>;
|
|
19
31
|
};
|
|
32
|
+
/**
|
|
33
|
+
* Validates tool output data against a Zod schema.
|
|
34
|
+
*
|
|
35
|
+
* @param schema The Zod schema to validate against
|
|
36
|
+
* @param output The output data to validate
|
|
37
|
+
* @param toolId Optional tool ID for better error messages
|
|
38
|
+
* @returns The validated data or a validation error
|
|
39
|
+
*/
|
|
40
|
+
export declare function validateToolOutput<T = any>(schema: ZodLikeSchema | undefined, output: unknown, toolId?: string, suspendCalled?: boolean): {
|
|
41
|
+
data: T | unknown;
|
|
42
|
+
error?: ValidationError<T>;
|
|
43
|
+
};
|
|
20
44
|
//# sourceMappingURL=validation.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"validation.d.ts","sourceRoot":"","sources":["../../src/tools/validation.ts"],"names":[],"mappings":"AAAA,OAAO,
|
|
1
|
+
{"version":3,"file":"validation.d.ts","sourceRoot":"","sources":["../../src/tools/validation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,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,uBAAuB,CAAC,CAAC,GAAG,GAAG,EAC7C,MAAM,EAAE,aAAa,GAAG,SAAS,EACjC,WAAW,EAAE,OAAO,EACpB,MAAM,CAAC,EAAE,MAAM,GACd;IAAE,IAAI,EAAE,CAAC,GAAG,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,CAAA;CAAE,CAyBnD;AA8BD;;;;;;;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,CA6BnD;AAED;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAAC,CAAC,GAAG,GAAG,EACxC,MAAM,EAAE,aAAa,GAAG,SAAS,EACjC,MAAM,EAAE,OAAO,EACf,MAAM,CAAC,EAAE,MAAM,EACf,aAAa,CAAC,EAAE,OAAO,GACtB;IAAE,IAAI,EAAE,CAAC,GAAG,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,CAAA;CAAE,CAyBnD"}
|