openai-ws-opencode 0.1.0

package/dist/plugin.js

@@ -0,0 +1,142 @@
+import crypto from "node:crypto";
+import { CODEX_API_BASE, CODEX_API_ENDPOINT, INTERNAL_AGENT_HEADER, INTERNAL_MODEL_HEADER, INTERNAL_PREFIX_HASH_HEADER, INTERNAL_SESSION_HEADER, OPENAI_API_BASE, PROVIDER_ID, } from "./constants.js";
+import { oauthMethods } from "./auth/oauth.js";
+import { extractAccountId, refreshAccessToken, tokenExpiry } from "./auth/tokens.js";
+import { resolveModelsBestEffort } from "./models/resolve.js";
+import { prepareHttpFallbackBody } from "./transport/body.js";
+import { bridgeWebSocket } from "./transport/bridge.js";
+import { closeConnections } from "./transport/pool.js";
+import { extractTransportContext, httpAuthHeaders, transportIdentity } from "./transport/headers.js";
+function stableHash(value) {
+    return crypto.createHash("sha256").update(JSON.stringify(value ?? "")).digest("hex").slice(0, 32);
+}
+async function resolveOAuthAuth(auth, client) {
+    const expiryBufferMs = 5 * 60 * 1000;
+    if (auth.access && auth.expires && auth.expires > Date.now() + expiryBufferMs) {
+        return { accessToken: auth.access, accountId: auth.accountId };
+    }
+    const tokens = await refreshAccessToken(auth.refresh);
+    const accountId = extractAccountId(tokens) ?? auth.accountId;
+    await client.auth.set({
+        path: { id: PROVIDER_ID },
+        body: {
+            type: "oauth",
+            refresh: tokens.refresh_token ?? auth.refresh,
+            access: tokens.access_token,
+            expires: tokenExpiry(tokens.expires_in),
+            ...(accountId ? { accountId } : {}),
+        },
+    });
+    return { accessToken: tokens.access_token, accountId };
+}
+function shouldBridge(url, init) {
+    return (init?.method?.toUpperCase() === "POST" &&
+        (url.pathname.includes("/v1/responses") || url.pathname.includes("/backend-api/codex/responses")) &&
+        typeof init.body === "string");
+}
+const OpenAIWebSocketPlugin = async ({ client }) => {
+    const hooks = {
+        "chat.headers": async (input, output) => {
+            if (input.model.providerID !== PROVIDER_ID)
+                return;
+            const headers = (output.headers ??= {});
+            headers[INTERNAL_SESSION_HEADER] = input.sessionID;
+            headers[INTERNAL_AGENT_HEADER] = input.agent;
+            headers[INTERNAL_MODEL_HEADER] = input.model.id ?? input.model.modelID;
+            headers[INTERNAL_PREFIX_HASH_HEADER] = stableHash(input.message);
+        },
+        event: async ({ event }) => {
+            if (event.type === "session.deleted") {
+                const deletedSessionID = event.properties?.info?.id;
+                closeConnections((conn) => conn.activeSessionID === deletedSessionID || conn.lastSessionID === deletedSessionID);
+                return;
+            }
+            const eventType = String(event.type);
+            if (eventType === "server.instance.disposed" || eventType === "global.disposed") {
+                closeConnections();
+            }
+        },
+        auth: {
+            provider: PROVIDER_ID,
+            async loader(getAuth, provider) {
+                const auth = (await getAuth());
+                if (provider)
+                    provider.models = (await resolveModelsBestEffort(provider.models));
+                if (auth?.type === "api" && auth.key) {
+                    const apiKey = auth.key;
+                    const identity = transportIdentity({ type: "api", apiKey });
+                    return {
+                        apiKey,
+                        baseURL: OPENAI_API_BASE,
+                        async fetch(input, init) {
+                            const url = typeof input === "string" ? new URL(input) : input instanceof URL ? input : new URL(input.url);
+                            const context = extractTransportContext(init?.headers);
+                            if (shouldBridge(url, init)) {
+                                try {
+                                    const body = JSON.parse(init?.body);
+                                    if (body.stream !== false) {
+                                        return bridgeWebSocket(identity.wsUrl, identity.wsHeaders, body, false, context, init?.signal ?? undefined);
+                                    }
+                                }
+                                catch { }
+                            }
+                            if (init?.method?.toUpperCase() === "POST" && typeof init.body === "string") {
+                                try {
+                                    init = { ...init, body: JSON.stringify(prepareHttpFallbackBody(JSON.parse(init.body), false)) };
+                                }
+                                catch { }
+                            }
+                            return globalThis.fetch(input, {
+                                ...init,
+                                headers: httpAuthHeaders(context.forwardHeaders, { type: "api", apiKey }),
+                            });
+                        },
+                    };
+                }
+                if (auth?.type === "oauth") {
+                    const initialAuth = await resolveOAuthAuth(auth, client);
+                    return {
+                        apiKey: initialAuth.accessToken,
+                        baseURL: CODEX_API_BASE,
+                        async fetch(input, init) {
+                            const currentAuth = (await getAuth());
+                            if (currentAuth?.type !== "oauth")
+                                throw new Error("OpenAI WebSocket OAuth auth is missing");
+                            const { accessToken, accountId } = await resolveOAuthAuth(currentAuth, client);
+                            const identity = transportIdentity({ type: "oauth", accessToken, accountId });
+                            const url = typeof input === "string" ? new URL(input) : input instanceof URL ? input : new URL(input.url);
+                            const context = extractTransportContext(init?.headers);
+                            if (shouldBridge(url, init)) {
+                                try {
+                                    const body = JSON.parse(init?.body);
+                                    if (body.stream !== false) {
+                                        return bridgeWebSocket(identity.wsUrl, identity.wsHeaders, body, true, context, init?.signal ?? undefined);
+                                    }
+                                }
+                                catch { }
+                            }
+                            if (init?.method?.toUpperCase() === "POST" && typeof init.body === "string") {
+                                try {
+                                    init = { ...init, body: JSON.stringify(prepareHttpFallbackBody(JSON.parse(init.body), true)) };
+                                }
+                                catch { }
+                            }
+                            const rewrittenUrl = url.pathname.includes("/v1/responses") || url.pathname.includes("/chat/completions")
+                                ? new URL(CODEX_API_ENDPOINT)
+                                : url;
+                            return globalThis.fetch(rewrittenUrl, {
+                                ...init,
+                                headers: httpAuthHeaders(context.forwardHeaders, { type: "oauth", accessToken, accountId }),
+                            });
+                        },
+                    };
+                }
+                throw new Error("OpenAI WebSocket auth is missing; run `opencode auth login openai-ws`.");
+            },
+            methods: oauthMethods,
+        },
+    };
+    return hooks;
+};
+export default OpenAIWebSocketPlugin;
+//# sourceMappingURL=plugin.js.map

