openai-ws-opencode 0.1.6 → 0.1.8

package/dist/transport/bun-websocket.js

@@ -0,0 +1,281 @@
+import crypto from "node:crypto";
+import { EventEmitter } from "node:events";
+import WebSocket from "ws";
+const CONNECTING = 0;
+const OPEN = 1;
+const CLOSING = 2;
+const CLOSED = 3;
+export function loadDefaultWebSocketConstructor() {
+    const connect = globalThis.Bun?.connect;
+    return typeof connect === "function" ? BunWebSocket : WebSocket;
+}
+class BunWebSocket extends EventEmitter {
+    url;
+    options;
+    readyState = CONNECTING;
+    #socket = null;
+    #handshakeBuffer = Buffer.alloc(0);
+    #frameBuffer = Buffer.alloc(0);
+    #fragments = [];
+    #fragmentOpcode = null;
+    #closed = false;
+    constructor(url, options) {
+        super();
+        this.url = url;
+        this.options = options;
+        void this.#connect();
+    }
+    send(data) {
+        if (this.readyState !== OPEN || !this.#socket)
+            throw new Error("WebSocket is not open");
+        this.#socket.write(encodeClientFrame(0x1, Buffer.from(data, "utf8")));
+    }
+    ping() {
+        if (this.readyState !== OPEN || !this.#socket)
+            return;
+        this.#socket.write(encodeClientFrame(0x9));
+    }
+    close(code = 1000, reason = "") {
+        if (this.readyState === CLOSED || this.readyState === CLOSING)
+            return;
+        this.readyState = CLOSING;
+        if (this.#socket) {
+            const payload = Buffer.alloc(2 + Buffer.byteLength(reason));
+            payload.writeUInt16BE(code, 0);
+            payload.write(reason, 2);
+            try {
+                this.#socket.write(encodeClientFrame(0x8, payload));
+            }
+            catch { }
+            this.#endSocket();
+        }
+        else {
+            this.#finishClose(code, reason);
+        }
+    }
+    terminate() {
+        this.#endSocket();
+        this.readyState = CLOSED;
+    }
+    async #connect() {
+        const parsed = new URL(this.url);
+        const connect = globalThis.Bun?.connect;
+        if (typeof connect !== "function") {
+            this.#emitError(new Error("Bun.connect is unavailable"));
+            return;
+        }
+        try {
+            this.#socket = await connect({
+                hostname: parsed.hostname,
+                port: Number(parsed.port || (parsed.protocol === "wss:" ? 443 : 80)),
+                tls: parsed.protocol === "wss:",
+                socket: {
+                    open: (socket) => {
+                        this.#socket = socket;
+                        queueMicrotask(() => socket.write(this.#handshakeRequest(parsed)));
+                    },
+                    data: (_socket, data) => this.#handleData(Buffer.from(data)),
+                    close: (_socket, error) => {
+                        if (error)
+                            this.#emitError(error);
+                        this.#finishClose();
+                    },
+                    error: (_socket, error) => this.#emitError(error),
+                    connectError: (_socket, error) => this.#emitError(error),
+                    end: () => this.#finishClose(),
+                    timeout: () => this.#emitError(new Error("WebSocket socket timed out")),
+                },
+            });
+        }
+        catch (error) {
+            this.#emitError(error instanceof Error ? error : new Error(String(error)));
+        }
+    }
+    #handshakeRequest(parsed) {
+        const key = crypto.randomBytes(16).toString("base64");
+        const path = `${parsed.pathname || "/"}${parsed.search}`;
+        const host = parsed.port ? `${parsed.hostname}:${parsed.port}` : parsed.hostname;
+        const headers = {
+            Host: host,
+            Upgrade: "websocket",
+            Connection: "Upgrade",
+            "Sec-WebSocket-Key": key,
+            "Sec-WebSocket-Version": "13",
+            ...this.options.headers,
+        };
+        return `GET ${path} HTTP/1.1\r\n${Object.entries(headers)
+            .map(([name, value]) => `${name}: ${value}`)
+            .join("\r\n")}\r\n\r\n`;
+    }
+    #handleData(data) {
+        if (this.readyState === CONNECTING) {
+            this.#handshakeBuffer = Buffer.concat([this.#handshakeBuffer, data]);
+            const split = this.#handshakeBuffer.indexOf("\r\n\r\n");
+            if (split < 0)
+                return;
+            const head = this.#handshakeBuffer.subarray(0, split).toString("utf8");
+            const rest = this.#handshakeBuffer.subarray(split + 4);
+            this.#handshakeBuffer = Buffer.alloc(0);
+            const response = parseHandshake(head);
+            if (response.status !== 101) {
+                this.#emitError(new Error(`WebSocket upgrade failed with HTTP ${response.status}`));
+                this.#endSocket();
+                return;
+            }
+            this.readyState = OPEN;
+            this.emit("upgrade", { headers: response.headers });
+            this.emit("open");
+            if (rest.length)
+                this.#handleFrames(rest);
+            return;
+        }
+        this.#handleFrames(data);
+    }
+    #handleFrames(data) {
+        this.#frameBuffer = Buffer.concat([this.#frameBuffer, data]);
+        while (this.#frameBuffer.length >= 2) {
+            const first = this.#frameBuffer[0];
+            const second = this.#frameBuffer[1];
+            const fin = (first & 0x80) !== 0;
+            let length = second & 0x7f;
+            let offset = 2;
+            if (length === 126) {
+                if (this.#frameBuffer.length < offset + 2)
+                    return;
+                length = this.#frameBuffer.readUInt16BE(offset);
+                offset += 2;
+            }
+            else if (length === 127) {
+                if (this.#frameBuffer.length < offset + 8)
+                    return;
+                const bigLength = this.#frameBuffer.readBigUInt64BE(offset);
+                if (bigLength > BigInt(Number.MAX_SAFE_INTEGER)) {
+                    this.#emitError(new Error("WebSocket frame is too large"));
+                    this.#endSocket();
+                    return;
+                }
+                length = Number(bigLength);
+                offset += 8;
+            }
+            const masked = (second & 0x80) !== 0;
+            const maskOffset = offset;
+            if (masked)
+                offset += 4;
+            if (this.#frameBuffer.length < offset + length)
+                return;
+            let payload = Buffer.from(this.#frameBuffer.subarray(offset, offset + length));
+            if (masked)
+                payload = unmask(payload, this.#frameBuffer.subarray(maskOffset, maskOffset + 4));
+            this.#frameBuffer = this.#frameBuffer.subarray(offset + length);
+            this.#handleFrame(first & 0x0f, fin, payload);
+        }
+    }
+    #handleFrame(opcode, fin, payload) {
+        if (opcode === 0x8) {
+            const code = payload.length >= 2 ? payload.readUInt16BE(0) : undefined;
+            const reason = payload.length > 2 ? payload.subarray(2).toString("utf8") : undefined;
+            this.#endSocket();
+            this.#finishClose(code, reason);
+            return;
+        }
+        if (opcode === 0x9) {
+            this.#socket?.write(encodeClientFrame(0xa, payload));
+            return;
+        }
+        if (opcode === 0xa) {
+            this.emit("pong");
+            return;
+        }
+        if (opcode === 0x0) {
+            this.#fragments.push(payload);
+            if (fin)
+                this.#emitMessage(this.#fragmentOpcode ?? 0x1, Buffer.concat(this.#fragments));
+            if (fin)
+                this.#resetFragments();
+            return;
+        }
+        if (opcode !== 0x1 && opcode !== 0x2)
+            return;
+        if (!fin) {
+            this.#fragmentOpcode = opcode;
+            this.#fragments = [payload];
+            return;
+        }
+        this.#emitMessage(opcode, payload);
+    }
+    #emitMessage(opcode, payload) {
+        if (opcode === 0x1)
+            this.emit("message", payload.toString("utf8"), false);
+        else
+            this.emit("message", payload, true);
+    }
+    #resetFragments() {
+        this.#fragmentOpcode = null;
+        this.#fragments = [];
+    }
+    #emitError(error) {
+        queueMicrotask(() => this.emit("error", error));
+    }
+    #endSocket() {
+        try {
+            this.#socket?.end?.();
+        }
+        catch {
+            try {
+                this.#socket?.close?.();
+            }
+            catch { }
+        }
+    }
+    #finishClose(code, reason) {
+        if (this.#closed)
+            return;
+        this.#closed = true;
+        this.readyState = CLOSED;
+        this.emit("close", code, reason);
+    }
+}
+function parseHandshake(text) {
+    const lines = text.split(/\r?\n/);
+    const status = Number(lines[0]?.match(/\s(\d{3})\s/)?.[1] ?? 0);
+    const headers = new Headers();
+    for (const line of lines.slice(1)) {
+        const index = line.indexOf(":");
+        if (index <= 0)
+            continue;
+        headers.append(line.slice(0, index).trim(), line.slice(index + 1).trim());
+    }
+    return { status, headers };
+}
+function encodeClientFrame(opcode, payload = Buffer.alloc(0)) {
+    const length = payload.length;
+    const lengthBytes = length < 126 ? 0 : length <= 0xffff ? 2 : 8;
+    const frame = Buffer.alloc(2 + lengthBytes + 4 + length);
+    frame[0] = 0x80 | opcode;
+    if (length < 126) {
+        frame[1] = 0x80 | length;
+    }
+    else if (length <= 0xffff) {
+        frame[1] = 0x80 | 126;
+        frame.writeUInt16BE(length, 2);
+    }
+    else {
+        frame[1] = 0x80 | 127;
+        frame.writeBigUInt64BE(BigInt(length), 2);
+    }
+    const maskOffset = 2 + lengthBytes;
+    const dataOffset = maskOffset + 4;
+    const mask = crypto.randomBytes(4);
+    mask.copy(frame, maskOffset);
+    for (let index = 0; index < payload.length; index++) {
+        frame[dataOffset + index] = payload[index] ^ mask[index % 4];
+    }
+    return frame;
+}
+function unmask(payload, mask) {
+    const output = Buffer.alloc(payload.length);
+    for (let index = 0; index < payload.length; index++)
+        output[index] = payload[index] ^ mask[index % 4];
+    return output;
+}
+//# sourceMappingURL=bun-websocket.js.map

