ai-gateway-proxy 0.0.1 → 0.1.0

package/README.md

@@ -0,0 +1,62 @@
+# ai-gateway-proxy
+
+A lightweight proxy handler for [Vercel AI Gateway](https://vercel.com/docs/ai-gateway).
+
+## Features
+
+- **Request/Response Hooks** - Transform requests and responses with `beforeRequest`, `afterResponse`, and `onError` hooks
+- **Streaming Support** - Full SSE streaming with content aggregation using AI SDK V3 protocol
+- **Framework Agnostic** - Works with Next.js, Hono, Express, and other frameworks
+
+## Installation
+
+```bash
+npm install ai-gateway-proxy
+```
+
+## Quick Start
+
+### Next.js App Router
+
+Create a catch-all route at `app/api/ai/[...segments]/route.ts`:
+
+```typescript
+import { createGatewayProxy } from "ai-gateway-proxy";
+
+export const { GET, POST, PUT, DELETE, PATCH } = createGatewayProxy();
+```
+
+### Hono / Other Frameworks
+
+For frameworks without catch-all route support, use the `extractPath` option:
+
+```typescript
+import { Hono } from "hono";
+import { createGatewayProxy } from "ai-gateway-proxy";
+
+const app = new Hono();
+
+const proxy = createGatewayProxy({
+  extractPath: (request) => {
+    const url = new URL(request.url);
+    return url.pathname.replace(/^\/api\/ai\//, "");
+  },
+});
+
+app.all("/api/ai/*", (c) => proxy(c.req.raw));
+```
+
+## Authentication
+
+The proxy automatically handles authentication using one of the following methods (in priority order):
+
+1. **API Key** - Set `AI_GATEWAY_API_KEY` environment variable
+2. **OIDC** - Automatic when deployed on Vercel
+
+## Documentation
+
+For full documentation including configuration options, hooks, and API reference, see the [documentation](https://sferadev.com/docs/packages/ai-gateway-proxy).
+
+## License
+
+MIT

package/dist/index.cjs

@@ -235,24 +235,159 @@ var providerUtils = require('@ai-sdk/provider-utils');
     });
 }
 
