@bastani/atomic 0.9.4 → 0.9.5-alpha.10

package/dist/core/copilot-anthropic-sse-repair.js

@@ -0,0 +1,340 @@
+/**
+ * Narrow repair for GitHub Copilot Anthropic Messages SSE streams that carry a
+ * complete Anthropic turn but omit the required terminal `message_stop` event.
+ *
+ * The wrapper is deliberately conservative: it is only installed for Copilot
+ * `/v1/messages` event streams, passes original SSE frames through unchanged,
+ * and appends/inserts a single synthetic `message_stop` only after observing the
+ * same terminal evidence Anthropic's parser requires for a complete message.
+ */
+import { isCopilotApiHost } from "./copilot-hosts.js";
+export const COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT = 'event: message_stop\ndata: {"type":"message_stop"}\n\n';
+const COPILOT_ANTHROPIC_MESSAGES_PATH = "/v1/messages";
+const KNOWN_ANTHROPIC_SSE_EVENT_NAMES = new Set([
+    "message_start",
+    "content_block_start",
+    "content_block_delta",
+    "content_block_stop",
+    "message_delta",
+    "message_stop",
+    "ping",
+    "error",
+]);
+function createRepairState() {
+    return {
+        sawMessageStart: false,
+        sawMessageStop: false,
+        sawTerminalStopReason: false,
+        sawErrorEvent: false,
+        malformedOrUncertain: false,
+        sawDoneSentinel: false,
+        insertedSyntheticStop: false,
+        openContentBlockIndexes: new Set(),
+    };
+}
+function isPlainObject(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function isNonNegativeInteger(value) {
+    return typeof value === "number" && Number.isInteger(value) && value >= 0;
+}
+function markMalformed(state) {
+    state.malformedOrUncertain = true;
+}
+function shouldSynthesizeMessageStop(state) {
+    return (state.sawMessageStart &&
+        !state.sawMessageStop &&
+        state.sawTerminalStopReason &&
+        !state.sawErrorEvent &&
+        !state.malformedOrUncertain &&
+        !state.sawDoneSentinel &&
+        !state.insertedSyntheticStop &&
+        state.openContentBlockIndexes.size === 0);
+}
+function findSseEventBoundary(buffer) {
+    const candidates = ["\r\n\r\n", "\n\n", "\r\r"];
+    let bestIndex = -1;
+    let bestLength = 0;
+    for (const candidate of candidates) {
+        const index = buffer.indexOf(candidate);
+        if (index === -1)
+            continue;
+        if (bestIndex === -1 || index < bestIndex) {
+            bestIndex = index;
+            bestLength = candidate.length;
+        }
+    }
+    return bestIndex === -1 ? -1 : bestIndex + bestLength;
+}
+function stripEventSeparator(frame) {
+    if (frame.endsWith("\r\n\r\n"))
+        return frame.slice(0, -4);
+    if (frame.endsWith("\n\n"))
+        return frame.slice(0, -2);
+    if (frame.endsWith("\r\r"))
+        return frame.slice(0, -2);
+    return frame;
+}
+function missingFinalFrameSeparator(frame) {
+    if (frame.endsWith("\r\n"))
+        return "\r\n";
+    if (frame.endsWith("\n"))
+        return "\n";
+    if (frame.endsWith("\r"))
+        return "\r";
+    return "\n\n";
+}
+function hasKnownAnthropicEventName(eventName) {
+    return eventName !== undefined && KNOWN_ANTHROPIC_SSE_EVENT_NAMES.has(eventName);
+}
+function canObserveOpenMessageLifecycle(state) {
+    if (!state.sawMessageStart || state.sawMessageStop || state.sawTerminalStopReason) {
+        markMalformed(state);
+        return false;
+    }
+    return true;
+}
+function observeContentBlockStart(payload, state) {
+    if (!canObserveOpenMessageLifecycle(state))
+        return;
+    const index = payload.index;
+    if (!isNonNegativeInteger(index) || state.openContentBlockIndexes.has(index)) {
+        markMalformed(state);
+        return;
+    }
+    state.openContentBlockIndexes.add(index);
+}
+function observeContentBlockDelta(payload, state) {
+    if (!canObserveOpenMessageLifecycle(state))
+        return;
+    const index = payload.index;
+    if (!isNonNegativeInteger(index) || !state.openContentBlockIndexes.has(index)) {
+        markMalformed(state);
+    }
+}
+function observeContentBlockStop(payload, state) {
+    if (!canObserveOpenMessageLifecycle(state))
+        return;
+    const index = payload.index;
+    if (!isNonNegativeInteger(index) || !state.openContentBlockIndexes.delete(index)) {
+        markMalformed(state);
+    }
+}
+function observeMessageDelta(payload, state) {
+    if (!canObserveOpenMessageLifecycle(state))
+        return;
+    const delta = payload.delta;
+    if (!isPlainObject(delta))
+        return;
+    const stopReason = delta.stop_reason;
+    if (typeof stopReason === "string" && stopReason.length > 0) {
+        state.sawTerminalStopReason = true;
+    }
+}
+function observeAnthropicPayload(eventName, payload, state) {
+    if (!hasKnownAnthropicEventName(eventName)) {
+        markMalformed(state);
+        return;
+    }
+    const type = payload.type;
+    if (typeof type !== "string" || eventName !== type) {
+        markMalformed(state);
+        return;
+    }
+    switch (type) {
+        case "message_start":
+            if (state.sawMessageStart || state.sawMessageStop || state.sawTerminalStopReason)
+                markMalformed(state);
+            state.sawMessageStart = true;
+            break;
+        case "content_block_start":
+            observeContentBlockStart(payload, state);
+            break;
+        case "content_block_delta":
+            observeContentBlockDelta(payload, state);
+            break;
+        case "content_block_stop":
+            observeContentBlockStop(payload, state);
+            break;
+        case "message_delta":
+            observeMessageDelta(payload, state);
+            break;
+        case "message_stop":
+            if (!state.sawMessageStart || state.sawMessageStop || state.openContentBlockIndexes.size > 0) {
+                markMalformed(state);
+            }
+            state.sawMessageStop = true;
+            break;
+        case "ping":
+            break;
+        case "error":
+            state.sawErrorEvent = true;
+            break;
+    }
+}
+function observeSseFrame(frame, state) {
+    if (state.sawDoneSentinel) {
+        markMalformed(state);
+    }
+    const core = stripEventSeparator(frame);
+    if (core.length === 0)
+        return "empty";
+    let eventName;
+    const dataLines = [];
+    for (const line of core.split(/\r\n|\n|\r/)) {
+        if (line.length === 0 || line.startsWith(":"))
+            continue;
+        const separatorIndex = line.indexOf(":");
+        const field = separatorIndex === -1 ? line : line.slice(0, separatorIndex);
+        let value = separatorIndex === -1 ? "" : line.slice(separatorIndex + 1);
+        if (value.startsWith(" "))
+            value = value.slice(1);
+        if (field === "event") {
+            eventName = value;
+        }
+        else if (field === "data") {
+            dataLines.push(value);
+        }
+    }
+    if (dataLines.length === 0) {
+        if (eventName === "error")
+            state.sawErrorEvent = true;
+        return "empty";
+    }
+    const dataPayload = dataLines.join("\n");
+    if (dataPayload.trim() === "[DONE]") {
+        return "done";
+    }
+    let parsed;
+    try {
+        parsed = JSON.parse(dataPayload);
+    }
+    catch {
+        markMalformed(state);
+        return "data";
+    }
+    if (!isPlainObject(parsed)) {
+        markMalformed(state);
+        return "data";
+    }
+    observeAnthropicPayload(eventName, parsed, state);
+    return "data";
+}
+function drainCompleteFrames(buffer, state) {
+    let rest = buffer;
+    let output = "";
+    for (;;) {
+        const boundaryEnd = findSseEventBoundary(rest);
+        if (boundaryEnd === -1)
+            break;
+        const frame = rest.slice(0, boundaryEnd);
+        rest = rest.slice(boundaryEnd);
+        const observation = observeSseFrame(frame, state);
+        if (observation === "done") {
+            if (shouldSynthesizeMessageStop(state)) {
+                output += COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT;
+                state.insertedSyntheticStop = true;
+                state.sawMessageStop = true;
+            }
+            state.sawDoneSentinel = true;
+        }
+        output += frame;
+    }
+    return { output, rest };
+}
+/**
+ * Wrap a Copilot Anthropic Messages SSE byte stream and repair one narrow class
+ * of provider truncation: missing `message_stop` after a fully closed message.
+ * Original SSE frames are emitted unchanged; the only possible mutation is one
+ * synthetic terminal event at clean EOF or immediately before `[DONE]`.
+ */
+export function createCopilotAnthropicMessagesSseRepairStream(source) {
+    const reader = source.getReader();
+    const decoder = new TextDecoder();
+    const encoder = new TextEncoder();
+    const state = createRepairState();
+    let buffer = "";
+    return new ReadableStream({
+        async pull(controller) {
+            for (;;) {
+                const { done, value } = await reader.read();
+                if (done) {
+                    buffer += decoder.decode();
+                    const drained = drainCompleteFrames(buffer, state);
+                    buffer = drained.rest;
+                    let output = drained.output;
+                    if (buffer.length > 0) {
+                        const trailingFrame = buffer;
+                        const observation = observeSseFrame(trailingFrame, state);
+                        buffer = "";
+                        if (observation === "done") {
+                            if (shouldSynthesizeMessageStop(state)) {
+                                output += COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT;
+                                state.insertedSyntheticStop = true;
+                                state.sawMessageStop = true;
+                            }
+                            state.sawDoneSentinel = true;
+                        }
+                        output += trailingFrame;
+                        if (observation === "data" && shouldSynthesizeMessageStop(state)) {
+                            output += missingFinalFrameSeparator(trailingFrame);
+                        }
+                    }
+                    if (shouldSynthesizeMessageStop(state)) {
+                        output += COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT;
+                        state.insertedSyntheticStop = true;
+                        state.sawMessageStop = true;
+                    }
+                    if (output.length > 0) {
+                        controller.enqueue(encoder.encode(output));
+                    }
+                    controller.close();
+                    return;
+                }
+                buffer += decoder.decode(value, { stream: true });
+                const drained = drainCompleteFrames(buffer, state);
+                buffer = drained.rest;
+                if (drained.output.length > 0) {
+                    controller.enqueue(encoder.encode(drained.output));
+                    return;
+                }
+            }
+        },
+        cancel(reason) {
+            return reader.cancel(reason);
+        },
+    });
+}
+function isCopilotAnthropicMessagesUrl(url) {
+    if (!url)
+        return false;
+    try {
+        const parsed = new URL(url);
+        return isCopilotApiHost(parsed) && parsed.pathname === COPILOT_ANTHROPIC_MESSAGES_PATH;
+    }
+    catch {
+        return false;
+    }
+}
+function isEventStreamResponse(response) {
+    return (response.headers.get("content-type") ?? "").toLowerCase().includes("text/event-stream");
+}
+/**
+ * Scope the Anthropic SSE repair to GitHub Copilot `/v1/messages` event streams
+ * with a body. All other responses are returned as the original instance.
+ */
+export function maybeRepairCopilotAnthropicMessagesResponse(url, response) {
+    if (!isCopilotAnthropicMessagesUrl(url))
+        return response;
+    if (!isEventStreamResponse(response))
+        return response;
+    if (!response.body)
+        return response;
+    const transformed = createCopilotAnthropicMessagesSseRepairStream(response.body);
+    return new Response(transformed, {
+        status: response.status,
+        statusText: response.statusText,
+        headers: response.headers,
+    });
+}
+//# sourceMappingURL=copilot-anthropic-sse-repair.js.map

package/dist/core/copilot-anthropic-sse-repair.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"copilot-anthropic-sse-repair.js","sourceRoot":"","sources":["../../src/core/copilot-anthropic-sse-repair.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAkBtD,MAAM,CAAC,MAAM,8CAA8C,GACzD,wDAAwD,CAAC;AAE3D,MAAM,+BAA+B,GAAG,cAAc,CAAC;AACvD,MAAM,+BAA+B,GAAG,IAAI,GAAG,CAAC;IAC9C,eAAe;IACf,qBAAqB;IACrB,qBAAqB;IACrB,oBAAoB;IACpB,eAAe;IACf,cAAc;IACd,MAAM;IACN,OAAO;CACR,CAAC,CAAC;AAEH,SAAS,iBAAiB;IACxB,OAAO;QACL,eAAe,EAAE,KAAK;QACtB,cAAc,EAAE,KAAK;QACrB,qBAAqB,EAAE,KAAK;QAC5B,aAAa,EAAE,KAAK;QACpB,oBAAoB,EAAE,KAAK;QAC3B,eAAe,EAAE,KAAK;QACtB,qBAAqB,EAAE,KAAK;QAC5B,uBAAuB,EAAE,IAAI,GAAG,EAAU;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAgB;IACrC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,oBAAoB,CAAC,KAA4B;IACxD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,aAAa,CAAC,KAAkB;IACvC,KAAK,CAAC,oBAAoB,GAAG,IAAI,CAAC;AACpC,CAAC;AAED,SAAS,2BAA2B,CAAC,KAAkB;IACrD,OAAO,CACL,KAAK,CAAC,eAAe;QACrB,CAAC,KAAK,CAAC,cAAc;QACrB,KAAK,CAAC,qBAAqB;QAC3B,CAAC,KAAK,CAAC,aAAa;QACpB,CAAC,KAAK,CAAC,oBAAoB;QAC3B,CAAC,KAAK,CAAC,eAAe;QACtB,CAAC,KAAK,CAAC,qBAAqB;QAC5B,KAAK,CAAC,uBAAuB,CAAC,IAAI,KAAK,CAAC,CACzC,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAc;IAC1C,MAAM,UAAU,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IAChD,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;IACnB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,SAAS;QAC3B,IAAI,SAAS,KAAK,CAAC,CAAC,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;YAC1C,SAAS,GAAG,KAAK,CAAC;YAClB,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;QAChC,CAAC;IACH,CAAC;IACD,OAAO,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;AACxD,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAa;IACxC,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;QAAE,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1D,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACtD,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACtD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,0BAA0B,CAAC,KAAa;IAC/C,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IAC1C,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACtC,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACtC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,0BAA0B,CAAC,SAA6B;IAC/D,OAAO,SAAS,KAAK,SAAS,IAAI,+BAA+B,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,8BAA8B,CAAC,KAAkB;IACxD,IAAI,CAAC,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,qBAAqB,EAAE,CAAC;QAClF,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,wBAAwB,CAAC,OAAmB,EAAE,KAAkB;IACvE,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC;QAAE,OAAO;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7E,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO;IACT,CAAC;IACD,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,wBAAwB,CAAC,OAAmB,EAAE,KAAkB;IACvE,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC;QAAE,OAAO;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9E,aAAa,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,OAAmB,EAAE,KAAkB;IACtE,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC;QAAE,OAAO;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACjF,aAAa,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAmB,EAAE,KAAkB;IAClE,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC;QAAE,OAAO;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;QAAE,OAAO;IAClC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,CAAC;IACrC,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5D,KAAK,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACrC,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,SAA6B,EAAE,OAAmB,EAAE,KAAkB;IACrG,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3C,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAC1B,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;QACnD,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO;IACT,CAAC;IAED,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,eAAe;YAClB,IAAI,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,qBAAqB;gBAAE,aAAa,CAAC,KAAK,CAAC,CAAC;YACvG,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC;YAC7B,MAAM;QACR,KAAK,qBAAqB;YACxB,wBAAwB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACzC,MAAM;QACR,KAAK,qBAAqB;YACxB,wBAAwB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACzC,MAAM;QACR,KAAK,oBAAoB;YACvB,uBAAuB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACxC,MAAM;QACR,KAAK,eAAe;YAClB,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACpC,MAAM;QACR,KAAK,cAAc;YACjB,IAAI,CAAC,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,uBAAuB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC7F,aAAa,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YACD,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;YAC5B,MAAM;QACR,KAAK,MAAM;YACT,MAAM;QACR,KAAK,OAAO;YACV,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3B,MAAM;IACV,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,KAAa,EAAE,KAAkB;IACxD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;QAC1B,aAAa,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IAED,MAAM,IAAI,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACxC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAEtC,IAAI,SAA6B,CAAC;IAClC,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;QAC5C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS;QACxD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,cAAc,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;QAC3E,IAAI,KAAK,GAAG,cAAc,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;QACxE,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAElD,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YACtB,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;aAAM,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YAC5B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,IAAI,SAAS,KAAK,OAAO;YAAE,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;QACtD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzC,IAAI,WAAW,CAAC,IAAI,EAAE,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,MAAiB,CAAC;IACtB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAc,CAAC;IAChD,CAAC;IAAC,MAAM,CAAC;QACP,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,uBAAuB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAc,EAAE,KAAkB;IAC7D,IAAI,IAAI,GAAG,MAAM,CAAC;IAClB,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,SAAS,CAAC;QACR,MAAM,WAAW,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,WAAW,KAAK,CAAC,CAAC;YAAE,MAAM;QAE9B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;QACzC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC/B,MAAM,WAAW,GAAG,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAClD,IAAI,WAAW,KAAK,MAAM,EAAE,CAAC;YAC3B,IAAI,2BAA2B,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvC,MAAM,IAAI,8CAA8C,CAAC;gBACzD,KAAK,CAAC,qBAAqB,GAAG,IAAI,CAAC;gBACnC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;YAC9B,CAAC;YACD,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC;QAC/B,CAAC;QACD,MAAM,IAAI,KAAK,CAAC;IAClB,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,6CAA6C,CAC3D,MAAkC;IAElC,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;IAClC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,MAAM,KAAK,GAAG,iBAAiB,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,OAAO,IAAI,cAAc,CAAa;QACpC,KAAK,CAAC,IAAI,CAAC,UAAU;YACnB,SAAS,CAAC;gBACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;oBAC3B,MAAM,OAAO,GAAG,mBAAmB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;oBACnD,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;oBACtB,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;oBAE5B,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACtB,MAAM,aAAa,GAAG,MAAM,CAAC;wBAC7B,MAAM,WAAW,GAAG,eAAe,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;wBAC1D,MAAM,GAAG,EAAE,CAAC;wBACZ,IAAI,WAAW,KAAK,MAAM,EAAE,CAAC;4BAC3B,IAAI,2BAA2B,CAAC,KAAK,CAAC,EAAE,CAAC;gCACvC,MAAM,IAAI,8CAA8C,CAAC;gCACzD,KAAK,CAAC,qBAAqB,GAAG,IAAI,CAAC;gCACnC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;4BAC9B,CAAC;4BACD,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC;wBAC/B,CAAC;wBACD,MAAM,IAAI,aAAa,CAAC;wBACxB,IAAI,WAAW,KAAK,MAAM,IAAI,2BAA2B,CAAC,KAAK,CAAC,EAAE,CAAC;4BACjE,MAAM,IAAI,0BAA0B,CAAC,aAAa,CAAC,CAAC;wBACtD,CAAC;oBACH,CAAC;oBAED,IAAI,2BAA2B,CAAC,KAAK,CAAC,EAAE,CAAC;wBACvC,MAAM,IAAI,8CAA8C,CAAC;wBACzD,KAAK,CAAC,qBAAqB,GAAG,IAAI,CAAC;wBACnC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;oBAC9B,CAAC;oBAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACtB,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;oBAC7C,CAAC;oBACD,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnB,OAAO;gBACT,CAAC;gBAED,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,OAAO,GAAG,mBAAmB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBACnD,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;gBACtB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC9B,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;oBACnD,OAAO;gBACT,CAAC;YACH,CAAC;QACH,CAAC;QACD,MAAM,CAAC,MAAM;YACX,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,SAAS,6BAA6B,CAAC,GAAuB;IAC5D,IAAI,CAAC,GAAG;QAAE,OAAO,KAAK,CAAC;IACvB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5B,OAAO,gBAAgB,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,KAAK,+BAA+B,CAAC;IACzF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAkB;IAC/C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;AAClG,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,2CAA2C,CACzD,GAAuB,EACvB,QAAkB;IAElB,IAAI,CAAC,6BAA6B,CAAC,GAAG,CAAC;QAAE,OAAO,QAAQ,CAAC;IACzD,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IACtD,IAAI,CAAC,QAAQ,CAAC,IAAI;QAAE,OAAO,QAAQ,CAAC;IAEpC,MAAM,WAAW,GAAG,6CAA6C,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjF,OAAO,IAAI,QAAQ,CAAC,WAAW,EAAE;QAC/B,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;KAC1B,CAAC,CAAC;AACL,CAAC","sourcesContent":["/**\n * Narrow repair for GitHub Copilot Anthropic Messages SSE streams that carry a\n * complete Anthropic turn but omit the required terminal `message_stop` event.\n *\n * The wrapper is deliberately conservative: it is only installed for Copilot\n * `/v1/messages` event streams, passes original SSE frames through unchanged,\n * and appends/inserts a single synthetic `message_stop` only after observing the\n * same terminal evidence Anthropic's parser requires for a complete message.\n */\n\nimport { isCopilotApiHost } from \"./copilot-hosts.ts\";\n\ntype JsonObject = { [key: string]: JsonValue };\ntype JsonValue = string | number | boolean | null | JsonValue[] | JsonObject;\n\ntype SseObservation = \"data\" | \"done\" | \"empty\";\n\ninterface RepairState {\n  sawMessageStart: boolean;\n  sawMessageStop: boolean;\n  sawTerminalStopReason: boolean;\n  sawErrorEvent: boolean;\n  malformedOrUncertain: boolean;\n  sawDoneSentinel: boolean;\n  insertedSyntheticStop: boolean;\n  openContentBlockIndexes: Set<number>;\n}\n\nexport const COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT =\n  'event: message_stop\\ndata: {\"type\":\"message_stop\"}\\n\\n';\n\nconst COPILOT_ANTHROPIC_MESSAGES_PATH = \"/v1/messages\";\nconst KNOWN_ANTHROPIC_SSE_EVENT_NAMES = new Set([\n  \"message_start\",\n  \"content_block_start\",\n  \"content_block_delta\",\n  \"content_block_stop\",\n  \"message_delta\",\n  \"message_stop\",\n  \"ping\",\n  \"error\",\n]);\n\nfunction createRepairState(): RepairState {\n  return {\n    sawMessageStart: false,\n    sawMessageStop: false,\n    sawTerminalStopReason: false,\n    sawErrorEvent: false,\n    malformedOrUncertain: false,\n    sawDoneSentinel: false,\n    insertedSyntheticStop: false,\n    openContentBlockIndexes: new Set<number>(),\n  };\n}\n\nfunction isPlainObject(value: JsonValue): value is JsonObject {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nfunction isNonNegativeInteger(value: JsonValue | undefined): value is number {\n  return typeof value === \"number\" && Number.isInteger(value) && value >= 0;\n}\n\nfunction markMalformed(state: RepairState): void {\n  state.malformedOrUncertain = true;\n}\n\nfunction shouldSynthesizeMessageStop(state: RepairState): boolean {\n  return (\n    state.sawMessageStart &&\n    !state.sawMessageStop &&\n    state.sawTerminalStopReason &&\n    !state.sawErrorEvent &&\n    !state.malformedOrUncertain &&\n    !state.sawDoneSentinel &&\n    !state.insertedSyntheticStop &&\n    state.openContentBlockIndexes.size === 0\n  );\n}\n\nfunction findSseEventBoundary(buffer: string): number {\n  const candidates = [\"\\r\\n\\r\\n\", \"\\n\\n\", \"\\r\\r\"];\n  let bestIndex = -1;\n  let bestLength = 0;\n  for (const candidate of candidates) {\n    const index = buffer.indexOf(candidate);\n    if (index === -1) continue;\n    if (bestIndex === -1 || index < bestIndex) {\n      bestIndex = index;\n      bestLength = candidate.length;\n    }\n  }\n  return bestIndex === -1 ? -1 : bestIndex + bestLength;\n}\n\nfunction stripEventSeparator(frame: string): string {\n  if (frame.endsWith(\"\\r\\n\\r\\n\")) return frame.slice(0, -4);\n  if (frame.endsWith(\"\\n\\n\")) return frame.slice(0, -2);\n  if (frame.endsWith(\"\\r\\r\")) return frame.slice(0, -2);\n  return frame;\n}\n\nfunction missingFinalFrameSeparator(frame: string): string {\n  if (frame.endsWith(\"\\r\\n\")) return \"\\r\\n\";\n  if (frame.endsWith(\"\\n\")) return \"\\n\";\n  if (frame.endsWith(\"\\r\")) return \"\\r\";\n  return \"\\n\\n\";\n}\n\nfunction hasKnownAnthropicEventName(eventName: string | undefined): eventName is string {\n  return eventName !== undefined && KNOWN_ANTHROPIC_SSE_EVENT_NAMES.has(eventName);\n}\n\nfunction canObserveOpenMessageLifecycle(state: RepairState): boolean {\n  if (!state.sawMessageStart || state.sawMessageStop || state.sawTerminalStopReason) {\n    markMalformed(state);\n    return false;\n  }\n  return true;\n}\n\nfunction observeContentBlockStart(payload: JsonObject, state: RepairState): void {\n  if (!canObserveOpenMessageLifecycle(state)) return;\n  const index = payload.index;\n  if (!isNonNegativeInteger(index) || state.openContentBlockIndexes.has(index)) {\n    markMalformed(state);\n    return;\n  }\n  state.openContentBlockIndexes.add(index);\n}\n\nfunction observeContentBlockDelta(payload: JsonObject, state: RepairState): void {\n  if (!canObserveOpenMessageLifecycle(state)) return;\n  const index = payload.index;\n  if (!isNonNegativeInteger(index) || !state.openContentBlockIndexes.has(index)) {\n    markMalformed(state);\n  }\n}\n\nfunction observeContentBlockStop(payload: JsonObject, state: RepairState): void {\n  if (!canObserveOpenMessageLifecycle(state)) return;\n  const index = payload.index;\n  if (!isNonNegativeInteger(index) || !state.openContentBlockIndexes.delete(index)) {\n    markMalformed(state);\n  }\n}\n\nfunction observeMessageDelta(payload: JsonObject, state: RepairState): void {\n  if (!canObserveOpenMessageLifecycle(state)) return;\n  const delta = payload.delta;\n  if (!isPlainObject(delta)) return;\n  const stopReason = delta.stop_reason;\n  if (typeof stopReason === \"string\" && stopReason.length > 0) {\n    state.sawTerminalStopReason = true;\n  }\n}\n\nfunction observeAnthropicPayload(eventName: string | undefined, payload: JsonObject, state: RepairState): void {\n  if (!hasKnownAnthropicEventName(eventName)) {\n    markMalformed(state);\n    return;\n  }\n\n  const type = payload.type;\n  if (typeof type !== \"string\" || eventName !== type) {\n    markMalformed(state);\n    return;\n  }\n\n  switch (type) {\n    case \"message_start\":\n      if (state.sawMessageStart || state.sawMessageStop || state.sawTerminalStopReason) markMalformed(state);\n      state.sawMessageStart = true;\n      break;\n    case \"content_block_start\":\n      observeContentBlockStart(payload, state);\n      break;\n    case \"content_block_delta\":\n      observeContentBlockDelta(payload, state);\n      break;\n    case \"content_block_stop\":\n      observeContentBlockStop(payload, state);\n      break;\n    case \"message_delta\":\n      observeMessageDelta(payload, state);\n      break;\n    case \"message_stop\":\n      if (!state.sawMessageStart || state.sawMessageStop || state.openContentBlockIndexes.size > 0) {\n        markMalformed(state);\n      }\n      state.sawMessageStop = true;\n      break;\n    case \"ping\":\n      break;\n    case \"error\":\n      state.sawErrorEvent = true;\n      break;\n  }\n}\n\nfunction observeSseFrame(frame: string, state: RepairState): SseObservation {\n  if (state.sawDoneSentinel) {\n    markMalformed(state);\n  }\n\n  const core = stripEventSeparator(frame);\n  if (core.length === 0) return \"empty\";\n\n  let eventName: string | undefined;\n  const dataLines: string[] = [];\n  for (const line of core.split(/\\r\\n|\\n|\\r/)) {\n    if (line.length === 0 || line.startsWith(\":\")) continue;\n    const separatorIndex = line.indexOf(\":\");\n    const field = separatorIndex === -1 ? line : line.slice(0, separatorIndex);\n    let value = separatorIndex === -1 ? \"\" : line.slice(separatorIndex + 1);\n    if (value.startsWith(\" \")) value = value.slice(1);\n\n    if (field === \"event\") {\n      eventName = value;\n    } else if (field === \"data\") {\n      dataLines.push(value);\n    }\n  }\n\n  if (dataLines.length === 0) {\n    if (eventName === \"error\") state.sawErrorEvent = true;\n    return \"empty\";\n  }\n\n  const dataPayload = dataLines.join(\"\\n\");\n  if (dataPayload.trim() === \"[DONE]\") {\n    return \"done\";\n  }\n\n  let parsed: JsonValue;\n  try {\n    parsed = JSON.parse(dataPayload) as JsonValue;\n  } catch {\n    markMalformed(state);\n    return \"data\";\n  }\n\n  if (!isPlainObject(parsed)) {\n    markMalformed(state);\n    return \"data\";\n  }\n\n  observeAnthropicPayload(eventName, parsed, state);\n  return \"data\";\n}\n\nfunction drainCompleteFrames(buffer: string, state: RepairState): { output: string; rest: string } {\n  let rest = buffer;\n  let output = \"\";\n  for (;;) {\n    const boundaryEnd = findSseEventBoundary(rest);\n    if (boundaryEnd === -1) break;\n\n    const frame = rest.slice(0, boundaryEnd);\n    rest = rest.slice(boundaryEnd);\n    const observation = observeSseFrame(frame, state);\n    if (observation === \"done\") {\n      if (shouldSynthesizeMessageStop(state)) {\n        output += COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT;\n        state.insertedSyntheticStop = true;\n        state.sawMessageStop = true;\n      }\n      state.sawDoneSentinel = true;\n    }\n    output += frame;\n  }\n  return { output, rest };\n}\n\n/**\n * Wrap a Copilot Anthropic Messages SSE byte stream and repair one narrow class\n * of provider truncation: missing `message_stop` after a fully closed message.\n * Original SSE frames are emitted unchanged; the only possible mutation is one\n * synthetic terminal event at clean EOF or immediately before `[DONE]`.\n */\nexport function createCopilotAnthropicMessagesSseRepairStream(\n  source: ReadableStream<Uint8Array>,\n): ReadableStream<Uint8Array> {\n  const reader = source.getReader();\n  const decoder = new TextDecoder();\n  const encoder = new TextEncoder();\n  const state = createRepairState();\n  let buffer = \"\";\n\n  return new ReadableStream<Uint8Array>({\n    async pull(controller) {\n      for (;;) {\n        const { done, value } = await reader.read();\n        if (done) {\n          buffer += decoder.decode();\n          const drained = drainCompleteFrames(buffer, state);\n          buffer = drained.rest;\n          let output = drained.output;\n\n          if (buffer.length > 0) {\n            const trailingFrame = buffer;\n            const observation = observeSseFrame(trailingFrame, state);\n            buffer = \"\";\n            if (observation === \"done\") {\n              if (shouldSynthesizeMessageStop(state)) {\n                output += COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT;\n                state.insertedSyntheticStop = true;\n                state.sawMessageStop = true;\n              }\n              state.sawDoneSentinel = true;\n            }\n            output += trailingFrame;\n            if (observation === \"data\" && shouldSynthesizeMessageStop(state)) {\n              output += missingFinalFrameSeparator(trailingFrame);\n            }\n          }\n\n          if (shouldSynthesizeMessageStop(state)) {\n            output += COPILOT_ANTHROPIC_SYNTHETIC_MESSAGE_STOP_EVENT;\n            state.insertedSyntheticStop = true;\n            state.sawMessageStop = true;\n          }\n\n          if (output.length > 0) {\n            controller.enqueue(encoder.encode(output));\n          }\n          controller.close();\n          return;\n        }\n\n        buffer += decoder.decode(value, { stream: true });\n        const drained = drainCompleteFrames(buffer, state);\n        buffer = drained.rest;\n        if (drained.output.length > 0) {\n          controller.enqueue(encoder.encode(drained.output));\n          return;\n        }\n      }\n    },\n    cancel(reason) {\n      return reader.cancel(reason);\n    },\n  });\n}\n\nfunction isCopilotAnthropicMessagesUrl(url: string | undefined): boolean {\n  if (!url) return false;\n  try {\n    const parsed = new URL(url);\n    return isCopilotApiHost(parsed) && parsed.pathname === COPILOT_ANTHROPIC_MESSAGES_PATH;\n  } catch {\n    return false;\n  }\n}\n\nfunction isEventStreamResponse(response: Response): boolean {\n  return (response.headers.get(\"content-type\") ?? \"\").toLowerCase().includes(\"text/event-stream\");\n}\n\n/**\n * Scope the Anthropic SSE repair to GitHub Copilot `/v1/messages` event streams\n * with a body. All other responses are returned as the original instance.\n */\nexport function maybeRepairCopilotAnthropicMessagesResponse(\n  url: string | undefined,\n  response: Response,\n): Response {\n  if (!isCopilotAnthropicMessagesUrl(url)) return response;\n  if (!isEventStreamResponse(response)) return response;\n  if (!response.body) return response;\n\n  const transformed = createCopilotAnthropicMessagesSseRepairStream(response.body);\n  return new Response(transformed, {\n    status: response.status,\n    statusText: response.statusText,\n    headers: response.headers,\n  });\n}\n"]}

package/dist/core/copilot-gemini-reasoning.d.ts

@@ -103,16 +103,24 @@ export declare function restoreCopilotGeminiReasoningOpaque(payload: unknown, mo
  */
 export declare function maybeRewriteCopilotGeminiResponse(url: string | undefined, response: Response): Response;
 /**
- * Install a `globalThis.fetch` wrapper that rewrites CAPI Gemini SSE responses
- * to bridge `reasoning_opaque` into `reasoning_details` (see
- * {@link createCopilotGeminiSseStream}). Idempotent.
+ * Apply Copilot response-body transforms before pi-ai parses streamed provider
+ * output. Anthropic Messages repair gets first refusal for `/v1/messages`
+ * streams; other Copilot event streams keep the existing Gemini reasoning bridge.
+ */
+export declare function maybeRewriteCopilotProviderResponse(url: string | undefined, response: Response): Response;
+/**
+ * Install a `globalThis.fetch` wrapper that applies Copilot-only stream
+ * compatibility shims before pi-ai parses provider output. Currently this
+ * includes the Gemini `reasoning_opaque` bridge and the Anthropic Messages
+ * terminal `message_stop` repair. Idempotent.
  *
- * The OpenAI SDK used by the `openai-completions` provider resolves
- * `globalThis.fetch` at client-construction time, and a new client is built per
- * request, so wrapping the global before the first request is reliably picked
- * up. Non-Copilot hosts and non-event-stream responses are returned untouched,
- * keeping the blast radius to streaming CAPI Gemini turns only.
+ * SDK clients resolve `globalThis.fetch` at client-construction time, and a new
+ * client is built per request, so wrapping the global before the first request
+ * is reliably picked up. Non-Copilot hosts and non-event-stream responses are
+ * returned untouched by the underlying shims.
  */
+export declare function installCopilotResponseInterceptor(): void;
+/** Backward-compatible name for the original Gemini-only interceptor export. */
 export declare function installCopilotGeminiReasoningInterceptor(): void;
 export {};
 //# sourceMappingURL=copilot-gemini-reasoning.d.ts.map

package/dist/core/copilot-gemini-reasoning.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"copilot-gemini-reasoning.d.ts","sourceRoot":"","sources":["../../src/core/copilot-gemini-reasoning.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,8BAA8B,CAAC;AAG/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkDG;AAEH,KAAK,UAAU,GAAG;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAC;AAC/C,KAAK,SAAS,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,GAAG,SAAS,EAAE,GAAG,UAAU,CAAC;AAS7E;;;;;;;;;;;;GAYG;AACH,wBAAgB,mCAAmC,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAiC7E;AAED;;;;GAIG;AACH,wBAAgB,2BAA2B,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAWvE;AAeD;;;;;;;;GAQG;AACH,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,GACjC,cAAc,CAAC,UAAU,CAAC,CAyC5B;AAED;;;;;;;;;GASG;AACH,wBAAgB,mCAAmC,CACjD,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,GAAG,KAAK,GAAG,IAAI,CAAC,GACjD,OAAO,CA2BT;AAyBD;;;;;GAKG;AACH,wBAAgB,iCAAiC,CAC/C,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,QAAQ,EAAE,QAAQ,GACjB,QAAQ,CAWV;AAID;;;;;;;;;;GAUG;AACH,wBAAgB,wCAAwC,IAAI,IAAI,CAuB/D","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai/compat\";\nimport { isCopilotGeminiModel } from \"./copilot-gemini-payload-sanitizer.ts\";\n\n/**\n * Round-trips GitHub Copilot Gemini \"thought signatures\" so multi-turn tool use\n * does not silently die after the first tool call.\n *\n * Why this exists\n * ---------------\n * `github-copilot` Gemini models (e.g. `gemini-3.1-pro-preview`) are served\n * through Copilot's CAPI gateway, which proxies to Google's GenAI API. Gemini\n * is a thinking model: when it emits a function/tool call it also returns an\n * opaque **thought signature** that must be sent back, verbatim, on the next\n * request or Gemini refuses to continue the reasoning chain.\n *\n * CAPI carries that signature in a non-standard OpenAI-completions field named\n * **`reasoning_opaque`** (an encrypted blob) on the assistant message / streamed\n * delta, and on replay it reads the same `reasoning_opaque` back off the\n * assistant message and re-attaches the signature to each Gemini function-call\n * part (keyed by `tool_call.id`). The underlying OpenAI-completions client\n * (`@earendil-works/pi-ai`) does not understand `reasoning_opaque`; it captures\n * thought signatures only from the OpenRouter-style\n * `reasoning_details: [{ type: \"reasoning.encrypted\", id, data }]` shape, which\n * CAPI never emits. So the real Gemini thought signature was being dropped on\n * the way in and never replayed on the way out.\n *\n * With the signature missing, CAPI substitutes the sentinel\n * `skip_thought_signature_validator` on the first replayed function call, and\n * Gemini responds with an empty candidate / `finish_reason: \"stop\"` and zero\n * output tokens — the harness sees a degenerate empty completion, retries with\n * the same signature-less history, and eventually gives up: \"Gemini just stops\n * responding.\"\n *\n * What this does\n * --------------\n * Two gated, self-contained transforms bridge CAPI's `reasoning_opaque` to the\n * `reasoning_details` mechanism the client already round-trips:\n *\n *  - **Inbound** ({@link rewriteCopilotGeminiSseData} via\n *    {@link createCopilotGeminiSseStream}): rewrites the CAPI Gemini SSE\n *    stream so each streamed delta that carries both `reasoning_opaque` and a\n *    `tool_calls[].id` gains a\n *    `reasoning_details: [{ type: \"reasoning.encrypted\", id, data: <opaque> }]`\n *    entry. The client then stores it as the tool call's `thoughtSignature`.\n *    CAPI confirms `reasoning_opaque` rides on the same streamed delta as the\n *    first (id-bearing) tool-call chunk, so the association is exact.\n *  - **Outbound** ({@link restoreCopilotGeminiReasoningOpaque} from the\n *    `onPayload` hook): converts the `reasoning_details` the client re-emits on\n *    replayed assistant messages back into a single `reasoning_opaque` field on\n *    that assistant message, which is the only shape CAPI reads.\n *\n * Both transforms are gated to GitHub Copilot Gemini and are no-ops for every\n * other provider/model (and for Gemini turns that carry no thought signature).\n */\n\ntype JsonObject = { [key: string]: JsonValue };\ntype JsonValue = string | number | boolean | null | JsonValue[] | JsonObject;\n\n/** OpenRouter-style encrypted reasoning detail the pi-ai client round-trips. */\nconst REASONING_ENCRYPTED_TYPE = \"reasoning.encrypted\";\n\nfunction isPlainObject(value: unknown): value is JsonObject {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\n/**\n * Inject `reasoning_details` into a parsed CAPI Gemini streaming chunk so the\n * pi-ai OpenAI-completions parser captures the Gemini thought signature.\n *\n * For each `choices[].delta` that carries a non-empty `reasoning_opaque` string\n * and a `tool_calls[]` entry with an `id`, adds a single\n * `reasoning_details: [{ type: \"reasoning.encrypted\", id, data: <opaque> }]`\n * entry keyed by that tool-call id. Returns whether the chunk was mutated.\n *\n * No-op when the delta already has `reasoning_details`, has no id-bearing tool\n * call (e.g. argument-continuation deltas or pure-text thought chunks), or has\n * no `reasoning_opaque`.\n */\nexport function injectCopilotGeminiReasoningDetails(chunk: JsonValue): boolean {\n  if (!isPlainObject(chunk)) return false;\n  const choices = chunk.choices;\n  if (!Array.isArray(choices)) return false;\n\n  let mutated = false;\n  for (const choice of choices) {\n    if (!isPlainObject(choice)) continue;\n    const delta = choice.delta;\n    if (!isPlainObject(delta)) continue;\n\n    const opaque = delta.reasoning_opaque;\n    if (typeof opaque !== \"string\" || opaque.length === 0) continue;\n\n    // Already carries the encrypted detail (don't double-inject / clobber).\n    if (Array.isArray(delta.reasoning_details) && delta.reasoning_details.length > 0) {\n      continue;\n    }\n\n    const toolCalls = delta.tool_calls;\n    if (!Array.isArray(toolCalls)) continue;\n    const idBearing = toolCalls.find(\n      (call): call is JsonObject =>\n        isPlainObject(call) && typeof call.id === \"string\" && call.id.length > 0,\n    );\n    if (!idBearing) continue;\n\n    delta.reasoning_details = [\n      { type: REASONING_ENCRYPTED_TYPE, id: idBearing.id as string, data: opaque },\n    ];\n    mutated = true;\n  }\n  return mutated;\n}\n\n/**\n * Rewrite the JSON payload of a single SSE `data:` line. Returns the original\n * string unchanged when it is not a Gemini chunk that needs a thought signature\n * bridged, or when parsing fails (fail-open: never corrupt the stream).\n */\nexport function rewriteCopilotGeminiSseData(dataPayload: string): string {\n  // Cheap gate: only chunks that actually carry a thought signature are touched.\n  if (!dataPayload.includes(\"reasoning_opaque\")) return dataPayload;\n  let parsed: JsonValue;\n  try {\n    parsed = JSON.parse(dataPayload) as JsonValue;\n  } catch {\n    return dataPayload;\n  }\n  if (!injectCopilotGeminiReasoningDetails(parsed)) return dataPayload;\n  return JSON.stringify(parsed);\n}\n\n/** Rewrite one SSE line, preserving a trailing carriage return when present. */\nfunction rewriteSseLine(line: string): string {\n  const hasCr = line.endsWith(\"\\r\");\n  const core = hasCr ? line.slice(0, -1) : line;\n  if (!core.startsWith(\"data:\")) return line;\n  const payload = core.slice(\"data:\".length).trimStart();\n  if (payload.length === 0 || payload === \"[DONE]\") return line;\n  const rewritten = rewriteCopilotGeminiSseData(payload);\n  if (rewritten === payload) return line;\n  const rebuilt = `data: ${rewritten}`;\n  return hasCr ? `${rebuilt}\\r` : rebuilt;\n}\n\n/**\n * Wrap a CAPI Gemini SSE byte stream so `reasoning_opaque` is bridged into\n * `reasoning_details`. Buffers across chunk boundaries and rewrites whole lines\n * only; bytes that are not affected pass through unchanged.\n *\n * Implemented as a `ReadableStream` over the source reader (rather than a\n * `TransformStream` piped via `pipeThrough`) so the transform pulls lazily and\n * propagates cancellation to the upstream body.\n */\nexport function createCopilotGeminiSseStream(\n  source: ReadableStream<Uint8Array>,\n): ReadableStream<Uint8Array> {\n  const reader = source.getReader();\n  const decoder = new TextDecoder();\n  const encoder = new TextEncoder();\n  let buffer = \"\";\n\n  return new ReadableStream<Uint8Array>({\n    async pull(controller) {\n      // Loop until we emit at least one chunk or close, so a read that yields no\n      // complete line still makes progress without relying on the runtime to\n      // re-invoke pull after a no-enqueue return.\n      for (;;) {\n        const { done, value } = await reader.read();\n        if (done) {\n          buffer += decoder.decode();\n          if (buffer.length > 0) {\n            controller.enqueue(encoder.encode(rewriteSseLine(buffer)));\n            buffer = \"\";\n          }\n          controller.close();\n          return;\n        }\n        buffer += decoder.decode(value, { stream: true });\n        let newlineIndex = buffer.indexOf(\"\\n\");\n        let out = \"\";\n        while (newlineIndex !== -1) {\n          const line = buffer.slice(0, newlineIndex);\n          buffer = buffer.slice(newlineIndex + 1);\n          out += `${rewriteSseLine(line)}\\n`;\n          newlineIndex = buffer.indexOf(\"\\n\");\n        }\n        if (out.length > 0) {\n          controller.enqueue(encoder.encode(out));\n          return;\n        }\n      }\n    },\n    cancel(reason) {\n      return reader.cancel(reason);\n    },\n  });\n}\n\n/**\n * Convert the `reasoning_details` the pi-ai client re-emits on replayed\n * assistant messages back into the single `reasoning_opaque` field CAPI reads.\n *\n * For GitHub Copilot Gemini payloads, each assistant message that carries a\n * `reasoning_details` entry of `type: \"reasoning.encrypted\"` has its `data`\n * (the original CAPI thought-signature blob) promoted to `reasoning_opaque`,\n * and the now-redundant `reasoning_details` removed. No-op for every other\n * provider/model and for payloads without such messages.\n */\nexport function restoreCopilotGeminiReasoningOpaque(\n  payload: unknown,\n  model: Pick<Model<Api>, \"provider\" | \"api\" | \"id\">,\n): unknown {\n  if (!isCopilotGeminiModel(model)) return payload;\n  if (!isPlainObject(payload as JsonValue)) return payload;\n  const payloadObject = payload as JsonObject;\n  const messages = payloadObject.messages;\n  if (!Array.isArray(messages)) return payload;\n\n  let mutated = false;\n  const nextMessages = messages.map((message) => {\n    if (!isPlainObject(message) || message.role !== \"assistant\") return message;\n    const details = message.reasoning_details;\n    if (!Array.isArray(details) || details.length === 0) return message;\n    const encrypted = details.find(\n      (detail): detail is JsonObject =>\n        isPlainObject(detail) &&\n        detail.type === REASONING_ENCRYPTED_TYPE &&\n        typeof detail.data === \"string\" &&\n        detail.data.length > 0,\n    );\n    if (!encrypted) return message;\n    mutated = true;\n    const { reasoning_details: _omitted, ...rest } = message;\n    return { ...rest, reasoning_opaque: encrypted.data as string };\n  });\n\n  if (!mutated) return payload;\n  return { ...payloadObject, messages: nextMessages };\n}\n\n/** Whether the URL targets Copilot's CAPI gateway (`*.githubcopilot.com`). */\nfunction isCopilotApiHost(url: string): boolean {\n  try {\n    const host = new URL(url).hostname.toLowerCase();\n    // Exact host or a real subdomain only — never a look-alike suffix such as\n    // `notgithubcopilot.com` (CodeQL: incomplete URL substring sanitization).\n    return host === \"githubcopilot.com\" || host.endsWith(\".githubcopilot.com\");\n  } catch {\n    return false;\n  }\n}\n\n/** Resolve the request URL string from a `fetch` input argument. */\nfunction resolveRequestUrl(input: Parameters<typeof fetch>[0]): string | undefined {\n  if (typeof input === \"string\") return input;\n  if (input instanceof URL) return input.href;\n  if (typeof input === \"object\" && input !== null && \"url\" in input) {\n    const url = (input as { url?: unknown }).url;\n    if (typeof url === \"string\") return url;\n  }\n  return undefined;\n}\n\n/**\n * Rewrite a streaming CAPI Gemini response so its SSE body bridges\n * `reasoning_opaque` into `reasoning_details`. Returns the original response\n * untouched for non-Copilot hosts, non-event-stream responses, or bodyless\n * responses, keeping the blast radius to streaming CAPI Gemini turns only.\n */\nexport function maybeRewriteCopilotGeminiResponse(\n  url: string | undefined,\n  response: Response,\n): Response {\n  if (!url || !isCopilotApiHost(url)) return response;\n  const contentType = response.headers.get(\"content-type\") ?? \"\";\n  if (!contentType.includes(\"text/event-stream\")) return response;\n  if (!response.body) return response;\n  const transformed = createCopilotGeminiSseStream(response.body);\n  return new Response(transformed, {\n    status: response.status,\n    statusText: response.statusText,\n    headers: response.headers,\n  });\n}\n\nlet originalFetch: typeof fetch | undefined;\n\n/**\n * Install a `globalThis.fetch` wrapper that rewrites CAPI Gemini SSE responses\n * to bridge `reasoning_opaque` into `reasoning_details` (see\n * {@link createCopilotGeminiSseStream}). Idempotent.\n *\n * The OpenAI SDK used by the `openai-completions` provider resolves\n * `globalThis.fetch` at client-construction time, and a new client is built per\n * request, so wrapping the global before the first request is reliably picked\n * up. Non-Copilot hosts and non-event-stream responses are returned untouched,\n * keeping the blast radius to streaming CAPI Gemini turns only.\n */\nexport function installCopilotGeminiReasoningInterceptor(): void {\n  if (originalFetch) return;\n  if (typeof globalThis.fetch !== \"function\") return;\n  const base = globalThis.fetch;\n  originalFetch = base;\n  const boundFetch = base.bind(globalThis);\n\n  const wrapped = (async (input, init) => {\n    const response = await boundFetch(input, init);\n    try {\n      return maybeRewriteCopilotGeminiResponse(resolveRequestUrl(input), response);\n    } catch {\n      return response;\n    }\n  }) as typeof fetch;\n\n  // Preserve `fetch.preconnect` so the wrapper remains a drop-in replacement.\n  const preconnect = (base as { preconnect?: unknown }).preconnect;\n  if (typeof preconnect === \"function\") {\n    (wrapped as { preconnect?: unknown }).preconnect = preconnect.bind(globalThis);\n  }\n\n  globalThis.fetch = wrapped;\n}\n"]}
+{"version":3,"file":"copilot-gemini-reasoning.d.ts","sourceRoot":"","sources":["../../src/core/copilot-gemini-reasoning.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,8BAA8B,CAAC;AAK/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkDG;AAEH,KAAK,UAAU,GAAG;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAC;AAC/C,KAAK,SAAS,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,GAAG,SAAS,EAAE,GAAG,UAAU,CAAC;AAS7E;;;;;;;;;;;;GAYG;AACH,wBAAgB,mCAAmC,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAiC7E;AAED;;;;GAIG;AACH,wBAAgB,2BAA2B,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAWvE;AAeD;;;;;;;;GAQG;AACH,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,GACjC,cAAc,CAAC,UAAU,CAAC,CAyC5B;AAED;;;;;;;;;GASG;AACH,wBAAgB,mCAAmC,CACjD,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,GAAG,KAAK,GAAG,IAAI,CAAC,GACjD,OAAO,CA2BT;AAaD;;;;;GAKG;AACH,wBAAgB,iCAAiC,CAC/C,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,QAAQ,EAAE,QAAQ,GACjB,QAAQ,CAWV;AAED;;;;GAIG;AACH,wBAAgB,mCAAmC,CACjD,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,QAAQ,EAAE,QAAQ,GACjB,QAAQ,CAIV;AAID;;;;;;;;;;GAUG;AACH,wBAAgB,iCAAiC,IAAI,IAAI,CAuBxD;AAED,gFAAgF;AAChF,wBAAgB,wCAAwC,IAAI,IAAI,CAE/D","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai/compat\";\nimport { maybeRepairCopilotAnthropicMessagesResponse } from \"./copilot-anthropic-sse-repair.ts\";\nimport { isCopilotApiHost } from \"./copilot-hosts.ts\";\nimport { isCopilotGeminiModel } from \"./copilot-gemini-payload-sanitizer.ts\";\n\n/**\n * Round-trips GitHub Copilot Gemini \"thought signatures\" so multi-turn tool use\n * does not silently die after the first tool call.\n *\n * Why this exists\n * ---------------\n * `github-copilot` Gemini models (e.g. `gemini-3.1-pro-preview`) are served\n * through Copilot's CAPI gateway, which proxies to Google's GenAI API. Gemini\n * is a thinking model: when it emits a function/tool call it also returns an\n * opaque **thought signature** that must be sent back, verbatim, on the next\n * request or Gemini refuses to continue the reasoning chain.\n *\n * CAPI carries that signature in a non-standard OpenAI-completions field named\n * **`reasoning_opaque`** (an encrypted blob) on the assistant message / streamed\n * delta, and on replay it reads the same `reasoning_opaque` back off the\n * assistant message and re-attaches the signature to each Gemini function-call\n * part (keyed by `tool_call.id`). The underlying OpenAI-completions client\n * (`@earendil-works/pi-ai`) does not understand `reasoning_opaque`; it captures\n * thought signatures only from the OpenRouter-style\n * `reasoning_details: [{ type: \"reasoning.encrypted\", id, data }]` shape, which\n * CAPI never emits. So the real Gemini thought signature was being dropped on\n * the way in and never replayed on the way out.\n *\n * With the signature missing, CAPI substitutes the sentinel\n * `skip_thought_signature_validator` on the first replayed function call, and\n * Gemini responds with an empty candidate / `finish_reason: \"stop\"` and zero\n * output tokens — the harness sees a degenerate empty completion, retries with\n * the same signature-less history, and eventually gives up: \"Gemini just stops\n * responding.\"\n *\n * What this does\n * --------------\n * Two gated, self-contained transforms bridge CAPI's `reasoning_opaque` to the\n * `reasoning_details` mechanism the client already round-trips:\n *\n *  - **Inbound** ({@link rewriteCopilotGeminiSseData} via\n *    {@link createCopilotGeminiSseStream}): rewrites the CAPI Gemini SSE\n *    stream so each streamed delta that carries both `reasoning_opaque` and a\n *    `tool_calls[].id` gains a\n *    `reasoning_details: [{ type: \"reasoning.encrypted\", id, data: <opaque> }]`\n *    entry. The client then stores it as the tool call's `thoughtSignature`.\n *    CAPI confirms `reasoning_opaque` rides on the same streamed delta as the\n *    first (id-bearing) tool-call chunk, so the association is exact.\n *  - **Outbound** ({@link restoreCopilotGeminiReasoningOpaque} from the\n *    `onPayload` hook): converts the `reasoning_details` the client re-emits on\n *    replayed assistant messages back into a single `reasoning_opaque` field on\n *    that assistant message, which is the only shape CAPI reads.\n *\n * Both transforms are gated to GitHub Copilot Gemini and are no-ops for every\n * other provider/model (and for Gemini turns that carry no thought signature).\n */\n\ntype JsonObject = { [key: string]: JsonValue };\ntype JsonValue = string | number | boolean | null | JsonValue[] | JsonObject;\n\n/** OpenRouter-style encrypted reasoning detail the pi-ai client round-trips. */\nconst REASONING_ENCRYPTED_TYPE = \"reasoning.encrypted\";\n\nfunction isPlainObject(value: unknown): value is JsonObject {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\n/**\n * Inject `reasoning_details` into a parsed CAPI Gemini streaming chunk so the\n * pi-ai OpenAI-completions parser captures the Gemini thought signature.\n *\n * For each `choices[].delta` that carries a non-empty `reasoning_opaque` string\n * and a `tool_calls[]` entry with an `id`, adds a single\n * `reasoning_details: [{ type: \"reasoning.encrypted\", id, data: <opaque> }]`\n * entry keyed by that tool-call id. Returns whether the chunk was mutated.\n *\n * No-op when the delta already has `reasoning_details`, has no id-bearing tool\n * call (e.g. argument-continuation deltas or pure-text thought chunks), or has\n * no `reasoning_opaque`.\n */\nexport function injectCopilotGeminiReasoningDetails(chunk: JsonValue): boolean {\n  if (!isPlainObject(chunk)) return false;\n  const choices = chunk.choices;\n  if (!Array.isArray(choices)) return false;\n\n  let mutated = false;\n  for (const choice of choices) {\n    if (!isPlainObject(choice)) continue;\n    const delta = choice.delta;\n    if (!isPlainObject(delta)) continue;\n\n    const opaque = delta.reasoning_opaque;\n    if (typeof opaque !== \"string\" || opaque.length === 0) continue;\n\n    // Already carries the encrypted detail (don't double-inject / clobber).\n    if (Array.isArray(delta.reasoning_details) && delta.reasoning_details.length > 0) {\n      continue;\n    }\n\n    const toolCalls = delta.tool_calls;\n    if (!Array.isArray(toolCalls)) continue;\n    const idBearing = toolCalls.find(\n      (call): call is JsonObject =>\n        isPlainObject(call) && typeof call.id === \"string\" && call.id.length > 0,\n    );\n    if (!idBearing) continue;\n\n    delta.reasoning_details = [\n      { type: REASONING_ENCRYPTED_TYPE, id: idBearing.id as string, data: opaque },\n    ];\n    mutated = true;\n  }\n  return mutated;\n}\n\n/**\n * Rewrite the JSON payload of a single SSE `data:` line. Returns the original\n * string unchanged when it is not a Gemini chunk that needs a thought signature\n * bridged, or when parsing fails (fail-open: never corrupt the stream).\n */\nexport function rewriteCopilotGeminiSseData(dataPayload: string): string {\n  // Cheap gate: only chunks that actually carry a thought signature are touched.\n  if (!dataPayload.includes(\"reasoning_opaque\")) return dataPayload;\n  let parsed: JsonValue;\n  try {\n    parsed = JSON.parse(dataPayload) as JsonValue;\n  } catch {\n    return dataPayload;\n  }\n  if (!injectCopilotGeminiReasoningDetails(parsed)) return dataPayload;\n  return JSON.stringify(parsed);\n}\n\n/** Rewrite one SSE line, preserving a trailing carriage return when present. */\nfunction rewriteSseLine(line: string): string {\n  const hasCr = line.endsWith(\"\\r\");\n  const core = hasCr ? line.slice(0, -1) : line;\n  if (!core.startsWith(\"data:\")) return line;\n  const payload = core.slice(\"data:\".length).trimStart();\n  if (payload.length === 0 || payload === \"[DONE]\") return line;\n  const rewritten = rewriteCopilotGeminiSseData(payload);\n  if (rewritten === payload) return line;\n  const rebuilt = `data: ${rewritten}`;\n  return hasCr ? `${rebuilt}\\r` : rebuilt;\n}\n\n/**\n * Wrap a CAPI Gemini SSE byte stream so `reasoning_opaque` is bridged into\n * `reasoning_details`. Buffers across chunk boundaries and rewrites whole lines\n * only; bytes that are not affected pass through unchanged.\n *\n * Implemented as a `ReadableStream` over the source reader (rather than a\n * `TransformStream` piped via `pipeThrough`) so the transform pulls lazily and\n * propagates cancellation to the upstream body.\n */\nexport function createCopilotGeminiSseStream(\n  source: ReadableStream<Uint8Array>,\n): ReadableStream<Uint8Array> {\n  const reader = source.getReader();\n  const decoder = new TextDecoder();\n  const encoder = new TextEncoder();\n  let buffer = \"\";\n\n  return new ReadableStream<Uint8Array>({\n    async pull(controller) {\n      // Loop until we emit at least one chunk or close, so a read that yields no\n      // complete line still makes progress without relying on the runtime to\n      // re-invoke pull after a no-enqueue return.\n      for (;;) {\n        const { done, value } = await reader.read();\n        if (done) {\n          buffer += decoder.decode();\n          if (buffer.length > 0) {\n            controller.enqueue(encoder.encode(rewriteSseLine(buffer)));\n            buffer = \"\";\n          }\n          controller.close();\n          return;\n        }\n        buffer += decoder.decode(value, { stream: true });\n        let newlineIndex = buffer.indexOf(\"\\n\");\n        let out = \"\";\n        while (newlineIndex !== -1) {\n          const line = buffer.slice(0, newlineIndex);\n          buffer = buffer.slice(newlineIndex + 1);\n          out += `${rewriteSseLine(line)}\\n`;\n          newlineIndex = buffer.indexOf(\"\\n\");\n        }\n        if (out.length > 0) {\n          controller.enqueue(encoder.encode(out));\n          return;\n        }\n      }\n    },\n    cancel(reason) {\n      return reader.cancel(reason);\n    },\n  });\n}\n\n/**\n * Convert the `reasoning_details` the pi-ai client re-emits on replayed\n * assistant messages back into the single `reasoning_opaque` field CAPI reads.\n *\n * For GitHub Copilot Gemini payloads, each assistant message that carries a\n * `reasoning_details` entry of `type: \"reasoning.encrypted\"` has its `data`\n * (the original CAPI thought-signature blob) promoted to `reasoning_opaque`,\n * and the now-redundant `reasoning_details` removed. No-op for every other\n * provider/model and for payloads without such messages.\n */\nexport function restoreCopilotGeminiReasoningOpaque(\n  payload: unknown,\n  model: Pick<Model<Api>, \"provider\" | \"api\" | \"id\">,\n): unknown {\n  if (!isCopilotGeminiModel(model)) return payload;\n  if (!isPlainObject(payload as JsonValue)) return payload;\n  const payloadObject = payload as JsonObject;\n  const messages = payloadObject.messages;\n  if (!Array.isArray(messages)) return payload;\n\n  let mutated = false;\n  const nextMessages = messages.map((message) => {\n    if (!isPlainObject(message) || message.role !== \"assistant\") return message;\n    const details = message.reasoning_details;\n    if (!Array.isArray(details) || details.length === 0) return message;\n    const encrypted = details.find(\n      (detail): detail is JsonObject =>\n        isPlainObject(detail) &&\n        detail.type === REASONING_ENCRYPTED_TYPE &&\n        typeof detail.data === \"string\" &&\n        detail.data.length > 0,\n    );\n    if (!encrypted) return message;\n    mutated = true;\n    const { reasoning_details: _omitted, ...rest } = message;\n    return { ...rest, reasoning_opaque: encrypted.data as string };\n  });\n\n  if (!mutated) return payload;\n  return { ...payloadObject, messages: nextMessages };\n}\n\n/** Resolve the request URL string from a `fetch` input argument. */\nfunction resolveRequestUrl(input: Parameters<typeof fetch>[0]): string | undefined {\n  if (typeof input === \"string\") return input;\n  if (input instanceof URL) return input.href;\n  if (typeof input === \"object\" && input !== null && \"url\" in input) {\n    const url = (input as { url?: unknown }).url;\n    if (typeof url === \"string\") return url;\n  }\n  return undefined;\n}\n\n/**\n * Rewrite a streaming CAPI Gemini response so its SSE body bridges\n * `reasoning_opaque` into `reasoning_details`. Returns the original response\n * untouched for non-Copilot hosts, non-event-stream responses, or bodyless\n * responses, keeping the blast radius to streaming CAPI Gemini turns only.\n */\nexport function maybeRewriteCopilotGeminiResponse(\n  url: string | undefined,\n  response: Response,\n): Response {\n  if (!url || !isCopilotApiHost(url)) return response;\n  const contentType = response.headers.get(\"content-type\") ?? \"\";\n  if (!contentType.includes(\"text/event-stream\")) return response;\n  if (!response.body) return response;\n  const transformed = createCopilotGeminiSseStream(response.body);\n  return new Response(transformed, {\n    status: response.status,\n    statusText: response.statusText,\n    headers: response.headers,\n  });\n}\n\n/**\n * Apply Copilot response-body transforms before pi-ai parses streamed provider\n * output. Anthropic Messages repair gets first refusal for `/v1/messages`\n * streams; other Copilot event streams keep the existing Gemini reasoning bridge.\n */\nexport function maybeRewriteCopilotProviderResponse(\n  url: string | undefined,\n  response: Response,\n): Response {\n  const anthropicRepaired = maybeRepairCopilotAnthropicMessagesResponse(url, response);\n  if (anthropicRepaired !== response) return anthropicRepaired;\n  return maybeRewriteCopilotGeminiResponse(url, response);\n}\n\nlet originalFetch: typeof fetch | undefined;\n\n/**\n * Install a `globalThis.fetch` wrapper that applies Copilot-only stream\n * compatibility shims before pi-ai parses provider output. Currently this\n * includes the Gemini `reasoning_opaque` bridge and the Anthropic Messages\n * terminal `message_stop` repair. Idempotent.\n *\n * SDK clients resolve `globalThis.fetch` at client-construction time, and a new\n * client is built per request, so wrapping the global before the first request\n * is reliably picked up. Non-Copilot hosts and non-event-stream responses are\n * returned untouched by the underlying shims.\n */\nexport function installCopilotResponseInterceptor(): void {\n  if (originalFetch) return;\n  if (typeof globalThis.fetch !== \"function\") return;\n  const base = globalThis.fetch;\n  originalFetch = base;\n  const boundFetch = base.bind(globalThis);\n\n  const wrapped = (async (input, init) => {\n    const response = await boundFetch(input, init);\n    try {\n      return maybeRewriteCopilotProviderResponse(resolveRequestUrl(input), response);\n    } catch {\n      return response;\n    }\n  }) as typeof fetch;\n\n  // Preserve `fetch.preconnect` so the wrapper remains a drop-in replacement.\n  const preconnect = (base as { preconnect?: unknown }).preconnect;\n  if (typeof preconnect === \"function\") {\n    (wrapped as { preconnect?: unknown }).preconnect = preconnect.bind(globalThis);\n  }\n\n  globalThis.fetch = wrapped;\n}\n\n/** Backward-compatible name for the original Gemini-only interceptor export. */\nexport function installCopilotGeminiReasoningInterceptor(): void {\n  installCopilotResponseInterceptor();\n}\n"]}

package/dist/core/copilot-gemini-reasoning.js

@@ -1,3 +1,5 @@
+import { maybeRepairCopilotAnthropicMessagesResponse } from "./copilot-anthropic-sse-repair.js";
+import { isCopilotApiHost } from "./copilot-hosts.js";
 import { isCopilotGeminiModel } from "./copilot-gemini-payload-sanitizer.js";
 /** OpenRouter-style encrypted reasoning detail the pi-ai client round-trips. */
 const REASONING_ENCRYPTED_TYPE = "reasoning.encrypted";
@@ -175,18 +177,6 @@ export function restoreCopilotGeminiReasoningOpaque(payload, model) {
         return payload;
     return { ...payloadObject, messages: nextMessages };
 }
-/** Whether the URL targets Copilot's CAPI gateway (`*.githubcopilot.com`). */
-function isCopilotApiHost(url) {
-    try {
-        const host = new URL(url).hostname.toLowerCase();
-        // Exact host or a real subdomain only — never a look-alike suffix such as
-        // `notgithubcopilot.com` (CodeQL: incomplete URL substring sanitization).
-        return host === "githubcopilot.com" || host.endsWith(".githubcopilot.com");
-    }
-    catch {
-        return false;
-    }
-}
 /** Resolve the request URL string from a `fetch` input argument. */
 function resolveRequestUrl(input) {
     if (typeof input === "string")
@@ -221,19 +211,30 @@ export function maybeRewriteCopilotGeminiResponse(url, response) {
         headers: response.headers,
     });
 }