package/dist/plugin.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AACA,OAAO,MAAM,MAAM,aAAa,CAAA;AAChC,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,2BAA2B,EAC3B,uBAAuB,EACvB,eAAe,EACf,WAAW,GACZ,MAAM,gBAAgB,CAAA;AACvB,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAC9C,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,WAAW,EAAwB,MAAM,kBAAkB,CAAA;AAC1G,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAA;AAC7D,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAA;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAA;AACtD,OAAO,EAAE,uBAAuB,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAA;AAMpG,SAAS,UAAU,CAAC,KAAc;IAChC,OAAO,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;AACnG,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,IAAe,EAAE,MAAW;IAC1D,MAAM,cAAc,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAA;IACpC,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,EAAE,CAAC;QAC9E,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAA;IAChE,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IACrD,MAAM,SAAS,GAAG,gBAAgB,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,SAAS,CAAA;IAC5D,MAAM,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;QACpB,IAAI,EAAE,EAAE,EAAE,EAAE,WAAW,EAAE;QACzB,IAAI,EAAE;YACJ,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,OAAO;YAC7C,MAAM,EAAE,MAAM,CAAC,YAAY;YAC3B,OAAO,EAAE,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC;YACvC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACpC;KACF,CAAC,CAAA;IACF,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,YAAY,EAAE,SAAS,EAAE,CAAA;AACxD,CAAC;AAED,SAAS,YAAY,CAAC,GAAQ,EAAE,IAAkB;IAChD,OAAO,CACL,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,MAAM;QACtC,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,8BAA8B,CAAC,CAAC;QACjG,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAC9B,CAAA;AACH,CAAC;AAED,MAAM,qBAAqB,GAAW,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IACzD,MAAM,KAAK,GAAU;QACnB,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACtC,IAAI,KAAK,CAAC,KAAK,CAAC,UAAU,KAAK,WAAW;gBAAE,OAAM;YAClD,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,EAAE,CAAC,CAAA;YACvC,OAAO,CAAC,uBAAuB,CAAC,GAAG,KAAK,CAAC,SAAS,CAAA;YAClD,OAAO,CAAC,qBAAqB,CAAC,GAAG,KAAK,CAAC,KAAK,CAAA;YAC5C,OAAO,CAAC,qBAAqB,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,IAAK,KAAK,CAAC,KAAa,CAAC,OAAO,CAAA;YAC/E,OAAO,CAAC,2BAA2B,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;QAClE,CAAC;QAED,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;YACzB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBACrC,MAAM,gBAAgB,GAAI,KAAa,CAAC,UAAU,EAAE,IAAI,EAAE,EAAE,CAAA;gBAC5D,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,KAAK,gBAAgB,IAAI,IAAI,CAAC,aAAa,KAAK,gBAAgB,CAAC,CAAA;gBAChH,OAAM;YACR,CAAC;YACD,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YACpC,IAAI,SAAS,KAAK,0BAA0B,IAAI,SAAS,KAAK,iBAAiB,EAAE,CAAC;gBAChF,gBAAgB,EAAE,CAAA;YACpB,CAAC;QACH,CAAC;QAED,IAAI,EAAE;YACJ,QAAQ,EAAE,WAAW;YACrB,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ;gBAC5B,MAAM,IAAI,GAAG,CAAC,MAAM,OAAO,EAAE,CAAiB,CAAA;gBAC9C,IAAI,QAAQ;oBAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,MAAM,uBAAuB,CAAC,QAAQ,CAAC,MAAa,CAAC,CAAQ,CAAA;gBAE9F,IAAI,IAAI,EAAE,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;oBACrC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAA;oBACvB,MAAM,QAAQ,GAAG,iBAAiB,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;oBAC3D,OAAO;wBACL,MAAM;wBACN,OAAO,EAAE,eAAe;wBACxB,KAAK,CAAC,KAAK,CAAC,KAAwB,EAAE,IAAkB;4BACtD,MAAM,GAAG,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,YAAY,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;4BAC1G,MAAM,OAAO,GAAG,uBAAuB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;4BACtD,IAAI,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,CAAC;gCAC5B,IAAI,CAAC;oCACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,IAAc,CAA4B,CAAA;oCACxE,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;wCAC1B,OAAO,eAAe,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,SAAS,CAAC,CAAA;oCAC7G,CAAC;gCACH,CAAC;gCAAC,MAAM,CAAC,CAAA,CAAC;4BACZ,CAAC;4BAED,IAAI,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gCAC5E,IAAI,CAAC;oCACH,IAAI,GAAG,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,CAAA;gCACjG,CAAC;gCAAC,MAAM,CAAC,CAAA,CAAC;4BACZ,CAAC;4BACD,OAAO,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE;gCAC7B,GAAG,IAAI;gCACP,OAAO,EAAE,eAAe,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;6BAC1E,CAAC,CAAA;wBACJ,CAAC;qBACF,CAAA;gBACH,CAAC;gBAED,IAAI,IAAI,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC3B,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;oBACxD,OAAO;wBACL,MAAM,EAAE,WAAW,CAAC,WAAW;wBAC/B,OAAO,EAAE,cAAc;wBACvB,KAAK,CAAC,KAAK,CAAC,KAAwB,EAAE,IAAkB;4BACtD,MAAM,WAAW,GAAG,CAAC,MAAM,OAAO,EAAE,CAA0B,CAAA;4BAC9D,IAAI,WAAW,EAAE,IAAI,KAAK,OAAO;gCAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAA;4BAC5F,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,GAAG,MAAM,gBAAgB,CAAC,WAAW,EAAE,MAAM,CAAC,CAAA;4BAC9E,MAAM,QAAQ,GAAG,iBAAiB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAA;4BAC7E,MAAM,GAAG,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,YAAY,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;4BAC1G,MAAM,OAAO,GAAG,uBAAuB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;4BACtD,IAAI,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,CAAC;gCAC5B,IAAI,CAAC;oCACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,IAAc,CAA4B,CAAA;oCACxE,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;wCAC1B,OAAO,eAAe,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,SAAS,CAAC,CAAA;oCAC5G,CAAC;gCACH,CAAC;gCAAC,MAAM,CAAC,CAAA,CAAC;4BACZ,CAAC;4BAED,IAAI,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gCAC5E,IAAI,CAAC;oCACH,IAAI,GAAG,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAA;gCAChG,CAAC;gCAAC,MAAM,CAAC,CAAA,CAAC;4BACZ,CAAC;4BAED,MAAM,YAAY,GAChB,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,CAAC;gCAClF,CAAC,CAAC,IAAI,GAAG,CAAC,kBAAkB,CAAC;gCAC7B,CAAC,CAAC,GAAG,CAAA;4BACT,OAAO,UAAU,CAAC,KAAK,CAAC,YAAY,EAAE;gCACpC,GAAG,IAAI;gCACP,OAAO,EAAE,eAAe,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;6BAC5F,CAAC,CAAA;wBACJ,CAAC;qBACF,CAAA;gBACH,CAAC;gBAED,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC,CAAA;YAC3F,CAAC;YACD,OAAO,EAAE,YAAY;SACtB;KACF,CAAA;IAED,OAAO,KAAK,CAAA;AACd,CAAC,CAAA;AAED,eAAe,qBAAqB,CAAA"}

