zk-credits 0.2.0 → 0.2.2

package/dist/sidecar.js

@@ -1,5 +1,6 @@
 import { createServer } from 'node:http';
 import { codexModelsResponse } from './codex-profile.js';
+import { responsesEventStream, translateResponsesRequest } from './responses-bridge.js';
 import { PAYMENT_REQUIRED_HEADER, PAYMENT_RESPONSE_HEADER, } from '@zk-credits/x402-zk-prepaid';
 const MAX_REQUEST_BYTES = 2_000_000;
 const PAYMENT_HEADERS = [PAYMENT_REQUIRED_HEADER, PAYMENT_RESPONSE_HEADER, 'cache-control'];
@@ -81,6 +82,32 @@ async function relayGatewayResponse(res, upstream) {
         reader.releaseLock();
     }
 }
+async function relayResponsesGatewayResponse(res, upstream, responseModel) {
+    if (upstream.status < 200 || upstream.status >= 300) {
+        await relayGatewayResponse(res, upstream);
+        return;
+    }
+    const upstreamText = await upstream.text();
+    let completion;
+    try {
+        completion = JSON.parse(upstreamText);
+    }
+    catch {
+        throw new Error('Gateway returned an invalid chat completion');
+    }
+    const eventStream = responsesEventStream(completion, responseModel);
+    const headers = {
+        'Content-Type': 'text/event-stream; charset=utf-8',
+        'Cache-Control': 'no-cache',
+    };
+    for (const name of PAYMENT_HEADERS) {
+        const value = upstream.headers.get(name);
+        if (value)
+            headers[name] = value;
+    }
+    res.writeHead(upstream.status, headers);
+    res.end(eventStream);
+}
 /** Creates a loopback-only proxy whose sole spend path is x402 zk-prepaid. */
 export function createSidecarServer(options) {
     const gatewayBaseUrl = options.gatewayBaseUrl.replace(/\/$/u, '');
@@ -117,7 +144,7 @@ export function createSidecarServer(options) {
             sendJson(res, 200, options.metrics());
             return;
         }
-        if (req.method !== 'POST' || pathname !== '/v1/chat/completions') {
+        if (req.method !== 'POST' || (pathname !== '/v1/chat/completions' && pathname !== '/v1/responses')) {
             sendJson(res, 404, { error: 'unsupported_openai_path' });
             return;
         }
@@ -127,17 +154,34 @@ export function createSidecarServer(options) {
         }
         try {
             const body = await readJsonBody(req);
-            const rejection = completionRejection(body.parsed);
-            if (rejection) {
-                sendJson(res, 400, { error: rejection });
-                return;
+            const isResponses = pathname === '/v1/responses';
+            let outboundBody = body.raw;
+            let responseModel = '';
+            if (isResponses) {
+                const translation = translateResponsesRequest(body.parsed);
+                if (!translation.ok) {
+                    sendJson(res, 400, { error: translation.error });
+                    return;
+                }
+                outboundBody = JSON.stringify(translation.chatRequest);
+                responseModel = translation.responseModel;
+            }
+            else {
+                const rejection = completionRejection(body.parsed);
+                if (rejection) {
+                    sendJson(res, 400, { error: rejection });
+                    return;
+                }
             }
-            const upstream = await options.prepaidClient.fetch(`${gatewayBaseUrl}${pathname}`, {
+            const upstream = await options.prepaidClient.fetch(`${gatewayBaseUrl}/v1/chat/completions`, {
                 method: 'POST',
                 headers: { 'Content-Type': 'application/json' },
-                body: body.raw,
+                body: outboundBody,
             });
-            await relayGatewayResponse(res, upstream);
+            if (isResponses)
+                await relayResponsesGatewayResponse(res, upstream, responseModel);
+            else
+                await relayGatewayResponse(res, upstream);
         }
         catch (error) {
             if (!res.headersSent) {