@ai-sdk/provider-utils 2.2.1 → 2.2.2

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/provider-utils
 
+## 2.2.2
+
+### Patch Changes
+
+- b01120e: chore (provider-utils): update unified test server
+
 ## 2.2.1
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/provider-utils",
-  "version": "2.2.1",
+  "version": "2.2.2",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",

package/test/dist/index.d.mts

@@ -74,6 +74,7 @@ declare class TestServerCall$1 {
     private request;
     constructor(request: Request);
     getRequestBodyJson(): Promise<any>;
+    getRequestCredentials(): RequestCredentials;
     getRequestHeaders(): Record<string, string>;
     getRequestUrlSearchParams(): URLSearchParams;
 }
@@ -90,66 +91,44 @@ declare function describeWithTestServer(description: string, responses: Array<Te
     streamController: ReadableStreamDefaultController<string>;
 }) => void): void;
 
+type UrlResponse = {
+    type: 'json-value';
+    headers?: Record<string, string>;
+    body: JsonBodyType;
+} | {
+    type: 'stream-chunks';
+    headers?: Record<string, string>;
+    chunks: Array<string>;
+} | {
+    type: 'binary';
+    headers?: Record<string, string>;
+    body: Buffer;
+} | {
+    type: 'empty';
+    headers?: Record<string, string>;
+    status?: number;
+} | {
+    type: 'error';
+    headers?: Record<string, string>;
+    status?: number;
+    body?: string;
+} | {
+    type: 'controlled-stream';
+    headers?: Record<string, string>;
+    controller: TestResponseController;
+} | undefined;
+type UrlResponseParameter = UrlResponse | UrlResponse[] | ((options: {
+    callNumber: number;
+}) => UrlResponse);
 type UrlHandler = {
-    response?: {
-        type: 'json-value';
-        headers?: Record<string, string>;
-        body: JsonBodyType;
-    } | {
-        type: 'stream-chunks';
-        headers?: Record<string, string>;
-        chunks: Array<string>;
-    } | {
-        type: 'binary';
-        headers?: Record<string, string>;
-        body: Buffer;
-    } | {
-        type: 'empty';
-        headers?: Record<string, string>;
-        status?: number;
-    } | {
-        type: 'error';
-        headers?: Record<string, string>;
-        status?: number;
-        body?: string;
-    } | {
-        type: 'readable-stream';
-        headers?: Record<string, string>;
-        stream: ReadableStream;
-    } | undefined;
+    response: UrlResponseParameter;
 };
-type FullUrlHandler = {
-    response: {
-        type: 'json-value';
-        headers?: Record<string, string>;
-        body: JsonBodyType;
-    } | {
-        type: 'stream-chunks';
-        headers?: Record<string, string>;
-        chunks: Array<string>;
-    } | {
-        type: 'binary';
-        headers?: Record<string, string>;
-        body: Buffer;
-    } | {
-        type: 'error';
-        headers?: Record<string, string>;
-        status: number;
-        body?: string;
-    } | {
-        type: 'empty';
-        headers?: Record<string, string>;
-        status?: number;
-    } | {
-        type: 'readable-stream';
-        headers?: Record<string, string>;
-        stream: ReadableStream;
-    } | undefined;
-};
-type FullHandlers<URLS extends {
-    [url: string]: UrlHandler;
+type UrlHandlers<URLS extends {
+    [url: string]: {
+        response?: UrlResponseParameter;
+    };
 }> = {
-    [url in keyof URLS]: FullUrlHandler;
+    [url in keyof URLS]: UrlHandler;
 };
 declare class TestServerCall {
     private request;
@@ -161,10 +140,20 @@ declare class TestServerCall {
     get requestMethod(): string;
 }
 declare function createTestServer<URLS extends {
-    [url: string]: UrlHandler;
+    [url: string]: {
+        response?: UrlResponseParameter;
+    };
 }>(routes: URLS): {
-    urls: FullHandlers<URLS>;
+    urls: UrlHandlers<URLS>;
     calls: TestServerCall[];
 };
+declare class TestResponseController {
+    private readonly transformStream;
+    private readonly writer;
+    constructor();
+    get stream(): ReadableStream;
+    write(chunk: string): Promise<void>;
+    close(): Promise<void>;
+}
 
-export { type FullHandlers, type FullUrlHandler, JsonTestServer, StreamingTestServer, type TestServerJsonBodyType, type TestServerResponse, type UrlHandler, convertArrayToAsyncIterable, convertArrayToReadableStream, convertAsyncIterableToArray, convertReadableStreamToArray, convertResponseStreamToArray, createTestServer, describeWithTestServer, mockId, withTestServer };
+export { JsonTestServer, StreamingTestServer, TestResponseController, type TestServerJsonBodyType, type TestServerResponse, type UrlHandler, type UrlHandlers, type UrlResponse, convertArrayToAsyncIterable, convertArrayToReadableStream, convertAsyncIterableToArray, convertReadableStreamToArray, convertResponseStreamToArray, createTestServer, describeWithTestServer, mockId, withTestServer };

package/test/dist/index.d.ts

@@ -74,6 +74,7 @@ declare class TestServerCall$1 {
     private request;
     constructor(request: Request);
     getRequestBodyJson(): Promise<any>;
+    getRequestCredentials(): RequestCredentials;
     getRequestHeaders(): Record<string, string>;
     getRequestUrlSearchParams(): URLSearchParams;
 }
@@ -90,66 +91,44 @@ declare function describeWithTestServer(description: string, responses: Array<Te
     streamController: ReadableStreamDefaultController<string>;
 }) => void): void;
 
