xsai-codex 0.0.1 → 0.0.2

package/README.md

@@ -6,12 +6,15 @@ Requires Baseline 2025 [`Uint8Array.fromBase64`](https://developer.mozilla.org/e
 
 ## Usage
 
+> For more examples, please refer to the [xsAI Docs](https://xsai.js.org/docs/packages-ext/responses).
+
 ```ts
 import { responses } from '@xsai-ext/responses'
 import { authorizeCodexHeadless, createCodex } from 'xsai-codex'
 
 const auth = await authorizeCodexHeadless({
   onUserCode: ({ instructions }) => {
+    // Open https://auth.openai.com/codex/device and enter code: XXXX-XXXXX
     console.log(instructions)
   },
 })

package/dist/index.d.ts

@@ -10,6 +10,7 @@ interface CodexAuthTokens {
 }
 interface CodexHeadlessAuthorizeOptions {
     onUserCode?: (info: CodexUserCodeInfo) => Promise<void> | void;
+    signal?: AbortSignal;
 }
 interface CodexUserCodeInfo {
     instructions: string;

package/dist/index.js

@@ -1,4 +1,4 @@
-var version = "0.0.1";
+var version = "0.0.2";
 
 const CODEX_CLIENT_ID = "app_EMoamEEZ73f0CkXaXp7hrann";
 const CODEX_ISSUER = "https://auth.openai.com";
@@ -9,15 +9,22 @@ const CODEX_DEVICE_URL = `${CODEX_ISSUER}/codex/device`;
 const CODEX_DEVICE_REDIRECT_URL = `${CODEX_ISSUER}/deviceauth/callback`;
 const CODEX_OAUTH_POLLING_SAFETY_MARGIN_MS = 3e3;
 const CODEX_TOKEN_REFRESH_MARGIN_MS = 6e4;
-const CODEX_DUMMY_API_KEY = "codex-oauth";
 const CODEX_DEFAULT_ORIGINATOR = "xsai-codex";
 const CODEX_DEFAULT_USER_AGENT = `xsai-codex/${version}`;
 const CODEX_BASE_URL = "https://chatgpt.com/backend-api/codex/";
 
-const sleep = async (ms) => (
-  // eslint-disable-next-line @masknet/prefer-timer-id
-  new Promise((resolve) => setTimeout(resolve, ms))
-);
+const sleep = async (ms, signal) => {
+  if (signal?.aborted) {
+    throw signal.reason;
+  }
+  return new Promise((resolve, reject) => {
+    const timer = setTimeout(resolve, ms);
+    signal?.addEventListener("abort", () => {
+      clearTimeout(timer);
+      reject(signal.reason);
+    }, { once: true });
+  });
+};
 const formatResponseError = async (message, response) => {
   const body = await response.text().catch(() => "");
   const details = body.trim().slice(0, 500);
@@ -55,12 +62,19 @@ const extractAccountId = (tokens) => {
   const claims = parseJwtClaims(tokens.access_token);
   return claims === void 0 ? void 0 : extractAccountIdFromClaims(claims);
 };
-const toCodexAuthTokens = (tokens, accountId = extractAccountId(tokens)) => ({
-  access: tokens.access_token,
-  expires: Date.now() + (tokens.expires_in ?? 3600) * 1e3,
-  refresh: tokens.refresh_token,
-  ...accountId !== void 0 && accountId.length > 0 && { accountId }
-});
+const toCodexAuthTokens = (tokens, options = {}) => {
+  const accountId = options.accountId ?? extractAccountId(tokens);
+  const refresh = tokens.refresh_token ?? options.refreshToken;
+  if (refresh === void 0 || refresh.length === 0) {
+    throw new Error("Codex token response did not include a refresh token.");
+  }
+  return {
+    access: tokens.access_token,
+    ...accountId !== void 0 && accountId.length > 0 && { accountId },
+    expires: Date.now() + (tokens.expires_in ?? 3600) * 1e3,
+    refresh
+  };
+};
 const refreshCodexAccessToken = async (refreshToken) => {
   const tokens = await jsonFetch(
     CODEX_TOKEN_URL,
@@ -75,7 +89,7 @@ const refreshCodexAccessToken = async (refreshToken) => {
     },
     "Codex token refresh failed"
   );
-  return toCodexAuthTokens(tokens);
+  return toCodexAuthTokens(tokens, { refreshToken });
 };
 const authorizeCodexHeadless = async (options = {}) => {
   const deviceData = await jsonFetch(
@@ -86,7 +100,8 @@ const authorizeCodexHeadless = async (options = {}) => {
         "Content-Type": "application/json",
         "User-Agent": CODEX_DEFAULT_USER_AGENT
       },
-      method: "POST"
+      method: "POST",
+      signal: options.signal
     },
     "Failed to initiate Codex device authorization"
   );
@@ -107,7 +122,8 @@ const authorizeCodexHeadless = async (options = {}) => {
         "Content-Type": "application/json",
         "User-Agent": CODEX_DEFAULT_USER_AGENT
       },
-      method: "POST"
+      method: "POST",
+      signal: options.signal
     });
     if (response.ok) {
       const deviceToken = await response.json();
@@ -122,7 +138,8 @@ const authorizeCodexHeadless = async (options = {}) => {
             redirect_uri: CODEX_DEVICE_REDIRECT_URL
           }).toString(),
           headers: { "Content-Type": "application/x-www-form-urlencoded" },
-          method: "POST"
+          method: "POST",
+          signal: options.signal
         },
         "Codex token exchange failed"
       );
@@ -131,7 +148,7 @@ const authorizeCodexHeadless = async (options = {}) => {
     if (response.status !== 403 && response.status !== 404) {
       throw new Error(await formatResponseError("Codex device authorization failed", response));
     }
-    await sleep(interval + CODEX_OAUTH_POLLING_SAFETY_MARGIN_MS);
+    await sleep(interval + CODEX_OAUTH_POLLING_SAFETY_MARGIN_MS, options.signal);
   }
 };
 
@@ -151,10 +168,9 @@ const createCodex = async (options) => {
   return async (model) => {
     const auth = await resolveAuth();
     return {
-      apiKey: CODEX_DUMMY_API_KEY,
+      apiKey: auth.access,
       baseURL: CODEX_BASE_URL,
       headers: {
-        "Authorization": `Bearer ${auth.access}`,
         ...auth.accountId !== void 0 && auth.accountId.length > 0 && { "ChatGPT-Account-Id": auth.accountId },
         "originator": options.originator ?? CODEX_DEFAULT_ORIGINATOR,
         "User-Agent": options.userAgent ?? CODEX_DEFAULT_USER_AGENT,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "xsai-codex",
   "type": "module",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "OpenAI Codex provider for xsAI.",
   "author": "Moeru AI",
   "license": "MIT",