package/dist/testing.d.ts

@@ -0,0 +1,7 @@
+export { oauthTesting } from "./auth/oauth.js";
+export { prepareBody, prepareHttpFallbackBody } from "./transport/body.js";
+export { apiKeyWebSocketHeaders, oauthWebSocketHeaders, extractTransportContext, transportIdentity } from "./transport/headers.js";
+export { connectionPool, readyState, resetPoolForTesting, resetWebSocketConstructorForTesting, setWebSocketConstructorForTesting, } from "./transport/pool.js";
+export { bridgeWebSocket } from "./transport/bridge.js";
+export { resolveModels, resolveModelsBestEffort, resolveModelsFromCatalog, providerConfig } from "./models/resolve.js";
+//# sourceMappingURL=testing.d.ts.map

package/dist/testing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"testing.d.ts","sourceRoot":"","sources":["../src/testing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAC9C,OAAO,EAAE,WAAW,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAA;AAC1E,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAA;AAClI,OAAO,EACL,cAAc,EACd,UAAU,EACV,mBAAmB,EACnB,mCAAmC,EACnC,iCAAiC,GAClC,MAAM,qBAAqB,CAAA;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AACvD,OAAO,EAAE,aAAa,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAA"}

package/dist/testing.js

@@ -0,0 +1,7 @@
+export { oauthTesting } from "./auth/oauth.js";
+export { prepareBody, prepareHttpFallbackBody } from "./transport/body.js";
+export { apiKeyWebSocketHeaders, oauthWebSocketHeaders, extractTransportContext, transportIdentity } from "./transport/headers.js";
+export { connectionPool, readyState, resetPoolForTesting, resetWebSocketConstructorForTesting, setWebSocketConstructorForTesting, } from "./transport/pool.js";
+export { bridgeWebSocket } from "./transport/bridge.js";
+export { resolveModels, resolveModelsBestEffort, resolveModelsFromCatalog, providerConfig } from "./models/resolve.js";
+//# sourceMappingURL=testing.js.map