package/dist/transport/bun-websocket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bun-websocket.js","sourceRoot":"","sources":["../../src/transport/bun-websocket.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAA;AAChC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAC1C,OAAO,SAAS,MAAM,IAAI,CAAA;AAuB1B,MAAM,UAAU,GAAG,CAAC,CAAA;AACpB,MAAM,IAAI,GAAG,CAAC,CAAA;AACd,MAAM,OAAO,GAAG,CAAC,CAAA;AACjB,MAAM,MAAM,GAAG,CAAC,CAAA;AA6BhB,MAAM,UAAU,+BAA+B;IAC7C,MAAM,OAAO,GAAI,UAAwB,CAAC,GAAG,EAAE,OAAO,CAAA;IACtD,OAAO,OAAO,OAAO,KAAK,UAAU,CAAC,CAAC,CAAE,YAAgD,CAAC,CAAC,CAAE,SAA6C,CAAA;AAC3I,CAAC;AAED,MAAM,YAAa,SAAQ,YAAY;IAUlB;IACA;IAVnB,UAAU,GAAG,UAAU,CAAA;IACvB,OAAO,GAAqB,IAAI,CAAA;IAChC,gBAAgB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;IAClC,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;IAC9B,UAAU,GAAa,EAAE,CAAA;IACzB,eAAe,GAAkB,IAAI,CAAA;IACrC,OAAO,GAAG,KAAK,CAAA;IAEf,YACmB,GAAW,EACX,OAA4C;QAE7D,KAAK,EAAE,CAAA;QAHU,QAAG,GAAH,GAAG,CAAQ;QACX,YAAO,GAAP,OAAO,CAAqC;QAG7D,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAA;IACtB,CAAC;IAED,IAAI,CAAC,IAAY;QACf,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAA;QACvF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAA;IACvE,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAM;QACrD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAA;IAC5C,CAAC;IAED,KAAK,CAAC,IAAI,GAAG,IAAI,EAAE,MAAM,GAAG,EAAE;QAC5B,IAAI,IAAI,CAAC,UAAU,KAAK,MAAM,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO;YAAE,OAAM;QACrE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAA;QACzB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAA;YAC3D,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;YAC9B,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;YACxB,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAA;YACrD,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;YACV,IAAI,CAAC,UAAU,EAAE,CAAA;QACnB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;QACjC,CAAC;IACH,CAAC;IAED,SAAS;QACP,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;IAC1B,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAChC,MAAM,OAAO,GAAI,UAAwB,CAAC,GAAG,EAAE,OAAO,CAAA;QACtD,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAA;YACxD,OAAM;QACR,CAAC;QACD,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,GAAG,MAAM,OAAO,CAAC;gBAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpE,GAAG,EAAE,MAAM,CAAC,QAAQ,KAAK,MAAM;gBAC/B,MAAM,EAAE;oBACN,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;wBACf,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;wBACrB,cAAc,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;oBACpE,CAAC;oBACD,IAAI,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAC5D,KAAK,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;wBACxB,IAAI,KAAK;4BAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAA;wBACjC,IAAI,CAAC,YAAY,EAAE,CAAA;oBACrB,CAAC;oBACD,KAAK,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;oBACjD,YAAY,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;oBACxD,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE;oBAC9B,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;iBACxE;aACF,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;QAC5E,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,MAAW;QAC3B,MAAM,GAAG,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;QACrD,MAAM,IAAI,GAAG,GAAG,MAAM,CAAC,QAAQ,IAAI,GAAG,GAAG,MAAM,CAAC,MAAM,EAAE,CAAA;QACxD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAA;QAChF,MAAM,OAAO,GAA2B;YACtC,IAAI,EAAE,IAAI;YACV,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,SAAS;YACrB,mBAAmB,EAAE,GAAG;YACxB,uBAAuB,EAAE,IAAI;YAC7B,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO;SACxB,CAAA;QACD,OAAO,OAAO,IAAI,gBAAgB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;aACtD,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,KAAK,KAAK,EAAE,CAAC;aAC3C,IAAI,CAAC,MAAM,CAAC,UAAU,CAAA;IAC3B,CAAC;IAED,WAAW,CAAC,IAAY;QACtB,IAAI,IAAI,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;YACnC,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC,CAAA;YACpE,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;YACvD,IAAI,KAAK,GAAG,CAAC;gBAAE,OAAM;YACrB,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;YACtE,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;YACtD,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;YACvC,MAAM,QAAQ,GAAG,cAAc,CAAC,IAAI,CAAC,CAAA;YACrC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,sCAAsC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;gBACnF,IAAI,CAAC,UAAU,EAAE,CAAA;gBACjB,OAAM;YACR,CAAC;YACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;YACtB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAA;YACnD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACjB,IAAI,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;YACzC,OAAM;QACR,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;IAC1B,CAAC;IAED,aAAa,CAAC,IAAY;QACxB,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAA;QAC5D,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAClC,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACnC,MAAM,GAAG,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAA;YAChC,IAAI,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;YAC1B,IAAI,MAAM,GAAG,CAAC,CAAA;YACd,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnB,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,MAAM,GAAG,CAAC;oBAAE,OAAM;gBACjD,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;gBAC/C,MAAM,IAAI,CAAC,CAAA;YACb,CAAC;iBAAM,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC1B,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,MAAM,GAAG,CAAC;oBAAE,OAAM;gBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;gBAC3D,IAAI,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;oBAChD,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAA;oBAC1D,IAAI,CAAC,UAAU,EAAE,CAAA;oBACjB,OAAM;gBACR,CAAC;gBACD,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,CAAA;gBAC1B,MAAM,IAAI,CAAC,CAAA;YACb,CAAC;YACD,MAAM,MAAM,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAA;YACpC,MAAM,UAAU,GAAG,MAAM,CAAA;YACzB,IAAI,MAAM;gBAAE,MAAM,IAAI,CAAC,CAAA;YACvB,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,MAAM,GAAG,MAAM;gBAAE,OAAM;YACtD,IAAI,OAAO,GAAW,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC,CAAA;YACtF,IAAI,MAAM;gBAAE,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAA;YAC7F,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,CAAA;YAC/D,IAAI,CAAC,YAAY,CAAC,KAAK,GAAG,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;QAC/C,CAAC;IACH,CAAC;IAED,YAAY,CAAC,MAAc,EAAE,GAAY,EAAE,OAAe;QACxD,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;YACnB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YACtE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YACpF,IAAI,CAAC,UAAU,EAAE,CAAA;YACjB,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;YAC/B,OAAM;QACR,CAAC;QACD,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAA;YACpD,OAAM;QACR,CAAC;QACD,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACjB,OAAM;QACR,CAAC;QACD,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,IAAI,GAAG;gBAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,IAAI,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAA;YACvF,IAAI,GAAG;gBAAE,IAAI,CAAC,eAAe,EAAE,CAAA;YAC/B,OAAM;QACR,CAAC;QACD,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG;YAAE,OAAM;QAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,GAAG,MAAM,CAAA;YAC7B,IAAI,CAAC,UAAU,GAAG,CAAC,OAAO,CAAC,CAAA;YAC3B,OAAM;QACR,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACpC,CAAC;IAED,YAAY,CAAC,MAAc,EAAE,OAAe;QAC1C,IAAI,MAAM,KAAK,GAAG;YAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,CAAA;;YACpE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,CAAA;IAC1C,CAAC;IAED,eAAe;QACb,IAAI,CAAC,eAAe,GAAG,IAAI,CAAA;QAC3B,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;IACtB,CAAC;IAED,UAAU,CAAC,KAAY;QACrB,cAAc,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAA;IACjD,CAAC;IAED,UAAU;QACR,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CAAA;QACvB,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAA;YACzB,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC;IAED,YAAY,CAAC,IAAa,EAAE,MAAe;QACzC,IAAI,IAAI,CAAC,OAAO;YAAE,OAAM;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,CAAA;IAClC,CAAC;CACF;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IACjC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;IAC/D,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAA;IAC7B,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;QAC/B,IAAI,KAAK,IAAI,CAAC;YAAE,SAAQ;QACxB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;IAC3E,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAA;AAC5B,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc,EAAE,UAAsB,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC9E,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAA;IAC7B,MAAM,WAAW,GAAG,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAC/D,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,CAAA;IACxD,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,MAAM,CAAA;IACxB,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC;QACjB,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,MAAM,CAAA;IAC1B,CAAC;SAAM,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;QAC5B,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,GAAG,CAAA;QACrB,KAAK,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;IAChC,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,GAAG,CAAA;QACrB,KAAK,CAAC,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAA;IAC3C,CAAC;IACD,MAAM,UAAU,GAAG,CAAC,GAAG,WAAW,CAAA;IAClC,MAAM,UAAU,GAAG,UAAU,GAAG,CAAC,CAAA;IACjC,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAA;IAClC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,CAAA;IAC5B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QACpD,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;IAC9D,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC;AAED,SAAS,MAAM,CAAC,OAAmB,EAAE,IAAgB;IACnD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IAC3C,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE;QAAE,MAAM,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;IACrG,OAAO,MAAM,CAAA;AACf,CAAC"}