+/**
+ * Apply Copilot response-body transforms before pi-ai parses streamed provider
+ * output. Anthropic Messages repair gets first refusal for `/v1/messages`
+ * streams; other Copilot event streams keep the existing Gemini reasoning bridge.
+ */
+export function maybeRewriteCopilotProviderResponse(url, response) {
+    const anthropicRepaired = maybeRepairCopilotAnthropicMessagesResponse(url, response);
+    if (anthropicRepaired !== response)
+        return anthropicRepaired;
+    return maybeRewriteCopilotGeminiResponse(url, response);
+}
 let originalFetch;
 /**
- * Install a `globalThis.fetch` wrapper that rewrites CAPI Gemini SSE responses
- * to bridge `reasoning_opaque` into `reasoning_details` (see
- * {@link createCopilotGeminiSseStream}). Idempotent.
+ * Install a `globalThis.fetch` wrapper that applies Copilot-only stream
+ * compatibility shims before pi-ai parses provider output. Currently this
+ * includes the Gemini `reasoning_opaque` bridge and the Anthropic Messages
+ * terminal `message_stop` repair. Idempotent.
  *
- * The OpenAI SDK used by the `openai-completions` provider resolves
- * `globalThis.fetch` at client-construction time, and a new client is built per
- * request, so wrapping the global before the first request is reliably picked
- * up. Non-Copilot hosts and non-event-stream responses are returned untouched,
- * keeping the blast radius to streaming CAPI Gemini turns only.
+ * SDK clients resolve `globalThis.fetch` at client-construction time, and a new
+ * client is built per request, so wrapping the global before the first request
+ * is reliably picked up. Non-Copilot hosts and non-event-stream responses are
+ * returned untouched by the underlying shims.
  */
-export function installCopilotGeminiReasoningInterceptor() {
+export function installCopilotResponseInterceptor() {
     if (originalFetch)
         return;
     if (typeof globalThis.fetch !== "function")
@@ -244,7 +245,7 @@ export function installCopilotGeminiReasoningInterceptor() {
     const wrapped = (async (input, init) => {
         const response = await boundFetch(input, init);
         try {
-            return maybeRewriteCopilotGeminiResponse(resolveRequestUrl(input), response);
+            return maybeRewriteCopilotProviderResponse(resolveRequestUrl(input), response);
         }
         catch {
             return response;
@@ -257,4 +258,8 @@ export function installCopilotGeminiReasoningInterceptor() {
     }
     globalThis.fetch = wrapped;
 }
+/** Backward-compatible name for the original Gemini-only interceptor export. */
+export function installCopilotGeminiReasoningInterceptor() {
+    installCopilotResponseInterceptor();
+}
 //# sourceMappingURL=copilot-gemini-reasoning.js.map