package/dist/testing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testing.js","sourceRoot":"","sources":["../src/testing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAC9C,OAAO,EAAE,WAAW,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAA;AAC1E,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAA;AAClI,OAAO,EACL,cAAc,EACd,UAAU,EACV,mBAAmB,EACnB,mCAAmC,EACnC,iCAAiC,GAClC,MAAM,qBAAqB,CAAA;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AACvD,OAAO,EAAE,aAAa,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAA"}

package/dist/transport/body.d.ts

@@ -0,0 +1,3 @@
+export declare function prepareBody(requestBody: Record<string, unknown>, isOAuth: boolean): Record<string, unknown>;
+export declare function prepareHttpFallbackBody(requestBody: Record<string, unknown>, isOAuth: boolean): Record<string, unknown>;
+//# sourceMappingURL=body.d.ts.map

package/dist/transport/body.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"body.d.ts","sourceRoot":"","sources":["../../src/transport/body.ts"],"names":[],"mappings":"AAEA,wBAAgB,WAAW,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAY3G;AAED,wBAAgB,uBAAuB,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAKvH"}

package/dist/transport/body.js

@@ -0,0 +1,22 @@
+import { DEFAULT_INSTRUCTIONS } from "../constants.js";
+export function prepareBody(requestBody, isOAuth) {
+    const { stream: _stream, stream_options: _streamOptions, ...wsBody } = requestBody;
+    if (!wsBody.instructions)
+        wsBody.instructions = DEFAULT_INSTRUCTIONS;
+    if (isOAuth) {
+        if (wsBody.store === undefined)
+            wsBody.store = false;
+        delete wsBody.max_output_tokens;
+        delete wsBody.max_tokens;
+    }
+    return wsBody;
+}
+export function prepareHttpFallbackBody(requestBody, isOAuth) {
+    const next = { ...requestBody };
+    if (!next.instructions)
+        next.instructions = DEFAULT_INSTRUCTIONS;
+    if (isOAuth && next.store === undefined)
+        next.store = false;
+    return next;
+}
+//# sourceMappingURL=body.js.map

package/dist/transport/body.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"body.js","sourceRoot":"","sources":["../../src/transport/body.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAA;AAEtD,MAAM,UAAU,WAAW,CAAC,WAAoC,EAAE,OAAgB;IAChF,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,MAAM,EAAE,GAAG,WAAW,CAAA;IAElF,IAAI,CAAC,MAAM,CAAC,YAAY;QAAE,MAAM,CAAC,YAAY,GAAG,oBAAoB,CAAA;IAEpE,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;YAAE,MAAM,CAAC,KAAK,GAAG,KAAK,CAAA;QACpD,OAAO,MAAM,CAAC,iBAAiB,CAAA;QAC/B,OAAO,MAAM,CAAC,UAAU,CAAA;IAC1B,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,WAAoC,EAAE,OAAgB;IAC5F,MAAM,IAAI,GAAG,EAAE,GAAG,WAAW,EAAE,CAAA;IAC/B,IAAI,CAAC,IAAI,CAAC,YAAY;QAAE,IAAI,CAAC,YAAY,GAAG,oBAAoB,CAAA;IAChE,IAAI,OAAO,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS;QAAE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IAC3D,OAAO,IAAI,CAAA;AACb,CAAC"}