package/dist/transport/config.d.ts

@@ -0,0 +1,14 @@
+export declare const transportConfig: {
+    connectTimeoutMs: number;
+    maxReconnectAttempts: number;
+    reconnectBaseDelayMs: number;
+    reconnectMaxDelayMs: number;
+    idleEvictMs: number;
+    staleReuseMs: number;
+    heartbeatIntervalMs: number;
+    pongTimeoutMs: number;
+    responseIdleTimeoutMs: number;
+    maxConnectionsPerScope: number;
+    connectionMaxAgeMs: number;
+};
+//# sourceMappingURL=config.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/transport/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,eAAe;;;;;;;;;;;;CAY3B,CAAA"}

package/dist/transport/config.js

@@ -0,0 +1,14 @@
+export const transportConfig = {
+    connectTimeoutMs: 10_000,
+    maxReconnectAttempts: 3,
+    reconnectBaseDelayMs: 100,
+    reconnectMaxDelayMs: 2_000,
+    idleEvictMs: 120_000,
+    staleReuseMs: 60_000,
+    heartbeatIntervalMs: 30_000,
+    pongTimeoutMs: 10_000,
+    responseIdleTimeoutMs: 90_000,
+    maxConnectionsPerScope: 4,
+    connectionMaxAgeMs: 55 * 60 * 1000,
+};
+//# sourceMappingURL=config.js.map