+type UrlResponse = {
+    type: 'json-value';
+    headers?: Record<string, string>;
+    body: JsonBodyType;
+} | {
+    type: 'stream-chunks';
+    headers?: Record<string, string>;
+    chunks: Array<string>;
+} | {
+    type: 'binary';
+    headers?: Record<string, string>;
+    body: Buffer;
+} | {
+    type: 'empty';
+    headers?: Record<string, string>;
+    status?: number;
+} | {
+    type: 'error';
+    headers?: Record<string, string>;
+    status?: number;
+    body?: string;
+} | {
+    type: 'controlled-stream';
+    headers?: Record<string, string>;
+    controller: TestResponseController;
+} | undefined;
+type UrlResponseParameter = UrlResponse | UrlResponse[] | ((options: {
+    callNumber: number;
+}) => UrlResponse);
 type UrlHandler = {
-    response?: {
-        type: 'json-value';
-        headers?: Record<string, string>;
-        body: JsonBodyType;
-    } | {
-        type: 'stream-chunks';
-        headers?: Record<string, string>;
-        chunks: Array<string>;
-    } | {
-        type: 'binary';
-        headers?: Record<string, string>;
-        body: Buffer;
-    } | {
-        type: 'empty';
-        headers?: Record<string, string>;
-        status?: number;
-    } | {
-        type: 'error';
-        headers?: Record<string, string>;
-        status?: number;
-        body?: string;
-    } | {
-        type: 'readable-stream';
-        headers?: Record<string, string>;
-        stream: ReadableStream;
-    } | undefined;
+    response: UrlResponseParameter;
 };