package/dist/transport/bridge.d.ts

@@ -0,0 +1,3 @@
+import type { TransportContext } from "./headers.js";
+export declare function bridgeWebSocket(wsUrl: string, headers: Record<string, string>, requestBody: Record<string, unknown>, isOAuth: boolean, context?: TransportContext, signal?: AbortSignal): Response;
+//# sourceMappingURL=bridge.d.ts.map

package/dist/transport/bridge.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bridge.d.ts","sourceRoot":"","sources":["../../src/transport/bridge.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AAkBpD,wBAAgB,eAAe,CAC7B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAC/B,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACpC,OAAO,EAAE,OAAO,EAChB,OAAO,GAAE,gBAAqB,EAC9B,MAAM,CAAC,EAAE,WAAW,GACnB,QAAQ,CAyDV"}

package/dist/transport/bridge.js

@@ -0,0 +1,78 @@
+import { prepareBody } from "./body.js";
+import { acquireConnection, closeConnections } from "./pool.js";
+function abortPending(conn, error) {
+    const pending = conn.pending;
+    if (!pending || pending.done)
+        return;
+    pending.done = true;
+    if (conn.ws?.readyState === 1) {
+        try {
+            conn.ws.send(JSON.stringify({ type: "response.cancel" }));
+        }
+        catch { }
+    }
+    try {
+        pending.controller.error(error);
+    }
+    catch { }
+    conn.pending = null;
+    closeConnections((candidate) => candidate === conn);
+}
+export function bridgeWebSocket(wsUrl, headers, requestBody, isOAuth, context = {}, signal) {
+    const wsBody = prepareBody(requestBody, isOAuth);
+    const conn = acquireConnection(wsUrl, headers, context);
+    let finalized = false;
+    const cleanupAbort = () => {
+        if (signal)
+            signal.removeEventListener("abort", onAbort);
+    };
+    const onAbort = () => {
+        if (finalized)
+            return;
+        finalized = true;
+        cleanupAbort();
+        abortPending(conn, signal?.reason instanceof Error ? signal.reason : new DOMException("Aborted", "AbortError"));
+    };
+    const readable = new ReadableStream({
+        start(controller) {
+            conn.activeSessionID = context.sessionID;
+            conn.activeAgent = context.agent;
+            conn.lastModelID = context.modelID ?? conn.lastModelID;
+            conn.lastStablePrefixHash = context.stablePrefixHash ?? conn.lastStablePrefixHash;
+            conn.pending = {
+                body: wsBody,
+                controller,
+                done: false,
+                sent: false,
+                forwarded: false,
+            };
+            if (signal?.aborted) {
+                onAbort();
+                return;
+            }
+            if (signal)
+                signal.addEventListener("abort", onAbort, { once: true });
+            if (conn.ws?.readyState === 1) {
+                conn.ws.send(JSON.stringify({ type: "response.create", ...wsBody }));
+                conn.pending.sent = true;
+            }
+        },
+        cancel() {
+            if (finalized)
+                return;
+            finalized = true;
+            cleanupAbort();
+            abortPending(conn, new DOMException("Aborted", "AbortError"));
+        },
+    });
+    return new Response(readable, {
+        status: 200,
+        headers: {
+            "content-type": "text/event-stream",
+            "cache-control": "no-cache",
+            connection: "keep-alive",
+            "x-request-id": `ws-${Date.now()}`,
+        },
+    });
+}
+//# sourceMappingURL=bridge.js.map

