@ai-sdk/provider-utils 5.0.30 → 5.0.32

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/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ai-sdk/provider-utils",
3
- "version": "5.0.30",
3
+ "version": "5.0.32",
4
4
  "type": "module",
5
5
  "license": "Apache-2.0",
6
6
  "sideEffects": false,
@@ -0,0 +1,53 @@
1
+ const marker = Symbol.for('vercel.ai.providerStreamError');
2
+
3
+ export type ProviderStreamError = {
4
+ readonly message: string;
5
+ readonly type?: string;
6
+ readonly code?: string | number;
7
+ readonly statusCode?: number;
8
+ readonly isRetryable?: boolean;
9
+ readonly data: unknown;
10
+ };
11
+
12
+ /**
13
+ * Adds provider-owned status and retry metadata to a stream error payload
14
+ * without requiring provider packages to depend on AI SDK Core.
15
+ */
16
+ export function createProviderStreamError({
17
+ message,
18
+ type,
19
+ code,
20
+ statusCode,
21
+ isRetryable,
22
+ data,
23
+ }: {
24
+ message: string;
25
+ type?: string;
26
+ code?: string | number;
27
+ statusCode?: number;
28
+ isRetryable?: boolean;
29
+ data: unknown;
30
+ }): ProviderStreamError {
31
+ const error = {
32
+ message,
33
+ type,
34
+ code,
35
+ statusCode,
36
+ isRetryable,
37
+ data,
38
+ };
39
+
40
+ Object.defineProperty(error, marker, { value: true });
41
+
42
+ return error;
43
+ }
44
+
45
+ export function isProviderStreamError(
46
+ error: unknown,
47
+ ): error is ProviderStreamError {
48
+ return (
49
+ typeof error === 'object' &&
50
+ error != null &&
51
+ (error as Record<symbol, unknown>)[marker] === true
52
+ );
53
+ }
@@ -0,0 +1,9 @@
1
+ /**
2
+ * Symbol for exposing the UTF-8 input byte budget of an embedding model.
3
+ *
4
+ * This capability is experimental and intentionally lives outside the versioned
5
+ * embedding model specification.
6
+ */
7
+ export const EMBEDDING_MODEL_MAX_INPUT_BYTES_PER_CALL = Symbol.for(
8
+ 'vercel.ai.embeddingModel.maxInputBytesPerCall',
9
+ );
package/src/index.ts CHANGED
@@ -15,6 +15,11 @@ export {
15
15
  createToolNameMapping,
16
16
  type ToolNameMapping,
17
17
  } from './create-tool-name-mapping';
18
+ export {
19
+ createProviderStreamError,
20
+ isProviderStreamError,
21
+ type ProviderStreamError,
22
+ } from './create-provider-stream-error';
18
23
  export * from './delay';
19
24
  export { DelayedPromise } from './delayed-promise';
