assistant-stream 0.2.45 → 0.2.47
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/dist/core/AssistantStream.d.ts +1 -1
- package/dist/core/AssistantStream.d.ts.map +1 -1
- package/dist/core/AssistantStream.js +15 -19
- package/dist/core/AssistantStream.js.map +1 -1
- package/dist/core/AssistantStreamChunk.d.ts +2 -2
- package/dist/core/AssistantStreamChunk.d.ts.map +1 -1
- package/dist/core/AssistantStreamChunk.js +1 -0
- package/dist/core/AssistantStreamChunk.js.map +1 -1
- package/dist/core/accumulators/AssistantMessageStream.d.ts +2 -2
- package/dist/core/accumulators/AssistantMessageStream.d.ts.map +1 -1
- package/dist/core/accumulators/AssistantMessageStream.js +45 -50
- package/dist/core/accumulators/AssistantMessageStream.js.map +1 -1
- package/dist/core/accumulators/assistant-message-accumulator.d.ts +3 -3
- package/dist/core/accumulators/assistant-message-accumulator.d.ts.map +1 -1
- package/dist/core/accumulators/assistant-message-accumulator.js +339 -329
- package/dist/core/accumulators/assistant-message-accumulator.js.map +1 -1
- package/dist/core/index.d.ts +17 -16
- package/dist/core/index.d.ts.map +1 -1
- package/dist/core/index.js +10 -44
- package/dist/core/index.js.map +1 -1
- package/dist/core/modules/assistant-stream.d.ts +7 -7
- package/dist/core/modules/assistant-stream.d.ts.map +1 -1
- package/dist/core/modules/assistant-stream.js +159 -188
- package/dist/core/modules/assistant-stream.js.map +1 -1
- package/dist/core/modules/text.d.ts +2 -2
- package/dist/core/modules/text.d.ts.map +1 -1
- package/dist/core/modules/text.js +43 -47
- package/dist/core/modules/text.js.map +1 -1
- package/dist/core/modules/tool-call.d.ts +5 -5
- package/dist/core/modules/tool-call.d.ts.map +1 -1
- package/dist/core/modules/tool-call.js +88 -89
- package/dist/core/modules/tool-call.js.map +1 -1
- package/dist/core/object/ObjectStreamAccumulator.d.ts +2 -2
- package/dist/core/object/ObjectStreamAccumulator.d.ts.map +1 -1
- package/dist/core/object/ObjectStreamAccumulator.js +49 -58
- package/dist/core/object/ObjectStreamAccumulator.js.map +1 -1
- package/dist/core/object/ObjectStreamResponse.d.ts +2 -2
- package/dist/core/object/ObjectStreamResponse.d.ts.map +1 -1
- package/dist/core/object/ObjectStreamResponse.js +70 -74
- package/dist/core/object/ObjectStreamResponse.js.map +1 -1
- package/dist/core/object/createObjectStream.d.ts +2 -2
- package/dist/core/object/createObjectStream.d.ts.map +1 -1
- package/dist/core/object/createObjectStream.js +45 -56
- package/dist/core/object/createObjectStream.js.map +1 -1
- package/dist/core/object/types.d.ts +1 -1
- package/dist/core/object/types.d.ts.map +1 -1
- package/dist/core/object/types.js +1 -0
- package/dist/core/object/types.js.map +1 -1
- package/dist/core/serialization/PlainText.d.ts +3 -3
- package/dist/core/serialization/PlainText.d.ts.map +1 -1
- package/dist/core/serialization/PlainText.js +46 -47
- package/dist/core/serialization/PlainText.js.map +1 -1
- package/dist/core/serialization/assistant-transport/AssistantTransport.d.ts +3 -3
- package/dist/core/serialization/assistant-transport/AssistantTransport.d.ts.map +1 -1
- package/dist/core/serialization/assistant-transport/AssistantTransport.js +117 -112
- package/dist/core/serialization/assistant-transport/AssistantTransport.js.map +1 -1
- package/dist/core/serialization/data-stream/DataStream.d.ts +3 -3
- package/dist/core/serialization/data-stream/DataStream.d.ts.map +1 -1
- package/dist/core/serialization/data-stream/DataStream.js +355 -354
- package/dist/core/serialization/data-stream/DataStream.js.map +1 -1
- package/dist/core/serialization/data-stream/chunk-types.d.ts +2 -2
- package/dist/core/serialization/data-stream/chunk-types.d.ts.map +1 -1
- package/dist/core/serialization/data-stream/chunk-types.js +22 -26
- package/dist/core/serialization/data-stream/chunk-types.js.map +1 -1
- package/dist/core/serialization/data-stream/serialization.d.ts +1 -1
- package/dist/core/serialization/data-stream/serialization.d.ts.map +1 -1
- package/dist/core/serialization/data-stream/serialization.js +23 -28
- package/dist/core/serialization/data-stream/serialization.js.map +1 -1
- package/dist/core/serialization/ui-message-stream/UIMessageStream.d.ts +19 -0
- package/dist/core/serialization/ui-message-stream/UIMessageStream.d.ts.map +1 -0
- package/dist/core/serialization/ui-message-stream/UIMessageStream.js +231 -0
- package/dist/core/serialization/ui-message-stream/UIMessageStream.js.map +1 -0
- package/dist/core/serialization/ui-message-stream/chunk-types.d.ts +78 -0
- package/dist/core/serialization/ui-message-stream/chunk-types.d.ts.map +1 -0
- package/dist/core/serialization/ui-message-stream/chunk-types.js +2 -0
- package/dist/core/serialization/ui-message-stream/chunk-types.js.map +1 -0
- package/dist/core/tool/ToolCallReader.d.ts +4 -4
- package/dist/core/tool/ToolCallReader.d.ts.map +1 -1
- package/dist/core/tool/ToolCallReader.js +303 -303
- package/dist/core/tool/ToolCallReader.js.map +1 -1
- package/dist/core/tool/ToolExecutionStream.d.ts +5 -5
- package/dist/core/tool/ToolExecutionStream.d.ts.map +1 -1
- package/dist/core/tool/ToolExecutionStream.js +140 -143
- package/dist/core/tool/ToolExecutionStream.js.map +1 -1
- package/dist/core/tool/ToolResponse.d.ts +1 -1
- package/dist/core/tool/ToolResponse.d.ts.map +1 -1
- package/dist/core/tool/ToolResponse.js +25 -29
- package/dist/core/tool/ToolResponse.js.map +1 -1
- package/dist/core/tool/index.d.ts +5 -5
- package/dist/core/tool/index.d.ts.map +1 -1
- package/dist/core/tool/index.js +3 -13
- package/dist/core/tool/index.js.map +1 -1
- package/dist/core/tool/tool-types.d.ts +3 -3
- package/dist/core/tool/tool-types.d.ts.map +1 -1
- package/dist/core/tool/tool-types.js +1 -0
- package/dist/core/tool/tool-types.js.map +1 -1
- package/dist/core/tool/toolResultStream.d.ts +3 -3
- package/dist/core/tool/toolResultStream.d.ts.map +1 -1
- package/dist/core/tool/toolResultStream.js +118 -125
- package/dist/core/tool/toolResultStream.js.map +1 -1
- package/dist/core/tool/type-path-utils.js +1 -0
- package/dist/core/tool/type-path-utils.js.map +1 -1
- package/dist/core/utils/Counter.js +6 -10
- package/dist/core/utils/Counter.js.map +1 -1
- package/dist/core/utils/generateId.js +1 -8
- package/dist/core/utils/generateId.js.map +1 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts +1 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts.map +1 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.js +42 -43
- package/dist/core/utils/stream/AssistantMetaTransformStream.js.map +1 -1
- package/dist/core/utils/stream/AssistantTransformStream.d.ts +2 -2
- package/dist/core/utils/stream/AssistantTransformStream.d.ts.map +1 -1
- package/dist/core/utils/stream/AssistantTransformStream.js +35 -45
- package/dist/core/utils/stream/AssistantTransformStream.js.map +1 -1
- package/dist/core/utils/stream/LineDecoderStream.js +24 -26
- package/dist/core/utils/stream/LineDecoderStream.js.map +1 -1
- package/dist/core/utils/stream/PipeableTransformStream.js +10 -14
- package/dist/core/utils/stream/PipeableTransformStream.js.map +1 -1
- package/dist/core/utils/stream/SSE.d.ts +1 -1
- package/dist/core/utils/stream/SSE.d.ts.map +1 -1
- package/dist/core/utils/stream/SSE.js +90 -98
- package/dist/core/utils/stream/SSE.js.map +1 -1
- package/dist/core/utils/stream/UnderlyingReadable.js +1 -0
- package/dist/core/utils/stream/UnderlyingReadable.js.map +1 -1
- package/dist/core/utils/stream/merge.d.ts +1 -1
- package/dist/core/utils/stream/merge.d.ts.map +1 -1
- package/dist/core/utils/stream/merge.js +169 -81
- package/dist/core/utils/stream/merge.js.map +1 -1
- package/dist/core/utils/stream/path-utils.d.ts +2 -2
- package/dist/core/utils/stream/path-utils.d.ts.map +1 -1
- package/dist/core/utils/stream/path-utils.js +49 -56
- package/dist/core/utils/stream/path-utils.js.map +1 -1
- package/dist/core/utils/types.d.ts +1 -1
- package/dist/core/utils/types.d.ts.map +1 -1
- package/dist/core/utils/types.js +1 -0
- package/dist/core/utils/types.js.map +1 -1
- package/dist/core/utils/withPromiseOrValue.js +14 -14
- package/dist/core/utils/withPromiseOrValue.js.map +1 -1
- package/dist/index.d.ts +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +0 -1
- package/dist/index.js.map +1 -1
- package/dist/utils/AsyncIterableStream.js +15 -16
- package/dist/utils/AsyncIterableStream.js.map +1 -1
- package/dist/utils/index.d.ts +2 -2
- package/dist/utils/index.d.ts.map +1 -1
- package/dist/utils/index.js +1 -7
- package/dist/utils/index.js.map +1 -1
- package/dist/utils/json/fix-json.js +365 -343
- package/dist/utils/json/fix-json.js.map +1 -1
- package/dist/utils/json/index.d.ts +1 -1
- package/dist/utils/json/index.d.ts.map +1 -1
- package/dist/utils/json/index.js +1 -0
- package/dist/utils/json/index.js.map +1 -1
- package/dist/utils/json/is-json.d.ts +1 -1
- package/dist/utils/json/is-json.d.ts.map +1 -1
- package/dist/utils/json/is-json.js +21 -25
- package/dist/utils/json/is-json.js.map +1 -1
- package/dist/utils/json/json-value.js +1 -0
- package/dist/utils/json/json-value.js.map +1 -1
- package/dist/utils/json/parse-partial-json-object.d.ts +1 -1
- package/dist/utils/json/parse-partial-json-object.d.ts.map +1 -1
- package/dist/utils/json/parse-partial-json-object.js +61 -56
- package/dist/utils/json/parse-partial-json-object.js.map +1 -1
- package/dist/utils/promiseWithResolvers.js +10 -13
- package/dist/utils/promiseWithResolvers.js.map +1 -1
- package/dist/utils.d.ts +5 -5
- package/dist/utils.d.ts.map +1 -1
- package/dist/utils.js +4 -17
- package/dist/utils.js.map +1 -1
- package/package.json +26 -14
- package/src/core/index.ts +6 -0
- package/src/core/serialization/ui-message-stream/UIMessageStream.test.ts +370 -0
- package/src/core/serialization/ui-message-stream/UIMessageStream.ts +300 -0
- package/src/core/serialization/ui-message-stream/chunk-types.ts +60 -0
- package/dist/core/object/ObjectStream.test.d.ts +0 -2
- package/dist/core/object/ObjectStream.test.d.ts.map +0 -1
- package/dist/core/serialization/assistant-transport/AssistantTransport.test.d.ts +0 -2
- package/dist/core/serialization/assistant-transport/AssistantTransport.test.d.ts.map +0 -1
- package/dist/core/tool/toolResultStream.test.d.ts +0 -2
- package/dist/core/tool/toolResultStream.test.d.ts.map +0 -1
- package/dist/utils/json/parse-partial-json-object.test.d.ts +0 -2
- package/dist/utils/json/parse-partial-json-object.test.d.ts.map +0 -1
|
@@ -1,80 +1,76 @@
|
|
|
1
|
-
// src/core/object/ObjectStreamResponse.ts
|
|
2
1
|
import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream.js";
|
|
3
2
|
import { ObjectStreamAccumulator } from "./ObjectStreamAccumulator.js";
|
|
4
3
|
import { SSEDecoder, SSEEncoder } from "../utils/stream/SSE.js";
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
4
|
+
export class ObjectStreamEncoder extends PipeableTransformStream {
|
|
5
|
+
constructor() {
|
|
6
|
+
super((readable) => readable
|
|
7
|
+
.pipeThrough((() => {
|
|
8
|
+
class ObjectStreamTransformer {
|
|
9
|
+
#isFirstChunk = true;
|
|
10
|
+
start() {
|
|
11
|
+
// Nothing needed here since we initialize in the field declaration
|
|
12
|
+
}
|
|
13
|
+
transform(chunk, controller) {
|
|
14
|
+
if (this.#isFirstChunk &&
|
|
15
|
+
chunk.snapshot &&
|
|
16
|
+
Object.keys(chunk.snapshot).length > 0) {
|
|
17
|
+
// For the first chunk, if there's an initial state that's not empty,
|
|
18
|
+
// prepend a set operation for the initial state
|
|
19
|
+
controller.enqueue([
|
|
20
|
+
{ type: "set", path: [], value: chunk.snapshot },
|
|
21
|
+
...chunk.operations,
|
|
22
|
+
]);
|
|
23
|
+
}
|
|
24
|
+
else {
|
|
25
|
+
controller.enqueue(chunk.operations);
|
|
26
|
+
}
|
|
27
|
+
this.#isFirstChunk = false;
|
|
28
|
+
}
|
|
13
29
|
}
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
}
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
new
|
|
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
|
-
var fromObjectStreamResponse = (response) => {
|
|
63
|
-
if (!response.ok)
|
|
64
|
-
throw new Error(`Response failed, status ${response.status}`);
|
|
65
|
-
if (!response.body) throw new Error("Response body is null");
|
|
66
|
-
if (response.headers.get("Content-Type") !== "text/event-stream") {
|
|
67
|
-
throw new Error("Response is not an event stream");
|
|
68
|
-
}
|
|
69
|
-
if (response.headers.get("Assistant-Stream-Format") !== "object-stream/v0") {
|
|
70
|
-
throw new Error("Unsupported Assistant-Stream-Format header");
|
|
71
|
-
}
|
|
72
|
-
return response.body.pipeThrough(new ObjectStreamDecoder());
|
|
73
|
-
};
|
|
74
|
-
export {
|
|
75
|
-
ObjectStreamDecoder,
|
|
76
|
-
ObjectStreamEncoder,
|
|
77
|
-
ObjectStreamResponse,
|
|
78
|
-
fromObjectStreamResponse
|
|
30
|
+
return new TransformStream(new ObjectStreamTransformer());
|
|
31
|
+
})())
|
|
32
|
+
.pipeThrough(new SSEEncoder()));
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
export class ObjectStreamDecoder extends PipeableTransformStream {
|
|
36
|
+
constructor() {
|
|
37
|
+
const accumulator = new ObjectStreamAccumulator();
|
|
38
|
+
super((readable) => readable
|
|
39
|
+
.pipeThrough(new SSEDecoder())
|
|
40
|
+
.pipeThrough(new TransformStream({
|
|
41
|
+
transform(operations, controller) {
|
|
42
|
+
accumulator.append(operations);
|
|
43
|
+
controller.enqueue({
|
|
44
|
+
snapshot: accumulator.state,
|
|
45
|
+
operations,
|
|
46
|
+
});
|
|
47
|
+
},
|
|
48
|
+
})));
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
export class ObjectStreamResponse extends Response {
|
|
52
|
+
constructor(body) {
|
|
53
|
+
super(body.pipeThrough(new ObjectStreamEncoder()), {
|
|
54
|
+
headers: new Headers({
|
|
55
|
+
"Content-Type": "text/event-stream",
|
|
56
|
+
"Cache-Control": "no-cache",
|
|
57
|
+
Connection: "keep-alive",
|
|
58
|
+
"Assistant-Stream-Format": "object-stream/v0",
|
|
59
|
+
}),
|
|
60
|
+
});
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
export const fromObjectStreamResponse = (response) => {
|
|
64
|
+
if (!response.ok)
|
|
65
|
+
throw new Error(`Response failed, status ${response.status}`);
|
|
66
|
+
if (!response.body)
|
|
67
|
+
throw new Error("Response body is null");
|
|
68
|
+
if (response.headers.get("Content-Type") !== "text/event-stream") {
|
|
69
|
+
throw new Error("Response is not an event stream");
|
|
70
|
+
}
|
|
71
|
+
if (response.headers.get("Assistant-Stream-Format") !== "object-stream/v0") {
|
|
72
|
+
throw new Error("Unsupported Assistant-Stream-Format header");
|
|
73
|
+
}
|
|
74
|
+
return response.body.pipeThrough(new ObjectStreamDecoder());
|
|
79
75
|
};
|
|
80
76
|
//# sourceMappingURL=ObjectStreamResponse.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../src/core/object/ObjectStreamResponse.ts"],"
|
|
1
|
+
{"version":3,"file":"ObjectStreamResponse.js","sourceRoot":"","sources":["../../../src/core/object/ObjectStreamResponse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,mDAAgD;AAClF,OAAO,EAAE,uBAAuB,EAAE,qCAAkC;AACpE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,+BAA4B;AAG7D,MAAM,OAAO,mBAAoB,SAAQ,uBAGxC;IACC;QACE,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE,CACjB,QAAQ;aACL,WAAW,CACV,CAAC,GAAG,EAAE;YACJ,MAAM,uBAAuB;gBAI3B,aAAa,GAAG,IAAI,CAAC;gBAErB,KAAK;oBACH,mEAAmE;gBACrE,CAAC;gBAED,SAAS,CACP,KAAwB,EACxB,UAEC;oBAED,IACE,IAAI,CAAC,aAAa;wBAClB,KAAK,CAAC,QAAQ;wBACd,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EACtC,CAAC;wBACD,qEAAqE;wBACrE,gDAAgD;wBAChD,UAAU,CAAC,OAAO,CAAC;4BACjB,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE;4BAChD,GAAG,KAAK,CAAC,UAAU;yBACpB,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;oBACvC,CAAC;oBACD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC7B,CAAC;aACF;YACD,OAAO,IAAI,eAAe,CAAC,IAAI,uBAAuB,EAAE,CAAC,CAAC;QAC5D,CAAC,CAAC,EAAE,CACL;aACA,WAAW,CAAC,IAAI,UAAU,EAAE,CAAC,CACjC,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,mBAAoB,SAAQ,uBAGxC;IACC;QACE,MAAM,WAAW,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAClD,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE,CACjB,QAAQ;aACL,WAAW,CAAC,IAAI,UAAU,EAAoC,CAAC;aAC/D,WAAW,CACV,IAAI,eAAe,CAGjB;YACA,SAAS,CAAC,UAAU,EAAE,UAAU;gBAC9B,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAC/B,UAAU,CAAC,OAAO,CAAC;oBACjB,QAAQ,EAAE,WAAW,CAAC,KAAK;oBAC3B,UAAU;iBACX,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CACH,CACJ,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,oBAAqB,SAAQ,QAAQ;IAChD,YAAY,IAAuC;QACjD,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,mBAAmB,EAAE,CAAC,EAAE;YACjD,OAAO,EAAE,IAAI,OAAO,CAAC;gBACnB,cAAc,EAAE,mBAAmB;gBACnC,eAAe,EAAE,UAAU;gBAC3B,UAAU,EAAE,YAAY;gBACxB,yBAAyB,EAAE,kBAAkB;aAC9C,CAAC;SACH,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACtC,QAAkB,EACiB,EAAE;IACrC,IAAI,CAAC,QAAQ,CAAC,EAAE;QACd,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAChE,IAAI,CAAC,QAAQ,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC7D,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,mBAAmB,EAAE,CAAC;QACjE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,KAAK,kBAAkB,EAAE,CAAC;QAC3E,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC;AAC9D,CAAC,CAAC"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { ReadonlyJSONValue } from "../../utils";
|
|
2
|
-
import { ObjectStreamOperation, ObjectStreamChunk } from "./types";
|
|
1
|
+
import { ReadonlyJSONValue } from "../../utils.js";
|
|
2
|
+
import { ObjectStreamOperation, ObjectStreamChunk } from "./types.js";
|
|
3
3
|
type ObjectStreamController = {
|
|
4
4
|
readonly abortSignal: AbortSignal;
|
|
5
5
|
enqueue(operations: readonly ObjectStreamOperation[]): void;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"createObjectStream.d.ts","sourceRoot":"","sources":["../../../src/core/object/createObjectStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,
|
|
1
|
+
{"version":3,"file":"createObjectStream.d.ts","sourceRoot":"","sources":["../../../src/core/object/createObjectStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,uBAAoB;AAGhD,OAAO,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,mBAAgB;AAEnE,KAAK,sBAAsB,GAAG;IAC5B,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAElC,OAAO,CAAC,UAAU,EAAE,SAAS,qBAAqB,EAAE,GAAG,IAAI,CAAC;CAC7D,CAAC;AAuDF,KAAK,yBAAyB,GAAG;IAC/B,OAAO,EAAE,CAAC,UAAU,EAAE,sBAAsB,KAAK,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC1E,YAAY,CAAC,EAAE,iBAAiB,CAAC;CAClC,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,4BAGhC,yBAAyB,sCAc3B,CAAC"}
|
|
@@ -1,63 +1,52 @@
|
|
|
1
|
-
// src/core/object/createObjectStream.ts
|
|
2
1
|
import { withPromiseOrValue } from "../utils/withPromiseOrValue.js";
|
|
3
2
|
import { ObjectStreamAccumulator } from "./ObjectStreamAccumulator.js";
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
}
|
|
11
|
-
constructor(controller, defaultValue) {
|
|
12
|
-
this._controller = controller;
|
|
13
|
-
this._accumulator = new ObjectStreamAccumulator(defaultValue);
|
|
14
|
-
}
|
|
15
|
-
enqueue(operations) {
|
|
16
|
-
this._accumulator.append(operations);
|
|
17
|
-
this._controller.enqueue({
|
|
18
|
-
snapshot: this._accumulator.state,
|
|
19
|
-
operations
|
|
20
|
-
});
|
|
21
|
-
}
|
|
22
|
-
__internalError(error) {
|
|
23
|
-
this._controller.error(error);
|
|
24
|
-
}
|
|
25
|
-
__internalClose() {
|
|
26
|
-
this._controller.close();
|
|
27
|
-
}
|
|
28
|
-
__internalCancel(reason) {
|
|
29
|
-
this._abortController.abort(reason);
|
|
30
|
-
}
|
|
31
|
-
};
|
|
32
|
-
var getStreamControllerPair = (defaultValue) => {
|
|
33
|
-
let controller;
|
|
34
|
-
const stream = new ReadableStream({
|
|
35
|
-
start(c) {
|
|
36
|
-
controller = new ObjectStreamControllerImpl(c, defaultValue);
|
|
37
|
-
},
|
|
38
|
-
cancel(reason) {
|
|
39
|
-
controller.__internalCancel(reason);
|
|
3
|
+
class ObjectStreamControllerImpl {
|
|
4
|
+
_controller;
|
|
5
|
+
_abortController = new AbortController();
|
|
6
|
+
_accumulator;
|
|
7
|
+
get abortSignal() {
|
|
8
|
+
return this._abortController.signal;
|
|
40
9
|
}
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
(e) => {
|
|
55
|
-
controller.__internalError(e);
|
|
10
|
+
constructor(controller, defaultValue) {
|
|
11
|
+
this._controller = controller;
|
|
12
|
+
this._accumulator = new ObjectStreamAccumulator(defaultValue);
|
|
13
|
+
}
|
|
14
|
+
enqueue(operations) {
|
|
15
|
+
this._accumulator.append(operations);
|
|
16
|
+
this._controller.enqueue({
|
|
17
|
+
snapshot: this._accumulator.state,
|
|
18
|
+
operations,
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
__internalError(error) {
|
|
22
|
+
this._controller.error(error);
|
|
56
23
|
}
|
|
57
|
-
|
|
58
|
-
|
|
24
|
+
__internalClose() {
|
|
25
|
+
this._controller.close();
|
|
26
|
+
}
|
|
27
|
+
__internalCancel(reason) {
|
|
28
|
+
this._abortController.abort(reason);
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
const getStreamControllerPair = (defaultValue) => {
|
|
32
|
+
let controller;
|
|
33
|
+
const stream = new ReadableStream({
|
|
34
|
+
start(c) {
|
|
35
|
+
controller = new ObjectStreamControllerImpl(c, defaultValue);
|
|
36
|
+
},
|
|
37
|
+
cancel(reason) {
|
|
38
|
+
controller.__internalCancel(reason);
|
|
39
|
+
},
|
|
40
|
+
});
|
|
41
|
+
return [stream, controller];
|
|
59
42
|
};
|
|
60
|
-
export {
|
|
61
|
-
|
|
43
|
+
export const createObjectStream = ({ execute, defaultValue = {}, }) => {
|
|
44
|
+
const [stream, controller] = getStreamControllerPair(defaultValue);
|
|
45
|
+
withPromiseOrValue(() => execute(controller), () => {
|
|
46
|
+
controller.__internalClose();
|
|
47
|
+
}, (e) => {
|
|
48
|
+
controller.__internalError(e);
|
|
49
|
+
});
|
|
50
|
+
return stream;
|
|
62
51
|
};
|
|
63
52
|
//# sourceMappingURL=createObjectStream.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../src/core/object/createObjectStream.ts"],"
|
|
1
|
+
{"version":3,"file":"createObjectStream.js","sourceRoot":"","sources":["../../../src/core/object/createObjectStream.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,uCAAoC;AACjE,OAAO,EAAE,uBAAuB,EAAE,qCAAkC;AASpE,MAAM,0BAA0B;IACtB,WAAW,CAAqD;IAChE,gBAAgB,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,YAAY,CAA0B;IAE9C,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;IACtC,CAAC;IAED,YACE,UAA8D,EAC9D,YAA+B;QAE/B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,uBAAuB,CAAC,YAAY,CAAC,CAAC;IAChE,CAAC;IAED,OAAO,CAAC,UAA4C;QAClD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAErC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;YACvB,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK;YACjC,UAAU;SACX,CAAC,CAAC;IACL,CAAC;IAED,eAAe,CAAC,KAAc;QAC5B,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,eAAe;QACb,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,gBAAgB,CAAC,MAAgB;QAC/B,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;CACF;AAED,MAAM,uBAAuB,GAAG,CAAC,YAA+B,EAAE,EAAE;IAClE,IAAI,UAAuC,CAAC;IAC5C,MAAM,MAAM,GAAG,IAAI,cAAc,CAAoB;QACnD,KAAK,CAAC,CAAC;YACL,UAAU,GAAG,IAAI,0BAA0B,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC/D,CAAC;QACD,MAAM,CAAC,MAAe;YACpB,UAAU,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,CAAC,MAAM,EAAE,UAAU,CAAU,CAAC;AACvC,CAAC,CAAC;AAOF,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,EACjC,OAAO,EACP,YAAY,GAAG,EAAE,GACS,EAAE,EAAE;IAC9B,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,uBAAuB,CAAC,YAAY,CAAC,CAAC;IAEnE,kBAAkB,CAChB,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,EACzB,GAAG,EAAE;QACH,UAAU,CAAC,eAAe,EAAE,CAAC;IAC/B,CAAC,EACD,CAAC,CAAU,EAAE,EAAE;QACb,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/core/object/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/core/object/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,uBAAoB;AAEhD,MAAM,MAAM,qBAAqB,GAC7B;IACE,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;IACrB,QAAQ,CAAC,IAAI,EAAE,SAAS,MAAM,EAAE,CAAC;IACjC,QAAQ,CAAC,KAAK,EAAE,iBAAiB,CAAC;CACnC,GACD;IACE,QAAQ,CAAC,IAAI,EAAE,aAAa,CAAC;IAC7B,QAAQ,CAAC,IAAI,EAAE,SAAS,MAAM,EAAE,CAAC;IACjC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;CACxB,CAAC;AAEN,MAAM,MAAM,iBAAiB,GAAG;IAC9B,QAAQ,CAAC,QAAQ,EAAE,iBAAiB,CAAC;IACrC,QAAQ,CAAC,UAAU,EAAE,SAAS,qBAAqB,EAAE,CAAC;CACvD,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/core/object/types.ts"],"names":[],"mappings":""}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { AssistantStreamEncoder } from "../AssistantStream";
|
|
2
|
-
import { AssistantStreamChunk } from "../AssistantStreamChunk";
|
|
3
|
-
import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream";
|
|
1
|
+
import { AssistantStreamEncoder } from "../AssistantStream.js";
|
|
2
|
+
import { AssistantStreamChunk } from "../AssistantStreamChunk.js";
|
|
3
|
+
import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream.js";
|
|
4
4
|
export declare class PlainTextEncoder extends PipeableTransformStream<AssistantStreamChunk, Uint8Array<ArrayBuffer>> implements AssistantStreamEncoder {
|
|
5
5
|
headers: Headers;
|
|
6
6
|
constructor();
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"PlainText.d.ts","sourceRoot":"","sources":["../../../src/core/serialization/PlainText.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,
|
|
1
|
+
{"version":3,"file":"PlainText.d.ts","sourceRoot":"","sources":["../../../src/core/serialization/PlainText.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,8BAA2B;AAC5D,OAAO,EAAE,oBAAoB,EAAE,mCAAgC;AAE/D,OAAO,EAAE,uBAAuB,EAAE,mDAAgD;AAElF,qBAAa,gBACX,SAAQ,uBAAuB,CAAC,oBAAoB,EAAE,UAAU,CAAC,WAAW,CAAC,CAC7E,YAAW,sBAAsB;IAEjC,OAAO,UAGJ;;CAuCJ;AAED,qBAAa,gBAAiB,SAAQ,uBAAuB,CAC3D,UAAU,CAAC,WAAW,CAAC,EACvB,oBAAoB,CACrB;;CAcA"}
|
|
@@ -1,51 +1,50 @@
|
|
|
1
|
-
// src/core/serialization/PlainText.ts
|
|
2
1
|
import { AssistantTransformStream } from "../utils/stream/AssistantTransformStream.js";
|
|
3
2
|
import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream.js";
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
});
|
|
9
|
-
constructor() {
|
|
10
|
-
super((readable) => {
|
|
11
|
-
const transform = new TransformStream({
|
|
12
|
-
transform(chunk, controller) {
|
|
13
|
-
const type = chunk.type;
|
|
14
|
-
switch (type) {
|
|
15
|
-
case "text-delta":
|
|
16
|
-
controller.enqueue(chunk.textDelta);
|
|
17
|
-
break;
|
|
18
|
-
case "part-start":
|
|
19
|
-
case "part-finish":
|
|
20
|
-
case "step-start":
|
|
21
|
-
case "step-finish":
|
|
22
|
-
case "message-finish":
|
|
23
|
-
case "error":
|
|
24
|
-
break;
|
|
25
|
-
default:
|
|
26
|
-
const unsupportedType = type;
|
|
27
|
-
throw new Error(`unsupported chunk type: ${unsupportedType}`);
|
|
28
|
-
}
|
|
29
|
-
}
|
|
30
|
-
});
|
|
31
|
-
return readable.pipeThrough(transform).pipeThrough(new TextEncoderStream());
|
|
3
|
+
export class PlainTextEncoder extends PipeableTransformStream {
|
|
4
|
+
headers = new Headers({
|
|
5
|
+
"Content-Type": "text/plain; charset=utf-8",
|
|
6
|
+
"x-vercel-ai-data-stream": "v1",
|
|
32
7
|
});
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
8
|
+
constructor() {
|
|
9
|
+
super((readable) => {
|
|
10
|
+
const transform = new TransformStream({
|
|
11
|
+
transform(chunk, controller) {
|
|
12
|
+
const type = chunk.type;
|
|
13
|
+
switch (type) {
|
|
14
|
+
case "text-delta":
|
|
15
|
+
controller.enqueue(chunk.textDelta);
|
|
16
|
+
break;
|
|
17
|
+
case "part-start":
|
|
18
|
+
case "part-finish":
|
|
19
|
+
case "step-start":
|
|
20
|
+
case "step-finish":
|
|
21
|
+
case "message-finish":
|
|
22
|
+
case "error":
|
|
23
|
+
break;
|
|
24
|
+
default:
|
|
25
|
+
const unsupportedType = type;
|
|
26
|
+
throw new Error(`unsupported chunk type: ${unsupportedType}`);
|
|
27
|
+
}
|
|
28
|
+
},
|
|
29
|
+
});
|
|
30
|
+
return readable
|
|
31
|
+
.pipeThrough(transform)
|
|
32
|
+
.pipeThrough(new TextEncoderStream());
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
export class PlainTextDecoder extends PipeableTransformStream {
|
|
37
|
+
constructor() {
|
|
38
|
+
super((readable) => {
|
|
39
|
+
const transform = new AssistantTransformStream({
|
|
40
|
+
transform(chunk, controller) {
|
|
41
|
+
controller.appendText(chunk);
|
|
42
|
+
},
|
|
43
|
+
});
|
|
44
|
+
return readable
|
|
45
|
+
.pipeThrough(new TextDecoderStream())
|
|
46
|
+
.pipeThrough(transform);
|
|
47
|
+
});
|
|
48
|
+
}
|
|
49
|
+
}
|
|
51
50
|
//# sourceMappingURL=PlainText.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"
|
|
1
|
+
{"version":3,"file":"PlainText.js","sourceRoot":"","sources":["../../../src/core/serialization/PlainText.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,wBAAwB,EAAE,oDAAiD;AACpF,OAAO,EAAE,uBAAuB,EAAE,mDAAgD;AAElF,MAAM,OAAO,gBACX,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,CAA+B;gBAClE,SAAS,CAAC,KAAK,EAAE,UAAU;oBACzB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;oBACxB,QAAQ,IAAI,EAAE,CAAC;wBACb,KAAK,YAAY;4BACf,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;4BACpC,MAAM;wBAER,KAAK,YAAY,CAAC;wBAClB,KAAK,aAAa,CAAC;wBACnB,KAAK,YAAY,CAAC;wBAClB,KAAK,aAAa,CAAC;wBACnB,KAAK,gBAAgB,CAAC;wBACtB,KAAK,OAAO;4BACV,MAAM;wBAER;4BACE,MAAM,eAAe,GAOA,IAAI,CAAC;4BAC1B,MAAM,IAAI,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,QAAQ;iBACZ,WAAW,CAAC,SAAS,CAAC;iBACtB,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,gBAAiB,SAAQ,uBAGrC;IACC;QACE,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,MAAM,SAAS,GAAG,IAAI,wBAAwB,CAAS;gBACrD,SAAS,CAAC,KAAK,EAAE,UAAU;oBACzB,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC/B,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,QAAQ;iBACZ,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;iBACpC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { AssistantStreamChunk } from "../../AssistantStreamChunk";
|
|
2
|
-
import { PipeableTransformStream } from "../../utils/stream/PipeableTransformStream";
|
|
3
|
-
import { AssistantStreamEncoder } from "../../AssistantStream";
|
|
1
|
+
import { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
|
|
2
|
+
import { PipeableTransformStream } from "../../utils/stream/PipeableTransformStream.js";
|
|
3
|
+
import { AssistantStreamEncoder } from "../../AssistantStream.js";
|
|
4
4
|
/**
|
|
5
5
|
* AssistantTransportEncoder encodes AssistantStreamChunks into SSE format
|
|
6
6
|
* and emits [DONE] when the stream completes.
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AssistantTransport.d.ts","sourceRoot":"","sources":["../../../../src/core/serialization/assistant-transport/AssistantTransport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,
|
|
1
|
+
{"version":3,"file":"AssistantTransport.d.ts","sourceRoot":"","sources":["../../../../src/core/serialization/assistant-transport/AssistantTransport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,sCAAmC;AAClE,OAAO,EAAE,uBAAuB,EAAE,sDAAmD;AAErF,OAAO,EAAE,sBAAsB,EAAE,iCAA8B;AAE/D;;;GAGG;AACH,qBAAa,yBACX,SAAQ,uBAAuB,CAAC,oBAAoB,EAAE,UAAU,CAAC,WAAW,CAAC,CAC7E,YAAW,sBAAsB;IAEjC,OAAO,UAIJ;;CAkBJ;AAoED;;;GAGG;AACH,qBAAa,yBAA0B,SAAQ,uBAAuB,CACpE,UAAU,CAAC,WAAW,CAAC,EACvB,oBAAoB,CACrB;;CAsCA"}
|