package/dist/transport/bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bridge.js","sourceRoot":"","sources":["../../src/transport/bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAA;AACvC,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAyB,MAAM,WAAW,CAAA;AAGtF,SAAS,YAAY,CAAC,IAAsB,EAAE,KAAY;IACxD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;IAC5B,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI;QAAE,OAAM;IACpC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAA;IACnB,IAAI,IAAI,CAAC,EAAE,EAAE,UAAU,KAAK,CAAC,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAA;QAC3D,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IACD,IAAI,CAAC;QACH,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACjC,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;IACV,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;IACnB,gBAAgB,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,KAAK,IAAI,CAAC,CAAA;AACrD,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,KAAa,EACb,OAA+B,EAC/B,WAAoC,EACpC,OAAgB,EAChB,UAA4B,EAAE,EAC9B,MAAoB;IAEpB,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA;IAChD,MAAM,IAAI,GAAG,iBAAiB,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;IACvD,IAAI,SAAS,GAAG,KAAK,CAAA;IAErB,MAAM,YAAY,GAAG,GAAG,EAAE;QACxB,IAAI,MAAM;YAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;IAC1D,CAAC,CAAA;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,IAAI,SAAS;YAAE,OAAM;QACrB,SAAS,GAAG,IAAI,CAAA;QAChB,YAAY,EAAE,CAAA;QACd,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAA;IACjH,CAAC,CAAA;IAED,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAa;QAC9C,KAAK,CAAC,UAAU;YACd,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,SAAS,CAAA;YACxC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAA;YAChC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,WAAW,CAAA;YACtD,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,IAAI,IAAI,CAAC,oBAAoB,CAAA;YACjF,IAAI,CAAC,OAAO,GAAG;gBACb,IAAI,EAAE,MAAM;gBACZ,UAAU;gBACV,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,KAAK;gBACX,SAAS,EAAE,KAAK;aACjB,CAAA;YAED,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,OAAO,EAAE,CAAA;gBACT,OAAM;YACR,CAAC;YACD,IAAI,MAAM;gBAAE,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YACrE,IAAI,IAAI,CAAC,EAAE,EAAE,UAAU,KAAK,CAAC,EAAE,CAAC;gBAC9B,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC,CAAA;gBACpE,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAA;YAC1B,CAAC;QACH,CAAC;QACD,MAAM;YACJ,IAAI,SAAS;gBAAE,OAAM;YACrB,SAAS,GAAG,IAAI,CAAA;YAChB,YAAY,EAAE,CAAA;YACd,YAAY,CAAC,IAAI,EAAE,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAA;QAC/D,CAAC;KACF,CAAC,CAAA;IAEF,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE;QAC5B,MAAM,EAAE,GAAG;QACX,OAAO,EAAE;YACP,cAAc,EAAE,mBAAmB;YACnC,eAAe,EAAE,UAAU;YAC3B,UAAU,EAAE,YAAY;YACxB,cAAc,EAAE,MAAM,IAAI,CAAC,GAAG,EAAE,EAAE;SACnC;KACF,CAAC,CAAA;AACJ,CAAC"}

package/dist/transport/headers.d.ts

@@ -0,0 +1,32 @@
+export type TransportContext = {
+    sessionID?: string;
+    agent?: string;
+    modelID?: string;
+    stablePrefixHash?: string;
+};
+export declare function apiKeyWebSocketHeaders(apiKey: string): Record<string, string>;
+export declare function oauthWebSocketHeaders(accessToken: string, accountId?: string): Record<string, string>;
+export declare function httpAuthHeaders(baseHeaders: HeadersInit | undefined, auth: {
+    type: "api";
+    apiKey: string;
+} | {
+    type: "oauth";
+    accessToken: string;
+    accountId?: string;
+}): Headers;
+export declare function transportIdentity(auth: {
+    type: "api";
+    apiKey: string;
+} | {
+    type: "oauth";
+    accessToken: string;
+    accountId?: string;
+}): {
+    isOAuth: boolean;
+    wsUrl: string;
+    wsHeaders: Record<string, string>;
+};
+export declare function extractTransportContext(input?: HeadersInit): TransportContext & {
+    forwardHeaders: Headers;
+};
+//# sourceMappingURL=headers.d.ts.map

package/dist/transport/headers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"headers.d.ts","sourceRoot":"","sources":["../../src/transport/headers.ts"],"names":[],"mappings":"AAWA,MAAM,MAAM,gBAAgB,GAAG;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,gBAAgB,CAAC,EAAE,MAAM,CAAA;CAC1B,CAAA;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAK7E;AAED,wBAAgB,qBAAqB,CAAC,WAAW,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAOrG;AAED,wBAAgB,eAAe,CAC7B,WAAW,EAAE,WAAW,GAAG,SAAS,EACpC,IAAI,EAAE;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GACjG,OAAO,CAWT;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE;;;;EAanI;AAED,wBAAgB,uBAAuB,CAAC,KAAK,CAAC,EAAE,WAAW,GAAG,gBAAgB,GAAG;IAAE,cAAc,EAAE,OAAO,CAAA;CAAE,CAa3G"}

package/dist/transport/headers.js