20
25
  export {
@@ -24,6 +29,7 @@ export {
24
29
  } from './detect-media-type';
25
30
  export { downloadBlob } from './download-blob';
26
31
  export { DownloadError } from './download-error';
32
+ export { EMBEDDING_MODEL_MAX_INPUT_BYTES_PER_CALL as EXPERIMENTAL_EMBEDDING_MODEL_MAX_INPUT_BYTES_PER_CALL } from './embedding-model-capabilities';
27
33
  export { fetchWithValidatedRedirects } from './fetch-with-validated-redirects';
28
34
  export { extractLines } from './extract-lines';
29
35
  export * from './extract-response-headers';
@@ -54,6 +60,7 @@ export {
54
60
  export { type MaybePromiseLike } from './maybe-promise-like';
55
61
  export { mediaTypeToExtension } from './media-type-to-extension';
56
62
  export { normalizeHeaders } from './normalize-headers';
63
+ export { normalizeBatchRequestCounts } from './normalize-batch-request-counts';
57
64
  export * from './parse-json';
58
65
  export { parseJsonEventStream } from './parse-json-event-stream';
59
66
  export { parseProviderOptions } from './parse-provider-options';
@@ -0,0 +1,42 @@
1
+ import type { Experimental_BatchV4Status as BatchV4Status } from '@ai-sdk/provider';
2
+
3
+ /**
4
+ * Normalizes complete batch request counts.
5
+ *
6
+ * Returns `undefined` when any count is missing, is not a non-negative safe
7
+ * integer, or when the item counts do not add up to the total.
8
+ */
9
+ export function normalizeBatchRequestCounts({
10
+ total,
11
+ pending,
12
+ completed,
13
+ failed,
14
+ }: {
15
+ total: number | null | undefined;
16
+ pending: number | null | undefined;
17
+ completed: number | null | undefined;
18
+ failed: number | null | undefined;
19
+ }): BatchV4Status['requestCounts'] | undefined {
20
+ if (
21
+ isNonNegativeSafeInteger(total) &&
22
+ isNonNegativeSafeInteger(pending) &&
23
+ isNonNegativeSafeInteger(completed) &&
24
+ isNonNegativeSafeInteger(failed) &&
25
+ pending + completed + failed === total
26
+ ) {
27
+ return {
28
+ total,
29
+ pending,
30
+ completed,
31
+ failed,
32
+ };
33
+ }
34
+
35
+ return undefined;
36
+ }
37
+
38
+ function isNonNegativeSafeInteger(
39
+ value: number | null | undefined,
40
+ ): value is number {
41
+ return value != null && Number.isSafeInteger(value) && value >= 0;
42
+ }
@@ -149,6 +149,73 @@ export const createJsonResponseHandler =
149
149
  };
150
150
  };
151
151
 
152
+ export const createJsonLinesResponseHandler =
153
+ <T>(responseSchema: FlexibleSchema<T>): ResponseHandler<AsyncGenerator<T>> =>
154
+ async ({ response }) => {
155
+ const responseHeaders = extractResponseHeaders(response);
156
+
157
+ if (response.body == null) {
158
+ throw new EmptyResponseBodyError({});
159
+ }
160
+
161
+ return {
162
+ responseHeaders,
163
+ value: parseJsonLines({
164
+ stream: response.body,
165
+ schema: responseSchema,
166
+ }),
167
+ };
168
+ };
169
+
170
+ async function* parseJsonLines<T>({
171
+ stream,
172
+ schema,
173
+ }: {
174
+ stream: ReadableStream<Uint8Array>;
175
+ schema: FlexibleSchema<T>;
176
+ }): AsyncGenerator<T> {
177
+ const reader = stream.getReader();
178
+ const decoder = new TextDecoder();
179
+ let buffer = '';
180
+ let finished = false;
181
+
182
+ try {
183
+ while (true) {
184
+ const { done, value } = await reader.read();
185
+
186
+ if (done) {
187
+ finished = true;
188
+ buffer += decoder.decode();
189
+ break;
190
+ }
191
+
192
+ buffer += decoder.decode(value, { stream: true });
193
+
194
+ let lineEnd = buffer.indexOf('\n');
195
+ while (lineEnd !== -1) {
196
+ const line = buffer.slice(0, lineEnd).replace(/\r$/, '');
197
+ buffer = buffer.slice(lineEnd + 1);
198
+
199
+ if (line.trim().length > 0) {
200
+ yield await parseJSON({ text: line, schema });
201
+ }
202
+
203
+ lineEnd = buffer.indexOf('\n');
204
+ }
205
+ }
206
+
207
+ const finalLine = buffer.replace(/\r$/, '');
208
+ if (finalLine.trim().length > 0) {
209
+ yield await parseJSON({ text: finalLine, schema });
210
+ }
211
+ } finally {
212
+ if (!finished) {
213
+ await reader.cancel().catch(() => {});
214
+ }
215
+ reader.releaseLock();
216
+ }
217
+ }
218
+
152
219
  export const createBinaryResponseHandler =
153
220
  (): ResponseHandler<Uint8Array> =>
154
221
  async ({ response, url, requestBodyValues }) => {
@@ -14,6 +14,11 @@ export type ToolApprovalRequest = {
14
14
  */
15
15
  toolCallId: string;
16
16
 
17
+ /**
18
+ * Reason why the tool call requires approval.
19
+ */
20
+ reason?: string;
21
+
17
22
  /**
18
23
  * Flag indicating whether the tool was automatically approved or denied.
19
24
  *