package/dist/transport/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/transport/config.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,gBAAgB,EAAE,MAAM;IACxB,oBAAoB,EAAE,CAAC;IACvB,oBAAoB,EAAE,GAAG;IACzB,mBAAmB,EAAE,KAAK;IAC1B,WAAW,EAAE,OAAO;IACpB,YAAY,EAAE,MAAM;IACpB,mBAAmB,EAAE,MAAM;IAC3B,aAAa,EAAE,MAAM;IACrB,qBAAqB,EAAE,MAAM;IAC7B,sBAAsB,EAAE,CAAC;IACzB,kBAAkB,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI;CACnC,CAAA"}

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

@@ -1 +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"}
+{"version":3,"file":"headers.d.ts","sourceRoot":"","sources":["../../src/transport/headers.ts"],"names":[],"mappings":"AAYA,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;AAiBD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAM7E;AAED,wBAAgB,qBAAqB,CAAC,WAAW,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAMrG;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,CAaT;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

@@ -1,16 +1,29 @@
-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";
+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, USER_AGENT, } from "../constants.js";
+function defaultHeaders() {
+    return {
+        originator: CODEX_ORIGINATOR,
+        "OpenAI-Beta": OPENAI_WS_BETA,
+        "User-Agent": USER_AGENT,
+    };
+}
+function organizationProjectHeaders() {
+    return {
+        ...(process.env.OPENAI_ORGANIZATION ? { "OpenAI-Organization": process.env.OPENAI_ORGANIZATION } : {}),
+        ...(process.env.OPENAI_PROJECT ? { "OpenAI-Project": process.env.OPENAI_PROJECT } : {}),
+    };
+}
 export function apiKeyWebSocketHeaders(apiKey) {
     return {
+        ...defaultHeaders(),
+        ...organizationProjectHeaders(),
         Authorization: `Bearer ${apiKey}`,
-        "OpenAI-Beta": OPENAI_WS_BETA,
     };
 }
 export function oauthWebSocketHeaders(accessToken, accountId) {
     return {
+        ...defaultHeaders(),
         Authorization: `Bearer ${accessToken}`,
         ...(accountId ? { "ChatGPT-Account-Id": accountId } : {}),
-        originator: CODEX_ORIGINATOR,
-        "OpenAI-Beta": OPENAI_WS_BETA,
     };
 }
 export function httpAuthHeaders(baseHeaders, auth) {
@@ -18,11 +31,15 @@ export function httpAuthHeaders(baseHeaders, auth) {
     headers.delete("authorization");
     headers.delete("Authorization");
     headers.set("Authorization", `Bearer ${auth.type === "api" ? auth.apiKey : auth.accessToken}`);
+    for (const [key, value] of Object.entries(defaultHeaders()))
+        headers.set(key, value);
+    if (auth.type === "api") {
+        for (const [key, value] of Object.entries(organizationProjectHeaders()))
+            headers.set(key, value);
+    }
     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;
 }

package/dist/transport/headers.js.map

@@ -1 +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"}
+{"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,EACb,UAAU,GACX,MAAM,iBAAiB,CAAA;AASxB,SAAS,cAAc;IACrB,OAAO;QACL,UAAU,EAAE,gBAAgB;QAC5B,aAAa,EAAE,cAAc;QAC7B,YAAY,EAAE,UAAU;KACzB,CAAA;AACH,CAAC;AAED,SAAS,0BAA0B;IACjC,OAAO;QACL,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,qBAAqB,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACtG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACxF,CAAA;AACH,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,OAAO;QACL,GAAG,cAAc,EAAE;QACnB,GAAG,0BAA0B,EAAE;QAC/B,aAAa,EAAE,UAAU,MAAM,EAAE;KAClC,CAAA;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,WAAmB,EAAE,SAAkB;IAC3E,OAAO;QACL,GAAG,cAAc,EAAE;QACnB,aAAa,EAAE,UAAU,WAAW,EAAE;QACtC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1D,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,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAAE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;IACpF,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QACxB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,0BAA0B,EAAE,CAAC;YAAE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;IAClG,CAAC;IACD,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;IACvE,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

@@ -1,31 +1,28 @@
+import { type WebSocketConstructor, type WebSocketLike } from "./bun-websocket.js";
 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 PendingMetadata = {
+    responseId?: string;
 };
-type WebSocketConstructor = new (url: string, options: {
-    headers: Record<string, string>;
-}) => WebSocketLike;
 export interface PendingRequest {
     body: Record<string, unknown>;
     controller: ReadableStreamDefaultController<Uint8Array>;
+    onFinalize?: () => void;
     done: boolean;
     sent: boolean;
-    forwarded: boolean;
+    processedAckSent: boolean;
+    idleTimer: ReturnType<typeof setTimeout> | null;
+    metadata: PendingMetadata;
 }
 export interface PooledConnection {
+    id: string;
     ws: WebSocketLike | null;
     wsUrl: string;
     headers: Record<string, string>;
     scopeKey: string;
+    scopeHash: string;
+    contextKey: string;
     busy: boolean;
+    warm: boolean;
     pending: PendingRequest | null;
     activeSessionID?: string;
     lastSessionID?: string;
@@ -33,8 +30,14 @@ export interface PooledConnection {
     lastAgent?: string;
     lastModelID?: string;
     lastStablePrefixHash?: string;
+    lastResponseID?: string;
+    turnState?: string;
+    serverModel?: string;
+    modelsEtag?: string;
+    serverReasoningIncluded?: boolean;
     generation: number;
     reconnectAttempts: number;
+    createdAt: number;
     lastActivityAt: number;
     lastCloseCode?: number;
     lastCloseReason?: string;
@@ -54,8 +57,9 @@ export declare const readyState: {
     readonly CLOSED: 3;
 };
 export declare function sendPending(conn: PooledConnection): boolean;
-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 acquireConnection(wsUrl: string, headers: Record<string, string>, context: TransportContext, signal?: AbortSignal): PooledConnection | Promise<PooledConnection>;
+export declare function ensureWarmConnection(wsUrl: string, headers: Record<string, string>): PooledConnection | undefined;
+export declare function closeConnections(predicate?: (conn: PooledConnection) => boolean, message?: string): void;
 export declare function resetPoolForTesting(): void;
 export declare function setWebSocketConstructorForTesting(ctor: WebSocketConstructor): void;
 export declare function resetWebSocketConstructorForTesting(): void;

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

@@ -1 +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;AAY5G,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,cAAc,EAAE,MAAM,CAAA;IACtB,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,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,cAAc,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IACpD,SAAS,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IAC/C,MAAM,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,IAAI,CAAA;CAC5B;AAED,eAAO,MAAM,cAAc,EAAE,gBAAgB,EAAO,CAAA;AAEpD,eAAO,MAAM,UAAU;;;;;CAKb,CAAA;AA8LV,wBAAgB,WAAW,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAc3D;AAqJD,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAC/B,OAAO,EAAE,gBAAgB,GACxB,gBAAgB,CAmBlB;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"}
+{"version":3,"file":"pool.d.ts","sourceRoot":"","sources":["../../src/transport/pool.ts"],"names":[],"mappings":"AAYA,OAAO,EAAmC,KAAK,oBAAoB,EAAE,KAAK,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAEnH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AAIpD,KAAK,eAAe,GAAG;IACrB,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB,CAAA;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC7B,UAAU,EAAE,+BAA+B,CAAC,UAAU,CAAC,CAAA;IACvD,UAAU,CAAC,EAAE,MAAM,IAAI,CAAA;IACvB,IAAI,EAAE,OAAO,CAAA;IACb,IAAI,EAAE,OAAO,CAAA;IACb,gBAAgB,EAAE,OAAO,CAAA;IACzB,SAAS,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IAC/C,QAAQ,EAAE,eAAe,CAAA;CAC1B;AAED,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAA;IACV,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,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,MAAM,CAAA;IAClB,IAAI,EAAE,OAAO,CAAA;IACb,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,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,uBAAuB,CAAC,EAAE,OAAO,CAAA;IACjC,UAAU,EAAE,MAAM,CAAA;IAClB,iBAAiB,EAAE,MAAM,CAAA;IACzB,SAAS,EAAE,MAAM,CAAA;IACjB,cAAc,EAAE,MAAM,CAAA;IACtB,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,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,cAAc,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IACpD,SAAS,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAA;IAC/C,MAAM,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,IAAI,CAAA;CAC5B;AAED,eAAO,MAAM,cAAc,EAAE,gBAAgB,EAAO,CAAA;AAiBpD,eAAO,MAAM,UAAU;;;;;CAKb,CAAA;AAuWV,wBAAgB,WAAW,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAiB3D;AAwRD,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAC/B,OAAO,EAAE,gBAAgB,EACzB,MAAM,CAAC,EAAE,WAAW,GACnB,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAI9C;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,gCAclF;AAED,wBAAgB,gBAAgB,CAAC,SAAS,GAAE,CAAC,IAAI,EAAE,gBAAgB,KAAK,OAAoB,EAAE,OAAO,SAAqB,QAKzH;AAED,wBAAgB,mBAAmB,SAmBlC;AAED,wBAAgB,iCAAiC,CAAC,IAAI,EAAE,oBAAoB,QAE3E;AAED,wBAAgB,mCAAmC,SAElD"}