@ai-sdk/provider-utils 0.0.0-85f9a635-20240518005312

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/index.mjs ADDED
@@ -0,0 +1,424 @@
1
+ // src/extract-response-headers.ts
2
+ function extractResponseHeaders(response) {
3
+ const headers = {};
4
+ response.headers.forEach((value, key) => {
5
+ headers[key] = value;
6
+ });
7
+ return headers;
8
+ }
9
+
10
+ // src/generate-id.ts
11
+ import { customAlphabet } from "nanoid/non-secure";
12
+ var generateId = customAlphabet(
13
+ "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
14
+ 7
15
+ );
16
+
17
+ // src/get-error-message.ts
18
+ function getErrorMessage(error) {
19
+ if (error == null) {
20
+ return "unknown error";
21
+ }
22
+ if (typeof error === "string") {
23
+ return error;
24
+ }
25
+ if (error instanceof Error) {
26
+ return error.message;
27
+ }
28
+ return JSON.stringify(error);
29
+ }
30
+
31
+ // src/is-abort-error.ts
32
+ function isAbortError(error) {
33
+ return error instanceof DOMException && (error.name === "AbortError" || error.name === "TimeoutError");
34
+ }
35
+
36
+ // src/load-api-key.ts
37
+ import { LoadAPIKeyError } from "@ai-sdk/provider";
38
+ function loadApiKey({
39
+ apiKey,
40
+ environmentVariableName,
41
+ apiKeyParameterName = "apiKey",
42
+ description
43
+ }) {
44
+ if (typeof apiKey === "string") {
45
+ return apiKey;
46
+ }
47
+ if (apiKey != null) {
48
+ throw new LoadAPIKeyError({
49
+ message: `${description} API key must be a string.`
50
+ });
51
+ }
52
+ if (typeof process === "undefined") {
53
+ throw new LoadAPIKeyError({
54
+ message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter. Environment variables is not supported in this environment.`
55
+ });
56
+ }
57
+ apiKey = process.env[environmentVariableName];
58
+ if (apiKey == null) {
59
+ throw new LoadAPIKeyError({
60
+ message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter or the ${environmentVariableName} environment variable.`
61
+ });
62
+ }
63
+ if (typeof apiKey !== "string") {
64
+ throw new LoadAPIKeyError({
65
+ message: `${description} API key must be a string. The value of the ${environmentVariableName} environment variable is not a string.`
66
+ });
67
+ }
68
+ return apiKey;
69
+ }
70
+
71
+ // src/parse-json.ts
72
+ import { JSONParseError, TypeValidationError as TypeValidationError2 } from "@ai-sdk/provider";
73
+ import SecureJSON from "secure-json-parse";
74
+
75
+ // src/validate-types.ts
76
+ import { TypeValidationError } from "@ai-sdk/provider";
77
+ function validateTypes({
78
+ value,
79
+ schema
80
+ }) {
81
+ try {
82
+ return schema.parse(value);
83
+ } catch (error) {
84
+ throw new TypeValidationError({ value, cause: error });
85
+ }
86
+ }
87
+ function safeValidateTypes({
88
+ value,
89
+ schema
90
+ }) {
91
+ try {
92
+ const validationResult = schema.safeParse(value);
93
+ if (validationResult.success) {
94
+ return {
95
+ success: true,
96
+ value: validationResult.data
97
+ };
98
+ }
99
+ return {
100
+ success: false,
101
+ error: new TypeValidationError({
102
+ value,
103
+ cause: validationResult.error
104
+ })
105
+ };
106
+ } catch (error) {
107
+ return {
108
+ success: false,
109
+ error: TypeValidationError.isTypeValidationError(error) ? error : new TypeValidationError({ value, cause: error })
110
+ };
111
+ }
112
+ }
113
+
114
+ // src/parse-json.ts
115
+ function parseJSON({
116
+ text,
117
+ schema
118
+ }) {
119
+ try {
120
+ const value = SecureJSON.parse(text);
121
+ if (schema == null) {
122
+ return value;
123
+ }
124
+ return validateTypes({ value, schema });
125
+ } catch (error) {
126
+ if (JSONParseError.isJSONParseError(error) || TypeValidationError2.isTypeValidationError(error)) {
127
+ throw error;
128
+ }
129
+ throw new JSONParseError({ text, cause: error });
130
+ }
131
+ }
132
+ function safeParseJSON({
133
+ text,
134
+ schema
135
+ }) {
136
+ try {
137
+ const value = SecureJSON.parse(text);
138
+ if (schema == null) {
139
+ return {
140
+ success: true,
141
+ value
142
+ };
143
+ }
144
+ return safeValidateTypes({ value, schema });
145
+ } catch (error) {
146
+ return {
147
+ success: false,
148
+ error: JSONParseError.isJSONParseError(error) ? error : new JSONParseError({ text, cause: error })
149
+ };
150
+ }
151
+ }
152
+ function isParseableJson(input) {
153
+ try {
154
+ SecureJSON.parse(input);
155
+ return true;
156
+ } catch (e) {
157
+ return false;
158
+ }
159
+ }
160
+
161
+ // src/post-to-api.ts
162
+ import { APICallError } from "@ai-sdk/provider";
163
+ var postJsonToApi = async ({
164
+ url,
165
+ headers,
166
+ body,
167
+ failedResponseHandler,
168
+ successfulResponseHandler,
169
+ abortSignal
170
+ }) => postToApi({
171
+ url,
172
+ headers: {
173
+ ...headers,
174
+ "Content-Type": "application/json"
175
+ },
176
+ body: {
177
+ content: JSON.stringify(body),
178
+ values: body
179
+ },
180
+ failedResponseHandler,
181
+ successfulResponseHandler,
182
+ abortSignal
183
+ });
184
+ var postToApi = async ({
185
+ url,
186
+ headers = {},
187
+ body,
188
+ successfulResponseHandler,
189
+ failedResponseHandler,
190
+ abortSignal
191
+ }) => {
192
+ try {
193
+ const definedHeaders = Object.fromEntries(
194
+ Object.entries(headers).filter(([_key, value]) => value != null)
195
+ );
196
+ const response = await fetch(url, {
197
+ method: "POST",
198
+ headers: definedHeaders,
199
+ body: body.content,
200
+ signal: abortSignal
201
+ });
202
+ const responseHeaders = extractResponseHeaders(response);
203
+ if (!response.ok) {
204
+ let errorInformation;
205
+ try {
206
+ errorInformation = await failedResponseHandler({
207
+ response,
208
+ url,
209
+ requestBodyValues: body.values
210
+ });
211
+ } catch (error) {
212
+ if (isAbortError(error) || APICallError.isAPICallError(error)) {
213
+ throw error;
214
+ }
215
+ throw new APICallError({
216
+ message: "Failed to process error response",
217
+ cause: error,
218
+ statusCode: response.status,
219
+ url,
220
+ responseHeaders,
221
+ requestBodyValues: body.values
222
+ });
223
+ }
224
+ throw errorInformation.value;
225
+ }
226
+ try {
227
+ return await successfulResponseHandler({
228
+ response,
229
+ url,
230
+ requestBodyValues: body.values
231
+ });
232
+ } catch (error) {
233
+ if (error instanceof Error) {
234
+ if (isAbortError(error) || APICallError.isAPICallError(error)) {
235
+ throw error;
236
+ }
237
+ }
238
+ throw new APICallError({
239
+ message: "Failed to process successful response",
240
+ cause: error,
241
+ statusCode: response.status,
242
+ url,
243
+ responseHeaders,
244
+ requestBodyValues: body.values
245
+ });
246
+ }
247
+ } catch (error) {
248
+ if (isAbortError(error)) {
249
+ throw error;
250
+ }
251
+ if (error instanceof TypeError && error.message === "fetch failed") {
252
+ const cause = error.cause;
253
+ if (cause != null) {
254
+ throw new APICallError({
255
+ message: `Cannot connect to API: ${cause.message}`,
256
+ cause,
257
+ url,
258
+ requestBodyValues: body.values,
259
+ isRetryable: true
260
+ // retry when network error
261
+ });
262
+ }
263
+ }
264
+ throw error;
265
+ }
266
+ };
267
+
268
+ // src/response-handler.ts
269
+ import { APICallError as APICallError2, EmptyResponseBodyError } from "@ai-sdk/provider";
270
+ import {
271
+ EventSourceParserStream
272
+ } from "eventsource-parser/stream";
273
+ var createJsonErrorResponseHandler = ({
274
+ errorSchema,
275
+ errorToMessage,
276
+ isRetryable
277
+ }) => async ({ response, url, requestBodyValues }) => {
278
+ const responseBody = await response.text();
279
+ const responseHeaders = extractResponseHeaders(response);
280
+ if (responseBody.trim() === "") {
281
+ return {
282
+ responseHeaders,
283
+ value: new APICallError2({
284
+ message: response.statusText,
285
+ url,
286
+ requestBodyValues,
287
+ statusCode: response.status,
288
+ responseHeaders,
289
+ responseBody,
290
+ isRetryable: isRetryable == null ? void 0 : isRetryable(response)
291
+ })
292
+ };
293
+ }
294
+ try {
295
+ const parsedError = parseJSON({
296
+ text: responseBody,
297
+ schema: errorSchema
298
+ });
299
+ return {
300
+ responseHeaders,
301
+ value: new APICallError2({
302
+ message: errorToMessage(parsedError),
303
+ url,
304
+ requestBodyValues,
305
+ statusCode: response.status,
306
+ responseHeaders,
307
+ responseBody,
308
+ data: parsedError,
309
+ isRetryable: isRetryable == null ? void 0 : isRetryable(response, parsedError)
310
+ })
311
+ };
312
+ } catch (parseError) {
313
+ return {
314
+ responseHeaders,
315
+ value: new APICallError2({
316
+ message: response.statusText,
317
+ url,
318
+ requestBodyValues,
319
+ statusCode: response.status,
320
+ responseHeaders,
321
+ responseBody,
322
+ isRetryable: isRetryable == null ? void 0 : isRetryable(response)
323
+ })
324
+ };
325
+ }
326
+ };
327
+ var createEventSourceResponseHandler = (chunkSchema) => async ({ response }) => {
328
+ const responseHeaders = extractResponseHeaders(response);
329
+ if (response.body == null) {
330
+ throw new EmptyResponseBodyError({});
331
+ }
332
+ return {
333
+ responseHeaders,
334
+ value: response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new EventSourceParserStream()).pipeThrough(
335
+ new TransformStream({
336
+ transform({ data }, controller) {
337
+ if (data === "[DONE]") {
338
+ return;
339
+ }
340
+ controller.enqueue(
341
+ safeParseJSON({
342
+ text: data,
343
+ schema: chunkSchema
344
+ })
345
+ );
346
+ }
347
+ })
348
+ )
349
+ };
350
+ };
351
+ var createEventSourcePassThroughHandler = (chunkSchema) => async ({ response }) => {
352
+ const responseHeaders = extractResponseHeaders(response);
353
+ if (response.body == null) {
354
+ throw new EmptyResponseBodyError({});
355
+ }
356
+ return {
357
+ responseHeaders,
358
+ value: response.body
359
+ };
360
+ };
361
+ var createJsonResponseHandler = (responseSchema) => async ({ response, url, requestBodyValues }) => {
362
+ const responseBody = await response.text();
363
+ const parsedResult = safeParseJSON({
364
+ text: responseBody,
365
+ schema: responseSchema
366
+ });
367
+ const responseHeaders = extractResponseHeaders(response);
368
+ if (!parsedResult.success) {
369
+ throw new APICallError2({
370
+ message: "Invalid JSON response",
371
+ cause: parsedResult.error,
372
+ statusCode: response.status,
373
+ responseHeaders,
374
+ responseBody,
375
+ url,
376
+ requestBodyValues
377
+ });
378
+ }
379
+ return {
380
+ responseHeaders,
381
+ value: parsedResult.value
382
+ };
383
+ };
384
+
385
+ // src/uint8-utils.ts
386
+ function convertBase64ToUint8Array(base64String) {
387
+ const base64Url = base64String.replace(/-/g, "+").replace(/_/g, "/");
388
+ const latin1string = globalThis.atob(base64Url);
389
+ return Uint8Array.from(latin1string, (byte) => byte.codePointAt(0));
390
+ }
391
+ function convertUint8ArrayToBase64(array) {
392
+ let latin1string = "";
393
+ for (let i = 0; i < array.length; i++) {
394
+ latin1string += String.fromCodePoint(array[i]);
395
+ }
396
+ return globalThis.btoa(latin1string);
397
+ }
398
+
399
+ // src/without-trailing-slash.ts
400
+ function withoutTrailingSlash(url) {
401
+ return url == null ? void 0 : url.replace(/\/$/, "");
402
+ }
403
+ export {
404
+ convertBase64ToUint8Array,
405
+ convertUint8ArrayToBase64,
406
+ createEventSourcePassThroughHandler,
407
+ createEventSourceResponseHandler,
408
+ createJsonErrorResponseHandler,
409
+ createJsonResponseHandler,
410
+ extractResponseHeaders,
411
+ generateId,
412
+ getErrorMessage,
413
+ isAbortError,
414
+ isParseableJson,
415
+ loadApiKey,
416
+ parseJSON,
417
+ postJsonToApi,
418
+ postToApi,
419
+ safeParseJSON,
420
+ safeValidateTypes,
421
+ validateTypes,
422
+ withoutTrailingSlash
423
+ };
424
+ //# sourceMappingURL=index.mjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/extract-response-headers.ts","../src/generate-id.ts","../src/get-error-message.ts","../src/is-abort-error.ts","../src/load-api-key.ts","../src/parse-json.ts","../src/validate-types.ts","../src/post-to-api.ts","../src/response-handler.ts","../src/uint8-utils.ts","../src/without-trailing-slash.ts"],"sourcesContent":["/**\nExtracts the headers from a response object and returns them as a key-value object.\n\n@param response - The response object to extract headers from.\n@returns The headers as a key-value object.\n*/\nexport function extractResponseHeaders(\n response: Response,\n): Record<string, string> {\n const headers: Record<string, string> = {};\n response.headers.forEach((value, key) => {\n headers[key] = value;\n });\n return headers;\n}\n","import { customAlphabet } from 'nanoid/non-secure';\n\n/**\n * Generates a 7-character random string to use for IDs. Not secure.\n */\nexport const generateId = customAlphabet(\n '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',\n 7,\n);\n","export function getErrorMessage(error: unknown | undefined) {\n if (error == null) {\n return 'unknown error';\n }\n\n if (typeof error === 'string') {\n return error;\n }\n\n if (error instanceof Error) {\n return error.message;\n }\n\n return JSON.stringify(error);\n}\n","export function isAbortError(error: unknown): error is DOMException {\n return (\n error instanceof DOMException &&\n (error.name === 'AbortError' || error.name === 'TimeoutError')\n );\n}\n","import { LoadAPIKeyError } from '@ai-sdk/provider';\n\nexport function loadApiKey({\n apiKey,\n environmentVariableName,\n apiKeyParameterName = 'apiKey',\n description,\n}: {\n apiKey: string | undefined;\n environmentVariableName: string;\n apiKeyParameterName?: string;\n description: string;\n}): string {\n if (typeof apiKey === 'string') {\n return apiKey;\n }\n\n if (apiKey != null) {\n throw new LoadAPIKeyError({\n message: `${description} API key must be a string.`,\n });\n }\n\n if (typeof process === 'undefined') {\n throw new LoadAPIKeyError({\n message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter. Environment variables is not supported in this environment.`,\n });\n }\n\n apiKey = process.env[environmentVariableName];\n\n if (apiKey == null) {\n throw new LoadAPIKeyError({\n message: `${description} API key is missing. Pass it using the '${apiKeyParameterName}' parameter or the ${environmentVariableName} environment variable.`,\n });\n }\n\n if (typeof apiKey !== 'string') {\n throw new LoadAPIKeyError({\n message: `${description} API key must be a string. The value of the ${environmentVariableName} environment variable is not a string.`,\n });\n }\n\n return apiKey;\n}\n","import { JSONParseError, TypeValidationError } from '@ai-sdk/provider';\nimport SecureJSON from 'secure-json-parse';\nimport { ZodSchema } from 'zod';\nimport { safeValidateTypes, validateTypes } from './validate-types';\n\n/**\n * Parses a JSON string into an unknown object.\n *\n * @param text - The JSON string to parse.\n * @returns {unknown} - The parsed JSON object.\n */\nexport function parseJSON({ text }: { text: string }): unknown;\n/**\n * Parses a JSON string into a strongly-typed object using the provided schema.\n *\n * @template T - The type of the object to parse the JSON into.\n * @param {string} text - The JSON string to parse.\n * @param {Schema<T>} schema - The schema to use for parsing the JSON.\n * @returns {T} - The parsed object.\n */\nexport function parseJSON<T>({\n text,\n schema,\n}: {\n text: string;\n schema: ZodSchema<T>;\n}): T;\nexport function parseJSON<T>({\n text,\n schema,\n}: {\n text: string;\n schema?: ZodSchema<T>;\n}): T {\n try {\n const value = SecureJSON.parse(text);\n\n if (schema == null) {\n return value;\n }\n\n return validateTypes({ value, schema });\n } catch (error) {\n if (\n JSONParseError.isJSONParseError(error) ||\n TypeValidationError.isTypeValidationError(error)\n ) {\n throw error;\n }\n\n throw new JSONParseError({ text, cause: error });\n }\n}\n\nexport type ParseResult<T> =\n | { success: true; value: T }\n | { success: false; error: JSONParseError | TypeValidationError };\n\n/**\n * Safely parses a JSON string and returns the result as an object of type `unknown`.\n *\n * @param text - The JSON string to parse.\n * @returns {object} Either an object with `success: true` and the parsed data, or an object with `success: false` and the error that occurred.\n */\nexport function safeParseJSON({ text }: { text: string }): ParseResult<unknown>;\n/**\n * Safely parses a JSON string into a strongly-typed object, using a provided schema to validate the object.\n *\n * @template T - The type of the object to parse the JSON into.\n * @param {string} text - The JSON string to parse.\n * @param {Schema<T>} schema - The schema to use for parsing the JSON.\n * @returns An object with either a `success` flag and the parsed and typed data, or a `success` flag and an error object.\n */\nexport function safeParseJSON<T>({\n text,\n schema,\n}: {\n text: string;\n schema: ZodSchema<T>;\n}): ParseResult<T>;\nexport function safeParseJSON<T>({\n text,\n schema,\n}: {\n text: string;\n schema?: ZodSchema<T>;\n}):\n | { success: true; value: T }\n | { success: false; error: JSONParseError | TypeValidationError } {\n try {\n const value = SecureJSON.parse(text);\n\n if (schema == null) {\n return {\n success: true,\n value: value as T,\n };\n }\n\n return safeValidateTypes({ value, schema });\n } catch (error) {\n return {\n success: false,\n error: JSONParseError.isJSONParseError(error)\n ? error\n : new JSONParseError({ text, cause: error }),\n };\n }\n}\n\nexport function isParseableJson(input: string): boolean {\n try {\n SecureJSON.parse(input);\n return true;\n } catch {\n return false;\n }\n}\n","import { TypeValidationError } from '@ai-sdk/provider';\nimport { ZodSchema } from 'zod';\n\n/**\n * Validates the types of an unknown object using a schema and\n * return a strongly-typed object.\n *\n * @template T - The type of the object to validate.\n * @param {string} options.value - The object to validate.\n * @param {Schema<T>} options.schema - The schema to use for validating the JSON.\n * @returns {T} - The typed object.\n */\nexport function validateTypes<T>({\n value,\n schema,\n}: {\n value: unknown;\n schema: ZodSchema<T>;\n}): T {\n try {\n return schema.parse(value);\n } catch (error) {\n throw new TypeValidationError({ value, cause: error });\n }\n}\n\n/**\n * Safely validates the types of an unknown object using a schema and\n * return a strongly-typed object.\n *\n * @template T - The type of the object to validate.\n * @param {string} options.value - The JSON object to validate.\n * @param {Schema<T>} options.schema - The schema to use for validating the JSON.\n * @returns An object with either a `success` flag and the parsed and typed data, or a `success` flag and an error object.\n */\nexport function safeValidateTypes<T>({\n value,\n schema,\n}: {\n value: unknown;\n schema: ZodSchema<T>;\n}):\n | { success: true; value: T }\n | { success: false; error: TypeValidationError } {\n try {\n const validationResult = schema.safeParse(value);\n\n if (validationResult.success) {\n return {\n success: true,\n value: validationResult.data,\n };\n }\n\n return {\n success: false,\n error: new TypeValidationError({\n value,\n cause: validationResult.error,\n }),\n };\n } catch (error) {\n return {\n success: false,\n error: TypeValidationError.isTypeValidationError(error)\n ? error\n : new TypeValidationError({ value, cause: error }),\n };\n }\n}\n","import { APICallError } from '@ai-sdk/provider';\nimport { extractResponseHeaders } from './extract-response-headers';\nimport { isAbortError } from './is-abort-error';\nimport { ResponseHandler } from './response-handler';\n\nexport const postJsonToApi = async <T>({\n url,\n headers,\n body,\n failedResponseHandler,\n successfulResponseHandler,\n abortSignal,\n}: {\n url: string;\n headers?: Record<string, string | undefined>;\n body: unknown;\n failedResponseHandler: ResponseHandler<APICallError>;\n successfulResponseHandler: ResponseHandler<T>;\n abortSignal?: AbortSignal;\n}) =>\n postToApi({\n url,\n headers: {\n ...headers,\n 'Content-Type': 'application/json',\n },\n body: {\n content: JSON.stringify(body),\n values: body,\n },\n failedResponseHandler,\n successfulResponseHandler,\n abortSignal,\n });\n\nexport const postToApi = async <T>({\n url,\n headers = {},\n body,\n successfulResponseHandler,\n failedResponseHandler,\n abortSignal,\n}: {\n url: string;\n headers?: Record<string, string | undefined>;\n body: {\n content: string | FormData | Uint8Array;\n values: unknown;\n };\n failedResponseHandler: ResponseHandler<Error>;\n successfulResponseHandler: ResponseHandler<T>;\n abortSignal?: AbortSignal;\n}) => {\n try {\n // remove undefined headers:\n const definedHeaders = Object.fromEntries(\n Object.entries(headers).filter(([_key, value]) => value != null),\n ) as Record<string, string>;\n\n const response = await fetch(url, {\n method: 'POST',\n headers: definedHeaders,\n body: body.content,\n signal: abortSignal,\n });\n\n const responseHeaders = extractResponseHeaders(response);\n\n if (!response.ok) {\n let errorInformation: {\n value: Error;\n responseHeaders?: Record<string, string> | undefined;\n };\n\n try {\n errorInformation = await failedResponseHandler({\n response,\n url,\n requestBodyValues: body.values,\n });\n } catch (error) {\n if (isAbortError(error) || APICallError.isAPICallError(error)) {\n throw error;\n }\n\n throw new APICallError({\n message: 'Failed to process error response',\n cause: error,\n statusCode: response.status,\n url,\n responseHeaders,\n requestBodyValues: body.values,\n });\n }\n\n throw errorInformation.value;\n }\n\n try {\n return await successfulResponseHandler({\n response,\n url,\n requestBodyValues: body.values,\n });\n } catch (error) {\n if (error instanceof Error) {\n if (isAbortError(error) || APICallError.isAPICallError(error)) {\n throw error;\n }\n }\n\n throw new APICallError({\n message: 'Failed to process successful response',\n cause: error,\n statusCode: response.status,\n url,\n responseHeaders,\n requestBodyValues: body.values,\n });\n }\n } catch (error) {\n if (isAbortError(error)) {\n throw error;\n }\n\n // unwrap original error when fetch failed (for easier debugging):\n if (error instanceof TypeError && error.message === 'fetch failed') {\n const cause = (error as any).cause;\n\n if (cause != null) {\n // Failed to connect to server:\n throw new APICallError({\n message: `Cannot connect to API: ${cause.message}`,\n cause,\n url,\n requestBodyValues: body.values,\n isRetryable: true, // retry when network error\n });\n }\n }\n\n throw error;\n }\n};\n","import { APICallError, EmptyResponseBodyError } from '@ai-sdk/provider';\nimport {\n EventSourceParserStream,\n ParsedEvent,\n} from 'eventsource-parser/stream';\nimport { ZodSchema } from 'zod';\nimport { extractResponseHeaders } from './extract-response-headers';\nimport { ParseResult, parseJSON, safeParseJSON } from './parse-json';\n\nexport type ResponseHandler<RETURN_TYPE> = (options: {\n url: string;\n requestBodyValues: unknown;\n response: Response;\n}) => PromiseLike<{\n value: RETURN_TYPE;\n responseHeaders?: Record<string, string>;\n}>;\n\nexport const createJsonErrorResponseHandler =\n <T>({\n errorSchema,\n errorToMessage,\n isRetryable,\n }: {\n errorSchema: ZodSchema<T>;\n errorToMessage: (error: T) => string;\n isRetryable?: (response: Response, error?: T) => boolean;\n }): ResponseHandler<APICallError> =>\n async ({ response, url, requestBodyValues }) => {\n const responseBody = await response.text();\n const responseHeaders = extractResponseHeaders(response);\n\n // Some providers return an empty response body for some errors:\n if (responseBody.trim() === '') {\n return {\n responseHeaders,\n value: new APICallError({\n message: response.statusText,\n url,\n requestBodyValues,\n statusCode: response.status,\n responseHeaders,\n responseBody,\n isRetryable: isRetryable?.(response),\n }),\n };\n }\n\n // resilient parsing in case the response is not JSON or does not match the schema:\n try {\n const parsedError = parseJSON({\n text: responseBody,\n schema: errorSchema,\n });\n\n return {\n responseHeaders,\n value: new APICallError({\n message: errorToMessage(parsedError),\n url,\n requestBodyValues,\n statusCode: response.status,\n responseHeaders,\n responseBody,\n data: parsedError,\n isRetryable: isRetryable?.(response, parsedError),\n }),\n };\n } catch (parseError) {\n return {\n responseHeaders,\n value: new APICallError({\n message: response.statusText,\n url,\n requestBodyValues,\n statusCode: response.status,\n responseHeaders,\n responseBody,\n isRetryable: isRetryable?.(response),\n }),\n };\n }\n };\n\nexport const createEventSourceResponseHandler =\n <T>(\n chunkSchema: ZodSchema<T>,\n ): ResponseHandler<ReadableStream<ParseResult<T>>> =>\n async ({ response }: { response: Response }) => {\n const responseHeaders = extractResponseHeaders(response);\n\n if (response.body == null) {\n throw new EmptyResponseBodyError({});\n }\n\n return {\n responseHeaders,\n value: response.body\n .pipeThrough(new TextDecoderStream())\n .pipeThrough(new EventSourceParserStream())\n .pipeThrough(\n new TransformStream<ParsedEvent, ParseResult<T>>({\n transform({ data }, controller) {\n // ignore the 'DONE' event that e.g. OpenAI sends:\n if (data === '[DONE]') {\n return;\n }\n\n controller.enqueue(\n safeParseJSON({\n text: data,\n schema: chunkSchema,\n }),\n );\n },\n }),\n ),\n };\n };\n\nexport const createEventSourcePassThroughHandler =\n <T>(chunkSchema: ZodSchema<T>): ResponseHandler<ReadableStream<Uint8Array>> =>\n async ({ response }: { response: Response }) => {\n const responseHeaders = extractResponseHeaders(response);\n\n if (response.body == null) {\n throw new EmptyResponseBodyError({});\n }\n\n return {\n responseHeaders,\n value: response.body,\n };\n };\n\nexport const createJsonResponseHandler =\n <T>(responseSchema: ZodSchema<T>): ResponseHandler<T> =>\n async ({ response, url, requestBodyValues }) => {\n const responseBody = await response.text();\n\n const parsedResult = safeParseJSON({\n text: responseBody,\n schema: responseSchema,\n });\n\n const responseHeaders = extractResponseHeaders(response);\n\n if (!parsedResult.success) {\n throw new APICallError({\n message: 'Invalid JSON response',\n cause: parsedResult.error,\n statusCode: response.status,\n responseHeaders,\n responseBody,\n url,\n requestBodyValues,\n });\n }\n\n return {\n responseHeaders,\n value: parsedResult.value,\n };\n };\n","export function convertBase64ToUint8Array(base64String: string) {\n const base64Url = base64String.replace(/-/g, '+').replace(/_/g, '/');\n const latin1string = globalThis.atob(base64Url);\n return Uint8Array.from(latin1string, byte => byte.codePointAt(0)!);\n}\n\nexport function convertUint8ArrayToBase64(array: Uint8Array): string {\n let latin1string = '';\n\n // Note: regular for loop to support older JavaScript versions that\n // do not support for..of on Uint8Array\n for (let i = 0; i < array.length; i++) {\n latin1string += String.fromCodePoint(array[i]);\n }\n\n return globalThis.btoa(latin1string);\n}\n","export function withoutTrailingSlash(url: string | undefined) {\n return url?.replace(/\\/$/, '');\n}\n"],"mappings":";AAMO,SAAS,uBACd,UACwB;AACxB,QAAM,UAAkC,CAAC;AACzC,WAAS,QAAQ,QAAQ,CAAC,OAAO,QAAQ;AACvC,YAAQ,GAAG,IAAI;AAAA,EACjB,CAAC;AACD,SAAO;AACT;;;ACdA,SAAS,sBAAsB;AAKxB,IAAM,aAAa;AAAA,EACxB;AAAA,EACA;AACF;;;ACRO,SAAS,gBAAgB,OAA4B;AAC1D,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,iBAAiB,OAAO;AAC1B,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,KAAK,UAAU,KAAK;AAC7B;;;ACdO,SAAS,aAAa,OAAuC;AAClE,SACE,iBAAiB,iBAChB,MAAM,SAAS,gBAAgB,MAAM,SAAS;AAEnD;;;ACLA,SAAS,uBAAuB;AAEzB,SAAS,WAAW;AAAA,EACzB;AAAA,EACA;AAAA,EACA,sBAAsB;AAAA,EACtB;AACF,GAKW;AACT,MAAI,OAAO,WAAW,UAAU;AAC9B,WAAO;AAAA,EACT;AAEA,MAAI,UAAU,MAAM;AAClB,UAAM,IAAI,gBAAgB;AAAA,MACxB,SAAS,GAAG,WAAW;AAAA,IACzB,CAAC;AAAA,EACH;AAEA,MAAI,OAAO,YAAY,aAAa;AAClC,UAAM,IAAI,gBAAgB;AAAA,MACxB,SAAS,GAAG,WAAW,2CAA2C,mBAAmB;AAAA,IACvF,CAAC;AAAA,EACH;AAEA,WAAS,QAAQ,IAAI,uBAAuB;AAE5C,MAAI,UAAU,MAAM;AAClB,UAAM,IAAI,gBAAgB;AAAA,MACxB,SAAS,GAAG,WAAW,2CAA2C,mBAAmB,sBAAsB,uBAAuB;AAAA,IACpI,CAAC;AAAA,EACH;AAEA,MAAI,OAAO,WAAW,UAAU;AAC9B,UAAM,IAAI,gBAAgB;AAAA,MACxB,SAAS,GAAG,WAAW,+CAA+C,uBAAuB;AAAA,IAC/F,CAAC;AAAA,EACH;AAEA,SAAO;AACT;;;AC5CA,SAAS,gBAAgB,uBAAAA,4BAA2B;AACpD,OAAO,gBAAgB;;;ACDvB,SAAS,2BAA2B;AAY7B,SAAS,cAAiB;AAAA,EAC/B;AAAA,EACA;AACF,GAGM;AACJ,MAAI;AACF,WAAO,OAAO,MAAM,KAAK;AAAA,EAC3B,SAAS,OAAO;AACd,UAAM,IAAI,oBAAoB,EAAE,OAAO,OAAO,MAAM,CAAC;AAAA,EACvD;AACF;AAWO,SAAS,kBAAqB;AAAA,EACnC;AAAA,EACA;AACF,GAKmD;AACjD,MAAI;AACF,UAAM,mBAAmB,OAAO,UAAU,KAAK;AAE/C,QAAI,iBAAiB,SAAS;AAC5B,aAAO;AAAA,QACL,SAAS;AAAA,QACT,OAAO,iBAAiB;AAAA,MAC1B;AAAA,IACF;AAEA,WAAO;AAAA,MACL,SAAS;AAAA,MACT,OAAO,IAAI,oBAAoB;AAAA,QAC7B;AAAA,QACA,OAAO,iBAAiB;AAAA,MAC1B,CAAC;AAAA,IACH;AAAA,EACF,SAAS,OAAO;AACd,WAAO;AAAA,MACL,SAAS;AAAA,MACT,OAAO,oBAAoB,sBAAsB,KAAK,IAClD,QACA,IAAI,oBAAoB,EAAE,OAAO,OAAO,MAAM,CAAC;AAAA,IACrD;AAAA,EACF;AACF;;;AD1CO,SAAS,UAAa;AAAA,EAC3B;AAAA,EACA;AACF,GAGM;AACJ,MAAI;AACF,UAAM,QAAQ,WAAW,MAAM,IAAI;AAEnC,QAAI,UAAU,MAAM;AAClB,aAAO;AAAA,IACT;AAEA,WAAO,cAAc,EAAE,OAAO,OAAO,CAAC;AAAA,EACxC,SAAS,OAAO;AACd,QACE,eAAe,iBAAiB,KAAK,KACrCC,qBAAoB,sBAAsB,KAAK,GAC/C;AACA,YAAM;AAAA,IACR;AAEA,UAAM,IAAI,eAAe,EAAE,MAAM,OAAO,MAAM,CAAC;AAAA,EACjD;AACF;AA4BO,SAAS,cAAiB;AAAA,EAC/B;AAAA,EACA;AACF,GAKoE;AAClE,MAAI;AACF,UAAM,QAAQ,WAAW,MAAM,IAAI;AAEnC,QAAI,UAAU,MAAM;AAClB,aAAO;AAAA,QACL,SAAS;AAAA,QACT;AAAA,MACF;AAAA,IACF;AAEA,WAAO,kBAAkB,EAAE,OAAO,OAAO,CAAC;AAAA,EAC5C,SAAS,OAAO;AACd,WAAO;AAAA,MACL,SAAS;AAAA,MACT,OAAO,eAAe,iBAAiB,KAAK,IACxC,QACA,IAAI,eAAe,EAAE,MAAM,OAAO,MAAM,CAAC;AAAA,IAC/C;AAAA,EACF;AACF;AAEO,SAAS,gBAAgB,OAAwB;AACtD,MAAI;AACF,eAAW,MAAM,KAAK;AACtB,WAAO;AAAA,EACT,SAAQ;AACN,WAAO;AAAA,EACT;AACF;;;AErHA,SAAS,oBAAoB;AAKtB,IAAM,gBAAgB,OAAU;AAAA,EACrC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAQE,UAAU;AAAA,EACR;AAAA,EACA,SAAS;AAAA,IACP,GAAG;AAAA,IACH,gBAAgB;AAAA,EAClB;AAAA,EACA,MAAM;AAAA,IACJ,SAAS,KAAK,UAAU,IAAI;AAAA,IAC5B,QAAQ;AAAA,EACV;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAEI,IAAM,YAAY,OAAU;AAAA,EACjC;AAAA,EACA,UAAU,CAAC;AAAA,EACX;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAUM;AACJ,MAAI;AAEF,UAAM,iBAAiB,OAAO;AAAA,MAC5B,OAAO,QAAQ,OAAO,EAAE,OAAO,CAAC,CAAC,MAAM,KAAK,MAAM,SAAS,IAAI;AAAA,IACjE;AAEA,UAAM,WAAW,MAAM,MAAM,KAAK;AAAA,MAChC,QAAQ;AAAA,MACR,SAAS;AAAA,MACT,MAAM,KAAK;AAAA,MACX,QAAQ;AAAA,IACV,CAAC;AAED,UAAM,kBAAkB,uBAAuB,QAAQ;AAEvD,QAAI,CAAC,SAAS,IAAI;AAChB,UAAI;AAKJ,UAAI;AACF,2BAAmB,MAAM,sBAAsB;AAAA,UAC7C;AAAA,UACA;AAAA,UACA,mBAAmB,KAAK;AAAA,QAC1B,CAAC;AAAA,MACH,SAAS,OAAO;AACd,YAAI,aAAa,KAAK,KAAK,aAAa,eAAe,KAAK,GAAG;AAC7D,gBAAM;AAAA,QACR;AAEA,cAAM,IAAI,aAAa;AAAA,UACrB,SAAS;AAAA,UACT,OAAO;AAAA,UACP,YAAY,SAAS;AAAA,UACrB;AAAA,UACA;AAAA,UACA,mBAAmB,KAAK;AAAA,QAC1B,CAAC;AAAA,MACH;AAEA,YAAM,iBAAiB;AAAA,IACzB;AAEA,QAAI;AACF,aAAO,MAAM,0BAA0B;AAAA,QACrC;AAAA,QACA;AAAA,QACA,mBAAmB,KAAK;AAAA,MAC1B,CAAC;AAAA,IACH,SAAS,OAAO;AACd,UAAI,iBAAiB,OAAO;AAC1B,YAAI,aAAa,KAAK,KAAK,aAAa,eAAe,KAAK,GAAG;AAC7D,gBAAM;AAAA,QACR;AAAA,MACF;AAEA,YAAM,IAAI,aAAa;AAAA,QACrB,SAAS;AAAA,QACT,OAAO;AAAA,QACP,YAAY,SAAS;AAAA,QACrB;AAAA,QACA;AAAA,QACA,mBAAmB,KAAK;AAAA,MAC1B,CAAC;AAAA,IACH;AAAA,EACF,SAAS,OAAO;AACd,QAAI,aAAa,KAAK,GAAG;AACvB,YAAM;AAAA,IACR;AAGA,QAAI,iBAAiB,aAAa,MAAM,YAAY,gBAAgB;AAClE,YAAM,QAAS,MAAc;AAE7B,UAAI,SAAS,MAAM;AAEjB,cAAM,IAAI,aAAa;AAAA,UACrB,SAAS,0BAA0B,MAAM,OAAO;AAAA,UAChD;AAAA,UACA;AAAA,UACA,mBAAmB,KAAK;AAAA,UACxB,aAAa;AAAA;AAAA,QACf,CAAC;AAAA,MACH;AAAA,IACF;AAEA,UAAM;AAAA,EACR;AACF;;;AC/IA,SAAS,gBAAAC,eAAc,8BAA8B;AACrD;AAAA,EACE;AAAA,OAEK;AAcA,IAAM,iCACX,CAAI;AAAA,EACF;AAAA,EACA;AAAA,EACA;AACF,MAKA,OAAO,EAAE,UAAU,KAAK,kBAAkB,MAAM;AAC9C,QAAM,eAAe,MAAM,SAAS,KAAK;AACzC,QAAM,kBAAkB,uBAAuB,QAAQ;AAGvD,MAAI,aAAa,KAAK,MAAM,IAAI;AAC9B,WAAO;AAAA,MACL;AAAA,MACA,OAAO,IAAIC,cAAa;AAAA,QACtB,SAAS,SAAS;AAAA,QAClB;AAAA,QACA;AAAA,QACA,YAAY,SAAS;AAAA,QACrB;AAAA,QACA;AAAA,QACA,aAAa,2CAAc;AAAA,MAC7B,CAAC;AAAA,IACH;AAAA,EACF;AAGA,MAAI;AACF,UAAM,cAAc,UAAU;AAAA,MAC5B,MAAM;AAAA,MACN,QAAQ;AAAA,IACV,CAAC;AAED,WAAO;AAAA,MACL;AAAA,MACA,OAAO,IAAIA,cAAa;AAAA,QACtB,SAAS,eAAe,WAAW;AAAA,QACnC;AAAA,QACA;AAAA,QACA,YAAY,SAAS;AAAA,QACrB;AAAA,QACA;AAAA,QACA,MAAM;AAAA,QACN,aAAa,2CAAc,UAAU;AAAA,MACvC,CAAC;AAAA,IACH;AAAA,EACF,SAAS,YAAY;AACnB,WAAO;AAAA,MACL;AAAA,MACA,OAAO,IAAIA,cAAa;AAAA,QACtB,SAAS,SAAS;AAAA,QAClB;AAAA,QACA;AAAA,QACA,YAAY,SAAS;AAAA,QACrB;AAAA,QACA;AAAA,QACA,aAAa,2CAAc;AAAA,MAC7B,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEK,IAAM,mCACX,CACE,gBAEF,OAAO,EAAE,SAAS,MAA8B;AAC9C,QAAM,kBAAkB,uBAAuB,QAAQ;AAEvD,MAAI,SAAS,QAAQ,MAAM;AACzB,UAAM,IAAI,uBAAuB,CAAC,CAAC;AAAA,EACrC;AAEA,SAAO;AAAA,IACL;AAAA,IACA,OAAO,SAAS,KACb,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,wBAAwB,CAAC,EACzC;AAAA,MACC,IAAI,gBAA6C;AAAA,QAC/C,UAAU,EAAE,KAAK,GAAG,YAAY;AAE9B,cAAI,SAAS,UAAU;AACrB;AAAA,UACF;AAEA,qBAAW;AAAA,YACT,cAAc;AAAA,cACZ,MAAM;AAAA,cACN,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACJ;AACF;AAEK,IAAM,sCACX,CAAI,gBACJ,OAAO,EAAE,SAAS,MAA8B;AAC9C,QAAM,kBAAkB,uBAAuB,QAAQ;AAEvD,MAAI,SAAS,QAAQ,MAAM;AACzB,UAAM,IAAI,uBAAuB,CAAC,CAAC;AAAA,EACrC;AAEA,SAAO;AAAA,IACL;AAAA,IACA,OAAO,SAAS;AAAA,EAClB;AACF;AAEK,IAAM,4BACX,CAAI,mBACJ,OAAO,EAAE,UAAU,KAAK,kBAAkB,MAAM;AAC9C,QAAM,eAAe,MAAM,SAAS,KAAK;AAEzC,QAAM,eAAe,cAAc;AAAA,IACjC,MAAM;AAAA,IACN,QAAQ;AAAA,EACV,CAAC;AAED,QAAM,kBAAkB,uBAAuB,QAAQ;AAEvD,MAAI,CAAC,aAAa,SAAS;AACzB,UAAM,IAAIA,cAAa;AAAA,MACrB,SAAS;AAAA,MACT,OAAO,aAAa;AAAA,MACpB,YAAY,SAAS;AAAA,MACrB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL;AAAA,IACA,OAAO,aAAa;AAAA,EACtB;AACF;;;ACnKK,SAAS,0BAA0B,cAAsB;AAC9D,QAAM,YAAY,aAAa,QAAQ,MAAM,GAAG,EAAE,QAAQ,MAAM,GAAG;AACnE,QAAM,eAAe,WAAW,KAAK,SAAS;AAC9C,SAAO,WAAW,KAAK,cAAc,UAAQ,KAAK,YAAY,CAAC,CAAE;AACnE;AAEO,SAAS,0BAA0B,OAA2B;AACnE,MAAI,eAAe;AAInB,WAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,oBAAgB,OAAO,cAAc,MAAM,CAAC,CAAC;AAAA,EAC/C;AAEA,SAAO,WAAW,KAAK,YAAY;AACrC;;;AChBO,SAAS,qBAAqB,KAAyB;AAC5D,SAAO,2BAAK,QAAQ,OAAO;AAC7B;","names":["TypeValidationError","TypeValidationError","APICallError","APICallError"]}
package/package.json ADDED
@@ -0,0 +1,74 @@
1
+ {
2
+ "name": "@ai-sdk/provider-utils",
3
+ "version": "0.0.0-85f9a635-20240518005312",
4
+ "license": "Apache-2.0",
5
+ "sideEffects": false,
6
+ "main": "./dist/index.js",
7
+ "module": "./dist/index.mjs",
8
+ "types": "./dist/index.d.ts",
9
+ "files": [
10
+ "dist/**/*",
11
+ "test/dist/**/*"
12
+ ],
13
+ "exports": {
14
+ "./package.json": "./package.json",
15
+ ".": {
16
+ "types": "./dist/index.d.ts",
17
+ "import": "./dist/index.mjs",
18
+ "require": "./dist/index.js"
19
+ },
20
+ "./test": {
21
+ "types": "./test/dist/index.d.ts",
22
+ "import": "./test/dist/index.mjs",
23
+ "module": "./test/dist/index.mjs",
24
+ "require": "./test/dist/index.js"
25
+ }
26
+ },
27
+ "dependencies": {
28
+ "@ai-sdk/provider": "0.0.0-85f9a635-20240518005312",
29
+ "eventsource-parser": "1.1.2",
30
+ "nanoid": "3.3.6",
31
+ "secure-json-parse": "2.7.0"
32
+ },
33
+ "devDependencies": {
34
+ "@types/node": "^18",
35
+ "msw": "2.0.9",
36
+ "tsup": "^8",
37
+ "typescript": "5.1.3",
38
+ "zod": "3.22.4",
39
+ "@vercel/ai-tsconfig": "0.0.0"
40
+ },
41
+ "peerDependencies": {
42
+ "zod": "^3.0.0"
43
+ },
44
+ "peerDependenciesMeta": {
45
+ "zod": {
46
+ "optional": true
47
+ }
48
+ },
49
+ "engines": {
50
+ "node": ">=18"
51
+ },
52
+ "publishConfig": {
53
+ "access": "public"
54
+ },
55
+ "homepage": "https://sdk.vercel.ai/docs",
56
+ "repository": {
57
+ "type": "git",
58
+ "url": "git+https://github.com/vercel/ai.git"
59
+ },
60
+ "bugs": {
61
+ "url": "https://github.com/vercel/ai/issues"
62
+ },
63
+ "keywords": [
64
+ "ai"
65
+ ],
66
+ "scripts": {
67
+ "build": "tsup",
68
+ "clean": "rm -rf dist && rm -rf test/dist",
69
+ "dev": "tsup --watch",
70
+ "lint": "eslint \"./**/*.ts*\"",
71
+ "type-check": "tsc --noEmit",
72
+ "prettier-check": "prettier --check \"./**/*.ts*\""
73
+ }
74
+ }
@@ -0,0 +1,27 @@
1
+ import { SetupServer } from 'msw/node';
2
+
3
+ declare function convertStreamToArray<T>(stream: ReadableStream<T>): Promise<T[]>;
4
+
5
+ declare class JsonTestServer {
6
+ readonly server: SetupServer;
7
+ responseHeaders: Record<string, string>;
8
+ responseBodyJson: any;
9
+ request: Request | undefined;
10
+ constructor(url: string);
11
+ getRequestBodyJson(): Promise<any>;
12
+ getRequestHeaders(): Promise<Headers>;
13
+ setupTestEnvironment(): void;
14
+ }
15
+
16
+ declare class StreamingTestServer {
17
+ readonly server: SetupServer;
18
+ responseHeaders: Record<string, string>;
19
+ responseChunks: any[];
20
+ request: Request | undefined;
21
+ constructor(url: string);
22
+ getRequestBodyJson(): Promise<any>;
23
+ getRequestHeaders(): Promise<Headers>;
24
+ setupTestEnvironment(): void;
25
+ }
26
+
27
+ export { JsonTestServer, StreamingTestServer, convertStreamToArray };
@@ -0,0 +1,27 @@
1
+ import { SetupServer } from 'msw/node';
2
+
3
+ declare function convertStreamToArray<T>(stream: ReadableStream<T>): Promise<T[]>;
4
+
5
+ declare class JsonTestServer {
6
+ readonly server: SetupServer;
7
+ responseHeaders: Record<string, string>;
8
+ responseBodyJson: any;
9
+ request: Request | undefined;
10
+ constructor(url: string);
11
+ getRequestBodyJson(): Promise<any>;
12
+ getRequestHeaders(): Promise<Headers>;
13
+ setupTestEnvironment(): void;
14
+ }
15
+
16
+ declare class StreamingTestServer {
17
+ readonly server: SetupServer;
18
+ responseHeaders: Record<string, string>;
19
+ responseChunks: any[];
20
+ request: Request | undefined;
21
+ constructor(url: string);
22
+ getRequestBodyJson(): Promise<any>;
23
+ getRequestHeaders(): Promise<Headers>;
24
+ setupTestEnvironment(): void;
25
+ }
26
+
27
+ export { JsonTestServer, StreamingTestServer, convertStreamToArray };