assistant-stream 0.0.0 → 0.0.2

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.
@@ -0,0 +1,27 @@
1
+ type AssistantStreamChunk = {
2
+ type: "text-delta";
3
+ textDelta: string;
4
+ } | {
5
+ type: "tool-call-begin";
6
+ toolCallId: string;
7
+ toolName: string;
8
+ } | {
9
+ type: "tool-call-delta";
10
+ toolCallId: string;
11
+ argsTextDelta: string;
12
+ } | {
13
+ type: "tool-result";
14
+ toolCallId: string;
15
+ result: any;
16
+ };
17
+ declare class AssistantStream {
18
+ readonly readable: ReadableStream<AssistantStreamChunk>;
19
+ constructor(readable: ReadableStream<AssistantStreamChunk>);
20
+ toResponse(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): Response;
21
+ static fromResponse(response: Response, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
22
+ toByteStream(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): ReadableStream<Uint8Array>;
23
+ static fromByteStream(readable: ReadableStream<Uint8Array>, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
24
+ tee(): [AssistantStream, AssistantStream];
25
+ }
26
+
27
+ export { AssistantStream as A, type AssistantStreamChunk as a };
@@ -0,0 +1,27 @@
1
+ type AssistantStreamChunk = {
2
+ type: "text-delta";
3
+ textDelta: string;
4
+ } | {
5
+ type: "tool-call-begin";
6
+ toolCallId: string;
7
+ toolName: string;
8
+ } | {
9
+ type: "tool-call-delta";
10
+ toolCallId: string;
11
+ argsTextDelta: string;
12
+ } | {
13
+ type: "tool-result";
14
+ toolCallId: string;
15
+ result: any;
16
+ };
17
+ declare class AssistantStream {
18
+ readonly readable: ReadableStream<AssistantStreamChunk>;
19
+ constructor(readable: ReadableStream<AssistantStreamChunk>);
20
+ toResponse(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): Response;
21
+ static fromResponse(response: Response, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
22
+ toByteStream(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): ReadableStream<Uint8Array>;
23
+ static fromByteStream(readable: ReadableStream<Uint8Array>, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
24
+ tee(): [AssistantStream, AssistantStream];
25
+ }
26
+
27
+ export { AssistantStream as A, type AssistantStreamChunk as a };
@@ -0,0 +1,7 @@
1
+ import { TextStreamPart, CoreTool, ObjectStreamPart } from 'ai';
2
+ import { A as AssistantStream } from './AssistantStream-CYbWmRWJ.mjs';
3
+
4
+ declare const fromAISDKStreamText: (stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>) => AssistantStream;
5
+ declare const fromAISDKStreamObject: (stream: ReadableStream<ObjectStreamPart<unknown>>, toolName: string) => AssistantStream;
6
+
7
+ export { fromAISDKStreamObject, fromAISDKStreamText };
@@ -0,0 +1,7 @@
1
+ import { TextStreamPart, CoreTool, ObjectStreamPart } from 'ai';
2
+ import { A as AssistantStream } from './AssistantStream-CYbWmRWJ.js';
3
+
4
+ declare const fromAISDKStreamText: (stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>) => AssistantStream;
5
+ declare const fromAISDKStreamObject: (stream: ReadableStream<ObjectStreamPart<unknown>>, toolName: string) => AssistantStream;
6
+
7
+ export { fromAISDKStreamObject, fromAISDKStreamText };
package/dist/ai-sdk.js ADDED
@@ -0,0 +1,176 @@
1
+ "use strict";
2
+ var __defProp = Object.defineProperty;
3
+ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
4
+ var __getOwnPropNames = Object.getOwnPropertyNames;
5
+ var __hasOwnProp = Object.prototype.hasOwnProperty;
6
+ var __export = (target, all) => {
7
+ for (var name in all)
8
+ __defProp(target, name, { get: all[name], enumerable: true });
9
+ };
10
+ var __copyProps = (to, from, except, desc) => {
11
+ if (from && typeof from === "object" || typeof from === "function") {
12
+ for (let key of __getOwnPropNames(from))
13
+ if (!__hasOwnProp.call(to, key) && key !== except)
14
+ __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
15
+ }
16
+ return to;
17
+ };
18
+ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
19
+
20
+ // src/ai-sdk.ts
21
+ var ai_sdk_exports = {};
22
+ __export(ai_sdk_exports, {
23
+ fromAISDKStreamObject: () => fromAISDKStreamObject,
24
+ fromAISDKStreamText: () => fromAISDKStreamText
25
+ });
26
+ module.exports = __toCommonJS(ai_sdk_exports);
27
+
28
+ // src/core/AssistantStream.ts
29
+ var AssistantStream = class _AssistantStream {
30
+ constructor(readable) {
31
+ this.readable = readable;
32
+ this.readable = readable;
33
+ }
34
+ toResponse(transformer) {
35
+ return new Response(this.toByteStream(transformer));
36
+ }
37
+ static fromResponse(response, transformer) {
38
+ return _AssistantStream.fromByteStream(response.body, transformer);
39
+ }
40
+ toByteStream(transformer) {
41
+ return this.readable.pipeThrough(transformer);
42
+ }
43
+ static fromByteStream(readable, transformer) {
44
+ return new _AssistantStream(readable.pipeThrough(transformer));
45
+ }
46
+ tee() {
47
+ const [readable1, readable2] = this.readable.tee();
48
+ return [new _AssistantStream(readable1), new _AssistantStream(readable2)];
49
+ }
50
+ };
51
+
52
+ // src/core/utils/generateId.tsx
53
+ var import_non_secure = require("nanoid/non-secure");
54
+ var generateId = (0, import_non_secure.customAlphabet)(
55
+ "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
56
+ 7
57
+ );
58
+
59
+ // src/ai-sdk/index.ts
60
+ var fromAISDKStreamText = (stream) => {
61
+ const transformer = new TransformStream({
62
+ transform(chunk, controller) {
63
+ const { type } = chunk;
64
+ switch (type) {
65
+ case "text-delta": {
66
+ const { textDelta } = chunk;
67
+ controller.enqueue({
68
+ type: "text-delta",
69
+ textDelta
70
+ });
71
+ break;
72
+ }
73
+ case "tool-call-streaming-start": {
74
+ const { toolCallId, toolName } = chunk;
75
+ controller.enqueue({
76
+ type: "tool-call-begin",
77
+ toolCallId,
78
+ toolName
79
+ });
80
+ break;
81
+ }
82
+ case "tool-call-delta": {
83
+ const { toolCallId, argsTextDelta } = chunk;
84
+ controller.enqueue({
85
+ type: "tool-call-delta",
86
+ toolCallId,
87
+ argsTextDelta
88
+ });
89
+ break;
90
+ }
91
+ case "tool-result": {
92
+ const { toolCallId, result } = chunk;
93
+ controller.enqueue({
94
+ type: "tool-result",
95
+ toolCallId,
96
+ result
97
+ });
98
+ break;
99
+ }
100
+ case "tool-call": {
101
+ const { toolCallId, toolName, args } = chunk;
102
+ controller.enqueue({
103
+ type: "tool-call-begin",
104
+ toolCallId,
105
+ toolName
106
+ });
107
+ controller.enqueue({
108
+ type: "tool-call-delta",
109
+ toolCallId,
110
+ argsTextDelta: JSON.stringify(args)
111
+ });
112
+ break;
113
+ }
114
+ case "step-finish":
115
+ case "error":
116
+ case "finish": {
117
+ break;
118
+ }
119
+ default: {
120
+ const unhandledType = type;
121
+ throw new Error(`Unhandled chunk type: ${unhandledType}`);
122
+ }
123
+ }
124
+ }
125
+ });
126
+ return new AssistantStream(stream.pipeThrough(transformer));
127
+ };
128
+ var fromAISDKStreamObject = (stream, toolName) => {
129
+ const toolCallId = generateId();
130
+ const transformer = new TransformStream({
131
+ start(controller) {
132
+ controller.enqueue({
133
+ type: "tool-call-begin",
134
+ toolName,
135
+ toolCallId
136
+ });
137
+ },
138
+ transform(chunk, controller) {
139
+ const { type } = chunk;
140
+ switch (type) {
141
+ case "text-delta": {
142
+ const { textDelta } = chunk;
143
+ controller.enqueue({
144
+ type: "tool-call-delta",
145
+ toolCallId,
146
+ argsTextDelta: textDelta
147
+ });
148
+ break;
149
+ }
150
+ case "finish": {
151
+ controller.enqueue({
152
+ type: "tool-result",
153
+ toolCallId,
154
+ result: ""
155
+ });
156
+ break;
157
+ }
158
+ case "object":
159
+ case "error": {
160
+ break;
161
+ }
162
+ default: {
163
+ const unhandledType = type;
164
+ throw new Error(`Unhandled chunk type: ${unhandledType}`);
165
+ }
166
+ }
167
+ }
168
+ });
169
+ return new AssistantStream(stream.pipeThrough(transformer));
170
+ };
171
+ // Annotate the CommonJS export names for ESM import in node:
172
+ 0 && (module.exports = {
173
+ fromAISDKStreamObject,
174
+ fromAISDKStreamText
175
+ });
176
+ //# sourceMappingURL=ai-sdk.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/ai-sdk.ts","../src/core/AssistantStream.ts","../src/core/utils/generateId.tsx","../src/ai-sdk/index.ts"],"sourcesContent":["export * from \"./ai-sdk/index\";\n","export type AssistantStream2 = {\n // kind: AssistantStreamKind;\n readable: ReadableStream<AssistantStreamChunk>;\n toResponse(\n format: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n ): Response;\n toByteStream(\n format: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n ): ReadableStream<Uint8Array>;\n tee(): [AssistantStream, AssistantStream];\n};\n\nexport type AssistantStreamChunk =\n | {\n type: \"text-delta\";\n textDelta: string;\n }\n | {\n type: \"tool-call-begin\";\n toolCallId: string;\n toolName: string;\n }\n | {\n type: \"tool-call-delta\";\n toolCallId: string;\n argsTextDelta: string;\n }\n | {\n type: \"tool-result\";\n toolCallId: string;\n result: any;\n };\n\nexport class AssistantStream {\n constructor(public readonly readable: ReadableStream<AssistantStreamChunk>) {\n this.readable = readable;\n }\n\n toResponse(\n transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n ) {\n return new Response(this.toByteStream(transformer));\n }\n\n static fromResponse(\n response: Response,\n transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>,\n ) {\n return AssistantStream.fromByteStream(response.body!, transformer);\n }\n\n toByteStream(\n transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n ) {\n return this.readable.pipeThrough(transformer);\n }\n\n static fromByteStream(\n readable: ReadableStream<Uint8Array>,\n transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>,\n ) {\n return new AssistantStream(readable.pipeThrough(transformer));\n }\n\n tee(): [AssistantStream, AssistantStream] {\n const [readable1, readable2] = this.readable.tee();\n return [new AssistantStream(readable1), new AssistantStream(readable2)];\n }\n}\n","import { customAlphabet } from \"nanoid/non-secure\";\n\nexport const generateId = customAlphabet(\n \"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz\",\n 7,\n);\n","import type { TextStreamPart, CoreTool, ObjectStreamPart } from \"ai\";\nimport { AssistantStream, AssistantStreamChunk } from \"../core\";\nimport { generateId } from \"../core/utils/generateId\";\n\nexport const fromAISDKStreamText = (\n stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>,\n): AssistantStream => {\n const transformer = new TransformStream<\n TextStreamPart<Record<string, CoreTool>>,\n AssistantStreamChunk\n >({\n transform(chunk, controller) {\n const { type } = chunk;\n switch (type) {\n case \"text-delta\": {\n const { textDelta } = chunk;\n controller.enqueue({\n type: \"text-delta\",\n textDelta,\n });\n break;\n }\n case \"tool-call-streaming-start\": {\n const { toolCallId, toolName } = chunk;\n controller.enqueue({\n type: \"tool-call-begin\",\n toolCallId,\n toolName,\n });\n break;\n }\n case \"tool-call-delta\": {\n const { toolCallId, argsTextDelta } = chunk;\n controller.enqueue({\n type: \"tool-call-delta\",\n toolCallId,\n argsTextDelta,\n });\n break;\n }\n case \"tool-result\" as string: {\n const { toolCallId, result } = chunk as unknown as {\n toolCallId: string;\n result: unknown;\n };\n controller.enqueue({\n type: \"tool-result\",\n toolCallId,\n result,\n });\n break;\n }\n case \"tool-call\": {\n const { toolCallId, toolName, args } = chunk;\n controller.enqueue({\n type: \"tool-call-begin\",\n toolCallId,\n toolName,\n });\n controller.enqueue({\n type: \"tool-call-delta\",\n toolCallId,\n argsTextDelta: JSON.stringify(args),\n });\n break;\n }\n case \"step-finish\":\n case \"error\":\n case \"finish\": {\n break;\n }\n\n default: {\n const unhandledType: never = type;\n throw new Error(`Unhandled chunk type: ${unhandledType}`);\n }\n }\n },\n });\n\n return new AssistantStream(stream.pipeThrough(transformer));\n};\n\nexport const fromAISDKStreamObject = (\n stream: ReadableStream<ObjectStreamPart<unknown>>,\n toolName: string,\n): AssistantStream => {\n const toolCallId = generateId();\n const transformer = new TransformStream<\n ObjectStreamPart<unknown>,\n AssistantStreamChunk\n >({\n start(controller) {\n controller.enqueue({\n type: \"tool-call-begin\",\n toolName,\n toolCallId,\n });\n },\n transform(chunk, controller) {\n const { type } = chunk;\n switch (type) {\n case \"text-delta\": {\n const { textDelta } = chunk;\n controller.enqueue({\n type: \"tool-call-delta\",\n toolCallId,\n argsTextDelta: textDelta,\n });\n break;\n }\n case \"finish\": {\n controller.enqueue({\n type: \"tool-result\",\n toolCallId,\n result: \"\",\n });\n break;\n }\n\n case \"object\":\n case \"error\": {\n break;\n }\n\n default: {\n const unhandledType: never = type;\n throw new Error(`Unhandled chunk type: ${unhandledType}`);\n }\n }\n },\n });\n\n return new AssistantStream(stream.pipeThrough(transformer));\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACiCO,IAAM,kBAAN,MAAM,iBAAgB;AAAA,EAC3B,YAA4B,UAAgD;AAAhD;AAC1B,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,WACE,aACA;AACA,WAAO,IAAI,SAAS,KAAK,aAAa,WAAW,CAAC;AAAA,EACpD;AAAA,EAEA,OAAO,aACL,UACA,aACA;AACA,WAAO,iBAAgB,eAAe,SAAS,MAAO,WAAW;AAAA,EACnE;AAAA,EAEA,aACE,aACA;AACA,WAAO,KAAK,SAAS,YAAY,WAAW;AAAA,EAC9C;AAAA,EAEA,OAAO,eACL,UACA,aACA;AACA,WAAO,IAAI,iBAAgB,SAAS,YAAY,WAAW,CAAC;AAAA,EAC9D;AAAA,EAEA,MAA0C;AACxC,UAAM,CAAC,WAAW,SAAS,IAAI,KAAK,SAAS,IAAI;AACjD,WAAO,CAAC,IAAI,iBAAgB,SAAS,GAAG,IAAI,iBAAgB,SAAS,CAAC;AAAA,EACxE;AACF;;;ACpEA,wBAA+B;AAExB,IAAM,iBAAa;AAAA,EACxB;AAAA,EACA;AACF;;;ACDO,IAAM,sBAAsB,CACjC,WACoB;AACpB,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,6BAA6B;AAChC,gBAAM,EAAE,YAAY,SAAS,IAAI;AACjC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,mBAAmB;AACtB,gBAAM,EAAE,YAAY,cAAc,IAAI;AACtC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,eAAyB;AAC5B,gBAAM,EAAE,YAAY,OAAO,IAAI;AAI/B,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,aAAa;AAChB,gBAAM,EAAE,YAAY,UAAU,KAAK,IAAI;AACvC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe,KAAK,UAAU,IAAI;AAAA,UACpC,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK,UAAU;AACb;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;AAEO,IAAM,wBAAwB,CACnC,QACA,aACoB;AACpB,QAAM,aAAa,WAAW;AAC9B,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,MAAM,YAAY;AAChB,iBAAW,QAAQ;AAAA,QACjB,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe;AAAA,UACjB,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,UAAU;AACb,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,QAAQ;AAAA,UACV,CAAC;AACD;AAAA,QACF;AAAA,QAEA,KAAK;AAAA,QACL,KAAK,SAAS;AACZ;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;","names":[]}
@@ -0,0 +1,122 @@
1
+ import {
2
+ AssistantStream,
3
+ generateId
4
+ } from "./chunk-7CCRZC5D.mjs";
5
+
6
+ // src/ai-sdk/index.ts
7
+ var fromAISDKStreamText = (stream) => {
8
+ const transformer = new TransformStream({
9
+ transform(chunk, controller) {
10
+ const { type } = chunk;
11
+ switch (type) {
12
+ case "text-delta": {
13
+ const { textDelta } = chunk;
14
+ controller.enqueue({
15
+ type: "text-delta",
16
+ textDelta
17
+ });
18
+ break;
19
+ }
20
+ case "tool-call-streaming-start": {
21
+ const { toolCallId, toolName } = chunk;
22
+ controller.enqueue({
23
+ type: "tool-call-begin",
24
+ toolCallId,
25
+ toolName
26
+ });
27
+ break;
28
+ }
29
+ case "tool-call-delta": {
30
+ const { toolCallId, argsTextDelta } = chunk;
31
+ controller.enqueue({
32
+ type: "tool-call-delta",
33
+ toolCallId,
34
+ argsTextDelta
35
+ });
36
+ break;
37
+ }
38
+ case "tool-result": {
39
+ const { toolCallId, result } = chunk;
40
+ controller.enqueue({
41
+ type: "tool-result",
42
+ toolCallId,
43
+ result
44
+ });
45
+ break;
46
+ }
47
+ case "tool-call": {
48
+ const { toolCallId, toolName, args } = chunk;
49
+ controller.enqueue({
50
+ type: "tool-call-begin",
51
+ toolCallId,
52
+ toolName
53
+ });
54
+ controller.enqueue({
55
+ type: "tool-call-delta",
56
+ toolCallId,
57
+ argsTextDelta: JSON.stringify(args)
58
+ });
59
+ break;
60
+ }
61
+ case "step-finish":
62
+ case "error":
63
+ case "finish": {
64
+ break;
65
+ }
66
+ default: {
67
+ const unhandledType = type;
68
+ throw new Error(`Unhandled chunk type: ${unhandledType}`);
69
+ }
70
+ }
71
+ }
72
+ });
73
+ return new AssistantStream(stream.pipeThrough(transformer));
74
+ };
75
+ var fromAISDKStreamObject = (stream, toolName) => {
76
+ const toolCallId = generateId();
77
+ const transformer = new TransformStream({
78
+ start(controller) {
79
+ controller.enqueue({
80
+ type: "tool-call-begin",
81
+ toolName,
82
+ toolCallId
83
+ });
84
+ },
85
+ transform(chunk, controller) {
86
+ const { type } = chunk;
87
+ switch (type) {
88
+ case "text-delta": {
89
+ const { textDelta } = chunk;
90
+ controller.enqueue({
91
+ type: "tool-call-delta",
92
+ toolCallId,
93
+ argsTextDelta: textDelta
94
+ });
95
+ break;
96
+ }
97
+ case "finish": {
98
+ controller.enqueue({
99
+ type: "tool-result",
100
+ toolCallId,
101
+ result: ""
102
+ });
103
+ break;
104
+ }
105
+ case "object":
106
+ case "error": {
107
+ break;
108
+ }
109
+ default: {
110
+ const unhandledType = type;
111
+ throw new Error(`Unhandled chunk type: ${unhandledType}`);
112
+ }
113
+ }
114
+ }
115
+ });
116
+ return new AssistantStream(stream.pipeThrough(transformer));
117
+ };
118
+ export {
119
+ fromAISDKStreamObject,
120
+ fromAISDKStreamText
121
+ };
122
+ //# sourceMappingURL=ai-sdk.mjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/ai-sdk/index.ts"],"sourcesContent":["import type { TextStreamPart, CoreTool, ObjectStreamPart } from \"ai\";\nimport { AssistantStream, AssistantStreamChunk } from \"../core\";\nimport { generateId } from \"../core/utils/generateId\";\n\nexport const fromAISDKStreamText = (\n stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>,\n): AssistantStream => {\n const transformer = new TransformStream<\n TextStreamPart<Record<string, CoreTool>>,\n AssistantStreamChunk\n >({\n transform(chunk, controller) {\n const { type } = chunk;\n switch (type) {\n case \"text-delta\": {\n const { textDelta } = chunk;\n controller.enqueue({\n type: \"text-delta\",\n textDelta,\n });\n break;\n }\n case \"tool-call-streaming-start\": {\n const { toolCallId, toolName } = chunk;\n controller.enqueue({\n type: \"tool-call-begin\",\n toolCallId,\n toolName,\n });\n break;\n }\n case \"tool-call-delta\": {\n const { toolCallId, argsTextDelta } = chunk;\n controller.enqueue({\n type: \"tool-call-delta\",\n toolCallId,\n argsTextDelta,\n });\n break;\n }\n case \"tool-result\" as string: {\n const { toolCallId, result } = chunk as unknown as {\n toolCallId: string;\n result: unknown;\n };\n controller.enqueue({\n type: \"tool-result\",\n toolCallId,\n result,\n });\n break;\n }\n case \"tool-call\": {\n const { toolCallId, toolName, args } = chunk;\n controller.enqueue({\n type: \"tool-call-begin\",\n toolCallId,\n toolName,\n });\n controller.enqueue({\n type: \"tool-call-delta\",\n toolCallId,\n argsTextDelta: JSON.stringify(args),\n });\n break;\n }\n case \"step-finish\":\n case \"error\":\n case \"finish\": {\n break;\n }\n\n default: {\n const unhandledType: never = type;\n throw new Error(`Unhandled chunk type: ${unhandledType}`);\n }\n }\n },\n });\n\n return new AssistantStream(stream.pipeThrough(transformer));\n};\n\nexport const fromAISDKStreamObject = (\n stream: ReadableStream<ObjectStreamPart<unknown>>,\n toolName: string,\n): AssistantStream => {\n const toolCallId = generateId();\n const transformer = new TransformStream<\n ObjectStreamPart<unknown>,\n AssistantStreamChunk\n >({\n start(controller) {\n controller.enqueue({\n type: \"tool-call-begin\",\n toolName,\n toolCallId,\n });\n },\n transform(chunk, controller) {\n const { type } = chunk;\n switch (type) {\n case \"text-delta\": {\n const { textDelta } = chunk;\n controller.enqueue({\n type: \"tool-call-delta\",\n toolCallId,\n argsTextDelta: textDelta,\n });\n break;\n }\n case \"finish\": {\n controller.enqueue({\n type: \"tool-result\",\n toolCallId,\n result: \"\",\n });\n break;\n }\n\n case \"object\":\n case \"error\": {\n break;\n }\n\n default: {\n const unhandledType: never = type;\n throw new Error(`Unhandled chunk type: ${unhandledType}`);\n }\n }\n },\n });\n\n return new AssistantStream(stream.pipeThrough(transformer));\n};\n"],"mappings":";;;;;;AAIO,IAAM,sBAAsB,CACjC,WACoB;AACpB,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,6BAA6B;AAChC,gBAAM,EAAE,YAAY,SAAS,IAAI;AACjC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,mBAAmB;AACtB,gBAAM,EAAE,YAAY,cAAc,IAAI;AACtC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,eAAyB;AAC5B,gBAAM,EAAE,YAAY,OAAO,IAAI;AAI/B,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,aAAa;AAChB,gBAAM,EAAE,YAAY,UAAU,KAAK,IAAI;AACvC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe,KAAK,UAAU,IAAI;AAAA,UACpC,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK,UAAU;AACb;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;AAEO,IAAM,wBAAwB,CACnC,QACA,aACoB;AACpB,QAAM,aAAa,WAAW;AAC9B,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,MAAM,YAAY;AAChB,iBAAW,QAAQ;AAAA,QACjB,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe;AAAA,UACjB,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,UAAU;AACb,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,QAAQ;AAAA,UACV,CAAC;AACD;AAAA,QACF;AAAA,QAEA,KAAK;AAAA,QACL,KAAK,SAAS;AACZ;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;","names":[]}