assistant-stream 0.1.8 → 0.2.0
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/ai-sdk/package.json +2 -2
- package/dist/ai-sdk/index.js +7 -34
- package/dist/ai-sdk/index.js.map +1 -1
- package/dist/ai-sdk/language-model.js +5 -31
- package/dist/ai-sdk/language-model.js.map +1 -1
- package/dist/ai-sdk.js +4 -32
- package/dist/ai-sdk.js.map +1 -1
- package/dist/core/AssistantStream.js +3 -29
- package/dist/core/AssistantStream.js.map +1 -1
- package/dist/core/AssistantStreamChunk.js +0 -18
- package/dist/core/AssistantStreamChunk.js.map +1 -1
- package/dist/core/accumulators/AssistantMessageStream.js +9 -35
- package/dist/core/accumulators/AssistantMessageStream.js.map +1 -1
- package/dist/core/accumulators/assistant-message-accumulator.js +22 -48
- package/dist/core/accumulators/assistant-message-accumulator.js.map +1 -1
- package/dist/core/index.js +16 -46
- package/dist/core/index.js.map +1 -1
- package/dist/core/modules/assistant-stream.js +29 -52
- package/dist/core/modules/assistant-stream.js.map +1 -1
- package/dist/core/modules/text.js +6 -33
- package/dist/core/modules/text.js.map +1 -1
- package/dist/core/modules/tool-call.js +8 -35
- package/dist/core/modules/tool-call.js.map +1 -1
- package/dist/core/serialization/PlainText.js +9 -36
- package/dist/core/serialization/PlainText.js.map +1 -1
- package/dist/core/serialization/data-stream/DataStream.js +57 -79
- package/dist/core/serialization/data-stream/DataStream.js.map +1 -1
- package/dist/core/serialization/data-stream/chunk-types.js +2 -28
- package/dist/core/serialization/data-stream/chunk-types.js.map +1 -1
- package/dist/core/serialization/data-stream/serialization.js +6 -33
- package/dist/core/serialization/data-stream/serialization.js.map +1 -1
- package/dist/core/tool/ToolCallReader.js +30 -55
- package/dist/core/tool/ToolCallReader.js.map +1 -1
- package/dist/core/tool/ToolExecutionStream.js +18 -52
- package/dist/core/tool/ToolExecutionStream.js.map +1 -1
- package/dist/core/tool/ToolResponse.js +5 -31
- package/dist/core/tool/ToolResponse.js.map +1 -1
- package/dist/core/tool/index.js +9 -35
- package/dist/core/tool/index.js.map +1 -1
- package/dist/core/tool/tool-types.js +0 -18
- package/dist/core/tool/tool-types.js.map +1 -1
- package/dist/core/tool/toolResultStream.js +8 -35
- package/dist/core/tool/toolResultStream.js.map +1 -1
- package/dist/core/tool/type-path-utils.js +0 -18
- package/dist/core/tool/type-path-utils.js.map +1 -1
- package/dist/core/utils/Counter.js +4 -30
- package/dist/core/utils/Counter.js.map +1 -1
- package/dist/core/utils/generateId.js +4 -30
- package/dist/core/utils/generateId.js.map +1 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.js +4 -30
- package/dist/core/utils/stream/AssistantMetaTransformStream.js.map +1 -1
- package/dist/core/utils/stream/AssistantTransformStream.js +8 -32
- package/dist/core/utils/stream/AssistantTransformStream.js.map +1 -1
- package/dist/core/utils/stream/LineDecoderStream.js +4 -30
- package/dist/core/utils/stream/LineDecoderStream.js.map +1 -1
- package/dist/core/utils/stream/PipeableTransformStream.js +4 -30
- package/dist/core/utils/stream/PipeableTransformStream.js.map +1 -1
- package/dist/core/utils/stream/UnderlyingReadable.js +0 -18
- package/dist/core/utils/stream/UnderlyingReadable.js.map +1 -1
- package/dist/core/utils/stream/merge.js +5 -31
- package/dist/core/utils/stream/merge.js.map +1 -1
- package/dist/core/utils/stream/path-utils.js +10 -38
- package/dist/core/utils/stream/path-utils.js.map +1 -1
- package/dist/core/utils/types.js +0 -18
- package/dist/core/utils/types.js.map +1 -1
- package/dist/core/utils/withPromiseOrValue.js +2 -28
- package/dist/core/utils/withPromiseOrValue.js.map +1 -1
- package/dist/index.js +1 -24
- package/dist/index.js.map +1 -1
- package/dist/utils/AsyncIterableStream.js +2 -28
- package/dist/utils/AsyncIterableStream.js.map +1 -1
- package/dist/utils/json/fix-json.js +2 -28
- package/dist/utils/json/fix-json.js.map +1 -1
- package/dist/utils/json/is-json.js +2 -30
- package/dist/utils/json/is-json.js.map +1 -1
- package/dist/utils/json/json-value.js +0 -18
- package/dist/utils/json/json-value.js.map +1 -1
- package/dist/utils/json/parse-partial-json-object.js +12 -50
- package/dist/utils/json/parse-partial-json-object.js.map +1 -1
- package/dist/utils/promiseWithResolvers.js +3 -29
- package/dist/utils/promiseWithResolvers.js.map +1 -1
- package/dist/utils.js +9 -32
- package/dist/utils.js.map +1 -1
- package/package.json +5 -8
- package/utils/package.json +1 -1
- package/dist/ai-sdk/index.d.ts +0 -5
- package/dist/ai-sdk/index.d.ts.map +0 -1
- package/dist/ai-sdk/index.mjs +0 -173
- package/dist/ai-sdk/index.mjs.map +0 -1
- package/dist/ai-sdk/language-model.d.ts +0 -6
- package/dist/ai-sdk/language-model.d.ts.map +0 -1
- package/dist/ai-sdk/language-model.mjs +0 -106
- package/dist/ai-sdk/language-model.mjs.map +0 -1
- package/dist/ai-sdk.d.ts +0 -3
- package/dist/ai-sdk.d.ts.map +0 -1
- package/dist/ai-sdk.mjs +0 -7
- package/dist/ai-sdk.mjs.map +0 -1
- package/dist/core/AssistantStream.d.ts +0 -12
- package/dist/core/AssistantStream.d.ts.map +0 -1
- package/dist/core/AssistantStream.mjs +0 -21
- package/dist/core/AssistantStream.mjs.map +0 -1
- package/dist/core/AssistantStreamChunk.d.ts +0 -64
- package/dist/core/AssistantStreamChunk.d.ts.map +0 -1
- package/dist/core/AssistantStreamChunk.mjs +0 -1
- package/dist/core/AssistantStreamChunk.mjs.map +0 -1
- package/dist/core/accumulators/AssistantMessageStream.d.ts +0 -13
- package/dist/core/accumulators/AssistantMessageStream.d.ts.map +0 -1
- package/dist/core/accumulators/AssistantMessageStream.mjs +0 -54
- package/dist/core/accumulators/AssistantMessageStream.mjs.map +0 -1
- package/dist/core/accumulators/assistant-message-accumulator.d.ts +0 -8
- package/dist/core/accumulators/assistant-message-accumulator.d.ts.map +0 -1
- package/dist/core/accumulators/assistant-message-accumulator.mjs +0 -309
- package/dist/core/accumulators/assistant-message-accumulator.mjs.map +0 -1
- package/dist/core/index.d.ts +0 -11
- package/dist/core/index.d.ts.map +0 -1
- package/dist/core/index.mjs +0 -26
- package/dist/core/index.mjs.map +0 -1
- package/dist/core/modules/assistant-stream.d.ts +0 -31
- package/dist/core/modules/assistant-stream.d.ts.map +0 -1
- package/dist/core/modules/assistant-stream.mjs +0 -202
- package/dist/core/modules/assistant-stream.mjs.map +0 -1
- package/dist/core/modules/text.d.ts +0 -9
- package/dist/core/modules/text.d.ts.map +0 -1
- package/dist/core/modules/text.mjs +0 -52
- package/dist/core/modules/text.mjs.map +0 -1
- package/dist/core/modules/tool-call.d.ts +0 -13
- package/dist/core/modules/tool-call.d.ts.map +0 -1
- package/dist/core/modules/tool-call.mjs +0 -84
- package/dist/core/modules/tool-call.mjs.map +0 -1
- package/dist/core/serialization/PlainText.d.ts +0 -11
- package/dist/core/serialization/PlainText.d.ts.map +0 -1
- package/dist/core/serialization/PlainText.mjs +0 -44
- package/dist/core/serialization/PlainText.mjs.map +0 -1
- package/dist/core/serialization/data-stream/DataStream.d.ts +0 -11
- package/dist/core/serialization/data-stream/DataStream.d.ts.map +0 -1
- package/dist/core/serialization/data-stream/DataStream.mjs +0 -311
- package/dist/core/serialization/data-stream/DataStream.mjs.map +0 -1
- package/dist/core/serialization/data-stream/chunk-types.d.ts +0 -86
- package/dist/core/serialization/data-stream/chunk-types.d.ts.map +0 -1
- package/dist/core/serialization/data-stream/chunk-types.mjs +0 -24
- package/dist/core/serialization/data-stream/chunk-types.mjs.map +0 -1
- package/dist/core/serialization/data-stream/serialization.d.ts +0 -8
- package/dist/core/serialization/data-stream/serialization.d.ts.map +0 -1
- package/dist/core/serialization/data-stream/serialization.mjs +0 -30
- package/dist/core/serialization/data-stream/serialization.mjs.map +0 -1
- package/dist/core/tool/ToolCallReader.d.ts +0 -33
- package/dist/core/tool/ToolCallReader.d.ts.map +0 -1
- package/dist/core/tool/ToolCallReader.mjs +0 -315
- package/dist/core/tool/ToolCallReader.mjs.map +0 -1
- package/dist/core/tool/ToolExecutionStream.d.ts +0 -24
- package/dist/core/tool/ToolExecutionStream.d.ts.map +0 -1
- package/dist/core/tool/ToolExecutionStream.mjs +0 -132
- package/dist/core/tool/ToolExecutionStream.mjs.map +0 -1
- package/dist/core/tool/ToolResponse.d.ts +0 -17
- package/dist/core/tool/ToolResponse.d.ts.map +0 -1
- package/dist/core/tool/ToolResponse.mjs +0 -22
- package/dist/core/tool/ToolResponse.mjs.map +0 -1
- package/dist/core/tool/index.d.ts +0 -6
- package/dist/core/tool/index.d.ts.map +0 -1
- package/dist/core/tool/index.mjs +0 -14
- package/dist/core/tool/index.mjs.map +0 -1
- package/dist/core/tool/tool-types.d.ts +0 -76
- package/dist/core/tool/tool-types.d.ts.map +0 -1
- package/dist/core/tool/tool-types.mjs +0 -1
- package/dist/core/tool/tool-types.mjs.map +0 -1
- package/dist/core/tool/toolResultStream.d.ts +0 -6
- package/dist/core/tool/toolResultStream.d.ts.map +0 -1
- package/dist/core/tool/toolResultStream.mjs +0 -78
- package/dist/core/tool/toolResultStream.mjs.map +0 -1
- package/dist/core/tool/type-path-utils.d.ts +0 -23
- package/dist/core/tool/type-path-utils.d.ts.map +0 -1
- package/dist/core/tool/type-path-utils.mjs +0 -1
- package/dist/core/tool/type-path-utils.mjs.map +0 -1
- package/dist/core/utils/Counter.d.ts +0 -5
- package/dist/core/utils/Counter.d.ts.map +0 -1
- package/dist/core/utils/Counter.mjs +0 -11
- package/dist/core/utils/Counter.mjs.map +0 -1
- package/dist/core/utils/generateId.d.ts +0 -2
- package/dist/core/utils/generateId.d.ts.map +0 -1
- package/dist/core/utils/generateId.mjs +0 -10
- package/dist/core/utils/generateId.mjs.map +0 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts +0 -20
- package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts.map +0 -1
- package/dist/core/utils/stream/AssistantMetaTransformStream.mjs +0 -44
- package/dist/core/utils/stream/AssistantMetaTransformStream.mjs.map +0 -1
- package/dist/core/utils/stream/AssistantTransformStream.d.ts +0 -15
- package/dist/core/utils/stream/AssistantTransformStream.d.ts.map +0 -1
- package/dist/core/utils/stream/AssistantTransformStream.mjs +0 -46
- package/dist/core/utils/stream/AssistantTransformStream.mjs.map +0 -1
- package/dist/core/utils/stream/LineDecoderStream.d.ts +0 -5
- package/dist/core/utils/stream/LineDecoderStream.d.ts.map +0 -1
- package/dist/core/utils/stream/LineDecoderStream.mjs +0 -27
- package/dist/core/utils/stream/LineDecoderStream.mjs.map +0 -1
- package/dist/core/utils/stream/PipeableTransformStream.d.ts +0 -4
- package/dist/core/utils/stream/PipeableTransformStream.d.ts.map +0 -1
- package/dist/core/utils/stream/PipeableTransformStream.mjs +0 -15
- package/dist/core/utils/stream/PipeableTransformStream.mjs.map +0 -1
- package/dist/core/utils/stream/UnderlyingReadable.d.ts +0 -6
- package/dist/core/utils/stream/UnderlyingReadable.d.ts.map +0 -1
- package/dist/core/utils/stream/UnderlyingReadable.mjs +0 -1
- package/dist/core/utils/stream/UnderlyingReadable.mjs.map +0 -1
- package/dist/core/utils/stream/merge.d.ts +0 -9
- package/dist/core/utils/stream/merge.d.ts.map +0 -1
- package/dist/core/utils/stream/merge.mjs +0 -85
- package/dist/core/utils/stream/merge.mjs.map +0 -1
- package/dist/core/utils/stream/path-utils.d.ts +0 -12
- package/dist/core/utils/stream/path-utils.d.ts.map +0 -1
- package/dist/core/utils/stream/path-utils.mjs +0 -61
- package/dist/core/utils/stream/path-utils.mjs.map +0 -1
- package/dist/core/utils/types.d.ts +0 -102
- package/dist/core/utils/types.d.ts.map +0 -1
- package/dist/core/utils/types.mjs +0 -1
- package/dist/core/utils/types.mjs.map +0 -1
- package/dist/core/utils/withPromiseOrValue.d.ts +0 -2
- package/dist/core/utils/withPromiseOrValue.d.ts.map +0 -1
- package/dist/core/utils/withPromiseOrValue.mjs +0 -17
- package/dist/core/utils/withPromiseOrValue.mjs.map +0 -1
- package/dist/index.d.ts +0 -2
- package/dist/index.d.ts.map +0 -1
- package/dist/index.mjs +0 -3
- package/dist/index.mjs.map +0 -1
- package/dist/utils/AsyncIterableStream.d.ts +0 -3
- package/dist/utils/AsyncIterableStream.d.ts.map +0 -1
- package/dist/utils/AsyncIterableStream.mjs +0 -21
- package/dist/utils/AsyncIterableStream.mjs.map +0 -1
- package/dist/utils/json/fix-json.d.ts +0 -2
- package/dist/utils/json/fix-json.d.ts.map +0 -1
- package/dist/utils/json/fix-json.mjs +0 -362
- package/dist/utils/json/fix-json.mjs.map +0 -1
- package/dist/utils/json/is-json.d.ts +0 -5
- package/dist/utils/json/is-json.d.ts.map +0 -1
- package/dist/utils/json/is-json.mjs +0 -29
- package/dist/utils/json/is-json.mjs.map +0 -1
- package/dist/utils/json/json-value.d.ts +0 -6
- package/dist/utils/json/json-value.d.ts.map +0 -1
- package/dist/utils/json/json-value.mjs +0 -1
- package/dist/utils/json/json-value.mjs.map +0 -1
- package/dist/utils/json/parse-partial-json-object.d.ts +0 -14
- package/dist/utils/json/parse-partial-json-object.d.ts.map +0 -1
- package/dist/utils/json/parse-partial-json-object.mjs +0 -65
- package/dist/utils/json/parse-partial-json-object.mjs.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
- package/dist/utils/promiseWithResolvers.d.ts +0 -6
- package/dist/utils/promiseWithResolvers.d.ts.map +0 -1
- package/dist/utils/promiseWithResolvers.mjs +0 -15
- package/dist/utils/promiseWithResolvers.mjs.map +0 -1
- package/dist/utils.d.ts +0 -4
- package/dist/utils.d.ts.map +0 -1
- package/dist/utils.mjs +0 -14
- package/dist/utils.mjs.map +0 -1
|
@@ -1,54 +0,0 @@
|
|
|
1
|
-
// src/core/accumulators/AssistantMessageStream.ts
|
|
2
|
-
import { AssistantMessageAccumulator } from "./assistant-message-accumulator.mjs";
|
|
3
|
-
var AssistantMessageStream = class _AssistantMessageStream {
|
|
4
|
-
constructor(readable) {
|
|
5
|
-
this.readable = readable;
|
|
6
|
-
this.readable = readable;
|
|
7
|
-
}
|
|
8
|
-
static fromAssistantStream(stream) {
|
|
9
|
-
return new _AssistantMessageStream(
|
|
10
|
-
stream.pipeThrough(new AssistantMessageAccumulator())
|
|
11
|
-
);
|
|
12
|
-
}
|
|
13
|
-
async unstable_result() {
|
|
14
|
-
let last;
|
|
15
|
-
for await (const chunk of this) {
|
|
16
|
-
last = chunk;
|
|
17
|
-
}
|
|
18
|
-
if (!last) {
|
|
19
|
-
return {
|
|
20
|
-
role: "assistant",
|
|
21
|
-
status: { type: "complete", reason: "unknown" },
|
|
22
|
-
parts: [],
|
|
23
|
-
content: [],
|
|
24
|
-
metadata: {
|
|
25
|
-
unstable_data: [],
|
|
26
|
-
unstable_annotations: [],
|
|
27
|
-
steps: [],
|
|
28
|
-
custom: {}
|
|
29
|
-
}
|
|
30
|
-
};
|
|
31
|
-
}
|
|
32
|
-
return last;
|
|
33
|
-
}
|
|
34
|
-
[Symbol.asyncIterator]() {
|
|
35
|
-
const reader = this.readable.getReader();
|
|
36
|
-
return {
|
|
37
|
-
async next() {
|
|
38
|
-
const { done, value } = await reader.read();
|
|
39
|
-
return done ? { done: true, value: void 0 } : { done: false, value };
|
|
40
|
-
}
|
|
41
|
-
};
|
|
42
|
-
}
|
|
43
|
-
tee() {
|
|
44
|
-
const [readable1, readable2] = this.readable.tee();
|
|
45
|
-
return [
|
|
46
|
-
new _AssistantMessageStream(readable1),
|
|
47
|
-
new _AssistantMessageStream(readable2)
|
|
48
|
-
];
|
|
49
|
-
}
|
|
50
|
-
};
|
|
51
|
-
export {
|
|
52
|
-
AssistantMessageStream
|
|
53
|
-
};
|
|
54
|
-
//# sourceMappingURL=AssistantMessageStream.mjs.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../src/core/accumulators/AssistantMessageStream.ts"],"sourcesContent":["import { AssistantStream } from \"../AssistantStream\";\nimport { AssistantMessage } from \"../utils/types\";\nimport { AssistantMessageAccumulator } from \"./assistant-message-accumulator\";\n\nexport class AssistantMessageStream {\n constructor(public readonly readable: ReadableStream<AssistantMessage>) {\n this.readable = readable;\n }\n\n static fromAssistantStream(stream: AssistantStream) {\n return new AssistantMessageStream(\n stream.pipeThrough(new AssistantMessageAccumulator()),\n );\n }\n\n async unstable_result(): Promise<AssistantMessage> {\n let last: AssistantMessage | undefined;\n for await (const chunk of this) {\n last = chunk;\n }\n\n if (!last) {\n return {\n role: \"assistant\",\n status: { type: \"complete\", reason: \"unknown\" },\n parts: [],\n content: [],\n metadata: {\n unstable_data: [],\n unstable_annotations: [],\n steps: [],\n custom: {},\n },\n };\n }\n return last;\n }\n\n [Symbol.asyncIterator]() {\n const reader = this.readable.getReader();\n return {\n async next(): Promise<IteratorResult<AssistantMessage, undefined>> {\n const { done, value } = await reader.read();\n return done ? { done: true, value: undefined } : { done: false, value };\n },\n };\n }\n\n tee(): [AssistantMessageStream, AssistantMessageStream] {\n const [readable1, readable2] = this.readable.tee();\n return [\n new AssistantMessageStream(readable1),\n new AssistantMessageStream(readable2),\n ];\n }\n}\n"],"mappings":";AAEA,SAAS,mCAAmC;AAErC,IAAM,yBAAN,MAAM,wBAAuB;AAAA,EAClC,YAA4B,UAA4C;AAA5C;AAC1B,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,OAAO,oBAAoB,QAAyB;AAClD,WAAO,IAAI;AAAA,MACT,OAAO,YAAY,IAAI,4BAA4B,CAAC;AAAA,IACtD;AAAA,EACF;AAAA,EAEA,MAAM,kBAA6C;AACjD,QAAI;AACJ,qBAAiB,SAAS,MAAM;AAC9B,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM;AACT,aAAO;AAAA,QACL,MAAM;AAAA,QACN,QAAQ,EAAE,MAAM,YAAY,QAAQ,UAAU;AAAA,QAC9C,OAAO,CAAC;AAAA,QACR,SAAS,CAAC;AAAA,QACV,UAAU;AAAA,UACR,eAAe,CAAC;AAAA,UAChB,sBAAsB,CAAC;AAAA,UACvB,OAAO,CAAC;AAAA,UACR,QAAQ,CAAC;AAAA,QACX;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEA,CAAC,OAAO,aAAa,IAAI;AACvB,UAAM,SAAS,KAAK,SAAS,UAAU;AACvC,WAAO;AAAA,MACL,MAAM,OAA6D;AACjE,cAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,eAAO,OAAO,EAAE,MAAM,MAAM,OAAO,OAAU,IAAI,EAAE,MAAM,OAAO,MAAM;AAAA,MACxE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAwD;AACtD,UAAM,CAAC,WAAW,SAAS,IAAI,KAAK,SAAS,IAAI;AACjD,WAAO;AAAA,MACL,IAAI,wBAAuB,SAAS;AAAA,MACpC,IAAI,wBAAuB,SAAS;AAAA,IACtC;AAAA,EACF;AACF;","names":[]}
|
|
@@ -1,8 +0,0 @@
|
|
|
1
|
-
import { AssistantStreamChunk } from "../AssistantStreamChunk";
|
|
2
|
-
import { AssistantMessage } from "../utils/types";
|
|
3
|
-
export declare class AssistantMessageAccumulator extends TransformStream<AssistantStreamChunk, AssistantMessage> {
|
|
4
|
-
constructor({ initialMessage, }?: {
|
|
5
|
-
initialMessage?: AssistantMessage;
|
|
6
|
-
});
|
|
7
|
-
}
|
|
8
|
-
//# sourceMappingURL=assistant-message-accumulator.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"assistant-message-accumulator.d.ts","sourceRoot":"","sources":["../../../src/core/accumulators/assistant-message-accumulator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAG/D,OAAO,EACL,gBAAgB,EAQjB,MAAM,gBAAgB,CAAC;AAkTxB,qBAAa,2BAA4B,SAAQ,eAAe,CAC9D,oBAAoB,EACpB,gBAAgB,CACjB;gBACa,EACV,cAAc,GACf,GAAE;QACD,cAAc,CAAC,EAAE,gBAAgB,CAAC;KAC9B;CAmEP"}
|
|
@@ -1,309 +0,0 @@
|
|
|
1
|
-
// src/core/accumulators/assistant-message-accumulator.ts
|
|
2
|
-
import { generateId } from "../utils/generateId.mjs";
|
|
3
|
-
import { parsePartialJsonObject } from "../../utils/json/parse-partial-json-object.mjs";
|
|
4
|
-
var createInitialMessage = () => ({
|
|
5
|
-
role: "assistant",
|
|
6
|
-
status: { type: "running" },
|
|
7
|
-
parts: [],
|
|
8
|
-
get content() {
|
|
9
|
-
return this.parts;
|
|
10
|
-
},
|
|
11
|
-
metadata: {
|
|
12
|
-
unstable_data: [],
|
|
13
|
-
unstable_annotations: [],
|
|
14
|
-
steps: [],
|
|
15
|
-
custom: {}
|
|
16
|
-
}
|
|
17
|
-
});
|
|
18
|
-
var updatePartForPath = (message, chunk, updater) => {
|
|
19
|
-
if (message.parts.length === 0) {
|
|
20
|
-
throw new Error("No parts available to update.");
|
|
21
|
-
}
|
|
22
|
-
if (chunk.path.length !== 1)
|
|
23
|
-
throw new Error("Nested paths are not supported yet.");
|
|
24
|
-
const partIndex = chunk.path[0];
|
|
25
|
-
const updatedPart = updater(message.parts[partIndex]);
|
|
26
|
-
return {
|
|
27
|
-
...message,
|
|
28
|
-
parts: [
|
|
29
|
-
...message.parts.slice(0, partIndex),
|
|
30
|
-
updatedPart,
|
|
31
|
-
...message.parts.slice(partIndex + 1)
|
|
32
|
-
],
|
|
33
|
-
get content() {
|
|
34
|
-
return this.parts;
|
|
35
|
-
}
|
|
36
|
-
};
|
|
37
|
-
};
|
|
38
|
-
var handlePartStart = (message, chunk) => {
|
|
39
|
-
const partInit = chunk.part;
|
|
40
|
-
if (partInit.type === "text" || partInit.type === "reasoning") {
|
|
41
|
-
const newTextPart = {
|
|
42
|
-
type: partInit.type,
|
|
43
|
-
text: "",
|
|
44
|
-
status: { type: "running" }
|
|
45
|
-
};
|
|
46
|
-
return {
|
|
47
|
-
...message,
|
|
48
|
-
parts: [...message.parts, newTextPart],
|
|
49
|
-
get content() {
|
|
50
|
-
return this.parts;
|
|
51
|
-
}
|
|
52
|
-
};
|
|
53
|
-
} else if (partInit.type === "tool-call") {
|
|
54
|
-
const newToolCallPart = {
|
|
55
|
-
type: "tool-call",
|
|
56
|
-
state: "partial-call",
|
|
57
|
-
status: { type: "running", isArgsComplete: false },
|
|
58
|
-
toolCallId: partInit.toolCallId,
|
|
59
|
-
toolName: partInit.toolName,
|
|
60
|
-
argsText: "",
|
|
61
|
-
args: {}
|
|
62
|
-
};
|
|
63
|
-
return {
|
|
64
|
-
...message,
|
|
65
|
-
parts: [...message.parts, newToolCallPart],
|
|
66
|
-
get content() {
|
|
67
|
-
return this.parts;
|
|
68
|
-
}
|
|
69
|
-
};
|
|
70
|
-
} else if (partInit.type === "source") {
|
|
71
|
-
const newSourcePart = {
|
|
72
|
-
type: "source",
|
|
73
|
-
sourceType: partInit.sourceType,
|
|
74
|
-
id: partInit.id,
|
|
75
|
-
url: partInit.url,
|
|
76
|
-
...partInit.title ? { title: partInit.title } : void 0
|
|
77
|
-
};
|
|
78
|
-
return {
|
|
79
|
-
...message,
|
|
80
|
-
parts: [...message.parts, newSourcePart],
|
|
81
|
-
get content() {
|
|
82
|
-
return this.parts;
|
|
83
|
-
}
|
|
84
|
-
};
|
|
85
|
-
} else if (partInit.type === "file") {
|
|
86
|
-
const newFilePart = {
|
|
87
|
-
type: "file",
|
|
88
|
-
mimeType: partInit.mimeType,
|
|
89
|
-
data: partInit.data
|
|
90
|
-
};
|
|
91
|
-
return {
|
|
92
|
-
...message,
|
|
93
|
-
parts: [...message.parts, newFilePart],
|
|
94
|
-
get content() {
|
|
95
|
-
return this.parts;
|
|
96
|
-
}
|
|
97
|
-
};
|
|
98
|
-
} else {
|
|
99
|
-
throw new Error(`Unsupported part type: ${partInit.type}`);
|
|
100
|
-
}
|
|
101
|
-
};
|
|
102
|
-
var handleToolCallArgsTextFinish = (message, chunk) => {
|
|
103
|
-
return updatePartForPath(message, chunk, (part) => {
|
|
104
|
-
if (part.type !== "tool-call") {
|
|
105
|
-
throw new Error("Last is not a tool call");
|
|
106
|
-
}
|
|
107
|
-
return {
|
|
108
|
-
...part,
|
|
109
|
-
state: "call"
|
|
110
|
-
};
|
|
111
|
-
});
|
|
112
|
-
};
|
|
113
|
-
var handlePartFinish = (message, chunk) => {
|
|
114
|
-
return updatePartForPath(message, chunk, (part) => ({
|
|
115
|
-
...part,
|
|
116
|
-
status: { type: "complete", reason: "unknown" }
|
|
117
|
-
}));
|
|
118
|
-
};
|
|
119
|
-
var handleTextDelta = (message, chunk) => {
|
|
120
|
-
return updatePartForPath(message, chunk, (part) => {
|
|
121
|
-
if (part.type === "text" || part.type === "reasoning") {
|
|
122
|
-
return { ...part, text: part.text + chunk.textDelta };
|
|
123
|
-
} else if (part.type === "tool-call") {
|
|
124
|
-
const newArgsText = part.argsText + chunk.textDelta;
|
|
125
|
-
const newArgs = parsePartialJsonObject(newArgsText) ?? part.args;
|
|
126
|
-
return { ...part, argsText: newArgsText, args: newArgs };
|
|
127
|
-
} else {
|
|
128
|
-
throw new Error(
|
|
129
|
-
"text-delta received but part is neither text nor tool-call"
|
|
130
|
-
);
|
|
131
|
-
}
|
|
132
|
-
});
|
|
133
|
-
};
|
|
134
|
-
var handleResult = (message, chunk) => {
|
|
135
|
-
return updatePartForPath(message, chunk, (part) => {
|
|
136
|
-
if (part.type === "tool-call") {
|
|
137
|
-
return {
|
|
138
|
-
...part,
|
|
139
|
-
state: "result",
|
|
140
|
-
artifact: chunk.artifact,
|
|
141
|
-
result: chunk.result,
|
|
142
|
-
isError: chunk.isError ?? false,
|
|
143
|
-
status: { type: "complete", reason: "stop" }
|
|
144
|
-
};
|
|
145
|
-
} else {
|
|
146
|
-
throw new Error("Result chunk received but part is not a tool-call");
|
|
147
|
-
}
|
|
148
|
-
});
|
|
149
|
-
};
|
|
150
|
-
var handleMessageFinish = (message, chunk) => {
|
|
151
|
-
const newStatus = getStatus(chunk);
|
|
152
|
-
return { ...message, status: newStatus };
|
|
153
|
-
};
|
|
154
|
-
var getStatus = (chunk) => {
|
|
155
|
-
if (chunk.finishReason === "tool-calls") {
|
|
156
|
-
return {
|
|
157
|
-
type: "requires-action",
|
|
158
|
-
reason: "tool-calls"
|
|
159
|
-
};
|
|
160
|
-
} else if (chunk.finishReason === "stop" || chunk.finishReason === "unknown") {
|
|
161
|
-
return {
|
|
162
|
-
type: "complete",
|
|
163
|
-
reason: chunk.finishReason
|
|
164
|
-
};
|
|
165
|
-
} else {
|
|
166
|
-
return {
|
|
167
|
-
type: "incomplete",
|
|
168
|
-
reason: chunk.finishReason
|
|
169
|
-
};
|
|
170
|
-
}
|
|
171
|
-
};
|
|
172
|
-
var handleAnnotations = (message, chunk) => {
|
|
173
|
-
return {
|
|
174
|
-
...message,
|
|
175
|
-
metadata: {
|
|
176
|
-
...message.metadata,
|
|
177
|
-
unstable_annotations: [
|
|
178
|
-
...message.metadata.unstable_annotations,
|
|
179
|
-
...chunk.annotations
|
|
180
|
-
]
|
|
181
|
-
}
|
|
182
|
-
};
|
|
183
|
-
};
|
|
184
|
-
var handleData = (message, chunk) => {
|
|
185
|
-
return {
|
|
186
|
-
...message,
|
|
187
|
-
metadata: {
|
|
188
|
-
...message.metadata,
|
|
189
|
-
unstable_data: [...message.metadata.unstable_data, ...chunk.data]
|
|
190
|
-
}
|
|
191
|
-
};
|
|
192
|
-
};
|
|
193
|
-
var handleStepStart = (message, chunk) => {
|
|
194
|
-
return {
|
|
195
|
-
...message,
|
|
196
|
-
metadata: {
|
|
197
|
-
...message.metadata,
|
|
198
|
-
steps: [
|
|
199
|
-
...message.metadata.steps,
|
|
200
|
-
{ state: "started", messageId: chunk.messageId }
|
|
201
|
-
]
|
|
202
|
-
}
|
|
203
|
-
};
|
|
204
|
-
};
|
|
205
|
-
var handleStepFinish = (message, chunk) => {
|
|
206
|
-
const steps = message.metadata.steps.slice();
|
|
207
|
-
const lastIndex = steps.length - 1;
|
|
208
|
-
if (steps.length > 0 && steps[lastIndex]?.state === "started") {
|
|
209
|
-
steps[lastIndex] = {
|
|
210
|
-
...steps[lastIndex],
|
|
211
|
-
state: "finished",
|
|
212
|
-
finishReason: chunk.finishReason,
|
|
213
|
-
usage: chunk.usage,
|
|
214
|
-
isContinued: chunk.isContinued
|
|
215
|
-
};
|
|
216
|
-
} else {
|
|
217
|
-
steps.push({
|
|
218
|
-
state: "finished",
|
|
219
|
-
messageId: generateId(),
|
|
220
|
-
finishReason: chunk.finishReason,
|
|
221
|
-
usage: chunk.usage,
|
|
222
|
-
isContinued: chunk.isContinued
|
|
223
|
-
});
|
|
224
|
-
}
|
|
225
|
-
return {
|
|
226
|
-
...message,
|
|
227
|
-
metadata: {
|
|
228
|
-
...message.metadata,
|
|
229
|
-
steps
|
|
230
|
-
}
|
|
231
|
-
};
|
|
232
|
-
};
|
|
233
|
-
var handleErrorChunk = (message, chunk) => {
|
|
234
|
-
return {
|
|
235
|
-
...message,
|
|
236
|
-
status: { type: "incomplete", reason: "error", error: chunk.error }
|
|
237
|
-
};
|
|
238
|
-
};
|
|
239
|
-
var AssistantMessageAccumulator = class extends TransformStream {
|
|
240
|
-
constructor({
|
|
241
|
-
initialMessage
|
|
242
|
-
} = {}) {
|
|
243
|
-
let message = initialMessage ?? createInitialMessage();
|
|
244
|
-
super({
|
|
245
|
-
transform(chunk, controller) {
|
|
246
|
-
const type = chunk.type;
|
|
247
|
-
switch (type) {
|
|
248
|
-
case "part-start":
|
|
249
|
-
message = handlePartStart(message, chunk);
|
|
250
|
-
break;
|
|
251
|
-
case "tool-call-args-text-finish":
|
|
252
|
-
message = handleToolCallArgsTextFinish(message, chunk);
|
|
253
|
-
break;
|
|
254
|
-
case "part-finish":
|
|
255
|
-
message = handlePartFinish(message, chunk);
|
|
256
|
-
break;
|
|
257
|
-
case "text-delta":
|
|
258
|
-
message = handleTextDelta(message, chunk);
|
|
259
|
-
break;
|
|
260
|
-
case "result":
|
|
261
|
-
message = handleResult(message, chunk);
|
|
262
|
-
break;
|
|
263
|
-
case "message-finish":
|
|
264
|
-
message = handleMessageFinish(message, chunk);
|
|
265
|
-
break;
|
|
266
|
-
case "annotations":
|
|
267
|
-
message = handleAnnotations(message, chunk);
|
|
268
|
-
break;
|
|
269
|
-
case "data":
|
|
270
|
-
message = handleData(message, chunk);
|
|
271
|
-
break;
|
|
272
|
-
case "step-start":
|
|
273
|
-
message = handleStepStart(message, chunk);
|
|
274
|
-
break;
|
|
275
|
-
case "step-finish":
|
|
276
|
-
message = handleStepFinish(message, chunk);
|
|
277
|
-
break;
|
|
278
|
-
case "error":
|
|
279
|
-
message = handleErrorChunk(message, chunk);
|
|
280
|
-
break;
|
|
281
|
-
default: {
|
|
282
|
-
const unhandledType = type;
|
|
283
|
-
throw new Error(`Unsupported chunk type: ${unhandledType}`);
|
|
284
|
-
}
|
|
285
|
-
}
|
|
286
|
-
controller.enqueue(message);
|
|
287
|
-
},
|
|
288
|
-
flush(controller) {
|
|
289
|
-
if (message.status?.type === "running") {
|
|
290
|
-
const requiresAction = message.parts?.at(-1)?.type === "tool-call";
|
|
291
|
-
message = handleMessageFinish(message, {
|
|
292
|
-
type: "message-finish",
|
|
293
|
-
path: [],
|
|
294
|
-
finishReason: requiresAction ? "tool-calls" : "unknown",
|
|
295
|
-
usage: {
|
|
296
|
-
promptTokens: 0,
|
|
297
|
-
completionTokens: 0
|
|
298
|
-
}
|
|
299
|
-
});
|
|
300
|
-
controller.enqueue(message);
|
|
301
|
-
}
|
|
302
|
-
}
|
|
303
|
-
});
|
|
304
|
-
}
|
|
305
|
-
};
|
|
306
|
-
export {
|
|
307
|
-
AssistantMessageAccumulator
|
|
308
|
-
};
|
|
309
|
-
//# sourceMappingURL=assistant-message-accumulator.mjs.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../src/core/accumulators/assistant-message-accumulator.ts"],"sourcesContent":["import { AssistantStreamChunk } from \"../AssistantStreamChunk\";\nimport { generateId } from \"../utils/generateId\";\nimport { parsePartialJsonObject } from \"../../utils/json/parse-partial-json-object\";\nimport {\n AssistantMessage,\n AssistantMessageStatus,\n TextPart,\n ToolCallPart,\n SourcePart,\n AssistantMessagePart,\n ReasoningPart,\n FilePart,\n} from \"../utils/types\";\n\nconst createInitialMessage = (): AssistantMessage => ({\n role: \"assistant\",\n status: { type: \"running\" },\n parts: [],\n get content() {\n return this.parts;\n },\n metadata: {\n unstable_data: [],\n unstable_annotations: [],\n steps: [],\n custom: {},\n },\n});\n\nconst updatePartForPath = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk,\n updater: (part: AssistantMessagePart) => AssistantMessagePart,\n): AssistantMessage => {\n if (message.parts.length === 0) {\n throw new Error(\"No parts available to update.\");\n }\n\n if (chunk.path.length !== 1)\n throw new Error(\"Nested paths are not supported yet.\");\n\n const partIndex = chunk.path[0]!;\n const updatedPart = updater(message.parts[partIndex]!);\n return {\n ...message,\n parts: [\n ...message.parts.slice(0, partIndex),\n updatedPart,\n ...message.parts.slice(partIndex + 1),\n ],\n get content() {\n return this.parts;\n },\n };\n};\n\nconst handlePartStart = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { readonly type: \"part-start\" },\n): AssistantMessage => {\n const partInit = chunk.part;\n if (partInit.type === \"text\" || partInit.type === \"reasoning\") {\n const newTextPart: TextPart | ReasoningPart = {\n type: partInit.type,\n text: \"\",\n status: { type: \"running\" },\n };\n return {\n ...message,\n parts: [...message.parts, newTextPart],\n get content() {\n return this.parts;\n },\n };\n } else if (partInit.type === \"tool-call\") {\n const newToolCallPart: ToolCallPart = {\n type: \"tool-call\",\n state: \"partial-call\",\n status: { type: \"running\", isArgsComplete: false },\n toolCallId: partInit.toolCallId,\n toolName: partInit.toolName,\n argsText: \"\",\n args: {},\n };\n return {\n ...message,\n parts: [...message.parts, newToolCallPart],\n get content() {\n return this.parts;\n },\n };\n } else if (partInit.type === \"source\") {\n const newSourcePart: SourcePart = {\n type: \"source\",\n sourceType: partInit.sourceType,\n id: partInit.id,\n url: partInit.url,\n ...(partInit.title ? { title: partInit.title } : undefined),\n };\n return {\n ...message,\n parts: [...message.parts, newSourcePart],\n get content() {\n return this.parts;\n },\n };\n } else if (partInit.type === \"file\") {\n const newFilePart: FilePart = {\n type: \"file\",\n mimeType: partInit.mimeType,\n data: partInit.data,\n };\n return {\n ...message,\n parts: [...message.parts, newFilePart],\n get content() {\n return this.parts;\n },\n };\n } else {\n throw new Error(`Unsupported part type: ${partInit.type}`);\n }\n};\n\nconst handleToolCallArgsTextFinish = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & {\n readonly type: \"tool-call-args-text-finish\";\n },\n): AssistantMessage => {\n return updatePartForPath(message, chunk, (part) => {\n if (part.type !== \"tool-call\") {\n throw new Error(\"Last is not a tool call\");\n }\n return {\n ...part,\n state: \"call\",\n };\n });\n};\n\nconst handlePartFinish = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { readonly type: \"part-finish\" },\n): AssistantMessage => {\n return updatePartForPath(message, chunk, (part) => ({\n ...part,\n status: { type: \"complete\", reason: \"unknown\" },\n }));\n};\n\nconst handleTextDelta = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"text-delta\" },\n): AssistantMessage => {\n return updatePartForPath(message, chunk, (part) => {\n if (part.type === \"text\" || part.type === \"reasoning\") {\n return { ...part, text: part.text + chunk.textDelta };\n } else if (part.type === \"tool-call\") {\n const newArgsText = part.argsText + chunk.textDelta;\n\n // Fall back to existing args if parsing fails\n const newArgs = parsePartialJsonObject(newArgsText) ?? part.args;\n\n return { ...part, argsText: newArgsText, args: newArgs };\n } else {\n throw new Error(\n \"text-delta received but part is neither text nor tool-call\",\n );\n }\n });\n};\n\nconst handleResult = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"result\" },\n): AssistantMessage => {\n return updatePartForPath(message, chunk, (part) => {\n if (part.type === \"tool-call\") {\n return {\n ...part,\n state: \"result\",\n artifact: chunk.artifact,\n result: chunk.result,\n isError: chunk.isError ?? false,\n status: { type: \"complete\", reason: \"stop\" },\n };\n } else {\n throw new Error(\"Result chunk received but part is not a tool-call\");\n }\n });\n};\n\nconst handleMessageFinish = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"message-finish\" },\n): AssistantMessage => {\n const newStatus = getStatus(chunk);\n return { ...message, status: newStatus };\n};\n\nconst getStatus = (\n chunk:\n | (AssistantStreamChunk & { type: \"message-finish\" })\n | (AssistantStreamChunk & { type: \"step-finish\" }),\n): AssistantMessageStatus => {\n if (chunk.finishReason === \"tool-calls\") {\n return {\n type: \"requires-action\",\n reason: \"tool-calls\",\n };\n } else if (\n chunk.finishReason === \"stop\" ||\n chunk.finishReason === \"unknown\"\n ) {\n return {\n type: \"complete\",\n reason: chunk.finishReason,\n };\n } else {\n return {\n type: \"incomplete\",\n reason: chunk.finishReason,\n };\n }\n};\n\nconst handleAnnotations = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"annotations\" },\n): AssistantMessage => {\n return {\n ...message,\n metadata: {\n ...message.metadata,\n unstable_annotations: [\n ...message.metadata.unstable_annotations,\n ...chunk.annotations,\n ],\n },\n };\n};\n\nconst handleData = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"data\" },\n): AssistantMessage => {\n return {\n ...message,\n metadata: {\n ...message.metadata,\n unstable_data: [...message.metadata.unstable_data, ...chunk.data],\n },\n };\n};\n\nconst handleStepStart = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"step-start\" },\n): AssistantMessage => {\n return {\n ...message,\n metadata: {\n ...message.metadata,\n steps: [\n ...message.metadata.steps,\n { state: \"started\", messageId: chunk.messageId },\n ],\n },\n };\n};\n\nconst handleStepFinish = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"step-finish\" },\n): AssistantMessage => {\n const steps = message.metadata.steps.slice();\n const lastIndex = steps.length - 1;\n\n // Check if the previous step is a step-start (has state \"started\")\n if (steps.length > 0 && steps[lastIndex]?.state === \"started\") {\n steps[lastIndex] = {\n ...steps[lastIndex],\n state: \"finished\",\n finishReason: chunk.finishReason,\n usage: chunk.usage,\n isContinued: chunk.isContinued,\n };\n } else {\n // If no previous step-start exists, append a finished step\n steps.push({\n state: \"finished\",\n messageId: generateId(),\n finishReason: chunk.finishReason,\n usage: chunk.usage,\n isContinued: chunk.isContinued,\n });\n }\n\n return {\n ...message,\n metadata: {\n ...message.metadata,\n steps,\n },\n };\n};\n\nconst handleErrorChunk = (\n message: AssistantMessage,\n chunk: AssistantStreamChunk & { type: \"error\" },\n): AssistantMessage => {\n return {\n ...message,\n status: { type: \"incomplete\", reason: \"error\", error: chunk.error },\n };\n};\n\nexport class AssistantMessageAccumulator extends TransformStream<\n AssistantStreamChunk,\n AssistantMessage\n> {\n constructor({\n initialMessage,\n }: {\n initialMessage?: AssistantMessage;\n } = {}) {\n let message = initialMessage ?? createInitialMessage();\n super({\n transform(chunk, controller) {\n const type = chunk.type;\n switch (type) {\n case \"part-start\":\n message = handlePartStart(message, chunk);\n break;\n\n case \"tool-call-args-text-finish\":\n message = handleToolCallArgsTextFinish(message, chunk);\n break;\n\n case \"part-finish\":\n message = handlePartFinish(message, chunk);\n break;\n\n case \"text-delta\":\n message = handleTextDelta(message, chunk);\n break;\n case \"result\":\n message = handleResult(message, chunk);\n break;\n case \"message-finish\":\n message = handleMessageFinish(message, chunk);\n break;\n case \"annotations\":\n message = handleAnnotations(message, chunk);\n break;\n case \"data\":\n message = handleData(message, chunk);\n break;\n case \"step-start\":\n message = handleStepStart(message, chunk);\n break;\n case \"step-finish\":\n message = handleStepFinish(message, chunk);\n break;\n case \"error\":\n message = handleErrorChunk(message, chunk);\n break;\n default: {\n const unhandledType: never = type;\n throw new Error(`Unsupported chunk type: ${unhandledType}`);\n }\n }\n controller.enqueue(message);\n },\n flush(controller) {\n if (message.status?.type === \"running\") {\n // TODO this test isn't sound\n const requiresAction = message.parts?.at(-1)?.type === \"tool-call\";\n message = handleMessageFinish(message, {\n type: \"message-finish\",\n path: [],\n finishReason: requiresAction ? \"tool-calls\" : \"unknown\",\n usage: {\n promptTokens: 0,\n completionTokens: 0,\n },\n });\n controller.enqueue(message);\n }\n },\n });\n }\n}\n"],"mappings":";AACA,SAAS,kBAAkB;AAC3B,SAAS,8BAA8B;AAYvC,IAAM,uBAAuB,OAAyB;AAAA,EACpD,MAAM;AAAA,EACN,QAAQ,EAAE,MAAM,UAAU;AAAA,EAC1B,OAAO,CAAC;AAAA,EACR,IAAI,UAAU;AACZ,WAAO,KAAK;AAAA,EACd;AAAA,EACA,UAAU;AAAA,IACR,eAAe,CAAC;AAAA,IAChB,sBAAsB,CAAC;AAAA,IACvB,OAAO,CAAC;AAAA,IACR,QAAQ,CAAC;AAAA,EACX;AACF;AAEA,IAAM,oBAAoB,CACxB,SACA,OACA,YACqB;AACrB,MAAI,QAAQ,MAAM,WAAW,GAAG;AAC9B,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,MAAI,MAAM,KAAK,WAAW;AACxB,UAAM,IAAI,MAAM,qCAAqC;AAEvD,QAAM,YAAY,MAAM,KAAK,CAAC;AAC9B,QAAM,cAAc,QAAQ,QAAQ,MAAM,SAAS,CAAE;AACrD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,OAAO;AAAA,MACL,GAAG,QAAQ,MAAM,MAAM,GAAG,SAAS;AAAA,MACnC;AAAA,MACA,GAAG,QAAQ,MAAM,MAAM,YAAY,CAAC;AAAA,IACtC;AAAA,IACA,IAAI,UAAU;AACZ,aAAO,KAAK;AAAA,IACd;AAAA,EACF;AACF;AAEA,IAAM,kBAAkB,CACtB,SACA,UACqB;AACrB,QAAM,WAAW,MAAM;AACvB,MAAI,SAAS,SAAS,UAAU,SAAS,SAAS,aAAa;AAC7D,UAAM,cAAwC;AAAA,MAC5C,MAAM,SAAS;AAAA,MACf,MAAM;AAAA,MACN,QAAQ,EAAE,MAAM,UAAU;AAAA,IAC5B;AACA,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,CAAC,GAAG,QAAQ,OAAO,WAAW;AAAA,MACrC,IAAI,UAAU;AACZ,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF,WAAW,SAAS,SAAS,aAAa;AACxC,UAAM,kBAAgC;AAAA,MACpC,MAAM;AAAA,MACN,OAAO;AAAA,MACP,QAAQ,EAAE,MAAM,WAAW,gBAAgB,MAAM;AAAA,MACjD,YAAY,SAAS;AAAA,MACrB,UAAU,SAAS;AAAA,MACnB,UAAU;AAAA,MACV,MAAM,CAAC;AAAA,IACT;AACA,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,CAAC,GAAG,QAAQ,OAAO,eAAe;AAAA,MACzC,IAAI,UAAU;AACZ,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF,WAAW,SAAS,SAAS,UAAU;AACrC,UAAM,gBAA4B;AAAA,MAChC,MAAM;AAAA,MACN,YAAY,SAAS;AAAA,MACrB,IAAI,SAAS;AAAA,MACb,KAAK,SAAS;AAAA,MACd,GAAI,SAAS,QAAQ,EAAE,OAAO,SAAS,MAAM,IAAI;AAAA,IACnD;AACA,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,CAAC,GAAG,QAAQ,OAAO,aAAa;AAAA,MACvC,IAAI,UAAU;AACZ,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF,WAAW,SAAS,SAAS,QAAQ;AACnC,UAAM,cAAwB;AAAA,MAC5B,MAAM;AAAA,MACN,UAAU,SAAS;AAAA,MACnB,MAAM,SAAS;AAAA,IACjB;AACA,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,CAAC,GAAG,QAAQ,OAAO,WAAW;AAAA,MACrC,IAAI,UAAU;AACZ,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF,OAAO;AACL,UAAM,IAAI,MAAM,0BAA0B,SAAS,IAAI,EAAE;AAAA,EAC3D;AACF;AAEA,IAAM,+BAA+B,CACnC,SACA,UAGqB;AACrB,SAAO,kBAAkB,SAAS,OAAO,CAAC,SAAS;AACjD,QAAI,KAAK,SAAS,aAAa;AAC7B,YAAM,IAAI,MAAM,yBAAyB;AAAA,IAC3C;AACA,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO;AAAA,IACT;AAAA,EACF,CAAC;AACH;AAEA,IAAM,mBAAmB,CACvB,SACA,UACqB;AACrB,SAAO,kBAAkB,SAAS,OAAO,CAAC,UAAU;AAAA,IAClD,GAAG;AAAA,IACH,QAAQ,EAAE,MAAM,YAAY,QAAQ,UAAU;AAAA,EAChD,EAAE;AACJ;AAEA,IAAM,kBAAkB,CACtB,SACA,UACqB;AACrB,SAAO,kBAAkB,SAAS,OAAO,CAAC,SAAS;AACjD,QAAI,KAAK,SAAS,UAAU,KAAK,SAAS,aAAa;AACrD,aAAO,EAAE,GAAG,MAAM,MAAM,KAAK,OAAO,MAAM,UAAU;AAAA,IACtD,WAAW,KAAK,SAAS,aAAa;AACpC,YAAM,cAAc,KAAK,WAAW,MAAM;AAG1C,YAAM,UAAU,uBAAuB,WAAW,KAAK,KAAK;AAE5D,aAAO,EAAE,GAAG,MAAM,UAAU,aAAa,MAAM,QAAQ;AAAA,IACzD,OAAO;AACL,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAEA,IAAM,eAAe,CACnB,SACA,UACqB;AACrB,SAAO,kBAAkB,SAAS,OAAO,CAAC,SAAS;AACjD,QAAI,KAAK,SAAS,aAAa;AAC7B,aAAO;AAAA,QACL,GAAG;AAAA,QACH,OAAO;AAAA,QACP,UAAU,MAAM;AAAA,QAChB,QAAQ,MAAM;AAAA,QACd,SAAS,MAAM,WAAW;AAAA,QAC1B,QAAQ,EAAE,MAAM,YAAY,QAAQ,OAAO;AAAA,MAC7C;AAAA,IACF,OAAO;AACL,YAAM,IAAI,MAAM,mDAAmD;AAAA,IACrE;AAAA,EACF,CAAC;AACH;AAEA,IAAM,sBAAsB,CAC1B,SACA,UACqB;AACrB,QAAM,YAAY,UAAU,KAAK;AACjC,SAAO,EAAE,GAAG,SAAS,QAAQ,UAAU;AACzC;AAEA,IAAM,YAAY,CAChB,UAG2B;AAC3B,MAAI,MAAM,iBAAiB,cAAc;AACvC,WAAO;AAAA,MACL,MAAM;AAAA,MACN,QAAQ;AAAA,IACV;AAAA,EACF,WACE,MAAM,iBAAiB,UACvB,MAAM,iBAAiB,WACvB;AACA,WAAO;AAAA,MACL,MAAM;AAAA,MACN,QAAQ,MAAM;AAAA,IAChB;AAAA,EACF,OAAO;AACL,WAAO;AAAA,MACL,MAAM;AAAA,MACN,QAAQ,MAAM;AAAA,IAChB;AAAA,EACF;AACF;AAEA,IAAM,oBAAoB,CACxB,SACA,UACqB;AACrB,SAAO;AAAA,IACL,GAAG;AAAA,IACH,UAAU;AAAA,MACR,GAAG,QAAQ;AAAA,MACX,sBAAsB;AAAA,QACpB,GAAG,QAAQ,SAAS;AAAA,QACpB,GAAG,MAAM;AAAA,MACX;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,aAAa,CACjB,SACA,UACqB;AACrB,SAAO;AAAA,IACL,GAAG;AAAA,IACH,UAAU;AAAA,MACR,GAAG,QAAQ;AAAA,MACX,eAAe,CAAC,GAAG,QAAQ,SAAS,eAAe,GAAG,MAAM,IAAI;AAAA,IAClE;AAAA,EACF;AACF;AAEA,IAAM,kBAAkB,CACtB,SACA,UACqB;AACrB,SAAO;AAAA,IACL,GAAG;AAAA,IACH,UAAU;AAAA,MACR,GAAG,QAAQ;AAAA,MACX,OAAO;AAAA,QACL,GAAG,QAAQ,SAAS;AAAA,QACpB,EAAE,OAAO,WAAW,WAAW,MAAM,UAAU;AAAA,MACjD;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,mBAAmB,CACvB,SACA,UACqB;AACrB,QAAM,QAAQ,QAAQ,SAAS,MAAM,MAAM;AAC3C,QAAM,YAAY,MAAM,SAAS;AAGjC,MAAI,MAAM,SAAS,KAAK,MAAM,SAAS,GAAG,UAAU,WAAW;AAC7D,UAAM,SAAS,IAAI;AAAA,MACjB,GAAG,MAAM,SAAS;AAAA,MAClB,OAAO;AAAA,MACP,cAAc,MAAM;AAAA,MACpB,OAAO,MAAM;AAAA,MACb,aAAa,MAAM;AAAA,IACrB;AAAA,EACF,OAAO;AAEL,UAAM,KAAK;AAAA,MACT,OAAO;AAAA,MACP,WAAW,WAAW;AAAA,MACtB,cAAc,MAAM;AAAA,MACpB,OAAO,MAAM;AAAA,MACb,aAAa,MAAM;AAAA,IACrB,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL,GAAG;AAAA,IACH,UAAU;AAAA,MACR,GAAG,QAAQ;AAAA,MACX;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,mBAAmB,CACvB,SACA,UACqB;AACrB,SAAO;AAAA,IACL,GAAG;AAAA,IACH,QAAQ,EAAE,MAAM,cAAc,QAAQ,SAAS,OAAO,MAAM,MAAM;AAAA,EACpE;AACF;AAEO,IAAM,8BAAN,cAA0C,gBAG/C;AAAA,EACA,YAAY;AAAA,IACV;AAAA,EACF,IAEI,CAAC,GAAG;AACN,QAAI,UAAU,kBAAkB,qBAAqB;AACrD,UAAM;AAAA,MACJ,UAAU,OAAO,YAAY;AAC3B,cAAM,OAAO,MAAM;AACnB,gBAAQ,MAAM;AAAA,UACZ,KAAK;AACH,sBAAU,gBAAgB,SAAS,KAAK;AACxC;AAAA,UAEF,KAAK;AACH,sBAAU,6BAA6B,SAAS,KAAK;AACrD;AAAA,UAEF,KAAK;AACH,sBAAU,iBAAiB,SAAS,KAAK;AACzC;AAAA,UAEF,KAAK;AACH,sBAAU,gBAAgB,SAAS,KAAK;AACxC;AAAA,UACF,KAAK;AACH,sBAAU,aAAa,SAAS,KAAK;AACrC;AAAA,UACF,KAAK;AACH,sBAAU,oBAAoB,SAAS,KAAK;AAC5C;AAAA,UACF,KAAK;AACH,sBAAU,kBAAkB,SAAS,KAAK;AAC1C;AAAA,UACF,KAAK;AACH,sBAAU,WAAW,SAAS,KAAK;AACnC;AAAA,UACF,KAAK;AACH,sBAAU,gBAAgB,SAAS,KAAK;AACxC;AAAA,UACF,KAAK;AACH,sBAAU,iBAAiB,SAAS,KAAK;AACzC;AAAA,UACF,KAAK;AACH,sBAAU,iBAAiB,SAAS,KAAK;AACzC;AAAA,UACF,SAAS;AACP,kBAAM,gBAAuB;AAC7B,kBAAM,IAAI,MAAM,2BAA2B,aAAa,EAAE;AAAA,UAC5D;AAAA,QACF;AACA,mBAAW,QAAQ,OAAO;AAAA,MAC5B;AAAA,MACA,MAAM,YAAY;AAChB,YAAI,QAAQ,QAAQ,SAAS,WAAW;AAEtC,gBAAM,iBAAiB,QAAQ,OAAO,GAAG,EAAE,GAAG,SAAS;AACvD,oBAAU,oBAAoB,SAAS;AAAA,YACrC,MAAM;AAAA,YACN,MAAM,CAAC;AAAA,YACP,cAAc,iBAAiB,eAAe;AAAA,YAC9C,OAAO;AAAA,cACL,cAAc;AAAA,cACd,kBAAkB;AAAA,YACpB;AAAA,UACF,CAAC;AACD,qBAAW,QAAQ,OAAO;AAAA,QAC5B;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;","names":[]}
|
package/dist/core/index.d.ts
DELETED
|
@@ -1,11 +0,0 @@
|
|
|
1
|
-
export { createAssistantStream, createAssistantStreamResponse, } from "./modules/assistant-stream";
|
|
2
|
-
export { AssistantMessageAccumulator } from "./accumulators/assistant-message-accumulator";
|
|
3
|
-
export { AssistantStream } from "./AssistantStream";
|
|
4
|
-
export type { AssistantStreamController } from "./modules/assistant-stream";
|
|
5
|
-
export type { AssistantStreamChunk } from "./AssistantStreamChunk";
|
|
6
|
-
export { DataStreamDecoder, DataStreamEncoder, } from "./serialization/data-stream/DataStream";
|
|
7
|
-
export { PlainTextDecoder, PlainTextEncoder } from "./serialization/PlainText";
|
|
8
|
-
export { AssistantMessageStream } from "./accumulators/AssistantMessageStream";
|
|
9
|
-
export type { AssistantMessage } from "./utils/types";
|
|
10
|
-
export * from "./tool";
|
|
11
|
-
//# sourceMappingURL=index.d.ts.map
|
package/dist/core/index.d.ts.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,6BAA6B,GAC9B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,2BAA2B,EAAE,MAAM,8CAA8C,CAAC;AAC3F,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,YAAY,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAC5E,YAAY,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,EACL,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,wCAAwC,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC/E,OAAO,EAAE,sBAAsB,EAAE,MAAM,uCAAuC,CAAC;AAC/E,YAAY,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEtD,cAAc,QAAQ,CAAC"}
|
package/dist/core/index.mjs
DELETED
|
@@ -1,26 +0,0 @@
|
|
|
1
|
-
// src/core/index.ts
|
|
2
|
-
import {
|
|
3
|
-
createAssistantStream,
|
|
4
|
-
createAssistantStreamResponse
|
|
5
|
-
} from "./modules/assistant-stream.mjs";
|
|
6
|
-
import { AssistantMessageAccumulator } from "./accumulators/assistant-message-accumulator.mjs";
|
|
7
|
-
import { AssistantStream } from "./AssistantStream.mjs";
|
|
8
|
-
import {
|
|
9
|
-
DataStreamDecoder,
|
|
10
|
-
DataStreamEncoder
|
|
11
|
-
} from "./serialization/data-stream/DataStream.mjs";
|
|
12
|
-
import { PlainTextDecoder, PlainTextEncoder } from "./serialization/PlainText.mjs";
|
|
13
|
-
import { AssistantMessageStream } from "./accumulators/AssistantMessageStream.mjs";
|
|
14
|
-
export * from "./tool/index.mjs";
|
|
15
|
-
export {
|
|
16
|
-
AssistantMessageAccumulator,
|
|
17
|
-
AssistantMessageStream,
|
|
18
|
-
AssistantStream,
|
|
19
|
-
DataStreamDecoder,
|
|
20
|
-
DataStreamEncoder,
|
|
21
|
-
PlainTextDecoder,
|
|
22
|
-
PlainTextEncoder,
|
|
23
|
-
createAssistantStream,
|
|
24
|
-
createAssistantStreamResponse
|
|
25
|
-
};
|
|
26
|
-
//# sourceMappingURL=index.mjs.map
|
package/dist/core/index.mjs.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../../src/core/index.ts"],"sourcesContent":["export {\n createAssistantStream,\n createAssistantStreamResponse,\n} from \"./modules/assistant-stream\";\nexport { AssistantMessageAccumulator } from \"./accumulators/assistant-message-accumulator\";\nexport { AssistantStream } from \"./AssistantStream\";\nexport type { AssistantStreamController } from \"./modules/assistant-stream\";\nexport type { AssistantStreamChunk } from \"./AssistantStreamChunk\";\nexport {\n DataStreamDecoder,\n DataStreamEncoder,\n} from \"./serialization/data-stream/DataStream\";\nexport { PlainTextDecoder, PlainTextEncoder } from \"./serialization/PlainText\";\nexport { AssistantMessageStream } from \"./accumulators/AssistantMessageStream\";\nexport type { AssistantMessage } from \"./utils/types\";\n\nexport * from \"./tool\";\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,mCAAmC;AAC5C,SAAS,uBAAuB;AAGhC;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,kBAAkB,wBAAwB;AACnD,SAAS,8BAA8B;AAGvC,cAAc;","names":[]}
|
|
@@ -1,31 +0,0 @@
|
|
|
1
|
-
import { AssistantStream } from "../AssistantStream";
|
|
2
|
-
import { AssistantStreamChunk } from "../AssistantStreamChunk";
|
|
3
|
-
import { TextStreamController } from "./text";
|
|
4
|
-
import { ToolCallStreamController } from "./tool-call";
|
|
5
|
-
import { FilePart, SourcePart } from "../utils/types";
|
|
6
|
-
import { ReadonlyJSONObject, ReadonlyJSONValue } from "../../utils/json/json-value";
|
|
7
|
-
import { ToolResponseInit } from "../tool/ToolResponse";
|
|
8
|
-
type ToolCallPartInit = {
|
|
9
|
-
toolCallId?: string;
|
|
10
|
-
toolName: string;
|
|
11
|
-
argsText?: string;
|
|
12
|
-
args?: ReadonlyJSONObject;
|
|
13
|
-
response?: ToolResponseInit<ReadonlyJSONValue>;
|
|
14
|
-
};
|
|
15
|
-
export type AssistantStreamController = {
|
|
16
|
-
appendText(textDelta: string): void;
|
|
17
|
-
appendReasoning(reasoningDelta: string): void;
|
|
18
|
-
appendSource(options: SourcePart): void;
|
|
19
|
-
appendFile(options: FilePart): void;
|
|
20
|
-
addTextPart(): TextStreamController;
|
|
21
|
-
addToolCallPart(options: string): ToolCallStreamController;
|
|
22
|
-
addToolCallPart(options: ToolCallPartInit): ToolCallStreamController;
|
|
23
|
-
enqueue(chunk: AssistantStreamChunk): void;
|
|
24
|
-
merge(stream: AssistantStream): void;
|
|
25
|
-
close(): void;
|
|
26
|
-
};
|
|
27
|
-
export declare function createAssistantStream(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): AssistantStream;
|
|
28
|
-
export declare function createAssistantStreamController(): readonly [AssistantStream, AssistantStreamController];
|
|
29
|
-
export declare function createAssistantStreamResponse(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): Response;
|
|
30
|
-
export {};
|
|
31
|
-
//# sourceMappingURL=assistant-stream.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"assistant-stream.d.ts","sourceRoot":"","sources":["../../../src/core/modules/assistant-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,oBAAoB,EAAY,MAAM,yBAAyB,CAAC;AAEzE,OAAO,EAA8B,oBAAoB,EAAE,MAAM,QAAQ,CAAC;AAC1E,OAAO,EAEL,wBAAwB,EACzB,MAAM,aAAa,CAAC;AAOrB,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAEtD,OAAO,EACL,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAGxD,KAAK,gBAAgB,GAAG;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,kBAAkB,CAAC;IAC1B,QAAQ,CAAC,EAAE,gBAAgB,CAAC,iBAAiB,CAAC,CAAC;CAChD,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,eAAe,CAAC,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9C,YAAY,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,CAAC;IACxC,UAAU,CAAC,OAAO,EAAE,QAAQ,GAAG,IAAI,CAAC;IACpC,WAAW,IAAI,oBAAoB,CAAC;IACpC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,wBAAwB,CAAC;IAC3D,eAAe,CAAC,OAAO,EAAE,gBAAgB,GAAG,wBAAwB,CAAC;IACrE,OAAO,CAAC,KAAK,EAAE,oBAAoB,GAAG,IAAI,CAAC;IAC3C,KAAK,CAAC,MAAM,EAAE,eAAe,GAAG,IAAI,CAAC;IACrC,KAAK,IAAI,IAAI,CAAC;CACf,CAAC;AAsJF,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,CAAC,UAAU,EAAE,yBAAyB,KAAK,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,GAC5E,eAAe,CA6CjB;AAED,wBAAgB,+BAA+B,0DAa9C;AAED,wBAAgB,6BAA6B,CAC3C,QAAQ,EAAE,CAAC,UAAU,EAAE,yBAAyB,KAAK,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,YAM9E"}
|