+const DEFAULT_BASE_URL = "https://ai-gateway.vercel.sh/v1/ai";
+const PROTOCOL_VERSION = "0.0.1";
+/**
+ * Extracts the path from the request using either a custom extractor or route params.
+ */ function extractPathFromRequest(request, context, options) {
+    return _async_to_generator._(function*() {
+        if (options.extractPath) {
+            return options.extractPath(request);
+        }
+        if (context == null ? void 0 : context.params) {
+            const params = yield context.params;
+            const segments = params[options.segmentsParam];
+            return Array.isArray(segments) ? segments.join("/") : segments != null ? segments : "";
+        }
+        return "";
+    })();
+}
+/**
+ * Builds the upstream URL from base URL, path, and query params.
+ */ function buildUpstreamUrl(baseUrl, path, requestUrl) {
+    const searchParams = new URL(requestUrl).searchParams;
+    const queryString = searchParams.toString();
+    return `${baseUrl}/${path}${queryString ? `?${queryString}` : ""}`;
+}
+/**
+ * Builds the headers for the upstream request.
+ */ function buildRequestHeaders(auth, isStreaming, modelId, customHeaders) {
+    return _extends._({
+        "content-type": "application/json"
+    }, customHeaders, {
+        authorization: `Bearer ${auth.token}`,
+        "ai-gateway-auth-method": auth.authMethod,
+        "ai-gateway-protocol-version": PROTOCOL_VERSION,
+        "ai-language-model-id": modelId != null ? modelId : "",
+        "ai-language-model-streaming": isStreaming ? "true" : "false"
+    });
+}
+/**
+ * Parses the error response from the gateway.
+ */ function parseErrorBody(res) {
+    return _async_to_generator._(function*() {
+        try {
+            return yield res.json();
+        } catch (unused) {
+            return {
+                error: {
+                    message: res.statusText || "Unknown error",
+                    type: "gateway_error"
+                }
+            };
+        }
+    })();
+}
+/**
+ * Creates an error response, optionally applying the onError hook.
+ */ function createErrorResponse(res, errorBody, requestBody, onError) {
+    return _async_to_generator._(function*() {
+        if (onError) {
+            const result = yield onError({
+                request: requestBody,
+                error: errorBody,
+                status: res.status
+            });
+            if (result instanceof Response) {
+                return result;
+            }
+            return new Response(JSON.stringify(result), {
+                status: res.status,
+                statusText: res.statusText,
+                headers: {
+                    "content-type": "application/json"
+                }
+            });
+        }
+        return new Response(JSON.stringify(errorBody), {
+            status: res.status,
+            statusText: res.statusText,
+            headers: {
+                "content-type": "application/json"
+            }
+        });
+    })();
+}
+/**
+ * Handles a streaming response from the gateway.
+ */ function handleStreamingResponse(res, requestBody, afterResponse) {
+    if (!res.body) {
+        return new Response("No response body", {
+            status: 500
+        });
+    }
+    const contentType = res.headers.get("content-type") || "text/event-stream";
+    // Pass through directly if no afterResponse hook
+    if (!afterResponse) {
+        return new Response(res.body, {
+            status: res.status,
+            statusText: res.statusText,
+            headers: {
+                "content-type": contentType
+            }
+        });
+    }
+    // Transform stream to call afterResponse on finish
+    const transformedStream = createStreamTransformer(res.body, requestBody, afterResponse);
+    return new Response(transformedStream, {
+        status: res.status,
+        statusText: res.statusText,
+        headers: {
+            "content-type": contentType
+        }
+    });
+}
+/**
+ * Handles a non-streaming JSON response from the gateway.
+ */ function handleJsonResponse(res, requestBody, afterResponse) {
+    return _async_to_generator._(function*() {
+        let responseBody = yield res.json();
+        if (afterResponse) {
+            responseBody = yield afterResponse({
+                request: requestBody,
+                response: responseBody
+            });
+        }
+        return new Response(JSON.stringify(responseBody), {
+            status: res.status,
+            statusText: res.statusText,
+            headers: {
+                "content-type": "application/json"
+            }
+        });
+    })();
+}
 /**
  * Creates a proxy handler for the Vercel AI Gateway.
  *
  * Uses AI SDK V3 Language Model.
  * @see https://github.com/vercel/ai/blob/main/packages/gateway/src/gateway-language-model.ts
- */ const createGatewayProxy = ({ baseUrl = "https://ai-gateway.vercel.sh/v1/ai", headers, beforeRequest, afterResponse, onError } = {})=>{
-    const handler = (request, { params })=>_async_to_generator._(function*() {
-            var _ref;
+ */ const createGatewayProxy = ({ baseUrl = DEFAULT_BASE_URL, headers, segmentsParam = "segments", extractPath, beforeRequest, afterResponse, onError } = {})=>{
+    const handler = (request, context)=>_async_to_generator._(function*() {
+            // Authenticate with the gateway
             const auth = yield getGatewayAuthToken();
             if (!auth) {
                 return new Response("AI Gateway authentication not configured", {
                     status: 500
                 });
             }
-            const { segments } = yield params;
-            const path = (_ref = segments == null ? void 0 : segments.join("/")) != null ? _ref : "";
-            const searchParams = new URL(request.url).searchParams;
-            const upstream = `${baseUrl}/${path}${searchParams.toString() ? `?${searchParams.toString()}` : ""}`;
+            // Build upstream URL
+            const path = yield extractPathFromRequest(request, context, {
+                extractPath,
+                segmentsParam
+            });
+            const upstream = buildUpstreamUrl(baseUrl, path, request.url);
+            // Parse request body
             let requestBody;
             try {
                 requestBody = yield request.clone().json();
@@ -269,109 +404,29 @@ var providerUtils = require('@ai-sdk/provider-utils');
             }
             const isStreaming = request.headers.get("ai-language-model-streaming") === "true";
             try {
-                var _request_headers_get;
                 const res = yield fetch(upstream, {
                     method: request.method,
                     body: JSON.stringify(requestBody),
                     duplex: "half",
-                    headers: _extends._({
-                        "content-type": "application/json"
-                    }, headers, {
-                        authorization: `Bearer ${auth.token}`,
-                        "ai-gateway-auth-method": auth.authMethod,
-                        "ai-gateway-protocol-version": "0.0.1",
-                        "ai-language-model-id": (_request_headers_get = request.headers.get("ai-language-model-id")) != null ? _request_headers_get : "",
-                        "ai-language-model-streaming": isStreaming ? "true" : "false"
-                    })
+                    headers: buildRequestHeaders(auth, isStreaming, request.headers.get("ai-language-model-id"), headers)
                 });
                 // Handle error responses
                 if (!res.ok) {
-                    let errorBody;
-                    try {
-                        errorBody = yield res.json();
-                    } catch (unused) {
-                        errorBody = {
-                            error: {
-                                message: res.statusText || "Unknown error",
-                                type: "gateway_error"
-                            }
-                        };
-                    }
-                    if (onError) {
-                        const result = yield onError({
-                            request: requestBody,
-                            error: errorBody,
-                            status: res.status
-                        });
-                        // If onError returns a Response, use it directly
-                        if (result instanceof Response) {
-                            return result;
-                        }
-                        // Otherwise, return the modified error
-                        return new Response(JSON.stringify(result), {
-                            status: res.status,
-                            statusText: res.statusText,
-                            headers: {
-                                "content-type": "application/json"
-                            }
-                        });
-                    }
-                    return new Response(JSON.stringify(errorBody), {
-                        status: res.status,
-                        statusText: res.statusText,
-                        headers: {
-                            "content-type": "application/json"
-                        }
-                    });
+                    const errorBody = yield parseErrorBody(res);
+                    return createErrorResponse(res, errorBody, requestBody, onError);
                 }
-                // For streaming responses
+                // Handle successful responses
                 if (isStreaming) {
-                    if (!res.body) {
-                        return new Response("No response body", {
-                            status: 500
-                        });
-                    }
-                    // If no afterResponse hook, pass through directly
-                    if (!afterResponse) {
-                        return new Response(res.body, {
-                            status: res.status,
-                            statusText: res.statusText,
-                            headers: {
-                                "content-type": res.headers.get("content-type") || "text/event-stream"
-                            }
-                        });
-                    }
-                    // Transform the stream to call afterResponse on finish
-                    const transformedStream = createStreamTransformer(res.body, requestBody, afterResponse);
-                    return new Response(transformedStream, {
-                        status: res.status,
-                        statusText: res.statusText,
-                        headers: {
-                            "content-type": res.headers.get("content-type") || "text/event-stream"
-                        }
-                    });
+                    return handleStreamingResponse(res, requestBody, afterResponse);
                 }
-                // For non-streaming responses, parse and optionally transform
-                let responseBody = yield res.json();
-                if (afterResponse) {
-                    responseBody = yield afterResponse({
-                        request: requestBody,
-                        response: responseBody
-                    });
-                }
-                return new Response(JSON.stringify(responseBody), {
-                    status: res.status,
-                    statusText: res.statusText,
-                    headers: {
-                        "content-type": "application/json"
-                    }
-                });
+                return handleJsonResponse(res, requestBody, afterResponse);
             } catch (unused) {
                 return new Response("Error proxying request to AI Gateway", {
                     status: 500
                 });
             }
         })();
+    // Export handler for all HTTP methods
     handler.GET = handler;
     handler.POST = handler;
     handler.PUT = handler;

package/dist/index.d.ts

@@ -39,6 +39,30 @@ type CreateGatewayProxyOptions = {
     baseUrl?: string;
     /** Additional headers to send with requests. */
     headers?: Record<string, string>;
+    /**
+     * The name of the catch-all parameter in the route.
+     * @default "segments"
+     * @example
+     * // For Next.js route `app/api/ai/[...path]/route.ts`
+     * createGatewayProxy({ segmentsParam: "path" })
+     */
+    segmentsParam?: string;
+    /**
+     * Custom function to extract the path from the request.
+     * Use this for frameworks without catch-all route support (e.g., Hono, Express).
+     *
+     * When provided, `segmentsParam` is ignored.
+     *
+     * @example
+     * // Hono with a base path
+     * createGatewayProxy({
+     *   extractPath: (request) => {
+     *     const url = new URL(request.url);
+     *     return url.pathname.replace(/^\/api\/ai\//, "");
+     *   }
+     * })
+     */
+    extractPath?: (request: Request) => string | Promise<string>;
     /**
      * Called before the request is sent to the gateway.
      * Allows modification of the request body.
@@ -69,12 +93,11 @@ type CreateGatewayProxyOptions = {
     }) => GatewayError | Response | Promise<GatewayError | Response>;
 };
 /**
- * Next.js route handler function signature for the gateway proxy.
+ * Route handler function signature for the gateway proxy.
+ * Compatible with Next.js App Router and other frameworks.
  */
-type CreateGatewayProxyFn = (request: Request, context: {
-    params: Promise<{
-        segments?: string[];
-    }>;
+type CreateGatewayProxyFn = (request: Request, context?: {
+    params: Promise<Record<string, string | string[] | undefined>>;
 }) => Promise<Response>;
 /**
  * Result of createGatewayProxy - a handler function with HTTP method exports.
@@ -93,7 +116,7 @@ type CreateGatewayProxyResult = CreateGatewayProxyFn & {
  * Uses AI SDK V3 Language Model.
  * @see https://github.com/vercel/ai/blob/main/packages/gateway/src/gateway-language-model.ts
  */
-declare const createGatewayProxy: ({ baseUrl, headers, beforeRequest, afterResponse, onError, }?: CreateGatewayProxyOptions) => CreateGatewayProxyResult;
+declare const createGatewayProxy: ({ baseUrl, headers, segmentsParam, extractPath, beforeRequest, afterResponse, onError, }?: CreateGatewayProxyOptions) => CreateGatewayProxyResult;
 
 /**
  * Aggregator for collecting stream content.

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sources":["../src/auth.ts","../src/types.ts","../src/proxy.ts","../src/stream.ts"],"sourcesContent":["import { getVercelOidcToken } from \"@vercel/oidc\";\n\nexport type GatewayAuthToken = {\n\ttoken: string;\n\tauthMethod: \"api-key\" | \"oidc\";\n};\n\n/**\n * Get authentication token for the AI Gateway.\n * Tries API key first (AI_GATEWAY_API_KEY env var), then falls back to Vercel OIDC.\n */\nexport async function getGatewayAuthToken(): Promise<GatewayAuthToken | null> {\n\tconst apiKey = process.env.AI_GATEWAY_API_KEY;\n\tif (apiKey) {\n\t\treturn { token: apiKey, authMethod: \"api-key\" };\n\t}\n\n\ttry {\n\t\tconst oidcToken = await getVercelOidcToken();\n\t\treturn { token: oidcToken, authMethod: \"oidc\" };\n\t} catch {\n\t\treturn null;\n\t}\n}\n","import type {\n\tLanguageModelV3CallOptions,\n\tLanguageModelV3Content,\n\tLanguageModelV3FinishReason,\n\tLanguageModelV3Usage,\n\tSharedV3ProviderMetadata,\n\tSharedV3Warning,\n} from \"@ai-sdk/provider\";\n\n/**\n * The response body from the AI Gateway for non-streaming requests.\n * For streaming requests, this is aggregated from the stream events.\n */\nexport type GatewayResponse = {\n\tcontent: LanguageModelV3Content[];\n\tfinishReason: LanguageModelV3FinishReason;\n\tusage: LanguageModelV3Usage;\n\twarnings: SharedV3Warning[];\n\tproviderMetadata?: SharedV3ProviderMetadata;\n};\n\n/**\n * Error response from the AI Gateway.\n */\nexport type GatewayError = {\n\terror: {\n\t\tmessage: string;\n\t\ttype: string;\n\t\tparam?: unknown;\n\t};\n};\n\n/**\n * Options for creating a gateway proxy.\n */\nexport type CreateGatewayProxyOptions = {\n\t/** Base URL for the AI Gateway. Defaults to Vercel's gateway. */\n\tbaseUrl?: string;\n\t/** Additional headers to send with requests. */\n\theaders?: Record<string, string>;\n\t/**\n\t * Called before the request is sent to the gateway.\n\t * Allows modification of the request body.\n\t */\n\tbeforeRequest?: (context: {\n\t\trequest: LanguageModelV3CallOptions;\n\t}) => LanguageModelV3CallOptions | Promise<LanguageModelV3CallOptions>;\n\t/**\n\t * Called after a successful response is received from the gateway.\n\t * Allows modification of the response body.\n\t *\n\t * For streaming responses, this is called when the stream completes with\n\t * the aggregated response. The content has already been streamed, but you\n\t * can modify the finish event metadata (usage, finishReason, providerMetadata).\n\t */\n\tafterResponse?: (context: {\n\t\trequest: LanguageModelV3CallOptions;\n\t\tresponse: GatewayResponse;\n\t}) => GatewayResponse | Promise<GatewayResponse>;\n\t/**\n\t * Called when the gateway returns an error response.\n\t * Allows custom error handling or modification.\n\t */\n\tonError?: (context: {\n\t\trequest: LanguageModelV3CallOptions;\n\t\terror: GatewayError;\n\t\tstatus: number;\n\t}) => GatewayError | Response | Promise<GatewayError | Response>;\n};\n\n/**\n * Next.js route handler function signature for the gateway proxy.\n */\nexport type CreateGatewayProxyFn = (\n\trequest: Request,\n\tcontext: { params: Promise<{ segments?: string[] }> },\n) => Promise<Response>;\n\n/**\n * Result of createGatewayProxy - a handler function with HTTP method exports.\n */\nexport type CreateGatewayProxyResult = CreateGatewayProxyFn & {\n\tGET: CreateGatewayProxyFn;\n\tPOST: CreateGatewayProxyFn;\n\tPUT: CreateGatewayProxyFn;\n\tDELETE: CreateGatewayProxyFn;\n\tPATCH: CreateGatewayProxyFn;\n};\n","import type { LanguageModelV3CallOptions } from \"@ai-sdk/provider\";\nimport { getGatewayAuthToken } from \"./auth\";\nimport { createStreamTransformer } from \"./stream\";\nimport type {\n\tCreateGatewayProxyOptions,\n\tCreateGatewayProxyResult,\n\tGatewayError,\n\tGatewayResponse,\n} from \"./types\";\n\n/**\n * Creates a proxy handler for the Vercel AI Gateway.\n *\n * Uses AI SDK V3 Language Model.\n * @see https://github.com/vercel/ai/blob/main/packages/gateway/src/gateway-language-model.ts\n */\nexport const createGatewayProxy = ({\n\tbaseUrl = \"https://ai-gateway.vercel.sh/v1/ai\",\n\theaders,\n\tbeforeRequest,\n\tafterResponse,\n\tonError,\n}: CreateGatewayProxyOptions = {}): CreateGatewayProxyResult => {\n\tconst handler = async (\n\t\trequest: Request,\n\t\t{ params }: { params: Promise<{ segments?: string[] }> },\n\t) => {\n\t\tconst auth = await getGatewayAuthToken();\n\t\tif (!auth) {\n\t\t\treturn new Response(\"AI Gateway authentication not configured\", {\n\t\t\t\tstatus: 500,\n\t\t\t});\n\t\t}\n\n\t\tconst { segments } = await params;\n\t\tconst path = segments?.join(\"/\") ?? \"\";\n\t\tconst searchParams = new URL(request.url).searchParams;\n\t\tconst upstream = `${baseUrl}/${path}${searchParams.toString() ? `?${searchParams.toString()}` : \"\"}`;\n\n\t\tlet requestBody: LanguageModelV3CallOptions;\n\t\ttry {\n\t\t\trequestBody = await request.clone().json();\n\t\t} catch {\n\t\t\treturn new Response(\"Invalid JSON body\", { status: 400 });\n\t\t}\n\n\t\t// Apply beforeRequest hook\n\t\tif (beforeRequest) {\n\t\t\trequestBody = await beforeRequest({ request: requestBody });\n\t\t}\n\n\t\tconst isStreaming = request.headers.get(\"ai-language-model-streaming\") === \"true\";\n\n\t\ttry {\n\t\t\tconst res = await fetch(upstream, {\n\t\t\t\tmethod: request.method,\n\t\t\t\tbody: JSON.stringify(requestBody),\n\t\t\t\tduplex: \"half\",\n\t\t\t\theaders: {\n\t\t\t\t\t\"content-type\": \"application/json\",\n\t\t\t\t\t...headers,\n\t\t\t\t\tauthorization: `Bearer ${auth.token}`,\n\t\t\t\t\t\"ai-gateway-auth-method\": auth.authMethod,\n\t\t\t\t\t\"ai-gateway-protocol-version\": \"0.0.1\",\n\t\t\t\t\t\"ai-language-model-id\": request.headers.get(\"ai-language-model-id\") ?? \"\",\n\t\t\t\t\t\"ai-language-model-streaming\": isStreaming ? \"true\" : \"false\",\n\t\t\t\t},\n\t\t\t} as RequestInit);\n\n\t\t\t// Handle error responses\n\t\t\tif (!res.ok) {\n\t\t\t\tlet errorBody: GatewayError;\n\t\t\t\ttry {\n\t\t\t\t\terrorBody = await res.json();\n\t\t\t\t} catch {\n\t\t\t\t\terrorBody = {\n\t\t\t\t\t\terror: {\n\t\t\t\t\t\t\tmessage: res.statusText || \"Unknown error\",\n\t\t\t\t\t\t\ttype: \"gateway_error\",\n\t\t\t\t\t\t},\n\t\t\t\t\t};\n\t\t\t\t}\n\n\t\t\t\tif (onError) {\n\t\t\t\t\tconst result = await onError({\n\t\t\t\t\t\trequest: requestBody,\n\t\t\t\t\t\terror: errorBody,\n\t\t\t\t\t\tstatus: res.status,\n\t\t\t\t\t});\n\n\t\t\t\t\t// If onError returns a Response, use it directly\n\t\t\t\t\tif (result instanceof Response) {\n\t\t\t\t\t\treturn result;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Otherwise, return the modified error\n\t\t\t\t\treturn new Response(JSON.stringify(result), {\n\t\t\t\t\t\tstatus: res.status,\n\t\t\t\t\t\tstatusText: res.statusText,\n\t\t\t\t\t\theaders: { \"content-type\": \"application/json\" },\n\t\t\t\t\t});\n\t\t\t\t}\n\n\t\t\t\treturn new Response(JSON.stringify(errorBody), {\n\t\t\t\t\tstatus: res.status,\n\t\t\t\t\tstatusText: res.statusText,\n\t\t\t\t\theaders: { \"content-type\": \"application/json\" },\n\t\t\t\t});\n\t\t\t}\n\n\t\t\t// For streaming responses\n\t\t\tif (isStreaming) {\n\t\t\t\tif (!res.body) {\n\t\t\t\t\treturn new Response(\"No response body\", { status: 500 });\n\t\t\t\t}\n\n\t\t\t\t// If no afterResponse hook, pass through directly\n\t\t\t\tif (!afterResponse) {\n\t\t\t\t\treturn new Response(res.body, {\n\t\t\t\t\t\tstatus: res.status,\n\t\t\t\t\t\tstatusText: res.statusText,\n\t\t\t\t\t\theaders: {\n\t\t\t\t\t\t\t\"content-type\": res.headers.get(\"content-type\") || \"text/event-stream\",\n\t\t\t\t\t\t},\n\t\t\t\t\t});\n\t\t\t\t}\n\n\t\t\t\t// Transform the stream to call afterResponse on finish\n\t\t\t\tconst transformedStream = createStreamTransformer(res.body, requestBody, afterResponse);\n\n\t\t\t\treturn new Response(transformedStream, {\n\t\t\t\t\tstatus: res.status,\n\t\t\t\t\tstatusText: res.statusText,\n\t\t\t\t\theaders: {\n\t\t\t\t\t\t\"content-type\": res.headers.get(\"content-type\") || \"text/event-stream\",\n\t\t\t\t\t},\n\t\t\t\t});\n\t\t\t}\n\n\t\t\t// For non-streaming responses, parse and optionally transform\n\t\t\tlet responseBody: GatewayResponse = await res.json();\n\n\t\t\tif (afterResponse) {\n\t\t\t\tresponseBody = await afterResponse({\n\t\t\t\t\trequest: requestBody,\n\t\t\t\t\tresponse: responseBody,\n\t\t\t\t});\n\t\t\t}\n\n\t\t\treturn new Response(JSON.stringify(responseBody), {\n\t\t\t\tstatus: res.status,\n\t\t\t\tstatusText: res.statusText,\n\t\t\t\theaders: { \"content-type\": \"application/json\" },\n\t\t\t});\n\t\t} catch {\n\t\t\treturn new Response(\"Error proxying request to AI Gateway\", {\n\t\t\t\tstatus: 500,\n\t\t\t});\n\t\t}\n\t};\n\n\thandler.GET = handler;\n\thandler.POST = handler;\n\thandler.PUT = handler;\n\thandler.DELETE = handler;\n\thandler.PATCH = handler;\n\n\treturn handler;\n};\n","import type {\n\tLanguageModelV3CallOptions,\n\tLanguageModelV3Content,\n\tLanguageModelV3FinishReason,\n\tLanguageModelV3StreamPart,\n\tLanguageModelV3Usage,\n\tSharedV3ProviderMetadata,\n\tSharedV3Warning,\n} from \"@ai-sdk/provider\";\nimport { EventSourceParserStream } from \"@ai-sdk/provider-utils\";\nimport type { CreateGatewayProxyOptions, GatewayResponse } from \"./types\";\n\n/**\n * Aggregator for collecting stream content.\n * Tracks text deltas, reasoning, tool calls, and other content to build the final GatewayResponse.\n *\n * Uses AI SDK V3 protocol types.\n * @see https://github.com/vercel/ai/blob/main/packages/provider/src/language-model/v3/language-model-v3-stream-part.ts\n */\nexport class StreamContentAggregator {\n\tprivate content: LanguageModelV3Content[] = [];\n\tprivate warnings: SharedV3Warning[] = [];\n\n\t// Text content being built (keyed by id)\n\tprivate textContent = new Map<string, { type: \"text\"; text: string }>();\n\n\t// Reasoning content being built (keyed by id)\n\tprivate reasoningContent = new Map<string, { type: \"reasoning\"; text: string }>();\n\n\tprivate finishReason: LanguageModelV3FinishReason = {\n\t\tunified: \"stop\",\n\t\traw: undefined,\n\t};\n\n\tprivate usage: LanguageModelV3Usage = {\n\t\tinputTokens: {\n\t\t\ttotal: undefined,\n\t\t\tnoCache: undefined,\n\t\t\tcacheRead: undefined,\n\t\t\tcacheWrite: undefined,\n\t\t},\n\t\toutputTokens: {\n\t\t\ttotal: undefined,\n\t\t\ttext: undefined,\n\t\t\treasoning: undefined,\n\t\t},\n\t};\n\n\tprivate providerMetadata: SharedV3ProviderMetadata | undefined;\n\n\t/**\n\t * Process a stream part and update aggregated state.\n\t * Uses LanguageModelV3StreamPart type from @ai-sdk/provider.\n\t *\n\t * @see https://github.com/vercel/ai/blob/main/packages/provider/src/language-model/v3/language-model-v3-stream-part.ts\n\t */\n\tprocess(part: LanguageModelV3StreamPart): void {\n\t\tswitch (part.type) {\n\t\t\tcase \"stream-start\":\n\t\t\t\tthis.warnings.push(...part.warnings);\n\t\t\t\tbreak;\n\n\t\t\tcase \"text-start\":\n\t\t\t\tthis.textContent.set(part.id, { type: \"text\", text: \"\" });\n\t\t\t\tbreak;\n\n\t\t\tcase \"text-delta\":\n\t\t\t\t{\n\t\t\t\t\tconst text = this.textContent.get(part.id);\n\t\t\t\t\tif (text) {\n\t\t\t\t\t\ttext.text += part.delta;\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"text-end\":\n\t\t\t\t{\n\t\t\t\t\tconst text = this.textContent.get(part.id);\n\t\t\t\t\tif (text) {\n\t\t\t\t\t\tthis.content.push(text);\n\t\t\t\t\t\tthis.textContent.delete(part.id);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"reasoning-start\":\n\t\t\t\tthis.reasoningContent.set(part.id, { type: \"reasoning\", text: \"\" });\n\t\t\t\tbreak;\n\n\t\t\tcase \"reasoning-delta\":\n\t\t\t\t{\n\t\t\t\t\tconst reasoning = this.reasoningContent.get(part.id);\n\t\t\t\t\tif (reasoning) {\n\t\t\t\t\t\treasoning.text += part.delta;\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"reasoning-end\":\n\t\t\t\t{\n\t\t\t\t\tconst reasoning = this.reasoningContent.get(part.id);\n\t\t\t\t\tif (reasoning) {\n\t\t\t\t\t\tthis.content.push(reasoning);\n\t\t\t\t\t\tthis.reasoningContent.delete(part.id);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-input-start\":\n\t\t\tcase \"tool-input-delta\":\n\t\t\tcase \"tool-input-end\":\n\t\t\t\t// Tool input is streamed incrementally but we wait for the complete tool-call event\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-call\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-result\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-approval-request\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"file\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"source\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"finish\":\n\t\t\t\t// Finalize any remaining content\n\t\t\t\tfor (const [, text] of this.textContent) {\n\t\t\t\t\tthis.content.push(text);\n\t\t\t\t}\n\t\t\t\tthis.textContent.clear();\n\n\t\t\t\tfor (const [, reasoning] of this.reasoningContent) {\n\t\t\t\t\tthis.content.push(reasoning);\n\t\t\t\t}\n\t\t\t\tthis.reasoningContent.clear();\n\n\t\t\t\tthis.finishReason = part.finishReason;\n\t\t\t\tthis.usage = part.usage;\n\t\t\t\tthis.providerMetadata = part.providerMetadata;\n\t\t\t\tbreak;\n\t\t}\n\t}\n\n\t/**\n\t * Get the aggregated response.\n\t */\n\tgetResponse(): GatewayResponse {\n\t\treturn {\n\t\t\tcontent: this.content,\n\t\t\tfinishReason: this.finishReason,\n\t\t\tusage: this.usage,\n\t\t\twarnings: this.warnings,\n\t\t\tproviderMetadata: this.providerMetadata,\n\t\t};\n\t}\n}\n\n/**\n * Creates a transform stream that:\n * 1. Passes through all SSE events immediately\n * 2. Aggregates content using StreamContentAggregator\n * 3. Calls afterResponse when finish event is received\n * 4. Modifies the finish event based on afterResponse result\n *\n * Uses EventSourceParserStream from @ai-sdk/provider-utils for SSE parsing.\n * EventSourceParserStream is re-exported from the eventsource-parser package.\n * @see https://github.com/vercel/ai/blob/main/packages/provider-utils/src/parse-json-event-stream.ts\n * @see https://github.com/rexxars/eventsource-parser\n */\nexport function createStreamTransformer(\n\tbody: ReadableStream<Uint8Array>,\n\trequestBody: LanguageModelV3CallOptions,\n\tafterResponse: NonNullable<CreateGatewayProxyOptions[\"afterResponse\"]>,\n): ReadableStream<Uint8Array> {\n\tconst encoder = new TextEncoder();\n\tconst aggregator = new StreamContentAggregator();\n\n\t// Use EventSourceParserStream for SSE parsing (re-exported from eventsource-parser via @ai-sdk/provider-utils)\n\tconst eventStream = body\n\t\t.pipeThrough(new TextDecoderStream() as TransformStream<Uint8Array, string>)\n\t\t.pipeThrough(new EventSourceParserStream());\n\n\treturn new ReadableStream({\n\t\tasync start(controller) {\n\t\t\tconst reader = eventStream.getReader();\n\n\t\t\ttry {\n\t\t\t\twhile (true) {\n\t\t\t\t\tconst { done, value } = await reader.read();\n\t\t\t\t\tif (done) break;\n\n\t\t\t\t\tconst { data } = value;\n\n\t\t\t\t\t// Handle DONE marker (end of stream)\n\t\t\t\t\tif (data === \"DONE\" || data === \"[DONE]\") {\n\t\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${data}\\n\\n`));\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Parse the JSON event data\n\t\t\t\t\tlet parsed: LanguageModelV3StreamPart;\n\t\t\t\t\ttry {\n\t\t\t\t\t\tparsed = JSON.parse(data);\n\t\t\t\t\t} catch {\n\t\t\t\t\t\t// Not valid JSON, pass through as-is\n\t\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${data}\\n\\n`));\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Track content for aggregation\n\t\t\t\t\taggregator.process(parsed);\n\n\t\t\t\t\t// Handle finish event specially - call afterResponse hook\n\t\t\t\t\tif (parsed.type === \"finish\") {\n\t\t\t\t\t\tconst modifiedResponse = await afterResponse({\n\t\t\t\t\t\t\trequest: requestBody,\n\t\t\t\t\t\t\tresponse: aggregator.getResponse(),\n\t\t\t\t\t\t});\n\n\t\t\t\t\t\t// Emit modified finish event\n\t\t\t\t\t\tconst modifiedFinish: LanguageModelV3StreamPart = {\n\t\t\t\t\t\t\ttype: \"finish\",\n\t\t\t\t\t\t\tfinishReason: modifiedResponse.finishReason,\n\t\t\t\t\t\t\tusage: modifiedResponse.usage,\n\t\t\t\t\t\t\tproviderMetadata: modifiedResponse.providerMetadata,\n\t\t\t\t\t\t};\n\n\t\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${JSON.stringify(modifiedFinish)}\\n\\n`));\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Pass through event unchanged\n\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${data}\\n\\n`));\n\t\t\t\t}\n\n\t\t\t\tcontroller.close();\n\t\t\t} catch (error) {\n\t\t\t\tcontroller.error(error);\n\t\t\t} finally {\n\t\t\t\treader.releaseLock();\n\t\t\t}\n\t\t},\n\t});\n}\n"],"names":[],"mappings":";;AAAO,KAAA,gBAAA;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACO,iBAAA,mBAAA,IAAA,OAAA,CAAA,gBAAA;;ACPP;AACA;AACA;AACA;AACO,KAAA,eAAA;AACP,aAAA,sBAAA;AACA,kBAAA,2BAAA;AACA,WAAA,oBAAA;AACA,cAAA,eAAA;AACA,uBAAA,wBAAA;AACA;AACA;AACA;AACA;AACO,KAAA,YAAA;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,KAAA,yBAAA;AACP;AACA;AACA;AACA,cAAA,MAAA;AACA;AACA;AACA;AACA;AACA;AACA,iBAAA,0BAAA;AACA,UAAA,0BAAA,GAAA,OAAA,CAAA,0BAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,iBAAA,0BAAA;AACA,kBAAA,eAAA;AACA,UAAA,eAAA,GAAA,OAAA,CAAA,eAAA;AACA;AACA;AACA;AACA;AACA;AACA,iBAAA,0BAAA;AACA,eAAA,YAAA;AACA;AACA,UAAA,YAAA,GAAA,QAAA,GAAA,OAAA,CAAA,YAAA,GAAA,QAAA;AACA;AACA;AACA;AACA;AACO,KAAA,oBAAA,aAAA,OAAA;AACP,YAAA,OAAA;AACA;AACA;AACA,MAAA,OAAA,CAAA,QAAA;AACA;AACA;AACA;AACO,KAAA,wBAAA,GAAA,oBAAA;AACP,SAAA,oBAAA;AACA,UAAA,oBAAA;AACA,SAAA,oBAAA;AACA,YAAA,oBAAA;AACA,WAAA,oBAAA;AACA;;AC3EA;AACA;AACA;AACA;AACA;AACA;AACO,cAAA,kBAAA,kEAAA,yBAAA,KAAA,wBAAA;;ACLP;AACA;AACA;AACA;AACA;AACA;AACA;AACO,cAAA,uBAAA;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,kBAAA,yBAAA;AACA;AACA;AACA;AACA,mBAAA,eAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,iBAAA,uBAAA,OAAA,cAAA,CAAA,UAAA,gBAAA,0BAAA,iBAAA,WAAA,CAAA,yBAAA,qBAAA,cAAA,CAAA,UAAA;;;;"}
+{"version":3,"file":"index.d.ts","sources":["../src/auth.ts","../src/types.ts","../src/proxy.ts","../src/stream.ts"],"sourcesContent":["import { getVercelOidcToken } from \"@vercel/oidc\";\n\nexport type GatewayAuthToken = {\n\ttoken: string;\n\tauthMethod: \"api-key\" | \"oidc\";\n};\n\n/**\n * Get authentication token for the AI Gateway.\n * Tries API key first (AI_GATEWAY_API_KEY env var), then falls back to Vercel OIDC.\n */\nexport async function getGatewayAuthToken(): Promise<GatewayAuthToken | null> {\n\tconst apiKey = process.env.AI_GATEWAY_API_KEY;\n\tif (apiKey) {\n\t\treturn { token: apiKey, authMethod: \"api-key\" };\n\t}\n\n\ttry {\n\t\tconst oidcToken = await getVercelOidcToken();\n\t\treturn { token: oidcToken, authMethod: \"oidc\" };\n\t} catch {\n\t\treturn null;\n\t}\n}\n","import type {\n\tLanguageModelV3CallOptions,\n\tLanguageModelV3Content,\n\tLanguageModelV3FinishReason,\n\tLanguageModelV3Usage,\n\tSharedV3ProviderMetadata,\n\tSharedV3Warning,\n} from \"@ai-sdk/provider\";\n\n/**\n * The response body from the AI Gateway for non-streaming requests.\n * For streaming requests, this is aggregated from the stream events.\n */\nexport type GatewayResponse = {\n\tcontent: LanguageModelV3Content[];\n\tfinishReason: LanguageModelV3FinishReason;\n\tusage: LanguageModelV3Usage;\n\twarnings: SharedV3Warning[];\n\tproviderMetadata?: SharedV3ProviderMetadata;\n};\n\n/**\n * Error response from the AI Gateway.\n */\nexport type GatewayError = {\n\terror: {\n\t\tmessage: string;\n\t\ttype: string;\n\t\tparam?: unknown;\n\t};\n};\n\n/**\n * Options for creating a gateway proxy.\n */\nexport type CreateGatewayProxyOptions = {\n\t/** Base URL for the AI Gateway. Defaults to Vercel's gateway. */\n\tbaseUrl?: string;\n\t/** Additional headers to send with requests. */\n\theaders?: Record<string, string>;\n\t/**\n\t * The name of the catch-all parameter in the route.\n\t * @default \"segments\"\n\t * @example\n\t * // For Next.js route `app/api/ai/[...path]/route.ts`\n\t * createGatewayProxy({ segmentsParam: \"path\" })\n\t */\n\tsegmentsParam?: string;\n\t/**\n\t * Custom function to extract the path from the request.\n\t * Use this for frameworks without catch-all route support (e.g., Hono, Express).\n\t *\n\t * When provided, `segmentsParam` is ignored.\n\t *\n\t * @example\n\t * // Hono with a base path\n\t * createGatewayProxy({\n\t *   extractPath: (request) => {\n\t *     const url = new URL(request.url);\n\t *     return url.pathname.replace(/^\\/api\\/ai\\//, \"\");\n\t *   }\n\t * })\n\t */\n\textractPath?: (request: Request) => string | Promise<string>;\n\t/**\n\t * Called before the request is sent to the gateway.\n\t * Allows modification of the request body.\n\t */\n\tbeforeRequest?: (context: {\n\t\trequest: LanguageModelV3CallOptions;\n\t}) => LanguageModelV3CallOptions | Promise<LanguageModelV3CallOptions>;\n\t/**\n\t * Called after a successful response is received from the gateway.\n\t * Allows modification of the response body.\n\t *\n\t * For streaming responses, this is called when the stream completes with\n\t * the aggregated response. The content has already been streamed, but you\n\t * can modify the finish event metadata (usage, finishReason, providerMetadata).\n\t */\n\tafterResponse?: (context: {\n\t\trequest: LanguageModelV3CallOptions;\n\t\tresponse: GatewayResponse;\n\t}) => GatewayResponse | Promise<GatewayResponse>;\n\t/**\n\t * Called when the gateway returns an error response.\n\t * Allows custom error handling or modification.\n\t */\n\tonError?: (context: {\n\t\trequest: LanguageModelV3CallOptions;\n\t\terror: GatewayError;\n\t\tstatus: number;\n\t}) => GatewayError | Response | Promise<GatewayError | Response>;\n};\n\n/**\n * Route handler function signature for the gateway proxy.\n * Compatible with Next.js App Router and other frameworks.\n */\nexport type CreateGatewayProxyFn = (\n\trequest: Request,\n\tcontext?: { params: Promise<Record<string, string | string[] | undefined>> },\n) => Promise<Response>;\n\n/**\n * Result of createGatewayProxy - a handler function with HTTP method exports.\n */\nexport type CreateGatewayProxyResult = CreateGatewayProxyFn & {\n\tGET: CreateGatewayProxyFn;\n\tPOST: CreateGatewayProxyFn;\n\tPUT: CreateGatewayProxyFn;\n\tDELETE: CreateGatewayProxyFn;\n\tPATCH: CreateGatewayProxyFn;\n};\n","import type { LanguageModelV3CallOptions } from \"@ai-sdk/provider\";\nimport { type GatewayAuthToken, getGatewayAuthToken } from \"./auth\";\nimport { createStreamTransformer } from \"./stream\";\nimport type {\n\tCreateGatewayProxyOptions,\n\tCreateGatewayProxyResult,\n\tGatewayError,\n\tGatewayResponse,\n} from \"./types\";\n\nconst DEFAULT_BASE_URL = \"https://ai-gateway.vercel.sh/v1/ai\";\nconst PROTOCOL_VERSION = \"0.0.1\";\n\n/**\n * Extracts the path from the request using either a custom extractor or route params.\n */\nasync function extractPathFromRequest(\n\trequest: Request,\n\tcontext: { params: Promise<Record<string, string | string[] | undefined>> } | undefined,\n\toptions: { extractPath?: CreateGatewayProxyOptions[\"extractPath\"]; segmentsParam: string },\n): Promise<string> {\n\tif (options.extractPath) {\n\t\treturn options.extractPath(request);\n\t}\n\n\tif (context?.params) {\n\t\tconst params = await context.params;\n\t\tconst segments = params[options.segmentsParam];\n\t\treturn Array.isArray(segments) ? segments.join(\"/\") : (segments ?? \"\");\n\t}\n\n\treturn \"\";\n}\n\n/**\n * Builds the upstream URL from base URL, path, and query params.\n */\nfunction buildUpstreamUrl(baseUrl: string, path: string, requestUrl: string): string {\n\tconst searchParams = new URL(requestUrl).searchParams;\n\tconst queryString = searchParams.toString();\n\treturn `${baseUrl}/${path}${queryString ? `?${queryString}` : \"\"}`;\n}\n\n/**\n * Builds the headers for the upstream request.\n */\nfunction buildRequestHeaders(\n\tauth: GatewayAuthToken,\n\tisStreaming: boolean,\n\tmodelId: string | null,\n\tcustomHeaders?: Record<string, string>,\n): Record<string, string> {\n\treturn {\n\t\t\"content-type\": \"application/json\",\n\t\t...customHeaders,\n\t\tauthorization: `Bearer ${auth.token}`,\n\t\t\"ai-gateway-auth-method\": auth.authMethod,\n\t\t\"ai-gateway-protocol-version\": PROTOCOL_VERSION,\n\t\t\"ai-language-model-id\": modelId ?? \"\",\n\t\t\"ai-language-model-streaming\": isStreaming ? \"true\" : \"false\",\n\t};\n}\n\n/**\n * Parses the error response from the gateway.\n */\nasync function parseErrorBody(res: Response): Promise<GatewayError> {\n\ttry {\n\t\treturn await res.json();\n\t} catch {\n\t\treturn {\n\t\t\terror: {\n\t\t\t\tmessage: res.statusText || \"Unknown error\",\n\t\t\t\ttype: \"gateway_error\",\n\t\t\t},\n\t\t};\n\t}\n}\n\n/**\n * Creates an error response, optionally applying the onError hook.\n */\nasync function createErrorResponse(\n\tres: Response,\n\terrorBody: GatewayError,\n\trequestBody: LanguageModelV3CallOptions,\n\tonError: CreateGatewayProxyOptions[\"onError\"],\n): Promise<Response> {\n\tif (onError) {\n\t\tconst result = await onError({\n\t\t\trequest: requestBody,\n\t\t\terror: errorBody,\n\t\t\tstatus: res.status,\n\t\t});\n\n\t\tif (result instanceof Response) {\n\t\t\treturn result;\n\t\t}\n\n\t\treturn new Response(JSON.stringify(result), {\n\t\t\tstatus: res.status,\n\t\t\tstatusText: res.statusText,\n\t\t\theaders: { \"content-type\": \"application/json\" },\n\t\t});\n\t}\n\n\treturn new Response(JSON.stringify(errorBody), {\n\t\tstatus: res.status,\n\t\tstatusText: res.statusText,\n\t\theaders: { \"content-type\": \"application/json\" },\n\t});\n}\n\n/**\n * Handles a streaming response from the gateway.\n */\nfunction handleStreamingResponse(\n\tres: Response,\n\trequestBody: LanguageModelV3CallOptions,\n\tafterResponse: CreateGatewayProxyOptions[\"afterResponse\"],\n): Response {\n\tif (!res.body) {\n\t\treturn new Response(\"No response body\", { status: 500 });\n\t}\n\n\tconst contentType = res.headers.get(\"content-type\") || \"text/event-stream\";\n\n\t// Pass through directly if no afterResponse hook\n\tif (!afterResponse) {\n\t\treturn new Response(res.body, {\n\t\t\tstatus: res.status,\n\t\t\tstatusText: res.statusText,\n\t\t\theaders: { \"content-type\": contentType },\n\t\t});\n\t}\n\n\t// Transform stream to call afterResponse on finish\n\tconst transformedStream = createStreamTransformer(res.body, requestBody, afterResponse);\n\n\treturn new Response(transformedStream, {\n\t\tstatus: res.status,\n\t\tstatusText: res.statusText,\n\t\theaders: { \"content-type\": contentType },\n\t});\n}\n\n/**\n * Handles a non-streaming JSON response from the gateway.\n */\nasync function handleJsonResponse(\n\tres: Response,\n\trequestBody: LanguageModelV3CallOptions,\n\tafterResponse: CreateGatewayProxyOptions[\"afterResponse\"],\n): Promise<Response> {\n\tlet responseBody: GatewayResponse = await res.json();\n\n\tif (afterResponse) {\n\t\tresponseBody = await afterResponse({\n\t\t\trequest: requestBody,\n\t\t\tresponse: responseBody,\n\t\t});\n\t}\n\n\treturn new Response(JSON.stringify(responseBody), {\n\t\tstatus: res.status,\n\t\tstatusText: res.statusText,\n\t\theaders: { \"content-type\": \"application/json\" },\n\t});\n}\n\n/**\n * Creates a proxy handler for the Vercel AI Gateway.\n *\n * Uses AI SDK V3 Language Model.\n * @see https://github.com/vercel/ai/blob/main/packages/gateway/src/gateway-language-model.ts\n */\nexport const createGatewayProxy = ({\n\tbaseUrl = DEFAULT_BASE_URL,\n\theaders,\n\tsegmentsParam = \"segments\",\n\textractPath,\n\tbeforeRequest,\n\tafterResponse,\n\tonError,\n}: CreateGatewayProxyOptions = {}): CreateGatewayProxyResult => {\n\tconst handler = async (\n\t\trequest: Request,\n\t\tcontext?: { params: Promise<Record<string, string | string[] | undefined>> },\n\t) => {\n\t\t// Authenticate with the gateway\n\t\tconst auth = await getGatewayAuthToken();\n\t\tif (!auth) {\n\t\t\treturn new Response(\"AI Gateway authentication not configured\", { status: 500 });\n\t\t}\n\n\t\t// Build upstream URL\n\t\tconst path = await extractPathFromRequest(request, context, { extractPath, segmentsParam });\n\t\tconst upstream = buildUpstreamUrl(baseUrl, path, request.url);\n\n\t\t// Parse request body\n\t\tlet requestBody: LanguageModelV3CallOptions;\n\t\ttry {\n\t\t\trequestBody = await request.clone().json();\n\t\t} catch {\n\t\t\treturn new Response(\"Invalid JSON body\", { status: 400 });\n\t\t}\n\n\t\t// Apply beforeRequest hook\n\t\tif (beforeRequest) {\n\t\t\trequestBody = await beforeRequest({ request: requestBody });\n\t\t}\n\n\t\tconst isStreaming = request.headers.get(\"ai-language-model-streaming\") === \"true\";\n\n\t\ttry {\n\t\t\tconst res = await fetch(upstream, {\n\t\t\t\tmethod: request.method,\n\t\t\t\tbody: JSON.stringify(requestBody),\n\t\t\t\tduplex: \"half\",\n\t\t\t\theaders: buildRequestHeaders(\n\t\t\t\t\tauth,\n\t\t\t\t\tisStreaming,\n\t\t\t\t\trequest.headers.get(\"ai-language-model-id\"),\n\t\t\t\t\theaders,\n\t\t\t\t),\n\t\t\t} as RequestInit);\n\n\t\t\t// Handle error responses\n\t\t\tif (!res.ok) {\n\t\t\t\tconst errorBody = await parseErrorBody(res);\n\t\t\t\treturn createErrorResponse(res, errorBody, requestBody, onError);\n\t\t\t}\n\n\t\t\t// Handle successful responses\n\t\t\tif (isStreaming) {\n\t\t\t\treturn handleStreamingResponse(res, requestBody, afterResponse);\n\t\t\t}\n\n\t\t\treturn handleJsonResponse(res, requestBody, afterResponse);\n\t\t} catch {\n\t\t\treturn new Response(\"Error proxying request to AI Gateway\", { status: 500 });\n\t\t}\n\t};\n\n\t// Export handler for all HTTP methods\n\thandler.GET = handler;\n\thandler.POST = handler;\n\thandler.PUT = handler;\n\thandler.DELETE = handler;\n\thandler.PATCH = handler;\n\n\treturn handler;\n};\n","import type {\n\tLanguageModelV3CallOptions,\n\tLanguageModelV3Content,\n\tLanguageModelV3FinishReason,\n\tLanguageModelV3StreamPart,\n\tLanguageModelV3Usage,\n\tSharedV3ProviderMetadata,\n\tSharedV3Warning,\n} from \"@ai-sdk/provider\";\nimport { EventSourceParserStream } from \"@ai-sdk/provider-utils\";\nimport type { CreateGatewayProxyOptions, GatewayResponse } from \"./types\";\n\n/**\n * Aggregator for collecting stream content.\n * Tracks text deltas, reasoning, tool calls, and other content to build the final GatewayResponse.\n *\n * Uses AI SDK V3 protocol types.\n * @see https://github.com/vercel/ai/blob/main/packages/provider/src/language-model/v3/language-model-v3-stream-part.ts\n */\nexport class StreamContentAggregator {\n\tprivate content: LanguageModelV3Content[] = [];\n\tprivate warnings: SharedV3Warning[] = [];\n\n\t// Text content being built (keyed by id)\n\tprivate textContent = new Map<string, { type: \"text\"; text: string }>();\n\n\t// Reasoning content being built (keyed by id)\n\tprivate reasoningContent = new Map<string, { type: \"reasoning\"; text: string }>();\n\n\tprivate finishReason: LanguageModelV3FinishReason = {\n\t\tunified: \"stop\",\n\t\traw: undefined,\n\t};\n\n\tprivate usage: LanguageModelV3Usage = {\n\t\tinputTokens: {\n\t\t\ttotal: undefined,\n\t\t\tnoCache: undefined,\n\t\t\tcacheRead: undefined,\n\t\t\tcacheWrite: undefined,\n\t\t},\n\t\toutputTokens: {\n\t\t\ttotal: undefined,\n\t\t\ttext: undefined,\n\t\t\treasoning: undefined,\n\t\t},\n\t};\n\n\tprivate providerMetadata: SharedV3ProviderMetadata | undefined;\n\n\t/**\n\t * Process a stream part and update aggregated state.\n\t * Uses LanguageModelV3StreamPart type from @ai-sdk/provider.\n\t *\n\t * @see https://github.com/vercel/ai/blob/main/packages/provider/src/language-model/v3/language-model-v3-stream-part.ts\n\t */\n\tprocess(part: LanguageModelV3StreamPart): void {\n\t\tswitch (part.type) {\n\t\t\tcase \"stream-start\":\n\t\t\t\tthis.warnings.push(...part.warnings);\n\t\t\t\tbreak;\n\n\t\t\tcase \"text-start\":\n\t\t\t\tthis.textContent.set(part.id, { type: \"text\", text: \"\" });\n\t\t\t\tbreak;\n\n\t\t\tcase \"text-delta\":\n\t\t\t\t{\n\t\t\t\t\tconst text = this.textContent.get(part.id);\n\t\t\t\t\tif (text) {\n\t\t\t\t\t\ttext.text += part.delta;\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"text-end\":\n\t\t\t\t{\n\t\t\t\t\tconst text = this.textContent.get(part.id);\n\t\t\t\t\tif (text) {\n\t\t\t\t\t\tthis.content.push(text);\n\t\t\t\t\t\tthis.textContent.delete(part.id);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"reasoning-start\":\n\t\t\t\tthis.reasoningContent.set(part.id, { type: \"reasoning\", text: \"\" });\n\t\t\t\tbreak;\n\n\t\t\tcase \"reasoning-delta\":\n\t\t\t\t{\n\t\t\t\t\tconst reasoning = this.reasoningContent.get(part.id);\n\t\t\t\t\tif (reasoning) {\n\t\t\t\t\t\treasoning.text += part.delta;\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"reasoning-end\":\n\t\t\t\t{\n\t\t\t\t\tconst reasoning = this.reasoningContent.get(part.id);\n\t\t\t\t\tif (reasoning) {\n\t\t\t\t\t\tthis.content.push(reasoning);\n\t\t\t\t\t\tthis.reasoningContent.delete(part.id);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-input-start\":\n\t\t\tcase \"tool-input-delta\":\n\t\t\tcase \"tool-input-end\":\n\t\t\t\t// Tool input is streamed incrementally but we wait for the complete tool-call event\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-call\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-result\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"tool-approval-request\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"file\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"source\":\n\t\t\t\tthis.content.push(part);\n\t\t\t\tbreak;\n\n\t\t\tcase \"finish\":\n\t\t\t\t// Finalize any remaining content\n\t\t\t\tfor (const [, text] of this.textContent) {\n\t\t\t\t\tthis.content.push(text);\n\t\t\t\t}\n\t\t\t\tthis.textContent.clear();\n\n\t\t\t\tfor (const [, reasoning] of this.reasoningContent) {\n\t\t\t\t\tthis.content.push(reasoning);\n\t\t\t\t}\n\t\t\t\tthis.reasoningContent.clear();\n\n\t\t\t\tthis.finishReason = part.finishReason;\n\t\t\t\tthis.usage = part.usage;\n\t\t\t\tthis.providerMetadata = part.providerMetadata;\n\t\t\t\tbreak;\n\t\t}\n\t}\n\n\t/**\n\t * Get the aggregated response.\n\t */\n\tgetResponse(): GatewayResponse {\n\t\treturn {\n\t\t\tcontent: this.content,\n\t\t\tfinishReason: this.finishReason,\n\t\t\tusage: this.usage,\n\t\t\twarnings: this.warnings,\n\t\t\tproviderMetadata: this.providerMetadata,\n\t\t};\n\t}\n}\n\n/**\n * Creates a transform stream that:\n * 1. Passes through all SSE events immediately\n * 2. Aggregates content using StreamContentAggregator\n * 3. Calls afterResponse when finish event is received\n * 4. Modifies the finish event based on afterResponse result\n *\n * Uses EventSourceParserStream from @ai-sdk/provider-utils for SSE parsing.\n * EventSourceParserStream is re-exported from the eventsource-parser package.\n * @see https://github.com/vercel/ai/blob/main/packages/provider-utils/src/parse-json-event-stream.ts\n * @see https://github.com/rexxars/eventsource-parser\n */\nexport function createStreamTransformer(\n\tbody: ReadableStream<Uint8Array>,\n\trequestBody: LanguageModelV3CallOptions,\n\tafterResponse: NonNullable<CreateGatewayProxyOptions[\"afterResponse\"]>,\n): ReadableStream<Uint8Array> {\n\tconst encoder = new TextEncoder();\n\tconst aggregator = new StreamContentAggregator();\n\n\t// Use EventSourceParserStream for SSE parsing (re-exported from eventsource-parser via @ai-sdk/provider-utils)\n\tconst eventStream = body\n\t\t.pipeThrough(new TextDecoderStream() as TransformStream<Uint8Array, string>)\n\t\t.pipeThrough(new EventSourceParserStream());\n\n\treturn new ReadableStream({\n\t\tasync start(controller) {\n\t\t\tconst reader = eventStream.getReader();\n\n\t\t\ttry {\n\t\t\t\twhile (true) {\n\t\t\t\t\tconst { done, value } = await reader.read();\n\t\t\t\t\tif (done) break;\n\n\t\t\t\t\tconst { data } = value;\n\n\t\t\t\t\t// Handle DONE marker (end of stream)\n\t\t\t\t\tif (data === \"DONE\" || data === \"[DONE]\") {\n\t\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${data}\\n\\n`));\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Parse the JSON event data\n\t\t\t\t\tlet parsed: LanguageModelV3StreamPart;\n\t\t\t\t\ttry {\n\t\t\t\t\t\tparsed = JSON.parse(data);\n\t\t\t\t\t} catch {\n\t\t\t\t\t\t// Not valid JSON, pass through as-is\n\t\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${data}\\n\\n`));\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Track content for aggregation\n\t\t\t\t\taggregator.process(parsed);\n\n\t\t\t\t\t// Handle finish event specially - call afterResponse hook\n\t\t\t\t\tif (parsed.type === \"finish\") {\n\t\t\t\t\t\tconst modifiedResponse = await afterResponse({\n\t\t\t\t\t\t\trequest: requestBody,\n\t\t\t\t\t\t\tresponse: aggregator.getResponse(),\n\t\t\t\t\t\t});\n\n\t\t\t\t\t\t// Emit modified finish event\n\t\t\t\t\t\tconst modifiedFinish: LanguageModelV3StreamPart = {\n\t\t\t\t\t\t\ttype: \"finish\",\n\t\t\t\t\t\t\tfinishReason: modifiedResponse.finishReason,\n\t\t\t\t\t\t\tusage: modifiedResponse.usage,\n\t\t\t\t\t\t\tproviderMetadata: modifiedResponse.providerMetadata,\n\t\t\t\t\t\t};\n\n\t\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${JSON.stringify(modifiedFinish)}\\n\\n`));\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Pass through event unchanged\n\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${data}\\n\\n`));\n\t\t\t\t}\n\n\t\t\t\tcontroller.close();\n\t\t\t} catch (error) {\n\t\t\t\tcontroller.error(error);\n\t\t\t} finally {\n\t\t\t\treader.releaseLock();\n\t\t\t}\n\t\t},\n\t});\n}\n"],"names":[],"mappings":";;AAAO,KAAA,gBAAA;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACO,iBAAA,mBAAA,IAAA,OAAA,CAAA,gBAAA;;ACPP;AACA;AACA;AACA;AACO,KAAA,eAAA;AACP,aAAA,sBAAA;AACA,kBAAA,2BAAA;AACA,WAAA,oBAAA;AACA,cAAA,eAAA;AACA,uBAAA,wBAAA;AACA;AACA;AACA;AACA;AACO,KAAA,YAAA;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,KAAA,yBAAA;AACP;AACA;AACA;AACA,cAAA,MAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,4BAAA,OAAA,cAAA,OAAA;AACA;AACA;AACA;AACA;AACA;AACA,iBAAA,0BAAA;AACA,UAAA,0BAAA,GAAA,OAAA,CAAA,0BAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,iBAAA,0BAAA;AACA,kBAAA,eAAA;AACA,UAAA,eAAA,GAAA,OAAA,CAAA,eAAA;AACA;AACA;AACA;AACA;AACA;AACA,iBAAA,0BAAA;AACA,eAAA,YAAA;AACA;AACA,UAAA,YAAA,GAAA,QAAA,GAAA,OAAA,CAAA,YAAA,GAAA,QAAA;AACA;AACA;AACA;AACA;AACA;AACO,KAAA,oBAAA,aAAA,OAAA;AACP,YAAA,OAAA,CAAA,MAAA;AACA,MAAA,OAAA,CAAA,QAAA;AACA;AACA;AACA;AACO,KAAA,wBAAA,GAAA,oBAAA;AACP,SAAA,oBAAA;AACA,UAAA,oBAAA;AACA,SAAA,oBAAA;AACA,YAAA,oBAAA;AACA,WAAA,oBAAA;AACA;;AClGA;AACA;AACA;AACA;AACA;AACA;AACO,cAAA,kBAAA,8FAAA,yBAAA,KAAA,wBAAA;;ACLP;AACA;AACA;AACA;AACA;AACA;AACA;AACO,cAAA,uBAAA;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,kBAAA,yBAAA;AACA;AACA;AACA;AACA,mBAAA,eAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,iBAAA,uBAAA,OAAA,cAAA,CAAA,UAAA,gBAAA,0BAAA,iBAAA,WAAA,CAAA,yBAAA,qBAAA,cAAA,CAAA,UAAA;;;;"}

package/dist/index.mjs

@@ -233,24 +233,159 @@ import { EventSourceParserStream } from '@ai-sdk/provider-utils';
     });
 }
 
+const DEFAULT_BASE_URL = "https://ai-gateway.vercel.sh/v1/ai";
+const PROTOCOL_VERSION = "0.0.1";
+/**
+ * Extracts the path from the request using either a custom extractor or route params.
+ */ function extractPathFromRequest(request, context, options) {
+    return _(function*() {
+        if (options.extractPath) {
+            return options.extractPath(request);
+        }
+        if (context == null ? void 0 : context.params) {
+            const params = yield context.params;
+            const segments = params[options.segmentsParam];
+            return Array.isArray(segments) ? segments.join("/") : segments != null ? segments : "";
+        }
+        return "";
+    })();
+}
+/**
+ * Builds the upstream URL from base URL, path, and query params.
+ */ function buildUpstreamUrl(baseUrl, path, requestUrl) {
+    const searchParams = new URL(requestUrl).searchParams;
+    const queryString = searchParams.toString();
+    return `${baseUrl}/${path}${queryString ? `?${queryString}` : ""}`;
+}
+/**
+ * Builds the headers for the upstream request.
+ */ function buildRequestHeaders(auth, isStreaming, modelId, customHeaders) {
+    return _$1({
+        "content-type": "application/json"
+    }, customHeaders, {
+        authorization: `Bearer ${auth.token}`,
+        "ai-gateway-auth-method": auth.authMethod,
+        "ai-gateway-protocol-version": PROTOCOL_VERSION,
+        "ai-language-model-id": modelId != null ? modelId : "",
+        "ai-language-model-streaming": isStreaming ? "true" : "false"
+    });
+}
+/**
+ * Parses the error response from the gateway.
+ */ function parseErrorBody(res) {
+    return _(function*() {
+        try {
+            return yield res.json();
+        } catch (unused) {
+            return {
+                error: {
+                    message: res.statusText || "Unknown error",
+                    type: "gateway_error"
+                }
+            };
+        }
+    })();
+}
+/**
+ * Creates an error response, optionally applying the onError hook.
+ */ function createErrorResponse(res, errorBody, requestBody, onError) {
+    return _(function*() {
+        if (onError) {
+            const result = yield onError({
+                request: requestBody,
+                error: errorBody,
+                status: res.status
+            });
+            if (result instanceof Response) {
+                return result;
+            }
+            return new Response(JSON.stringify(result), {
+                status: res.status,
+                statusText: res.statusText,
+                headers: {
+                    "content-type": "application/json"
+                }
+            });
+        }
+        return new Response(JSON.stringify(errorBody), {
+            status: res.status,
+            statusText: res.statusText,
+            headers: {
+                "content-type": "application/json"
+            }
+        });
+    })();
+}
+/**
+ * Handles a streaming response from the gateway.
+ */ function handleStreamingResponse(res, requestBody, afterResponse) {
+    if (!res.body) {
+        return new Response("No response body", {
+            status: 500
+        });
+    }
+    const contentType = res.headers.get("content-type") || "text/event-stream";
+    // Pass through directly if no afterResponse hook
+    if (!afterResponse) {
+        return new Response(res.body, {
+            status: res.status,
+            statusText: res.statusText,
+            headers: {
+                "content-type": contentType
+            }
+        });
+    }
+    // Transform stream to call afterResponse on finish
+    const transformedStream = createStreamTransformer(res.body, requestBody, afterResponse);
+    return new Response(transformedStream, {
+        status: res.status,
+        statusText: res.statusText,
+        headers: {
+            "content-type": contentType
+        }
+    });
+}
+/**
+ * Handles a non-streaming JSON response from the gateway.
+ */ function handleJsonResponse(res, requestBody, afterResponse) {
+    return _(function*() {
+        let responseBody = yield res.json();
+        if (afterResponse) {
+            responseBody = yield afterResponse({
+                request: requestBody,
+                response: responseBody
+            });
+        }
+        return new Response(JSON.stringify(responseBody), {
+            status: res.status,
+            statusText: res.statusText,
+            headers: {
+                "content-type": "application/json"
+            }
+        });
+    })();
+}
 /**
  * Creates a proxy handler for the Vercel AI Gateway.
  *
  * Uses AI SDK V3 Language Model.
  * @see https://github.com/vercel/ai/blob/main/packages/gateway/src/gateway-language-model.ts
- */ const createGatewayProxy = ({ baseUrl = "https://ai-gateway.vercel.sh/v1/ai", headers, beforeRequest, afterResponse, onError } = {})=>{
-    const handler = (request, { params })=>_(function*() {
-            var _ref;
+ */ const createGatewayProxy = ({ baseUrl = DEFAULT_BASE_URL, headers, segmentsParam = "segments", extractPath, beforeRequest, afterResponse, onError } = {})=>{
+    const handler = (request, context)=>_(function*() {
+            // Authenticate with the gateway
             const auth = yield getGatewayAuthToken();
             if (!auth) {
                 return new Response("AI Gateway authentication not configured", {
                     status: 500
                 });
             }
-            const { segments } = yield params;
-            const path = (_ref = segments == null ? void 0 : segments.join("/")) != null ? _ref : "";
-            const searchParams = new URL(request.url).searchParams;
-            const upstream = `${baseUrl}/${path}${searchParams.toString() ? `?${searchParams.toString()}` : ""}`;
+            // Build upstream URL
+            const path = yield extractPathFromRequest(request, context, {
+                extractPath,
+                segmentsParam
+            });
+            const upstream = buildUpstreamUrl(baseUrl, path, request.url);
+            // Parse request body
             let requestBody;
             try {
                 requestBody = yield request.clone().json();
@@ -267,109 +402,29 @@ import { EventSourceParserStream } from '@ai-sdk/provider-utils';
             }
             const isStreaming = request.headers.get("ai-language-model-streaming") === "true";
             try {
-                var _request_headers_get;
                 const res = yield fetch(upstream, {
                     method: request.method,
                     body: JSON.stringify(requestBody),
                     duplex: "half",
-                    headers: _$1({
-                        "content-type": "application/json"
-                    }, headers, {
-                        authorization: `Bearer ${auth.token}`,
-                        "ai-gateway-auth-method": auth.authMethod,
-                        "ai-gateway-protocol-version": "0.0.1",
-                        "ai-language-model-id": (_request_headers_get = request.headers.get("ai-language-model-id")) != null ? _request_headers_get : "",
-                        "ai-language-model-streaming": isStreaming ? "true" : "false"
-                    })
+                    headers: buildRequestHeaders(auth, isStreaming, request.headers.get("ai-language-model-id"), headers)
                 });
                 // Handle error responses
                 if (!res.ok) {
-                    let errorBody;
-                    try {
-                        errorBody = yield res.json();
-                    } catch (unused) {
-                        errorBody = {
-                            error: {
-                                message: res.statusText || "Unknown error",
-                                type: "gateway_error"
-                            }
-                        };
-                    }
-                    if (onError) {
-                        const result = yield onError({
-                            request: requestBody,
-                            error: errorBody,
-                            status: res.status
-                        });
-                        // If onError returns a Response, use it directly
-                        if (result instanceof Response) {
-                            return result;
-                        }
-                        // Otherwise, return the modified error
-                        return new Response(JSON.stringify(result), {
-                            status: res.status,
-                            statusText: res.statusText,
-                            headers: {
-                                "content-type": "application/json"
-                            }
-                        });
-                    }
-                    return new Response(JSON.stringify(errorBody), {
-                        status: res.status,
-                        statusText: res.statusText,
-                        headers: {
-                            "content-type": "application/json"
-                        }
-                    });
+                    const errorBody = yield parseErrorBody(res);
+                    return createErrorResponse(res, errorBody, requestBody, onError);
                 }
-                // For streaming responses
+                // Handle successful responses
                 if (isStreaming) {
-                    if (!res.body) {
-                        return new Response("No response body", {
-                            status: 500
-                        });
-                    }
-                    // If no afterResponse hook, pass through directly
-                    if (!afterResponse) {
-                        return new Response(res.body, {
-                            status: res.status,
-                            statusText: res.statusText,
-                            headers: {
-                                "content-type": res.headers.get("content-type") || "text/event-stream"
-                            }
-                        });
-                    }
-                    // Transform the stream to call afterResponse on finish
-                    const transformedStream = createStreamTransformer(res.body, requestBody, afterResponse);
-                    return new Response(transformedStream, {
-                        status: res.status,
-                        statusText: res.statusText,
-                        headers: {
-                            "content-type": res.headers.get("content-type") || "text/event-stream"
-                        }
-                    });
+                    return handleStreamingResponse(res, requestBody, afterResponse);
                 }
-                // For non-streaming responses, parse and optionally transform
-                let responseBody = yield res.json();
-                if (afterResponse) {
-                    responseBody = yield afterResponse({
-                        request: requestBody,
-                        response: responseBody
-                    });
-                }
-                return new Response(JSON.stringify(responseBody), {
-                    status: res.status,
-                    statusText: res.statusText,
-                    headers: {
-                        "content-type": "application/json"
-                    }
-                });
+                return handleJsonResponse(res, requestBody, afterResponse);
             } catch (unused) {
                 return new Response("Error proxying request to AI Gateway", {
                     status: 500
                 });
             }
         })();
+    // Export handler for all HTTP methods
     handler.GET = handler;
     handler.POST = handler;
     handler.PUT = handler;

package/package.json

@@ -1,9 +1,14 @@
 {
   "name": "ai-gateway-proxy",
-  "version": "0.0.1",
+  "version": "0.1.0",
   "type": "module",
   "sideEffects": false,
   "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/SferaDev/SferaDev.git",
+    "directory": "packages/ai-gateway-proxy"
+  },
   "exports": {
     ".": {
       "types": "./dist/index.d.ts",
@@ -34,7 +39,8 @@
     "@sferadev/tsconfig": "0.0.0"
   },
   "publishConfig": {
-    "access": "public"
+    "access": "public",
+    "provenance": true
   },
   "scripts": {
     "build": "bunchee",