@fractaal/pi-ai 0.86.9 → 1.6.0

package/dist/api/openai-codex-responses.js

@@ -5,8 +5,8 @@ import { appendAssistantMessageDiagnostic, createAssistantMessageDiagnostic, for
 import { formatProviderError, normalizeProviderError } from "../utils/error-body.js";
 import { AssistantMessageEventStream } from "../utils/event-stream.js";
 import { headersToRecord } from "../utils/headers.js";
+import { getIdentityHeaders } from "../utils/identity.js";
 import { resolveHttpProxyUrlForTarget } from "../utils/node-http-proxy.js";
-import { getPiUserAgent } from "../utils/pi-user-agent.js";
 import { isRetryableAssistantError, retryAssistantCall } from "../utils/retry.js";
 import { getSystemMessageText } from "../utils/text.js";
 import { getDeclaredTools, getInitialSystemMessage, normalizeContext, resolveTranscript, resolveTranscriptTools, } from "../utils/transcript.js";
@@ -14,7 +14,7 @@ import { uuidv7 } from "../utils/uuid.js";
 import { createGrammarToolInputProperties } from "./constrained-sampling.js";
 import { retainCodexCompactionInput, trimCodexCompactionToolOutputs } from "./openai-codex-compaction-history.js";
 import { clampOpenAIPromptCacheKey } from "./openai-prompt-cache.js";
-import { convertResponsesMessages, convertResponsesTools, processResponsesStream } from "./openai-responses-shared.js";
+import { convertResponsesMessages, convertResponsesTools, getResponsesToolSupport, processResponsesStream, } from "./openai-responses-shared.js";
 import { buildBaseOptions } from "./simple-options.js";
 // ============================================================================
 // Configuration
@@ -141,11 +141,15 @@ function sleep(ms, signal) {
             reject(new Error("Request was aborted"));
             return;
         }
-        const timeout = setTimeout(resolve, ms);
-        signal?.addEventListener("abort", () => {
+        const onAbort = () => {
             clearTimeout(timeout);
             reject(new Error("Request was aborted"));
-        });
+        };
+        const timeout = setTimeout(() => {
+            signal?.removeEventListener("abort", onAbort);
+            resolve();
+        }, ms);
+        signal?.addEventListener("abort", onAbort, { once: true });
     });
 }
 function normalizeTimeoutMs(value) {
@@ -174,7 +178,8 @@ function compressRequestBodyZstd(bodyJson) {
         const compressed = zlib.zstdCompressSync(bodyJson, {
             params: { [zlib.constants.ZSTD_c_compressionLevel]: REQUEST_COMPRESSION_ZSTD_LEVEL },
         });
-        return new Uint8Array(compressed.buffer, compressed.byteOffset, compressed.byteLength);
+        // A copy owns a plain ArrayBuffer, which fetch accepts as a body under every lib.
+        return new Uint8Array(compressed);
     }
     catch {
         return null;
@@ -223,8 +228,10 @@ export const stream = (model, context, options) => {
                 body = nextBody;
             }
             const websocketRequestId = codexSessionId || uuidv7();
-            const sseHeaders = buildSSEHeaders(model.headers, options?.headers, accountId, apiKey, codexSessionId);
-            const websocketHeaders = buildWebSocketHeaders(model.headers, options?.headers, accountId, apiKey, websocketRequestId);
+            // Codex expects an originator on every request; it is not attribution.
+            const identityHeaders = { originator: "pi", ...getIdentityHeaders(model, options?.identity) };
+            const sseHeaders = buildSSEHeaders(identityHeaders, model.headers, options?.headers, accountId, apiKey, codexSessionId);
+            const websocketHeaders = buildWebSocketHeaders(identityHeaders, model.headers, options?.headers, accountId, apiKey, websocketRequestId);
             const bodyJson = JSON.stringify(body);
             const httpTimeoutMs = normalizeTimeoutMs(options?.timeoutMs);
             const websocketConnectTimeoutMs = normalizeTimeoutMs(options?.websocketConnectTimeoutMs);
@@ -481,7 +488,11 @@ function buildRequestBody(model, context, options, cacheSessionId, grammarToolIn
     const supportsOpenAIGrammarTools = model.compat?.supportsOpenAIGrammarTools ?? false;
     const supportsAdditionalTools = model.compat?.supportsAdditionalTools ?? false;
     const supportsToolSearch = model.compat?.supportsToolSearch ?? false;
-    const transcriptTools = resolveTranscriptTools(context.messages, supportsAdditionalTools || supportsToolSearch);
+    const transcriptTools = resolveTranscriptTools(context.messages, getResponsesToolSupport({
+        supportsMidConvoSystemMessages: model.compat?.supportsMidConvoSystemMessages ?? false,
+        supportsAdditionalTools,
+        supportsToolSearch,
+    }));
     const checkpoint = options?.nativeCompactionCheckpoint;
     const checkpointInput = checkpoint ? [...(checkpoint.retainedInput ?? []), checkpoint.item] : [];
     const requestContext = options?.nativeCompaction
@@ -1422,8 +1433,12 @@ function extractAccountId(token) {
         throw new Error("Failed to extract accountId from token");
     }
 }
-function buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token) {
-    const headers = new Headers(initHeaders);
+function buildBaseCodexHeaders(defaultHeaders, initHeaders, additionalHeaders, accountId, token) {
+    // Defaults first so model and caller headers can override them, matching the other providers.
+    const headers = new Headers(defaultHeaders);
+    for (const [key, value] of Object.entries(initHeaders || {})) {
+        headers.set(key, value);
+    }
     for (const [key, value] of Object.entries(additionalHeaders || {})) {
         if (value === null) {
             headers.delete(key);
@@ -1441,12 +1456,10 @@ function buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token)
         headers.delete("Authorization");
         headers.delete("chatgpt-account-id");
     }
-    headers.set("originator", "pi");
-    headers.set("User-Agent", getPiUserAgent());
     return headers;
 }
-function buildSSEHeaders(initHeaders, additionalHeaders, accountId, token, sessionId) {
-    const headers = buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token);
+function buildSSEHeaders(defaultHeaders, initHeaders, additionalHeaders, accountId, token, sessionId) {
+    const headers = buildBaseCodexHeaders(defaultHeaders, initHeaders, additionalHeaders, accountId, token);
     headers.set("OpenAI-Beta", "responses=experimental");
     headers.set("accept", "text/event-stream");
     headers.set("content-type", "application/json");
@@ -1456,8 +1469,8 @@ function buildSSEHeaders(initHeaders, additionalHeaders, accountId, token, sessi
     }
     return headers;
 }
-function buildWebSocketHeaders(initHeaders, additionalHeaders, accountId, token, requestId) {
-    const headers = buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token);
+function buildWebSocketHeaders(defaultHeaders, initHeaders, additionalHeaders, accountId, token, requestId) {
+    const headers = buildBaseCodexHeaders(defaultHeaders, initHeaders, additionalHeaders, accountId, token);
     headers.delete("accept");
     headers.delete("content-type");
     headers.delete("OpenAI-Beta");