assistant-stream 0.3.15 → 0.3.17
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/README.md +6 -4
- package/dist/core/AssistantStream.d.ts +35 -31
- package/dist/core/AssistantStream.d.ts.map +1 -1
- package/dist/core/AssistantStream.js +39 -37
- package/dist/core/AssistantStream.js.map +1 -1
- package/dist/core/AssistantStreamChunk.d.ts +71 -78
- package/dist/core/AssistantStreamChunk.d.ts.map +1 -1
- package/dist/core/AssistantStreamChunk.js +0 -2
- package/dist/core/accumulators/AssistantMessageStream.d.ts +15 -11
- package/dist/core/accumulators/AssistantMessageStream.d.ts.map +1 -1
- package/dist/core/accumulators/AssistantMessageStream.js +52 -48
- package/dist/core/accumulators/AssistantMessageStream.js.map +1 -1
- package/dist/core/accumulators/TimingTracker.d.ts +15 -11
- package/dist/core/accumulators/TimingTracker.d.ts.map +1 -1
- package/dist/core/accumulators/TimingTracker.js +36 -43
- package/dist/core/accumulators/TimingTracker.js.map +1 -1
- package/dist/core/accumulators/assistant-message-accumulator.d.ts +21 -11
- package/dist/core/accumulators/assistant-message-accumulator.d.ts.map +1 -1
- package/dist/core/accumulators/assistant-message-accumulator.js +349 -390
- package/dist/core/accumulators/assistant-message-accumulator.js.map +1 -1
- package/dist/core/converters/toGenericMessages.d.ts +49 -47
- package/dist/core/converters/toGenericMessages.d.ts.map +1 -1
- package/dist/core/converters/toGenericMessages.js +122 -136
- package/dist/core/converters/toGenericMessages.js.map +1 -1
- package/dist/core/modules/assistant-stream.d.ts +62 -66
- package/dist/core/modules/assistant-stream.d.ts.map +1 -1
- package/dist/core/modules/assistant-stream.js +176 -182
- package/dist/core/modules/assistant-stream.js.map +1 -1
- package/dist/core/modules/text.d.ts +11 -7
- package/dist/core/modules/text.d.ts.map +1 -1
- package/dist/core/modules/text.js +45 -45
- package/dist/core/modules/text.js.map +1 -1
- package/dist/core/modules/tool-call.d.ts +15 -11
- package/dist/core/modules/tool-call.d.ts.map +1 -1
- package/dist/core/modules/tool-call.js +81 -96
- package/dist/core/modules/tool-call.js.map +1 -1
- package/dist/core/object/ObjectStreamAccumulator.d.ts +13 -9
- package/dist/core/object/ObjectStreamAccumulator.d.ts.map +1 -1
- package/dist/core/object/ObjectStreamAccumulator.js +44 -52
- package/dist/core/object/ObjectStreamAccumulator.js.map +1 -1
- package/dist/core/object/ObjectStreamResponse.d.ts +12 -8
- package/dist/core/object/ObjectStreamResponse.d.ts.map +1 -1
- package/dist/core/object/ObjectStreamResponse.js +52 -71
- package/dist/core/object/ObjectStreamResponse.js.map +1 -1
- package/dist/core/object/createObjectStream.d.ts +14 -8
- package/dist/core/object/createObjectStream.d.ts.map +1 -1
- package/dist/core/object/createObjectStream.js +50 -47
- package/dist/core/object/createObjectStream.js.map +1 -1
- package/dist/core/object/types.d.ts +14 -11
- package/dist/core/object/types.d.ts.map +1 -1
- package/dist/core/object/types.js +0 -2
- package/dist/core/serialization/PlainText.d.ts +11 -7
- package/dist/core/serialization/PlainText.d.ts.map +1 -1
- package/dist/core/serialization/PlainText.js +40 -48
- package/dist/core/serialization/PlainText.js.map +1 -1
- package/dist/core/serialization/assistant-transport/AssistantTransport.d.ts +11 -7
- package/dist/core/serialization/assistant-transport/AssistantTransport.d.ts.map +1 -1
- package/dist/core/serialization/assistant-transport/AssistantTransport.js +102 -117
- package/dist/core/serialization/assistant-transport/AssistantTransport.js.map +1 -1
- package/dist/core/serialization/data-stream/DataStream.d.ts +11 -7
- package/dist/core/serialization/data-stream/DataStream.d.ts.map +1 -1
- package/dist/core/serialization/data-stream/DataStream.js +320 -371
- package/dist/core/serialization/data-stream/DataStream.js.map +1 -1
- package/dist/core/serialization/data-stream/chunk-types.d.ts +101 -100
- package/dist/core/serialization/data-stream/chunk-types.d.ts.map +1 -1
- package/dist/core/serialization/data-stream/chunk-types.js +27 -23
- package/dist/core/serialization/data-stream/chunk-types.js.map +1 -1
- package/dist/core/serialization/data-stream/serialization.d.ts +9 -5
- package/dist/core/serialization/data-stream/serialization.d.ts.map +1 -1
- package/dist/core/serialization/data-stream/serialization.js +23 -24
- package/dist/core/serialization/data-stream/serialization.js.map +1 -1
- package/dist/core/serialization/ui-message-stream/UIMessageStream.d.ts +15 -12
- package/dist/core/serialization/ui-message-stream/UIMessageStream.d.ts.map +1 -1
- package/dist/core/serialization/ui-message-stream/UIMessageStream.js +205 -228
- package/dist/core/serialization/ui-message-stream/UIMessageStream.js.map +1 -1
- package/dist/core/serialization/ui-message-stream/chunk-types.d.ts +68 -65
- package/dist/core/serialization/ui-message-stream/chunk-types.d.ts.map +1 -1
- package/dist/core/serialization/ui-message-stream/chunk-types.js +0 -2
- package/dist/core/tool/ToolCallReader.d.ts +34 -30
- package/dist/core/tool/ToolCallReader.d.ts.map +1 -1
- package/dist/core/tool/ToolCallReader.js +264 -317
- package/dist/core/tool/ToolCallReader.js.map +1 -1
- package/dist/core/tool/ToolExecutionStream.d.ts +20 -17
- package/dist/core/tool/ToolExecutionStream.d.ts.map +1 -1
- package/dist/core/tool/ToolExecutionStream.js +123 -146
- package/dist/core/tool/ToolExecutionStream.js.map +1 -1
- package/dist/core/tool/ToolResponse.d.ts +41 -41
- package/dist/core/tool/ToolResponse.d.ts.map +1 -1
- package/dist/core/tool/ToolResponse.js +50 -56
- package/dist/core/tool/ToolResponse.js.map +1 -1
- package/dist/core/tool/schema-utils.d.ts +19 -15
- package/dist/core/tool/schema-utils.d.ts.map +1 -1
- package/dist/core/tool/schema-utils.js +50 -84
- package/dist/core/tool/schema-utils.js.map +1 -1
- package/dist/core/tool/tool-types.d.ts +103 -122
- package/dist/core/tool/tool-types.d.ts.map +1 -1
- package/dist/core/tool/tool-types.js +0 -2
- package/dist/core/tool/toolResultStream.d.ts +11 -9
- package/dist/core/tool/toolResultStream.d.ts.map +1 -1
- package/dist/core/tool/toolResultStream.js +127 -157
- package/dist/core/tool/toolResultStream.js.map +1 -1
- package/dist/core/tool/type-path-utils.d.ts +8 -19
- package/dist/core/tool/type-path-utils.d.ts.map +1 -1
- package/dist/core/tool/type-path-utils.js +0 -2
- package/dist/core/utils/Counter.d.ts +6 -3
- package/dist/core/utils/Counter.d.ts.map +1 -1
- package/dist/core/utils/Counter.js +10 -6
- package/dist/core/utils/Counter.js.map +1 -1
- package/dist/core/utils/generateId.d.ts +4 -1
- package/dist/core/utils/generateId.d.ts.map +1 -1
- package/dist/core/utils/generateId.js +5 -1
- package/dist/core/utils/generateId.js.map +1 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts +15 -11
- package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts.map +1 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.js +38 -42
- package/dist/core/utils/stream/AssistantMetaTransformStream.js.map +1 -1
- package/dist/core/utils/stream/AssistantTransformStream.d.ts +11 -8
- package/dist/core/utils/stream/AssistantTransformStream.d.ts.map +1 -1
- package/dist/core/utils/stream/AssistantTransformStream.js +37 -35
- package/dist/core/utils/stream/AssistantTransformStream.js.map +1 -1
- package/dist/core/utils/stream/LineDecoderStream.d.ts +6 -3
- package/dist/core/utils/stream/LineDecoderStream.d.ts.map +1 -1
- package/dist/core/utils/stream/LineDecoderStream.js +23 -26
- package/dist/core/utils/stream/LineDecoderStream.js.map +1 -1
- package/dist/core/utils/stream/PipeableTransformStream.d.ts +5 -2
- package/dist/core/utils/stream/PipeableTransformStream.d.ts.map +1 -1
- package/dist/core/utils/stream/PipeableTransformStream.js +14 -10
- package/dist/core/utils/stream/PipeableTransformStream.js.map +1 -1
- package/dist/core/utils/stream/SSE.d.ts +10 -6
- package/dist/core/utils/stream/SSE.d.ts.map +1 -1
- package/dist/core/utils/stream/SSE.js +79 -91
- package/dist/core/utils/stream/SSE.js.map +1 -1
- package/dist/core/utils/stream/UnderlyingReadable.d.ts +7 -4
- package/dist/core/utils/stream/UnderlyingReadable.d.ts.map +1 -1
- package/dist/core/utils/stream/UnderlyingReadable.js +0 -2
- package/dist/core/utils/stream/merge.d.ts +11 -7
- package/dist/core/utils/stream/merge.d.ts.map +1 -1
- package/dist/core/utils/stream/merge.js +68 -170
- package/dist/core/utils/stream/merge.js.map +1 -1
- package/dist/core/utils/stream/path-utils.d.ts +11 -7
- package/dist/core/utils/stream/path-utils.d.ts.map +1 -1
- package/dist/core/utils/stream/path-utils.js +46 -52
- package/dist/core/utils/stream/path-utils.js.map +1 -1
- package/dist/core/utils/types.d.ts +109 -113
- package/dist/core/utils/types.d.ts.map +1 -1
- package/dist/core/utils/types.js +0 -2
- package/dist/core/utils/withPromiseOrValue.d.ts +4 -1
- package/dist/core/utils/withPromiseOrValue.d.ts.map +1 -1
- package/dist/core/utils/withPromiseOrValue.js +12 -15
- package/dist/core/utils/withPromiseOrValue.js.map +1 -1
- package/dist/index.d.ts +23 -24
- package/dist/index.js +16 -16
- package/dist/node_modules/.pnpm/@types_json-schema@7.0.15/node_modules/@types/json-schema/index.d.ts +141 -0
- package/dist/node_modules/.pnpm/@types_json-schema@7.0.15/node_modules/@types/json-schema/index.d.ts.map +1 -0
- package/dist/resumable/ResumableStreamContext.d.ts +28 -24
- package/dist/resumable/ResumableStreamContext.d.ts.map +1 -1
- package/dist/resumable/ResumableStreamContext.js +99 -114
- package/dist/resumable/ResumableStreamContext.js.map +1 -1
- package/dist/resumable/constants.d.ts +4 -1
- package/dist/resumable/constants.d.ts.map +1 -1
- package/dist/resumable/constants.js +5 -1
- package/dist/resumable/constants.js.map +1 -1
- package/dist/resumable/createResumableAssistantStreamResponse.d.ts +22 -21
- package/dist/resumable/createResumableAssistantStreamResponse.d.ts.map +1 -1
- package/dist/resumable/createResumableAssistantStreamResponse.js +25 -32
- package/dist/resumable/createResumableAssistantStreamResponse.js.map +1 -1
- package/dist/resumable/errors.d.ts +8 -5
- package/dist/resumable/errors.d.ts.map +1 -1
- package/dist/resumable/errors.js +14 -12
- package/dist/resumable/errors.js.map +1 -1
- package/dist/resumable/index.d.ts +7 -7
- package/dist/resumable/index.js +5 -5
- package/dist/resumable/stores/InMemoryResumableStreamStore.d.ts +14 -10
- package/dist/resumable/stores/InMemoryResumableStreamStore.d.ts.map +1 -1
- package/dist/resumable/stores/InMemoryResumableStreamStore.js +159 -193
- package/dist/resumable/stores/InMemoryResumableStreamStore.js.map +1 -1
- package/dist/resumable/stores/ioredis.d.ts +9 -5
- package/dist/resumable/stores/ioredis.d.ts.map +1 -1
- package/dist/resumable/stores/ioredis.js +70 -82
- package/dist/resumable/stores/ioredis.js.map +1 -1
- package/dist/resumable/stores/redis-impl.d.ts +51 -48
- package/dist/resumable/stores/redis-impl.d.ts.map +1 -1
- package/dist/resumable/stores/redis-impl.js +176 -176
- package/dist/resumable/stores/redis-impl.js.map +1 -1
- package/dist/resumable/stores/redis.d.ts +31 -28
- package/dist/resumable/stores/redis.d.ts.map +1 -1
- package/dist/resumable/stores/redis.js +87 -95
- package/dist/resumable/stores/redis.js.map +1 -1
- package/dist/resumable/types.d.ts +29 -26
- package/dist/resumable/types.d.ts.map +1 -1
- package/dist/resumable/types.js +0 -2
- package/dist/utils/AsyncIterableStream.d.ts +5 -2
- package/dist/utils/AsyncIterableStream.d.ts.map +1 -1
- package/dist/utils/AsyncIterableStream.js +17 -16
- package/dist/utils/AsyncIterableStream.js.map +1 -1
- package/dist/utils/json/fix-json.d.ts +4 -1
- package/dist/utils/json/fix-json.d.ts.map +1 -1
- package/dist/utils/json/fix-json.js +286 -380
- package/dist/utils/json/fix-json.js.map +1 -1
- package/dist/utils/json/is-json.d.ts +8 -4
- package/dist/utils/json/is-json.d.ts.map +1 -1
- package/dist/utils/json/is-json.js +13 -21
- package/dist/utils/json/is-json.js.map +1 -1
- package/dist/utils/json/json-value.d.ts +7 -4
- package/dist/utils/json/json-value.d.ts.map +1 -1
- package/dist/utils/json/json-value.js +0 -2
- package/dist/utils/json/parse-partial-json-object.d.ts +11 -8
- package/dist/utils/json/parse-partial-json-object.d.ts.map +1 -1
- package/dist/utils/json/parse-partial-json-object.js +50 -62
- package/dist/utils/json/parse-partial-json-object.js.map +1 -1
- package/dist/utils/promiseWithResolvers.d.ts +7 -4
- package/dist/utils/promiseWithResolvers.d.ts.map +1 -1
- package/dist/utils/promiseWithResolvers.js +17 -10
- package/dist/utils/promiseWithResolvers.js.map +1 -1
- package/dist/utils.d.ts +6 -6
- package/dist/utils.js +5 -5
- package/package.json +3 -3
- package/src/core/modules/assistant-stream.test.ts +104 -0
- package/src/core/modules/assistant-stream.ts +13 -4
- package/dist/core/AssistantStreamChunk.js.map +0 -1
- package/dist/core/object/types.js.map +0 -1
- package/dist/core/serialization/ui-message-stream/chunk-types.js.map +0 -1
- package/dist/core/tool/tool-types.js.map +0 -1
- package/dist/core/tool/type-path-utils.js.map +0 -1
- package/dist/core/utils/stream/UnderlyingReadable.js.map +0 -1
- package/dist/core/utils/types.js.map +0 -1
- package/dist/index.d.ts.map +0 -1
- package/dist/index.js.map +0 -1
- package/dist/resumable/index.d.ts.map +0 -1
- package/dist/resumable/index.js.map +0 -1
- package/dist/resumable/types.js.map +0 -1
- package/dist/utils/json/json-value.js.map +0 -1
- package/dist/utils.d.ts.map +0 -1
- package/dist/utils.js.map +0 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"DataStream.js","sourceRoot":"","sources":["../../../../src/core/serialization/data-stream/DataStream.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,wBAAwB,EAAE,uDAAoD;AACvF,OAAO,EAAE,uBAAuB,EAAE,sDAAmD;AACrF,OAAO,EAAwB,yBAAyB,EAAE,yBAAsB;AAChF,OAAO,EAAE,iBAAiB,EAAE,gDAA6C;AACzE,OAAO,EACL,sBAAsB,EACtB,sBAAsB,GACvB,2BAAwB;AACzB,OAAO,EAEL,4BAA4B,GAC7B,2DAAwD;AAIzD,MAAM,OAAO,iBACX,SAAQ,uBAAsE;IAG9E,OAAO,GAAG,IAAI,OAAO,CAAC;QACpB,cAAc,EAAE,2BAA2B;QAC3C,yBAAyB,EAAE,IAAI;KAChC,CAAC,CAAC;IAEH;QACE,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,MAAM,SAAS,GAAG,IAAI,eAAe,CAGnC;gBACA,SAAS,CAAC,KAAK,EAAE,UAAU;oBACzB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;oBACxB,QAAQ,IAAI,EAAE,CAAC;wBACb,KAAK,YAAY,CAAC,CAAC,CAAC;4BAClB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;4BACxB,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gCAC9B,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC;gCAChC,UAAU,CAAC,OAAO,CAAC;oCACjB,IAAI,EAAE,yBAAyB,CAAC,aAAa;oCAC7C,KAAK;iCACN,CAAC,CAAC;4BACL,CAAC;4BACD,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gCAC3B,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC;gCAChC,UAAU,CAAC,OAAO,CAAC;oCACjB,IAAI,EAAE,yBAAyB,CAAC,MAAM;oCACtC,KAAK;iCACN,CAAC,CAAC;4BACL,CAAC;4BACD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gCACzB,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC;gCAChC,UAAU,CAAC,OAAO,CAAC;oCACjB,IAAI,EAAE,yBAAyB,CAAC,WAAW;oCAC3C,KAAK;iCACN,CAAC,CAAC;4BACL,CAAC;4BACD,MAAM;wBACR,CAAC;wBACD,KAAK,YAAY,CAAC,CAAC,CAAC;4BAClB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;4BACxB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gCAClB,KAAK,MAAM,CAAC,CAAC,CAAC;oCACZ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;wCAClB,UAAU,CAAC,OAAO,CAAC;4CACjB,IAAI,EAAE,yBAAyB,CAAC,YAAY;4CAC5C,KAAK,EAAE;gDACL,SAAS,EAAE,KAAK,CAAC,SAAS;gDAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;6CACxB;yCACF,CAAC,CAAC;oCACL,CAAC;yCAAM,CAAC;wCACN,UAAU,CAAC,OAAO,CAAC;4CACjB,IAAI,EAAE,yBAAyB,CAAC,SAAS;4CACzC,KAAK,EAAE,KAAK,CAAC,SAAS;yCACvB,CAAC,CAAC;oCACL,CAAC;oCACD,MAAM;gCACR,CAAC;gCACD,KAAK,WAAW,CAAC,CAAC,CAAC;oCACjB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;wCAClB,UAAU,CAAC,OAAO,CAAC;4CACjB,IAAI,EAAE,yBAAyB,CAAC,iBAAiB;4CACjD,KAAK,EAAE;gDACL,cAAc,EAAE,KAAK,CAAC,SAAS;gDAC/B,QAAQ,EAAE,IAAI,CAAC,QAAQ;6CACxB;yCACF,CAAC,CAAC;oCACL,CAAC;yCAAM,CAAC;wCACN,UAAU,CAAC,OAAO,CAAC;4CACjB,IAAI,EAAE,yBAAyB,CAAC,cAAc;4CAC9C,KAAK,EAAE,KAAK,CAAC,SAAS;yCACvB,CAAC,CAAC;oCACL,CAAC;oCACD,MAAM;gCACR,CAAC;gCACD,KAAK,WAAW,CAAC,CAAC,CAAC;oCACjB,UAAU,CAAC,OAAO,CAAC;wCACjB,IAAI,EAAE,yBAAyB,CAAC,qBAAqB;wCACrD,KAAK,EAAE;4CACL,UAAU,EAAE,IAAI,CAAC,UAAU;4CAC3B,aAAa,EAAE,KAAK,CAAC,SAAS;yCAC/B;qCACF,CAAC,CAAC;oCACH,MAAM;gCACR,CAAC;gCACD;oCACE,MAAM,IAAI,KAAK,CACb,yCAAyC,IAAI,CAAC,IAAI,EAAE,CACrD,CAAC;4BACN,CAAC;4BACD,MAAM;wBACR,CAAC;wBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;4BACd,yCAAyC;4BACzC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;4BACxB,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gCAC9B,MAAM,IAAI,KAAK,CACb,qDAAqD,IAAI,CAAC,IAAI,EAAE,CACjE,CAAC;4BACJ,CAAC;4BACD,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,cAAc;gCAC9C,KAAK,EAAE;oCACL,UAAU,EAAE,IAAI,CAAC,UAAU;oCAC3B,MAAM,EAAE,KAAK,CAAC,MAAM;oCACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oCACxB,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iCACrD;6BACF,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,KAAK,YAAY,CAAC,CAAC,CAAC;4BAClB,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,EAAE,GAAG,KAAK,CAAC;4BACjC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,SAAS;gCACzC,KAAK;6BACN,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,KAAK,aAAa,CAAC,CAAC,CAAC;4BACnB,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,EAAE,GAAG,KAAK,CAAC;4BACjC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,UAAU;gCAC1C,KAAK;6BACN,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;4BACtB,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,EAAE,GAAG,KAAK,CAAC;4BACjC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,aAAa;gCAC7C,KAAK;6BACN,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,KAAK,OAAO,CAAC,CAAC,CAAC;4BACb,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,KAAK;gCACrC,KAAK,EAAE,KAAK,CAAC,KAAK;6BACnB,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,KAAK,aAAa,CAAC,CAAC,CAAC;4BACnB,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,UAAU;gCAC1C,KAAK,EAAE,KAAK,CAAC,WAAW;6BACzB,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,KAAK,MAAM,CAAC,CAAC,CAAC;4BACZ,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,IAAI;gCACpC,KAAK,EAAE,KAAK,CAAC,IAAI;6BAClB,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBAED,KAAK,cAAc,CAAC,CAAC,CAAC;4BACpB,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,yBAAyB,CAAC,wBAAwB;gCACxD,KAAK,EAAE,KAAK,CAAC,UAAU;6BACxB,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBAED,sBAAsB;wBACtB,kGAAkG;wBAClG,KAAK,4BAA4B,CAAC;wBAClC,KAAK,aAAa;4BAChB,MAAM;wBAER,OAAO,CAAC,CAAC,CAAC;4BACR,MAAM,eAAe,GAAU,IAAI,CAAC;4BACpC,MAAM,IAAI,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;wBAChE,CAAC;oBACH,CAAC;gBACH,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,QAAQ;iBACZ,WAAW,CAAC,IAAI,4BAA4B,EAAE,CAAC;iBAC/C,WAAW,CAAC,SAAS,CAAC;iBACtB,WAAW,CAAC,IAAI,sBAAsB,EAAE,CAAC;iBACzC,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,6BAA6B,GAAG;IACpC,yBAAyB,CAAC,aAAa;IACvC,yBAAyB,CAAC,QAAQ;IAClC,yBAAyB,CAAC,SAAS;IACnC,yBAAyB,CAAC,cAAc;IACxC,yBAAyB,CAAC,MAAM;IAChC,yBAAyB,CAAC,KAAK;IAC/B,yBAAyB,CAAC,UAAU;IACpC,yBAAyB,CAAC,aAAa;IACvC,yBAAyB,CAAC,YAAY;IACtC,yBAAyB,CAAC,iBAAiB;IAC3C,yBAAyB,CAAC,WAAW;CACtC,CAAC;AAEF,MAAM,OAAO,iBAAkB,SAAQ,uBAGtC;IACC;QACE,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAoC,CAAC;YACxE,IAAI,sBAAwD,CAAC;YAC7D,MAAM,SAAS,GAAG,IAAI,wBAAwB,CAAkB;gBAC9D,SAAS,CAAC,KAAK,EAAE,UAAU;oBACzB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;oBAE9B,IAAI,6BAA6B,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;wBACjD,sBAAsB,EAAE,KAAK,EAAE,CAAC;wBAChC,sBAAsB,GAAG,SAAS,CAAC;oBACrC,CAAC;oBAED,QAAQ,IAAI,EAAE,CAAC;wBACb,KAAK,yBAAyB,CAAC,cAAc;4BAC3C,UAAU,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;4BAClC,MAAM;wBAER,KAAK,yBAAyB,CAAC,SAAS;4BACtC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;4BAC7B,MAAM;wBAER,KAAK,yBAAyB,CAAC,YAAY;4BACzC,UAAU;iCACP,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC;iCAC5B,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;4BAC/B,MAAM;wBAER,KAAK,yBAAyB,CAAC,iBAAiB;4BAC9C,UAAU;iCACP,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC;iCAC5B,eAAe,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;4BACzC,MAAM;wBAER,KAAK,yBAAyB,CAAC,aAAa,CAAC,CAAC,CAAC;4BAC7C,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;4BACjD,MAAM,IAAI,GAAG,QAAQ;gCACnB,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC;gCACnC,CAAC,CAAC,UAAU,CAAC;4BAEf,IAAI,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC;gCACrC,MAAM,IAAI,KAAK,CACb,uCAAuC,UAAU,EAAE,CACpD,CAAC;4BAEJ,MAAM,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC;gCAC9C,UAAU;gCACV,QAAQ;6BACT,CAAC,CAAC;4BACH,mBAAmB,CAAC,GAAG,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;4BAExD,sBAAsB,GAAG,kBAAkB,CAAC,QAAQ,CAAC;4BACrD,MAAM;wBACR,CAAC;wBAED,KAAK,yBAAyB,CAAC,qBAAqB,CAAC,CAAC,CAAC;4BACrD,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;4BAC5C,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;4BAC/D,IAAI,CAAC,kBAAkB;gCACrB,MAAM,IAAI,KAAK,CACb,0CAA0C,UAAU,EAAE,CACvD,CAAC;4BACJ,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;4BAClD,MAAM;wBACR,CAAC;wBAED,KAAK,yBAAyB,CAAC,cAAc,CAAC,CAAC,CAAC;4BAC9C,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;4BACxD,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;4BAC/D,IAAI,CAAC,kBAAkB;gCACrB,MAAM,IAAI,KAAK,CACb,iDAAiD,UAAU,EAAE,CAC9D,CAAC;4BACJ,kBAAkB,CAAC,WAAW,CAAC;gCAC7B,QAAQ;gCACR,MAAM;gCACN,OAAO;6BACR,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBAED,KAAK,yBAAyB,CAAC,QAAQ,CAAC,CAAC,CAAC;4BACxC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;4BAE7C,IAAI,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;4BAC7D,IAAI,kBAAkB,EAAE,CAAC;gCACvB,kBAAkB,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;4BACtC,CAAC;iCAAM,CAAC;gCACN,kBAAkB,GAAG,UAAU,CAAC,eAAe,CAAC;oCAC9C,UAAU;oCACV,QAAQ;oCACR,IAAI;iCACL,CAAC,CAAC;gCACH,mBAAmB,CAAC,GAAG,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;4BAC1D,CAAC;4BACD,MAAM;wBACR,CAAC;wBAED,KAAK,yBAAyB,CAAC,aAAa;4BAC1C,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,gBAAgB;gCACtB,IAAI,EAAE,EAAE;gCACR,GAAG,KAAK;6BACT,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,SAAS;4BACtC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,YAAY;gCAClB,IAAI,EAAE,EAAE;gCACR,GAAG,KAAK;6BACT,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,UAAU;4BACvC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,aAAa;gCACnB,IAAI,EAAE,EAAE;gCACR,GAAG,KAAK;6BACT,CAAC,CAAC;4BACH,MAAM;wBACR,KAAK,yBAAyB,CAAC,IAAI;4BACjC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,EAAE;gCACR,IAAI,EAAE,KAAK;6BACZ,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,UAAU;4BACvC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,aAAa;gCACnB,IAAI,EAAE,EAAE;gCACR,WAAW,EAAE,KAAK;6BACnB,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC;4BACtC,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,EAAE,GAAG,KAAK,CAAC;4BAC1C,MAAM,IAAI,GAAG,QAAQ;gCACnB,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC;gCACnC,CAAC,CAAC,UAAU,CAAC;4BACf,IAAI,CAAC,YAAY,CAAC;gCAChB,IAAI,EAAE,QAAQ;gCACd,GAAG,UAAU;6BACd,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBAED,KAAK,yBAAyB,CAAC,KAAK;4BAClC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,OAAO;gCACb,IAAI,EAAE,EAAE;gCACR,KAAK,EAAE,KAAK;6BACb,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,IAAI;4BACjC,UAAU,CAAC,UAAU,CAAC;gCACpB,IAAI,EAAE,MAAM;gCACZ,GAAG,KAAK;6BACT,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,WAAW;4BACxC,UAAU,CAAC,UAAU,CAAC;gCACpB,IAAI,EAAE,MAAM;gCACZ,GAAG,KAAK;6BACT,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,wBAAwB;4BACrD,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,cAAc;gCACpB,IAAI,EAAE,EAAE;gCACR,UAAU,EAAE,KAAK;6BAClB,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,yBAAyB,CAAC,kBAAkB,CAAC;wBAClD,KAAK,yBAAyB,CAAC,iBAAiB;4BAC9C,uBAAuB;4BACvB,MAAM;wBAER,OAAO,CAAC,CAAC,CAAC;4BACR,MAAM,eAAe,GAAU,IAAI,CAAC;4BACpC,MAAM,IAAI,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;wBAChE,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,KAAK;oBACH,sBAAsB,EAAE,KAAK,EAAE,CAAC;oBAChC,sBAAsB,GAAG,SAAS,CAAC;oBACnC,uGAAuG;oBACvG,mBAAmB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;oBAChE,mBAAmB,CAAC,KAAK,EAAE,CAAC;gBAC9B,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,QAAQ;iBACZ,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;iBACpC,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;iBACpC,WAAW,CAAC,IAAI,sBAAsB,EAAE,CAAC;iBACzC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
1
|
+
{"version":3,"file":"DataStream.js","names":["exhaustiveCheck"],"sources":["../../../../src/core/serialization/data-stream/DataStream.ts"],"sourcesContent":["import type { AssistantStreamChunk } from \"../../AssistantStreamChunk\";\nimport type { ToolCallStreamController } from \"../../modules/tool-call\";\nimport { AssistantTransformStream } from \"../../utils/stream/AssistantTransformStream\";\nimport { PipeableTransformStream } from \"../../utils/stream/PipeableTransformStream\";\nimport { type DataStreamChunk, DataStreamStreamChunkType } from \"./chunk-types\";\nimport { LineDecoderStream } from \"../../utils/stream/LineDecoderStream\";\nimport {\n DataStreamChunkDecoder,\n DataStreamChunkEncoder,\n} from \"./serialization\";\nimport {\n type AssistantMetaStreamChunk,\n AssistantMetaTransformStream,\n} from \"../../utils/stream/AssistantMetaTransformStream\";\nimport type { TextStreamController } from \"../../modules/text\";\nimport type { AssistantStreamEncoder } from \"../../AssistantStream\";\n\nexport class DataStreamEncoder\n extends PipeableTransformStream<AssistantStreamChunk, Uint8Array<ArrayBuffer>>\n implements AssistantStreamEncoder\n{\n headers = new Headers({\n \"Content-Type\": \"text/plain; charset=utf-8\",\n \"x-vercel-ai-data-stream\": \"v1\",\n });\n\n constructor() {\n super((readable) => {\n const transform = new TransformStream<\n AssistantMetaStreamChunk,\n DataStreamChunk\n >({\n transform(chunk, controller) {\n const type = chunk.type;\n switch (type) {\n case \"part-start\": {\n const part = chunk.part;\n if (part.type === \"tool-call\") {\n const { type, ...value } = part;\n controller.enqueue({\n type: DataStreamStreamChunkType.StartToolCall,\n value,\n });\n }\n if (part.type === \"source\") {\n const { type, ...value } = part;\n controller.enqueue({\n type: DataStreamStreamChunkType.Source,\n value,\n });\n }\n if (part.type === \"data\") {\n const { type, ...value } = part;\n controller.enqueue({\n type: DataStreamStreamChunkType.AuiDataPart,\n value,\n });\n }\n break;\n }\n case \"text-delta\": {\n const part = chunk.meta;\n switch (part.type) {\n case \"text\": {\n if (part.parentId) {\n controller.enqueue({\n type: DataStreamStreamChunkType.AuiTextDelta,\n value: {\n textDelta: chunk.textDelta,\n parentId: part.parentId,\n },\n });\n } else {\n controller.enqueue({\n type: DataStreamStreamChunkType.TextDelta,\n value: chunk.textDelta,\n });\n }\n break;\n }\n case \"reasoning\": {\n if (part.parentId) {\n controller.enqueue({\n type: DataStreamStreamChunkType.AuiReasoningDelta,\n value: {\n reasoningDelta: chunk.textDelta,\n parentId: part.parentId,\n },\n });\n } else {\n controller.enqueue({\n type: DataStreamStreamChunkType.ReasoningDelta,\n value: chunk.textDelta,\n });\n }\n break;\n }\n case \"tool-call\": {\n controller.enqueue({\n type: DataStreamStreamChunkType.ToolCallArgsTextDelta,\n value: {\n toolCallId: part.toolCallId,\n argsTextDelta: chunk.textDelta,\n },\n });\n break;\n }\n default:\n throw new Error(\n `Unsupported part type for text-delta: ${part.type}`,\n );\n }\n break;\n }\n case \"result\": {\n // Only tool-call parts can have results.\n const part = chunk.meta;\n if (part.type !== \"tool-call\") {\n throw new Error(\n `Result chunk on non-tool-call part not supported: ${part.type}`,\n );\n }\n controller.enqueue({\n type: DataStreamStreamChunkType.ToolCallResult,\n value: {\n toolCallId: part.toolCallId,\n result: chunk.result,\n artifact: chunk.artifact,\n ...(chunk.isError ? { isError: chunk.isError } : {}),\n },\n });\n break;\n }\n case \"step-start\": {\n const { type, ...value } = chunk;\n controller.enqueue({\n type: DataStreamStreamChunkType.StartStep,\n value,\n });\n break;\n }\n case \"step-finish\": {\n const { type, ...value } = chunk;\n controller.enqueue({\n type: DataStreamStreamChunkType.FinishStep,\n value,\n });\n break;\n }\n case \"message-finish\": {\n const { type, ...value } = chunk;\n controller.enqueue({\n type: DataStreamStreamChunkType.FinishMessage,\n value,\n });\n break;\n }\n case \"error\": {\n controller.enqueue({\n type: DataStreamStreamChunkType.Error,\n value: chunk.error,\n });\n break;\n }\n case \"annotations\": {\n controller.enqueue({\n type: DataStreamStreamChunkType.Annotation,\n value: chunk.annotations,\n });\n break;\n }\n case \"data\": {\n controller.enqueue({\n type: DataStreamStreamChunkType.Data,\n value: chunk.data,\n });\n break;\n }\n\n case \"update-state\": {\n controller.enqueue({\n type: DataStreamStreamChunkType.AuiUpdateStateOperations,\n value: chunk.operations,\n });\n break;\n }\n\n // TODO ignore for now\n // in the future, we should create a handler that waits for text parts to finish before continuing\n case \"tool-call-args-text-finish\":\n case \"part-finish\":\n break;\n\n default: {\n const exhaustiveCheck: never = type;\n throw new Error(`Unsupported chunk type: ${exhaustiveCheck}`);\n }\n }\n },\n });\n\n return readable\n .pipeThrough(new AssistantMetaTransformStream())\n .pipeThrough(transform)\n .pipeThrough(new DataStreamChunkEncoder())\n .pipeThrough(new TextEncoderStream());\n });\n }\n}\n\nconst TOOL_CALL_ARGS_CLOSING_CHUNKS = [\n DataStreamStreamChunkType.StartToolCall,\n DataStreamStreamChunkType.ToolCall,\n DataStreamStreamChunkType.TextDelta,\n DataStreamStreamChunkType.ReasoningDelta,\n DataStreamStreamChunkType.Source,\n DataStreamStreamChunkType.Error,\n DataStreamStreamChunkType.FinishStep,\n DataStreamStreamChunkType.FinishMessage,\n DataStreamStreamChunkType.AuiTextDelta,\n DataStreamStreamChunkType.AuiReasoningDelta,\n DataStreamStreamChunkType.AuiDataPart,\n];\n\nexport class DataStreamDecoder extends PipeableTransformStream<\n Uint8Array<ArrayBuffer>,\n AssistantStreamChunk\n> {\n constructor() {\n super((readable) => {\n const toolCallControllers = new Map<string, ToolCallStreamController>();\n let activeToolCallArgsText: TextStreamController | undefined;\n const transform = new AssistantTransformStream<DataStreamChunk>({\n transform(chunk, controller) {\n const { type, value } = chunk;\n\n if (TOOL_CALL_ARGS_CLOSING_CHUNKS.includes(type)) {\n activeToolCallArgsText?.close();\n activeToolCallArgsText = undefined;\n }\n\n switch (type) {\n case DataStreamStreamChunkType.ReasoningDelta:\n controller.appendReasoning(value);\n break;\n\n case DataStreamStreamChunkType.TextDelta:\n controller.appendText(value);\n break;\n\n case DataStreamStreamChunkType.AuiTextDelta:\n controller\n .withParentId(value.parentId)\n .appendText(value.textDelta);\n break;\n\n case DataStreamStreamChunkType.AuiReasoningDelta:\n controller\n .withParentId(value.parentId)\n .appendReasoning(value.reasoningDelta);\n break;\n\n case DataStreamStreamChunkType.StartToolCall: {\n const { toolCallId, toolName, parentId } = value;\n const ctrl = parentId\n ? controller.withParentId(parentId)\n : controller;\n\n if (toolCallControllers.has(toolCallId))\n throw new Error(\n `Encountered duplicate tool call id: ${toolCallId}`,\n );\n\n const toolCallController = ctrl.addToolCallPart({\n toolCallId,\n toolName,\n });\n toolCallControllers.set(toolCallId, toolCallController);\n\n activeToolCallArgsText = toolCallController.argsText;\n break;\n }\n\n case DataStreamStreamChunkType.ToolCallArgsTextDelta: {\n const { toolCallId, argsTextDelta } = value;\n const toolCallController = toolCallControllers.get(toolCallId);\n if (!toolCallController)\n throw new Error(\n `Encountered tool call with unknown id: ${toolCallId}`,\n );\n toolCallController.argsText.append(argsTextDelta);\n break;\n }\n\n case DataStreamStreamChunkType.ToolCallResult: {\n const { toolCallId, artifact, result, isError } = value;\n const toolCallController = toolCallControllers.get(toolCallId);\n if (!toolCallController)\n throw new Error(\n `Encountered tool call result with unknown id: ${toolCallId}`,\n );\n toolCallController.setResponse({\n artifact,\n result,\n isError,\n });\n break;\n }\n\n case DataStreamStreamChunkType.ToolCall: {\n const { toolCallId, toolName, args } = value;\n\n let toolCallController = toolCallControllers.get(toolCallId);\n if (toolCallController) {\n toolCallController.argsText.close();\n } else {\n toolCallController = controller.addToolCallPart({\n toolCallId,\n toolName,\n args,\n });\n toolCallControllers.set(toolCallId, toolCallController);\n }\n break;\n }\n\n case DataStreamStreamChunkType.FinishMessage:\n controller.enqueue({\n type: \"message-finish\",\n path: [],\n ...value,\n });\n break;\n\n case DataStreamStreamChunkType.StartStep:\n controller.enqueue({\n type: \"step-start\",\n path: [],\n ...value,\n });\n break;\n\n case DataStreamStreamChunkType.FinishStep:\n controller.enqueue({\n type: \"step-finish\",\n path: [],\n ...value,\n });\n break;\n case DataStreamStreamChunkType.Data:\n controller.enqueue({\n type: \"data\",\n path: [],\n data: value,\n });\n break;\n\n case DataStreamStreamChunkType.Annotation:\n controller.enqueue({\n type: \"annotations\",\n path: [],\n annotations: value,\n });\n break;\n\n case DataStreamStreamChunkType.Source: {\n const { parentId, ...sourceData } = value;\n const ctrl = parentId\n ? controller.withParentId(parentId)\n : controller;\n ctrl.appendSource({\n type: \"source\",\n ...sourceData,\n });\n break;\n }\n\n case DataStreamStreamChunkType.Error:\n controller.enqueue({\n type: \"error\",\n path: [],\n error: value,\n });\n break;\n\n case DataStreamStreamChunkType.File:\n controller.appendFile({\n type: \"file\",\n ...value,\n });\n break;\n\n case DataStreamStreamChunkType.AuiDataPart:\n controller.appendData({\n type: \"data\",\n ...value,\n });\n break;\n\n case DataStreamStreamChunkType.AuiUpdateStateOperations:\n controller.enqueue({\n type: \"update-state\",\n path: [],\n operations: value,\n });\n break;\n\n case DataStreamStreamChunkType.ReasoningSignature:\n case DataStreamStreamChunkType.RedactedReasoning:\n // ignore these for now\n break;\n\n default: {\n const exhaustiveCheck: never = type;\n throw new Error(`unsupported chunk type: ${exhaustiveCheck}`);\n }\n }\n },\n flush() {\n activeToolCallArgsText?.close();\n activeToolCallArgsText = undefined;\n // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return\n toolCallControllers.forEach((controller) => controller.close());\n toolCallControllers.clear();\n },\n });\n\n return readable\n .pipeThrough(new TextDecoderStream())\n .pipeThrough(new LineDecoderStream())\n .pipeThrough(new DataStreamChunkDecoder())\n .pipeThrough(transform);\n });\n }\n}\n"],"mappings":";;;;;;;AAiBA,IAAa,oBAAb,cACU,wBAEV;CACE,UAAU,IAAI,QAAQ;EACpB,gBAAgB;EAChB,2BAA2B;CAC7B,CAAC;CAED,cAAc;EACZ,OAAO,aAAa;GAClB,MAAM,YAAY,IAAI,gBAGpB,EACA,UAAU,OAAO,YAAY;IAC3B,MAAM,OAAO,MAAM;IACnB,QAAQ,MAAR;KACE,KAAK,cAAc;MACjB,MAAM,OAAO,MAAM;MACnB,IAAI,KAAK,SAAS,aAAa;OAC7B,MAAM,EAAE,MAAM,GAAG,UAAU;OAC3B,WAAW,QAAQ;QACjB,MAAA;QACA;OACF,CAAC;MACH;MACA,IAAI,KAAK,SAAS,UAAU;OAC1B,MAAM,EAAE,MAAM,GAAG,UAAU;OAC3B,WAAW,QAAQ;QACjB,MAAA;QACA;OACF,CAAC;MACH;MACA,IAAI,KAAK,SAAS,QAAQ;OACxB,MAAM,EAAE,MAAM,GAAG,UAAU;OAC3B,WAAW,QAAQ;QACjB,MAAA;QACA;OACF,CAAC;MACH;MACA;KACF;KACA,KAAK,cAAc;MACjB,MAAM,OAAO,MAAM;MACnB,QAAQ,KAAK,MAAb;OACE,KAAK;QACH,IAAI,KAAK,UACP,WAAW,QAAQ;SACjB,MAAA;SACA,OAAO;UACL,WAAW,MAAM;UACjB,UAAU,KAAK;SACjB;QACF,CAAC;aAED,WAAW,QAAQ;SACjB,MAAA;SACA,OAAO,MAAM;QACf,CAAC;QAEH;OAEF,KAAK;QACH,IAAI,KAAK,UACP,WAAW,QAAQ;SACjB,MAAA;SACA,OAAO;UACL,gBAAgB,MAAM;UACtB,UAAU,KAAK;SACjB;QACF,CAAC;aAED,WAAW,QAAQ;SACjB,MAAA;SACA,OAAO,MAAM;QACf,CAAC;QAEH;OAEF,KAAK;QACH,WAAW,QAAQ;SACjB,MAAA;SACA,OAAO;UACL,YAAY,KAAK;UACjB,eAAe,MAAM;SACvB;QACF,CAAC;QACD;OAEF,SACE,MAAM,IAAI,MACR,yCAAyC,KAAK,MAChD;MACJ;MACA;KACF;KACA,KAAK,UAAU;MAEb,MAAM,OAAO,MAAM;MACnB,IAAI,KAAK,SAAS,aAChB,MAAM,IAAI,MACR,qDAAqD,KAAK,MAC5D;MAEF,WAAW,QAAQ;OACjB,MAAA;OACA,OAAO;QACL,YAAY,KAAK;QACjB,QAAQ,MAAM;QACd,UAAU,MAAM;QAChB,GAAI,MAAM,UAAU,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;OACpD;MACF,CAAC;MACD;KACF;KACA,KAAK,cAAc;MACjB,MAAM,EAAE,MAAM,GAAG,UAAU;MAC3B,WAAW,QAAQ;OACjB,MAAA;OACA;MACF,CAAC;MACD;KACF;KACA,KAAK,eAAe;MAClB,MAAM,EAAE,MAAM,GAAG,UAAU;MAC3B,WAAW,QAAQ;OACjB,MAAA;OACA;MACF,CAAC;MACD;KACF;KACA,KAAK,kBAAkB;MACrB,MAAM,EAAE,MAAM,GAAG,UAAU;MAC3B,WAAW,QAAQ;OACjB,MAAA;OACA;MACF,CAAC;MACD;KACF;KACA,KAAK;MACH,WAAW,QAAQ;OACjB,MAAA;OACA,OAAO,MAAM;MACf,CAAC;MACD;KAEF,KAAK;MACH,WAAW,QAAQ;OACjB,MAAA;OACA,OAAO,MAAM;MACf,CAAC;MACD;KAEF,KAAK;MACH,WAAW,QAAQ;OACjB,MAAA;OACA,OAAO,MAAM;MACf,CAAC;MACD;KAGF,KAAK;MACH,WAAW,QAAQ;OACjB,MAAA;OACA,OAAO,MAAM;MACf,CAAC;MACD;KAKF,KAAK;KACL,KAAK,eACH;KAEF,SAEE,MAAM,IAAI,MAAM,2BAA2BA,MAAiB;IAEhE;GACF,EACF,CAAC;GAED,OAAO,SACJ,YAAY,IAAI,6BAA6B,CAAC,EAC9C,YAAY,SAAS,EACrB,YAAY,IAAI,uBAAuB,CAAC,EACxC,YAAY,IAAI,kBAAkB,CAAC;EACxC,CAAC;CACH;AACF;AAEA,MAAM,gCAAgC;;;;;;;;;;;;AAYtC;AAEA,IAAa,oBAAb,cAAuC,wBAGrC;CACA,cAAc;EACZ,OAAO,aAAa;GAClB,MAAM,sCAAsB,IAAI,IAAsC;GACtE,IAAI;GACJ,MAAM,YAAY,IAAI,yBAA0C;IAC9D,UAAU,OAAO,YAAY;KAC3B,MAAM,EAAE,MAAM,UAAU;KAExB,IAAI,8BAA8B,SAAS,IAAI,GAAG;MAChD,wBAAwB,MAAM;MAC9B,yBAAyB,KAAA;KAC3B;KAEA,QAAQ,MAAR;MACE,KAAA;OACE,WAAW,gBAAgB,KAAK;OAChC;MAEF,KAAA;OACE,WAAW,WAAW,KAAK;OAC3B;MAEF,KAAA;OACE,WACG,aAAa,MAAM,QAAQ,EAC3B,WAAW,MAAM,SAAS;OAC7B;MAEF,KAAA;OACE,WACG,aAAa,MAAM,QAAQ,EAC3B,gBAAgB,MAAM,cAAc;OACvC;MAEF,KAAA,KAA8C;OAC5C,MAAM,EAAE,YAAY,UAAU,aAAa;OAC3C,MAAM,OAAO,WACT,WAAW,aAAa,QAAQ,IAChC;OAEJ,IAAI,oBAAoB,IAAI,UAAU,GACpC,MAAM,IAAI,MACR,uCAAuC,YACzC;OAEF,MAAM,qBAAqB,KAAK,gBAAgB;QAC9C;QACA;OACF,CAAC;OACD,oBAAoB,IAAI,YAAY,kBAAkB;OAEtD,yBAAyB,mBAAmB;OAC5C;MACF;MAEA,KAAA,KAAsD;OACpD,MAAM,EAAE,YAAY,kBAAkB;OACtC,MAAM,qBAAqB,oBAAoB,IAAI,UAAU;OAC7D,IAAI,CAAC,oBACH,MAAM,IAAI,MACR,0CAA0C,YAC5C;OACF,mBAAmB,SAAS,OAAO,aAAa;OAChD;MACF;MAEA,KAAA,KAA+C;OAC7C,MAAM,EAAE,YAAY,UAAU,QAAQ,YAAY;OAClD,MAAM,qBAAqB,oBAAoB,IAAI,UAAU;OAC7D,IAAI,CAAC,oBACH,MAAM,IAAI,MACR,iDAAiD,YACnD;OACF,mBAAmB,YAAY;QAC7B;QACA;QACA;OACF,CAAC;OACD;MACF;MAEA,KAAA,KAAyC;OACvC,MAAM,EAAE,YAAY,UAAU,SAAS;OAEvC,IAAI,qBAAqB,oBAAoB,IAAI,UAAU;OAC3D,IAAI,oBACF,mBAAmB,SAAS,MAAM;YAC7B;QACL,qBAAqB,WAAW,gBAAgB;SAC9C;SACA;SACA;QACF,CAAC;QACD,oBAAoB,IAAI,YAAY,kBAAkB;OACxD;OACA;MACF;MAEA,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,GAAG;OACL,CAAC;OACD;MAEF,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,GAAG;OACL,CAAC;OACD;MAEF,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,GAAG;OACL,CAAC;OACD;MACF,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,MAAM;OACR,CAAC;OACD;MAEF,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,aAAa;OACf,CAAC;OACD;MAEF,KAAA,KAAuC;OACrC,MAAM,EAAE,UAAU,GAAG,eAAe;OAIpC,CAHa,WACT,WAAW,aAAa,QAAQ,IAChC,YACC,aAAa;QAChB,MAAM;QACN,GAAG;OACL,CAAC;OACD;MACF;MAEA,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,OAAO;OACT,CAAC;OACD;MAEF,KAAA;OACE,WAAW,WAAW;QACpB,MAAM;QACN,GAAG;OACL,CAAC;OACD;MAEF,KAAA;OACE,WAAW,WAAW;QACpB,MAAM;QACN,GAAG;OACL,CAAC;OACD;MAEF,KAAA;OACE,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,YAAY;OACd,CAAC;OACD;MAEF,KAAA;MACA,KAAA,KAEE;MAEF,SAEE,MAAM,IAAI,MAAM,2BAA2BA,MAAiB;KAEhE;IACF;IACA,QAAQ;KACN,wBAAwB,MAAM;KAC9B,yBAAyB,KAAA;KAEzB,oBAAoB,SAAS,eAAe,WAAW,MAAM,CAAC;KAC9D,oBAAoB,MAAM;IAC5B;GACF,CAAC;GAED,OAAO,SACJ,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,uBAAuB,CAAC,EACxC,YAAY,SAAS;EAC1B,CAAC;CACH;AACF"}
|
|
@@ -1,107 +1,108 @@
|
|
|
1
|
-
import
|
|
2
|
-
import
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
}[DataStreamStreamChunkType];
|
|
1
|
+
import { ReadonlyJSONObject, ReadonlyJSONValue } from "../../../utils/json/json-value.js";
|
|
2
|
+
import { ObjectStreamOperation } from "../../object/types.js";
|
|
3
|
+
|
|
4
|
+
//#region src/core/serialization/data-stream/chunk-types.d.ts
|
|
5
|
+
type DataStreamChunk = { [K in DataStreamStreamChunkType]: {
|
|
6
|
+
type: K;
|
|
7
|
+
value: DataStreamStreamChunkValue[K];
|
|
8
|
+
} }[DataStreamStreamChunkType];
|
|
9
9
|
type LanguageModelV1FinishReason = "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown";
|
|
10
10
|
type LanguageModelV1Usage = {
|
|
11
|
-
|
|
12
|
-
|
|
11
|
+
inputTokens: number;
|
|
12
|
+
outputTokens: number;
|
|
13
13
|
};
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
14
|
+
declare enum DataStreamStreamChunkType {
|
|
15
|
+
TextDelta = "0",
|
|
16
|
+
Data = "2",
|
|
17
|
+
Error = "3",
|
|
18
|
+
Annotation = "8",
|
|
19
|
+
ToolCall = "9",
|
|
20
|
+
ToolCallResult = "a",
|
|
21
|
+
StartToolCall = "b",
|
|
22
|
+
ToolCallArgsTextDelta = "c",
|
|
23
|
+
FinishMessage = "d",
|
|
24
|
+
FinishStep = "e",
|
|
25
|
+
StartStep = "f",
|
|
26
|
+
ReasoningDelta = "g",
|
|
27
|
+
Source = "h",
|
|
28
|
+
RedactedReasoning = "i",
|
|
29
|
+
ReasoningSignature = "j",
|
|
30
|
+
File = "k",
|
|
31
|
+
AuiUpdateStateOperations = "aui-state",
|
|
32
|
+
AuiTextDelta = "aui-text-delta",
|
|
33
|
+
AuiReasoningDelta = "aui-reasoning-delta",
|
|
34
|
+
AuiDataPart = "aui-data"
|
|
35
35
|
}
|
|
36
36
|
type DataStreamStreamChunkValue = {
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
37
|
+
[DataStreamStreamChunkType.TextDelta]: string;
|
|
38
|
+
[DataStreamStreamChunkType.Data]: ReadonlyJSONValue[];
|
|
39
|
+
[DataStreamStreamChunkType.Annotation]: ReadonlyJSONValue[];
|
|
40
|
+
[DataStreamStreamChunkType.ToolCall]: {
|
|
41
|
+
toolCallId: string;
|
|
42
|
+
toolName: string;
|
|
43
|
+
args: ReadonlyJSONObject;
|
|
44
|
+
};
|
|
45
|
+
[DataStreamStreamChunkType.StartToolCall]: {
|
|
46
|
+
toolCallId: string;
|
|
47
|
+
toolName: string;
|
|
48
|
+
parentId?: string;
|
|
49
|
+
};
|
|
50
|
+
[DataStreamStreamChunkType.ToolCallArgsTextDelta]: {
|
|
51
|
+
toolCallId: string;
|
|
52
|
+
argsTextDelta: string;
|
|
53
|
+
};
|
|
54
|
+
[DataStreamStreamChunkType.ToolCallResult]: {
|
|
55
|
+
toolCallId: string;
|
|
56
|
+
result: ReadonlyJSONValue;
|
|
57
|
+
artifact?: ReadonlyJSONValue | undefined;
|
|
58
|
+
isError?: boolean;
|
|
59
|
+
};
|
|
60
|
+
[DataStreamStreamChunkType.Error]: string;
|
|
61
|
+
[DataStreamStreamChunkType.FinishStep]: {
|
|
62
|
+
finishReason: LanguageModelV1FinishReason;
|
|
63
|
+
usage: LanguageModelV1Usage;
|
|
64
|
+
isContinued: boolean;
|
|
65
|
+
};
|
|
66
|
+
[DataStreamStreamChunkType.FinishMessage]: {
|
|
67
|
+
finishReason: LanguageModelV1FinishReason;
|
|
68
|
+
usage: LanguageModelV1Usage;
|
|
69
|
+
};
|
|
70
|
+
[DataStreamStreamChunkType.StartStep]: {
|
|
71
|
+
messageId: string;
|
|
72
|
+
};
|
|
73
|
+
[DataStreamStreamChunkType.ReasoningDelta]: string;
|
|
74
|
+
[DataStreamStreamChunkType.Source]: {
|
|
75
|
+
sourceType: "url";
|
|
76
|
+
id: string;
|
|
77
|
+
url: string;
|
|
78
|
+
title?: string;
|
|
79
|
+
parentId?: string;
|
|
80
|
+
};
|
|
81
|
+
[DataStreamStreamChunkType.RedactedReasoning]: {
|
|
82
|
+
data: string;
|
|
83
|
+
};
|
|
84
|
+
[DataStreamStreamChunkType.ReasoningSignature]: {
|
|
85
|
+
signature: string;
|
|
86
|
+
};
|
|
87
|
+
[DataStreamStreamChunkType.File]: {
|
|
88
|
+
data: string;
|
|
89
|
+
mimeType: string;
|
|
90
|
+
};
|
|
91
|
+
[DataStreamStreamChunkType.AuiUpdateStateOperations]: ObjectStreamOperation[];
|
|
92
|
+
[DataStreamStreamChunkType.AuiTextDelta]: {
|
|
93
|
+
textDelta: string;
|
|
94
|
+
parentId: string;
|
|
95
|
+
};
|
|
96
|
+
[DataStreamStreamChunkType.AuiReasoningDelta]: {
|
|
97
|
+
reasoningDelta: string;
|
|
98
|
+
parentId: string;
|
|
99
|
+
};
|
|
100
|
+
[DataStreamStreamChunkType.AuiDataPart]: {
|
|
101
|
+
name: string;
|
|
102
|
+
data: ReadonlyJSONValue;
|
|
103
|
+
parentId?: string;
|
|
104
|
+
};
|
|
105
105
|
};
|
|
106
|
-
|
|
106
|
+
//#endregion
|
|
107
|
+
export { DataStreamChunk, DataStreamStreamChunkType };
|
|
107
108
|
//# sourceMappingURL=chunk-types.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chunk-types.d.ts","
|
|
1
|
+
{"version":3,"file":"chunk-types.d.ts","names":[],"sources":["../../../../src/core/serialization/data-stream/chunk-types.ts"],"mappings":";;;;KAMY,eAAA,WACJ,yBAAA;EACJ,IAAA,EAAM,CAAA;EACN,KAAA,EAAO,0BAAA,CAA2B,CAAA;AAAA,IAEpC,yBAAA;AAAA,KAEG,2BAAA;AAAA,KASA,oBAAA;EACH,WAAA;EACA,YAAY;AAAA;AAAA,aAGF,yBAAA;EACV,SAAA;EACA,IAAA;EACA,KAAA;EACA,UAAA;EACA,QAAA;EACA,cAAA;EACA,aAAA;EACA,qBAAA;EACA,aAAA;EACA,UAAA;EACA,SAAA;EACA,cAAA;EACA,MAAA;EACA,iBAAA;EACA,kBAAA;EACA,IAAA;EAEA,wBAAA;EACA,YAAA;EACA,iBAAA;EACA,WAAA;AAAA;AAAA,KAEG,0BAAA;EAAA,CACF,yBAAA,CAA0B,SAAA;EAAA,CAC1B,yBAAA,CAA0B,IAAA,GAAO,iBAAA;EAAA,CACjC,yBAAA,CAA0B,UAAA,GAAa,iBAAA;EAAA,CACvC,yBAAA,CAA0B,QAAA;IACzB,UAAA;IACA,QAAA;IACA,IAAA,EAAM,kBAAA;EAAA;EAAA,CAEP,yBAAA,CAA0B,aAAA;IACzB,UAAA;IACA,QAAA;IACA,QAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,qBAAA;IACzB,UAAA;IACA,aAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,cAAA;IACzB,UAAA;IACA,MAAA,EAAQ,iBAAA;IAGR,QAAA,GAAW,iBAAA;IACX,OAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,KAAA;EAAA,CAC1B,yBAAA,CAA0B,UAAA;IACzB,YAAA,EAAc,2BAAA;IACd,KAAA,EAAO,oBAAA;IACP,WAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,aAAA;IACzB,YAAA,EAAc,2BAAA;IACd,KAAA,EAAO,oBAAA;EAAA;EAAA,CAER,yBAAA,CAA0B,SAAA;IACzB,SAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,cAAA;EAAA,CAC1B,yBAAA,CAA0B,MAAA;IACzB,UAAA;IACA,EAAA;IACA,GAAA;IACA,KAAA;IACA,QAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,iBAAA;IAAsB,IAAA;EAAA;EAAA,CAChD,yBAAA,CAA0B,kBAAA;IAAuB,SAAA;EAAA;EAAA,CACjD,yBAAA,CAA0B,IAAA;IAAS,IAAA;IAAc,QAAA;EAAA;EAAA,CAGjD,yBAAA,CAA0B,wBAAA,GAA2B,qBAAA;EAAA,CACrD,yBAAA,CAA0B,YAAA;IACzB,SAAA;IACA,QAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,iBAAA;IACzB,cAAA;IACA,QAAA;EAAA;EAAA,CAED,yBAAA,CAA0B,WAAA;IACzB,IAAA;IACA,IAAA,EAAM,iBAAA;IACN,QAAA;EAAA;AAAA"}
|
|
@@ -1,24 +1,28 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
1
|
+
//#region src/core/serialization/data-stream/chunk-types.ts
|
|
2
|
+
let DataStreamStreamChunkType = /* @__PURE__ */ function(DataStreamStreamChunkType) {
|
|
3
|
+
DataStreamStreamChunkType["TextDelta"] = "0";
|
|
4
|
+
DataStreamStreamChunkType["Data"] = "2";
|
|
5
|
+
DataStreamStreamChunkType["Error"] = "3";
|
|
6
|
+
DataStreamStreamChunkType["Annotation"] = "8";
|
|
7
|
+
DataStreamStreamChunkType["ToolCall"] = "9";
|
|
8
|
+
DataStreamStreamChunkType["ToolCallResult"] = "a";
|
|
9
|
+
DataStreamStreamChunkType["StartToolCall"] = "b";
|
|
10
|
+
DataStreamStreamChunkType["ToolCallArgsTextDelta"] = "c";
|
|
11
|
+
DataStreamStreamChunkType["FinishMessage"] = "d";
|
|
12
|
+
DataStreamStreamChunkType["FinishStep"] = "e";
|
|
13
|
+
DataStreamStreamChunkType["StartStep"] = "f";
|
|
14
|
+
DataStreamStreamChunkType["ReasoningDelta"] = "g";
|
|
15
|
+
DataStreamStreamChunkType["Source"] = "h";
|
|
16
|
+
DataStreamStreamChunkType["RedactedReasoning"] = "i";
|
|
17
|
+
DataStreamStreamChunkType["ReasoningSignature"] = "j";
|
|
18
|
+
DataStreamStreamChunkType["File"] = "k";
|
|
19
|
+
DataStreamStreamChunkType["AuiUpdateStateOperations"] = "aui-state";
|
|
20
|
+
DataStreamStreamChunkType["AuiTextDelta"] = "aui-text-delta";
|
|
21
|
+
DataStreamStreamChunkType["AuiReasoningDelta"] = "aui-reasoning-delta";
|
|
22
|
+
DataStreamStreamChunkType["AuiDataPart"] = "aui-data";
|
|
23
|
+
return DataStreamStreamChunkType;
|
|
24
|
+
}({});
|
|
25
|
+
//#endregion
|
|
26
|
+
export { DataStreamStreamChunkType };
|
|
27
|
+
|
|
24
28
|
//# sourceMappingURL=chunk-types.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chunk-types.js","
|
|
1
|
+
{"version":3,"file":"chunk-types.js","names":[],"sources":["../../../../src/core/serialization/data-stream/chunk-types.ts"],"sourcesContent":["import type {\n ReadonlyJSONObject,\n ReadonlyJSONValue,\n} from \"../../../utils/json/json-value\";\nimport type { ObjectStreamOperation } from \"../../object/types\";\n\nexport type DataStreamChunk = {\n [K in DataStreamStreamChunkType]: {\n type: K;\n value: DataStreamStreamChunkValue[K];\n };\n}[DataStreamStreamChunkType];\n\ntype LanguageModelV1FinishReason =\n | \"stop\"\n | \"length\"\n | \"content-filter\"\n | \"tool-calls\"\n | \"error\"\n | \"other\"\n | \"unknown\";\n\ntype LanguageModelV1Usage = {\n inputTokens: number;\n outputTokens: number;\n};\n\nexport enum DataStreamStreamChunkType {\n TextDelta = \"0\",\n Data = \"2\",\n Error = \"3\",\n Annotation = \"8\",\n ToolCall = \"9\",\n ToolCallResult = \"a\",\n StartToolCall = \"b\",\n ToolCallArgsTextDelta = \"c\",\n FinishMessage = \"d\",\n FinishStep = \"e\",\n StartStep = \"f\",\n ReasoningDelta = \"g\",\n Source = \"h\",\n RedactedReasoning = \"i\",\n ReasoningSignature = \"j\",\n File = \"k\",\n\n AuiUpdateStateOperations = \"aui-state\",\n AuiTextDelta = \"aui-text-delta\",\n AuiReasoningDelta = \"aui-reasoning-delta\",\n AuiDataPart = \"aui-data\",\n}\ntype DataStreamStreamChunkValue = {\n [DataStreamStreamChunkType.TextDelta]: string;\n [DataStreamStreamChunkType.Data]: ReadonlyJSONValue[];\n [DataStreamStreamChunkType.Annotation]: ReadonlyJSONValue[];\n [DataStreamStreamChunkType.ToolCall]: {\n toolCallId: string;\n toolName: string;\n args: ReadonlyJSONObject;\n };\n [DataStreamStreamChunkType.StartToolCall]: {\n toolCallId: string;\n toolName: string;\n parentId?: string;\n };\n [DataStreamStreamChunkType.ToolCallArgsTextDelta]: {\n toolCallId: string;\n argsTextDelta: string;\n };\n [DataStreamStreamChunkType.ToolCallResult]: {\n toolCallId: string;\n result: ReadonlyJSONValue;\n\n // aui-extensions\n artifact?: ReadonlyJSONValue | undefined;\n isError?: boolean;\n };\n [DataStreamStreamChunkType.Error]: string;\n [DataStreamStreamChunkType.FinishStep]: {\n finishReason: LanguageModelV1FinishReason;\n usage: LanguageModelV1Usage;\n isContinued: boolean;\n };\n [DataStreamStreamChunkType.FinishMessage]: {\n finishReason: LanguageModelV1FinishReason;\n usage: LanguageModelV1Usage;\n };\n [DataStreamStreamChunkType.StartStep]: {\n messageId: string;\n };\n [DataStreamStreamChunkType.ReasoningDelta]: string;\n [DataStreamStreamChunkType.Source]: {\n sourceType: \"url\";\n id: string;\n url: string;\n title?: string;\n parentId?: string;\n };\n [DataStreamStreamChunkType.RedactedReasoning]: { data: string };\n [DataStreamStreamChunkType.ReasoningSignature]: { signature: string };\n [DataStreamStreamChunkType.File]: { data: string; mimeType: string };\n\n // aui-extensions\n [DataStreamStreamChunkType.AuiUpdateStateOperations]: ObjectStreamOperation[];\n [DataStreamStreamChunkType.AuiTextDelta]: {\n textDelta: string;\n parentId: string;\n };\n [DataStreamStreamChunkType.AuiReasoningDelta]: {\n reasoningDelta: string;\n parentId: string;\n };\n [DataStreamStreamChunkType.AuiDataPart]: {\n name: string;\n data: ReadonlyJSONValue;\n parentId?: string;\n };\n};\n"],"mappings":";AA2BA,IAAY,4BAAL,yBAAA,2BAAA;CACL,0BAAA,eAAA;CACA,0BAAA,UAAA;CACA,0BAAA,WAAA;CACA,0BAAA,gBAAA;CACA,0BAAA,cAAA;CACA,0BAAA,oBAAA;CACA,0BAAA,mBAAA;CACA,0BAAA,2BAAA;CACA,0BAAA,mBAAA;CACA,0BAAA,gBAAA;CACA,0BAAA,eAAA;CACA,0BAAA,oBAAA;CACA,0BAAA,YAAA;CACA,0BAAA,uBAAA;CACA,0BAAA,wBAAA;CACA,0BAAA,UAAA;CAEA,0BAAA,8BAAA;CACA,0BAAA,kBAAA;CACA,0BAAA,uBAAA;CACA,0BAAA,iBAAA;;AACF,EAAA,CAAA,CAAA"}
|
|
@@ -1,8 +1,12 @@
|
|
|
1
|
-
import
|
|
2
|
-
|
|
3
|
-
|
|
1
|
+
import { DataStreamChunk } from "./chunk-types.js";
|
|
2
|
+
|
|
3
|
+
//#region src/core/serialization/data-stream/serialization.d.ts
|
|
4
|
+
declare class DataStreamChunkEncoder extends TransformStream<DataStreamChunk, string> {
|
|
5
|
+
constructor();
|
|
4
6
|
}
|
|
5
|
-
|
|
6
|
-
|
|
7
|
+
declare class DataStreamChunkDecoder extends TransformStream<string, DataStreamChunk> {
|
|
8
|
+
constructor();
|
|
7
9
|
}
|
|
10
|
+
//#endregion
|
|
11
|
+
export { DataStreamChunkDecoder, DataStreamChunkEncoder };
|
|
8
12
|
//# sourceMappingURL=serialization.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"serialization.d.ts","
|
|
1
|
+
{"version":3,"file":"serialization.d.ts","names":[],"sources":["../../../../src/core/serialization/data-stream/serialization.ts"],"mappings":";;;cAEa,sBAAA,SAA+B,eAAe,CACzD,eAAA;;;cAYW,sBAAA,SAA+B,eAAe,SAEzD,eAAA"}
|
|
@@ -1,25 +1,24 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
}
|
|
1
|
+
//#region src/core/serialization/data-stream/serialization.ts
|
|
2
|
+
var DataStreamChunkEncoder = class extends TransformStream {
|
|
3
|
+
constructor() {
|
|
4
|
+
super({ transform: (chunk, controller) => {
|
|
5
|
+
controller.enqueue(`${chunk.type}:${JSON.stringify(chunk.value)}\n`);
|
|
6
|
+
} });
|
|
7
|
+
}
|
|
8
|
+
};
|
|
9
|
+
var DataStreamChunkDecoder = class extends TransformStream {
|
|
10
|
+
constructor() {
|
|
11
|
+
super({ transform: (chunk, controller) => {
|
|
12
|
+
const index = chunk.indexOf(":");
|
|
13
|
+
if (index === -1) throw new Error("Invalid stream part");
|
|
14
|
+
controller.enqueue({
|
|
15
|
+
type: chunk.slice(0, index),
|
|
16
|
+
value: JSON.parse(chunk.slice(index + 1))
|
|
17
|
+
});
|
|
18
|
+
} });
|
|
19
|
+
}
|
|
20
|
+
};
|
|
21
|
+
//#endregion
|
|
22
|
+
export { DataStreamChunkDecoder, DataStreamChunkEncoder };
|
|
23
|
+
|
|
25
24
|
//# sourceMappingURL=serialization.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"serialization.js","
|
|
1
|
+
{"version":3,"file":"serialization.js","names":[],"sources":["../../../../src/core/serialization/data-stream/serialization.ts"],"sourcesContent":["import type { DataStreamChunk, DataStreamStreamChunkType } from \"./chunk-types\";\n\nexport class DataStreamChunkEncoder extends TransformStream<\n DataStreamChunk,\n string\n> {\n constructor() {\n super({\n transform: (chunk, controller) => {\n controller.enqueue(`${chunk.type}:${JSON.stringify(chunk.value)}\\n`);\n },\n });\n }\n}\n\nexport class DataStreamChunkDecoder extends TransformStream<\n string,\n DataStreamChunk\n> {\n constructor() {\n super({\n transform: (chunk, controller) => {\n const index = chunk.indexOf(\":\");\n if (index === -1) throw new Error(\"Invalid stream part\");\n controller.enqueue({\n type: chunk.slice(0, index) as DataStreamStreamChunkType,\n value: JSON.parse(chunk.slice(index + 1)),\n });\n },\n });\n }\n}\n"],"mappings":";AAEA,IAAa,yBAAb,cAA4C,gBAG1C;CACA,cAAc;EACZ,MAAM,EACJ,YAAY,OAAO,eAAe;GAChC,WAAW,QAAQ,GAAG,MAAM,KAAK,GAAG,KAAK,UAAU,MAAM,KAAK,EAAE,GAAG;EACrE,EACF,CAAC;CACH;AACF;AAEA,IAAa,yBAAb,cAA4C,gBAG1C;CACA,cAAc;EACZ,MAAM,EACJ,YAAY,OAAO,eAAe;GAChC,MAAM,QAAQ,MAAM,QAAQ,GAAG;GAC/B,IAAI,UAAU,IAAI,MAAM,IAAI,MAAM,qBAAqB;GACvD,WAAW,QAAQ;IACjB,MAAM,MAAM,MAAM,GAAG,KAAK;IAC1B,OAAO,KAAK,MAAM,MAAM,MAAM,QAAQ,CAAC,CAAC;GAC1C,CAAC;EACH,EACF,CAAC;CACH;AACF"}
|
|
@@ -1,19 +1,22 @@
|
|
|
1
|
-
import
|
|
1
|
+
import { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
|
|
2
2
|
import { PipeableTransformStream } from "../../utils/stream/PipeableTransformStream.js";
|
|
3
|
-
import
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
3
|
+
import { UIMessageStreamChunk, UIMessageStreamDataChunk } from "./chunk-types.js";
|
|
4
|
+
|
|
5
|
+
//#region src/core/serialization/ui-message-stream/UIMessageStream.d.ts
|
|
6
|
+
type UIMessageStreamDecoderOptions = {
|
|
7
|
+
onData?: (data: {
|
|
8
|
+
type: string;
|
|
9
|
+
name: string;
|
|
10
|
+
data: unknown;
|
|
11
|
+
transient?: boolean;
|
|
12
|
+
}) => void;
|
|
12
13
|
};
|
|
13
14
|
/**
|
|
14
15
|
* Decodes AI SDK v6 UI Message Stream format into AssistantStreamChunks.
|
|
15
16
|
*/
|
|
16
|
-
|
|
17
|
-
|
|
17
|
+
declare class UIMessageStreamDecoder extends PipeableTransformStream<Uint8Array<ArrayBuffer>, AssistantStreamChunk> {
|
|
18
|
+
constructor(options?: UIMessageStreamDecoderOptions);
|
|
18
19
|
}
|
|
20
|
+
//#endregion
|
|
21
|
+
export { type UIMessageStreamChunk, type UIMessageStreamDataChunk, UIMessageStreamDecoder, UIMessageStreamDecoderOptions };
|
|
19
22
|
//# sourceMappingURL=UIMessageStream.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"UIMessageStream.d.ts","
|
|
1
|
+
{"version":3,"file":"UIMessageStream.d.ts","names":[],"sources":["../../../../src/core/serialization/ui-message-stream/UIMessageStream.ts"],"mappings":";;;;;KAcY,6BAAA;EACV,MAAA,IAAU,IAAA;IACR,IAAA;IACA,IAAA;IACA,IAAA;IACA,SAAA;EAAA;AAAA;;;;cA6ES,sBAAA,SAA+B,uBAAA,CAC1C,UAAA,CAAW,WAAA,GACX,oBAAA;cAEY,OAAA,GAAS,6BAAA;AAAA"}
|