-type FullUrlHandler = {
-    response: {
-        type: 'json-value';
-        headers?: Record<string, string>;
-        body: JsonBodyType;
-    } | {
-        type: 'stream-chunks';
-        headers?: Record<string, string>;
-        chunks: Array<string>;
-    } | {
-        type: 'binary';
-        headers?: Record<string, string>;
-        body: Buffer;
-    } | {
-        type: 'error';
-        headers?: Record<string, string>;
-        status: number;
-        body?: string;
-    } | {
-        type: 'empty';
-        headers?: Record<string, string>;
-        status?: number;
-    } | {
-        type: 'readable-stream';
-        headers?: Record<string, string>;
-        stream: ReadableStream;
-    } | undefined;
-};
-type FullHandlers<URLS extends {
-    [url: string]: UrlHandler;
+type UrlHandlers<URLS extends {
+    [url: string]: {
+        response?: UrlResponseParameter;
+    };
 }> = {
-    [url in keyof URLS]: FullUrlHandler;
+    [url in keyof URLS]: UrlHandler;
 };
 declare class TestServerCall {
     private request;
@@ -161,10 +140,20 @@ declare class TestServerCall {
     get requestMethod(): string;
 }
 declare function createTestServer<URLS extends {
-    [url: string]: UrlHandler;
+    [url: string]: {
+        response?: UrlResponseParameter;
+    };
 }>(routes: URLS): {
-    urls: FullHandlers<URLS>;
+    urls: UrlHandlers<URLS>;
     calls: TestServerCall[];
 };
+declare class TestResponseController {
+    private readonly transformStream;
+    private readonly writer;
+    constructor();
+    get stream(): ReadableStream;
+    write(chunk: string): Promise<void>;
+    close(): Promise<void>;
+}
 
-export { type FullHandlers, type FullUrlHandler, JsonTestServer, StreamingTestServer, type TestServerJsonBodyType, type TestServerResponse, type UrlHandler, convertArrayToAsyncIterable, convertArrayToReadableStream, convertAsyncIterableToArray, convertReadableStreamToArray, convertResponseStreamToArray, createTestServer, describeWithTestServer, mockId, withTestServer };
+export { JsonTestServer, StreamingTestServer, TestResponseController, type TestServerJsonBodyType, type TestServerResponse, type UrlHandler, type UrlHandlers, type UrlResponse, convertArrayToAsyncIterable, convertArrayToReadableStream, convertAsyncIterableToArray, convertReadableStreamToArray, convertResponseStreamToArray, createTestServer, describeWithTestServer, mockId, withTestServer };

package/test/dist/index.js

@@ -59,6 +59,7 @@ var test_exports = {};
 __export(test_exports, {
   JsonTestServer: () => JsonTestServer,
   StreamingTestServer: () => StreamingTestServer,
+  TestResponseController: () => TestResponseController,
   convertArrayToAsyncIterable: () => convertArrayToAsyncIterable,
   convertArrayToReadableStream: () => convertArrayToReadableStream,
   convertAsyncIterableToArray: () => convertAsyncIterableToArray,
@@ -18622,6 +18623,10 @@ var TestServerCall = class {
     expect(this.request).toBeDefined();
     return JSON.parse(await this.request.text());
   }
+  getRequestCredentials() {
+    expect(this.request).toBeDefined();
+    return this.request.credentials;
+  }
   getRequestHeaders() {
     expect(this.request).toBeDefined();
     const requestHeaders = this.request.headers;
@@ -18821,10 +18826,11 @@ function createTestServer(routes) {
   const originalRoutes = structuredClone(routes);
   const mswServer = setupServer(
     ...Object.entries(routes).map(([url, handler]) => {
-      return http.all(url, ({ request, params }) => {
-        var _a4, _b2, _c2, _d, _e;
+      return http.all(url, ({ request }) => {
+        var _a4, _b2, _c2;
+        const callNumber = calls.length;
         calls.push(new TestServerCall2(request));
-        const response = handler.response;
+        const response = typeof handler.response === "function" ? handler.response({ callNumber }) : Array.isArray(handler.response) ? handler.response[callNumber] : handler.response;
         if (response === void 0) {
           return HttpResponse.json({ error: "Not Found" }, { status: 404 });
         }
@@ -18835,7 +18841,7 @@ function createTestServer(routes) {
               status: 200,
               headers: {
                 "Content-Type": "application/json",
-                ...(_a4 = handler.response) == null ? void 0 : _a4.headers
+                ...response.headers
               }
             });
           case "stream-chunks":
@@ -18853,9 +18859,9 @@ function createTestServer(routes) {
                 }
               }
             );
-          case "readable-stream": {
+          case "controlled-stream": {
             return new HttpResponse(
-              response.stream.pipeThrough(new TextEncoderStream()),
+              response.controller.stream.pipeThrough(new TextEncoderStream()),
               {
                 status: 200,
                 headers: {
@@ -18870,17 +18876,17 @@ function createTestServer(routes) {
           case "binary": {
             return HttpResponse.arrayBuffer(response.body, {
               status: 200,
-              headers: (_b2 = handler.response) == null ? void 0 : _b2.headers
+              headers: response.headers
             });
           }
           case "error":
-            return HttpResponse.text((_c2 = response.body) != null ? _c2 : "Error", {
-              status: (_d = response.status) != null ? _d : 500,
+            return HttpResponse.text((_a4 = response.body) != null ? _a4 : "Error", {
+              status: (_b2 = response.status) != null ? _b2 : 500,
               headers: response.headers
             });
           case "empty":
             return new HttpResponse(null, {
-              status: (_e = response.status) != null ? _e : 200
+              status: (_c2 = response.status) != null ? _c2 : 200
             });
           default: {
             const _exhaustiveCheck = handlerType;
@@ -18911,10 +18917,26 @@ function createTestServer(routes) {
     }
   };
 }
+var TestResponseController = class {
+  constructor() {
+    this.transformStream = new TransformStream();
+    this.writer = this.transformStream.writable.getWriter();
+  }
+  get stream() {
+    return this.transformStream.readable;
+  }
+  async write(chunk) {
+    await this.writer.write(chunk);
+  }
+  async close() {
+    await this.writer.close();
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   JsonTestServer,
   StreamingTestServer,
+  TestResponseController,
   convertArrayToAsyncIterable,
   convertArrayToReadableStream,
   convertAsyncIterableToArray,