@@ -0,0 +1,55 @@
+import { CODEX_ORIGINATOR, CODEX_WS_URL, INTERNAL_AGENT_HEADER, INTERNAL_MODEL_HEADER, INTERNAL_PREFIX_HASH_HEADER, INTERNAL_SESSION_HEADER, OPENAI_WS_BETA, OPENAI_WS_URL, } from "../constants.js";
+export function apiKeyWebSocketHeaders(apiKey) {
+    return {
+        Authorization: `Bearer ${apiKey}`,
+        "OpenAI-Beta": OPENAI_WS_BETA,
+    };
+}
+export function oauthWebSocketHeaders(accessToken, accountId) {
+    return {
+        Authorization: `Bearer ${accessToken}`,
+        ...(accountId ? { "ChatGPT-Account-Id": accountId } : {}),
+        originator: CODEX_ORIGINATOR,
+        "OpenAI-Beta": OPENAI_WS_BETA,
+    };
+}
+export function httpAuthHeaders(baseHeaders, auth) {
+    const headers = new Headers(baseHeaders);
+    headers.delete("authorization");
+    headers.delete("Authorization");
+    headers.set("Authorization", `Bearer ${auth.type === "api" ? auth.apiKey : auth.accessToken}`);
+    if (auth.type === "oauth") {
+        if (auth.accountId)
+            headers.set("ChatGPT-Account-Id", auth.accountId);
+        headers.set("originator", CODEX_ORIGINATOR);
+        headers.set("OpenAI-Beta", OPENAI_WS_BETA);
+    }
+    return headers;
+}
+export function transportIdentity(auth) {
+    if (auth.type === "oauth") {
+        return {
+            isOAuth: true,
+            wsUrl: CODEX_WS_URL,
+            wsHeaders: oauthWebSocketHeaders(auth.accessToken, auth.accountId),
+        };
+    }
+    return {
+        isOAuth: false,
+        wsUrl: OPENAI_WS_URL,
+        wsHeaders: apiKeyWebSocketHeaders(auth.apiKey),
+    };
+}
+export function extractTransportContext(input) {
+    const forwardHeaders = new Headers(input);
+    const sessionID = forwardHeaders.get(INTERNAL_SESSION_HEADER) ?? undefined;
+    const agent = forwardHeaders.get(INTERNAL_AGENT_HEADER) ?? undefined;
+    const modelID = forwardHeaders.get(INTERNAL_MODEL_HEADER) ?? undefined;
+    const stablePrefixHash = forwardHeaders.get(INTERNAL_PREFIX_HASH_HEADER) ?? undefined;
+    forwardHeaders.delete(INTERNAL_SESSION_HEADER);
+    forwardHeaders.delete(INTERNAL_AGENT_HEADER);
+    forwardHeaders.delete(INTERNAL_MODEL_HEADER);
+    forwardHeaders.delete(INTERNAL_PREFIX_HASH_HEADER);
+    return { sessionID, agent, modelID, stablePrefixHash, forwardHeaders };
+}
+//# sourceMappingURL=headers.js.map

package/dist/transport/headers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"headers.js","sourceRoot":"","sources":["../../src/transport/headers.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,YAAY,EACZ,qBAAqB,EACrB,qBAAqB,EACrB,2BAA2B,EAC3B,uBAAuB,EACvB,cAAc,EACd,aAAa,GACd,MAAM,iBAAiB,CAAA;AASxB,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,OAAO;QACL,aAAa,EAAE,UAAU,MAAM,EAAE;QACjC,aAAa,EAAE,cAAc;KAC9B,CAAA;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,WAAmB,EAAE,SAAkB;IAC3E,OAAO;QACL,aAAa,EAAE,UAAU,WAAW,EAAE;QACtC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,UAAU,EAAE,gBAAgB;QAC5B,aAAa,EAAE,cAAc;KAC9B,CAAA;AACH,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,WAAoC,EACpC,IAAkG;IAElG,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,WAAW,CAAC,CAAA;IACxC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;IAC/B,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;IAC/B,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAA;IAC9F,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC1B,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;QACrE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAA;QAC3C,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,cAAc,CAAC,CAAA;IAC5C,CAAC;IACD,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAkG;IAClI,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO;YACL,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,qBAAqB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC;SACnE,CAAA;IACH,CAAC;IACD,OAAO;QACL,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,sBAAsB,CAAC,IAAI,CAAC,MAAM,CAAC;KAC/C,CAAA;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,KAAmB;IACzD,MAAM,cAAc,GAAG,IAAI,OAAO,CAAC,KAAK,CAAC,CAAA;IACzC,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,SAAS,CAAA;IAC1E,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,SAAS,CAAA;IACpE,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,SAAS,CAAA;IACtE,MAAM,gBAAgB,GAAG,cAAc,CAAC,GAAG,CAAC,2BAA2B,CAAC,IAAI,SAAS,CAAA;IAErF,cAAc,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAA;IAC9C,cAAc,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAA;IAC5C,cAAc,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAA;IAC5C,cAAc,CAAC,MAAM,CAAC,2BAA2B,CAAC,CAAA;IAElD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,CAAA;AACxE,CAAC"}

package/dist/transport/pool.d.ts

