ampcode-connector 0.1.3 → 0.1.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ampcode-connector",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Proxy AmpCode through local OAuth subscriptions (Claude Code, Codex, Gemini CLI, Antigravity)",
   "license": "MIT",
   "repository": {
@@ -40,7 +40,7 @@
   },
   "devDependencies": {
     "@biomejs/biome": "^2.4.0",
-    "@types/bun": "latest"
+    "@types/bun": "^1.3.9"
   },
   "peerDependencies": {
     "typescript": "^5"

package/src/constants.ts

@@ -7,6 +7,28 @@ export const DEFAULT_ANTIGRAVITY_PROJECT = "rising-fact-p41fc";
 
 export const ANTHROPIC_API_URL = "https://api.anthropic.com";
 export const OPENAI_API_URL = "https://api.openai.com";
+export const CODEX_BASE_URL = "https://chatgpt.com/backend-api";
+
+/** Codex-specific headers required by the ChatGPT backend. */
+export const codexHeaders = {
+  BETA: "OpenAI-Beta",
+  ACCOUNT_ID: "chatgpt-account-id",
+  ORIGINATOR: "originator",
+  SESSION_ID: "session_id",
+  CONVERSATION_ID: "conversation_id",
+} as const;
+
+export const codexHeaderValues = {
+  BETA_RESPONSES: "responses=experimental",
+  ORIGINATOR: "codex_cli_rs",
+  VERSION: "0.101.0",
+  USER_AGENT: `codex_cli_rs/0.101.0 (${process.platform} ${process.arch})`,
+} as const;
+
+/** Map /v1/responses → /codex/responses for the ChatGPT backend. */
+export const codexPathMap: Record<string, string> = {
+  "/v1/responses": "/codex/responses",
+} as const;
 export const DEFAULT_AMP_UPSTREAM_URL = "https://ampcode.com";
 
 export const ANTHROPIC_TOKEN_URL = "https://platform.claude.com/v1/oauth/token";

package/src/providers/base.ts

@@ -36,15 +36,16 @@ export async function forward(opts: ForwardOptions): Promise<Response> {
   });
 
   const contentType = response.headers.get("Content-Type") ?? "application/json";
-  const isSSE = contentType.includes("text/event-stream") || opts.streaming;
-  if (isSSE) return sse.proxy(response, opts.rewrite);
 
   if (!response.ok) {
     const text = await response.text();
-    logger.error(`${opts.providerName} API error`, { error: text.slice(0, 200) });
+    logger.error(`${opts.providerName} API error (${response.status})`, { error: text.slice(0, 200) });
     return new Response(text, { status: response.status, headers: { "Content-Type": contentType } });
   }
 
+  const isSSE = contentType.includes("text/event-stream") || opts.streaming;
+  if (isSSE) return sse.proxy(response, opts.rewrite);
+
   if (opts.rewrite) {
     const text = await response.text();
     return new Response(opts.rewrite(text), { status: response.status, headers: { "Content-Type": contentType } });

package/src/providers/codex.ts

@@ -1,9 +1,13 @@
-/** Forwards requests to api.openai.com with Codex CLI OAuth token. */
+/** Forwards requests to chatgpt.com/backend-api/codex with Codex CLI OAuth token.
+ *
+ *  The ChatGPT backend requires specific headers (account-id from JWT, originator,
+ *  OpenAI-Beta) and a different URL path (/codex/responses) than api.openai.com. */
 
 import { codex as config } from "../auth/configs.ts";
 import * as oauth from "../auth/oauth.ts";
 import * as store from "../auth/store.ts";
-import { OPENAI_API_URL } from "../constants.ts";
+import { CODEX_BASE_URL, codexHeaders, codexHeaderValues, codexPathMap } from "../constants.ts";
+import { fromBase64url } from "../utils/encoding.ts";
 import type { Provider } from "./base.ts";
 import { denied, forward } from "./base.ts";
 
@@ -20,8 +24,11 @@ export const provider: Provider = {
     const accessToken = await oauth.token(config, account);
     if (!accessToken) return denied("OpenAI Codex");
 
+    const accountId = getAccountId(accessToken, account);
+    const codexPath = codexPathMap[sub] ?? sub;
+
     return forward({
-      url: `${OPENAI_API_URL}${sub}`,
+      url: `${CODEX_BASE_URL}${codexPath}`,
       body: body.forwardBody,
       streaming: body.stream,
       providerName: "OpenAI Codex",
@@ -30,7 +37,29 @@ export const provider: Provider = {
         "Content-Type": "application/json",
         Authorization: `Bearer ${accessToken}`,
         Accept: body.stream ? "text/event-stream" : "application/json",
+        Connection: "Keep-Alive",
+        [codexHeaders.BETA]: codexHeaderValues.BETA_RESPONSES,
+        [codexHeaders.ORIGINATOR]: codexHeaderValues.ORIGINATOR,
+        "User-Agent": codexHeaderValues.USER_AGENT,
+        Version: codexHeaderValues.VERSION,
+        ...(accountId ? { [codexHeaders.ACCOUNT_ID]: accountId } : {}),
       },
     });
   },
 };
+
+/** Extract chatgpt_account_id from JWT, falling back to stored credentials. */
+function getAccountId(accessToken: string, account: number): string | undefined {
+  const creds = store.get("codex", account);
+  if (creds?.accountId) return creds.accountId;
+
+  try {
+    const parts = accessToken.split(".");
+    if (parts.length < 2 || !parts[1]) return undefined;
+    const payload = JSON.parse(new TextDecoder().decode(fromBase64url(parts[1]))) as Record<string, unknown>;
+    const auth = payload["https://api.openai.com/auth"] as Record<string, unknown> | undefined;
+    return (auth?.["chatgpt_account_id"] as string) ?? undefined;
+  } catch {
+    return undefined;
+  }
+}