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.
- package/dist/AssistantStream-CYbWmRWJ.d.mts +27 -0
- package/dist/AssistantStream-CYbWmRWJ.d.ts +27 -0
- package/dist/ai-sdk.d.mts +7 -0
- package/dist/ai-sdk.d.ts +7 -0
- package/dist/ai-sdk.js +176 -0
- package/dist/ai-sdk.js.map +1 -0
- package/dist/ai-sdk.mjs +122 -0
- package/dist/ai-sdk.mjs.map +1 -0
- package/dist/chunk-7CCRZC5D.mjs +969 -0
- package/dist/chunk-7CCRZC5D.mjs.map +1 -0
- package/dist/index.d.mts +99 -0
- package/dist/index.d.ts +99 -0
- package/dist/index.js +1012 -0
- package/dist/index.js.map +1 -0
- package/dist/index.mjs +20 -0
- package/dist/index.mjs.map +1 -0
- package/package.json +59 -3
|
@@ -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 };
|
package/dist/ai-sdk.d.ts
ADDED
|
@@ -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":[]}
|
package/dist/ai-sdk.mjs
ADDED
|
@@ -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":[]}
|