@@ -0,0 +1,56 @@
+import type { TransportContext } from "./headers.js";
+type WebSocketLike = {
+    readyState: number;
+    send(data: string): void;
+    close(code?: number, reason?: string): void;
+    terminate?: () => void;
+    ping?: () => void;
+    on?(event: string, listener: (...args: any[]) => void): unknown;
+    off?(event: string, listener: (...args: any[]) => void): unknown;
+    addEventListener?(event: string, listener: (...args: any[]) => void): unknown;
+    removeEventListener?(event: string, listener: (...args: any[]) => void): unknown;
+};
+type WebSocketConstructor = new (url: string, options: {
+    headers: Record<string, string>;
+}) => WebSocketLike;
+export interface PendingRequest {
+    body: Record<string, unknown>;
+    controller: ReadableStreamDefaultController<Uint8Array>;
+    done: boolean;
+    sent: boolean;
+    forwarded: boolean;
+}
+export interface PooledConnection {
+    ws: WebSocketLike | null;
+    wsUrl: string;
+    headers: Record<string, string>;
+    scopeKey: string;
+    busy: boolean;
+    pending: PendingRequest | null;
+    activeSessionID?: string;
+    lastSessionID?: string;
+    activeAgent?: string;
+    lastAgent?: string;
+    lastModelID?: string;
+    lastStablePrefixHash?: string;
+    generation: number;
+    reconnectAttempts: number;
+    idleTimer: ReturnType<typeof setTimeout> | null;
+    connectTimer: ReturnType<typeof setTimeout> | null;
+    retryTimer: ReturnType<typeof setTimeout> | null;
+    detach: (() => void) | null;
+}
+export declare const connectionPool: PooledConnection[];
+export declare const readyState: {
+    readonly CONNECTING: 0;
+    readonly OPEN: 1;
+    readonly CLOSING: 2;
+    readonly CLOSED: 3;
+};
+export declare function acquireConnection(wsUrl: string, headers: Record<string, string>, context: TransportContext): PooledConnection;
+export declare function closeConnections(predicate?: (conn: PooledConnection) => boolean): void;
+export declare function resetPoolForTesting(): void;
+export declare function setWebSocketConstructorForTesting(ctor: WebSocketConstructor): void;
+export declare function resetWebSocketConstructorForTesting(): void;
+export {};
+//# sourceMappingURL=pool.d.ts.map

package/dist/transport/pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pool.d.ts","sourceRoot":"","sources":["../../src/transport/pool.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AAEpD,KAAK,aAAa,GAAG;IACnB,UAAU,EAAE,MAAM,CAAA;IAClB,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAA;IACxB,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IAC3C,SAAS,CAAC,EAAE,MAAM,IAAI,CAAA;IACtB,IAAI,CAAC,EAAE,MAAM,IAAI,CAAA;IACjB,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,IAAI,GAAG,OAAO,CAAA;IAC/D,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,IAAI,GAAG,OAAO,CAAA;IAChE,gBAAgB,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,IAAI,GAAG,OAAO,CAAA;IAC7E,mBAAmB,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,IAAI,GAAG,OAAO,CAAA;CACjF,CAAA;AAED,KAAK,oBAAoB,GAAG,KAAK,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE;IAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CAAE,KAAK,aAAa,CAAA;AAS5G,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC7B,UAAU,EAAE,+BAA+B,CAAC,UAAU,CAAC,CAAA;IACvD,IAAI,EAAE,OAAO,CAAA;IACb,IAAI,EAAE,OAAO,CAAA;IACb,SAAS,EAAE,OAAO,CAAA;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,aAAa,GAAG,IAAI,CAAA;IACxB,KAAK,EAAE,MAAM,CAAA;IACb,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC/B,QAAQ,EAAE,MAAM,CAAA;IAChB,IAAI,EAAE,OAAO,CAAA;IACb,OAAO,EAAE,cAAc,GAAG,IAAI,CAAA;IAC9B,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,oBAAoB,CAAC,EAAE,MAAM,CAAA;IAC7B,UAAU,EAAE,MAAM,CAAA;IAClB,iBAAiB,EAAE,MAAM,CAAA;IACzB,SAAS,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IAC/C,YAAY,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IAClD,UAAU,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IAChD,MAAM,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,IAAI,CAAA;CAC5B;AAED,eAAO,MAAM,cAAc,EAAE,gBAAgB,EAAO,CAAA;AAEpD,eAAO,MAAM,UAAU;;;;;CAKb,CAAA;AAiNV,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,gBAAgB,GAAG,gBAAgB,CAe7H;AAED,wBAAgB,gBAAgB,CAAC,SAAS,GAAE,CAAC,IAAI,EAAE,gBAAgB,KAAK,OAAoB,QAK3F;AAED,wBAAgB,mBAAmB,SAOlC;AAED,wBAAgB,iCAAiC,CAAC,IAAI,EAAE,oBAAoB,QAE3E;AAED,wBAAgB,mCAAmC,SAElD"}