@bxb1337/windsurf-fast-context 1.0.8 → 1.0.9

package/dist/cjs/model/devstral-language-model.js

@@ -56,7 +56,7 @@ class DevstralLanguageModel {
         const requestFrame = (0, connect_frame_js_1.connectFrameEncode)(requestPayload);
         const headers = createConnectHeaders(this.headers);
         const responseFrame = await this.transport.postUnary(`${this.baseURL}${API_SERVICE_PATH}${DEVSTRAL_STREAM_PATH}`, requestFrame, headers);
-        const responsePayloads = (0, connect_frame_js_1.connectFrameDecode)(responseFrame);
+        const { payloads: responsePayloads } = (0, connect_frame_js_1.connectFrameDecode)(responseFrame);
         const payloads = responsePayloads.length > 0 ? responsePayloads : [responseFrame];
         const content = payloads.flatMap((payload) => toV3Content((0, response_converter_js_1.convertResponse)(payload)));
         const unified = content.some((part) => part.type === 'tool-call')
@@ -110,7 +110,7 @@ class DevstralLanguageModel {
                     try {
                         safeEnqueue(controller, { type: 'stream-start', warnings: [] });
                         safeEnqueue(controller, { type: 'response-metadata', modelId: this.modelId });
-                        while (!isAborted(options.abortSignal)) {
+                        outerLoop: while (!isAborted(options.abortSignal)) {
                             const next = await reader.read();
                             if (next.done) {
                                 break;
@@ -162,6 +162,9 @@ class DevstralLanguageModel {
                                     });
                                     safeEnqueue(controller, part);
                                 }
+                                if (frameResult.isEndStream) {
+                                    break outerLoop;
+                                }
                             }
                         }
                         if (isAborted(options.abortSignal)) {
@@ -219,10 +222,11 @@ function readNextConnectFrame(buffer) {
         return null;
     }
     const frame = buffer.subarray(0, frameLength);
-    const decoded = (0, connect_frame_js_1.connectFrameDecode)(frame);
+    const { payloads, isEndStream } = (0, connect_frame_js_1.connectFrameDecode)(frame);
     return {
-        payload: decoded[0] ?? Buffer.alloc(0),
+        payload: payloads[0] ?? Buffer.alloc(0),
         rest: buffer.subarray(frameLength),
+        isEndStream,
     };
 }
 function safeEnqueue(controller, part) {

package/dist/cjs/model/devstral-language-model.test.js

@@ -66,8 +66,8 @@ function waitFor(ms) {
 }
 function decodeRequestPayload(body) {
     // Body is now a connect frame directly (gzip is inside the frame, not outside)
-    const decodedFrames = (0, connect_frame_js_1.connectFrameDecode)(body);
-    return decodedFrames[0] ?? Buffer.alloc(0);
+    const { payloads } = (0, connect_frame_js_1.connectFrameDecode)(body);
+    return payloads[0] ?? Buffer.alloc(0);
 }
 function extractToolsPayload(strings) {
     return strings.find((value) => value.startsWith('[{"type":"function"'));
@@ -467,6 +467,76 @@ async function collectStreamParts(stream) {
             },
         });
     });
+    (0, vitest_1.it)('stream terminates immediately on EndStreamResponse frame (flags=2)', async () => {
+        const jwt = makeJwt(4_300_000_003, 'end-stream-response');
+        const textFrame = (0, connect_frame_js_1.connectFrameEncode)(Buffer.from('hello world', 'utf8'));
+        const endStreamPayload = Buffer.alloc(0);
+        const endStreamFrame = Buffer.allocUnsafe(5 + endStreamPayload.length);
+        endStreamFrame.writeUInt8(2, 0);
+        endStreamFrame.writeUInt32BE(endStreamPayload.length, 1);
+        endStreamPayload.copy(endStreamFrame, 5);
+        let releaseHttpClose;
+        const waitForHttpClose = new Promise((resolve) => {
+            releaseHttpClose = resolve;
+        });
+        const fakeFetch = async (input) => {
+            const url = String(input);
+            if (url.endsWith('/GetUserJwt')) {
+                return new Response(Uint8Array.from(Buffer.from(jwt, 'utf8')), { status: 200 });
+            }
+            let sentTextFrame = false;
+            let sentEndStreamFrame = false;
+            return new Response(new ReadableStream({
+                async pull(controller) {
+                    if (!sentTextFrame) {
+                        controller.enqueue(Uint8Array.from(textFrame));
+                        sentTextFrame = true;
+                        return;
+                    }
+                    if (!sentEndStreamFrame) {
+                        controller.enqueue(Uint8Array.from(endStreamFrame));
+                        sentEndStreamFrame = true;
+                        return;
+                    }
+                    await waitForHttpClose;
+                    controller.close();
+                },
+            }), { status: 200 });
+        };
+        const model = new devstral_language_model_js_1.DevstralLanguageModel({
+            apiKey: 'end-stream-key',
+            fetch: fakeFetch,
+            baseURL: 'https://windsurf.test',
+        });
+        const result = await model.doStream({
+            prompt: [{ role: 'user', content: [{ type: 'text', text: 'Stream with end.' }] }],
+        });
+        const reader = result.stream.getReader();
+        const parts = [];
+        const resultType = await Promise.race([
+            (async () => {
+                while (true) {
+                    const next = await reader.read();
+                    if (next.done) {
+                        break;
+                    }
+                    parts.push(next.value);
+                }
+                return 'completed';
+            })(),
+            waitFor(100).then(() => 'timed-out'),
+        ]);
+        (0, vitest_1.expect)(resultType).toBe('completed');
+        (0, vitest_1.expect)(parts.map((p) => p.type)).toEqual([
+            'stream-start',
+            'response-metadata',
+            'text-start',
+            'text-delta',
+            'text-end',
+            'finish',
+        ]);
+        releaseHttpClose?.();
+    });
     (0, vitest_1.it)('abort stops stream mid-response', async () => {
         const controller = new AbortController();
         const jwt = makeJwt(4_300_000_002, 'abort');

package/dist/cjs/protocol/connect-frame.js

@@ -1,10 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.FLAG_END_STREAM = exports.FLAG_COMPRESSED = void 0;
 exports.connectFrameEncode = connectFrameEncode;
 exports.connectFrameDecode = connectFrameDecode;
 const node_zlib_1 = require("node:zlib");
 const FRAME_HEADER_SIZE = 5;
-const FLAG_COMPRESSED = 1;
+exports.FLAG_COMPRESSED = 1;
+exports.FLAG_END_STREAM = 2;
 /**
  * Encode a protobuf payload into a Connect-RPC frame.
  *
@@ -16,7 +18,7 @@ const FLAG_COMPRESSED = 1;
 function connectFrameEncode(payload, compress = true) {
     const payloadBytes = compress ? (0, node_zlib_1.gzipSync)(payload) : payload;
     const frame = Buffer.allocUnsafe(FRAME_HEADER_SIZE + payloadBytes.length);
-    frame.writeUInt8(compress ? FLAG_COMPRESSED : 0, 0);
+    frame.writeUInt8(compress ? exports.FLAG_COMPRESSED : 0, 0);
     frame.writeUInt32BE(payloadBytes.length, 1);
     payloadBytes.copy(frame, FRAME_HEADER_SIZE);
     return frame;
@@ -24,6 +26,7 @@ function connectFrameEncode(payload, compress = true) {
 function connectFrameDecode(buffer) {
     const decoded = [];
     let offset = 0;
+    let isEndStream = false;
     while (offset + FRAME_HEADER_SIZE <= buffer.length) {
         const flags = buffer.readUInt8(offset);
         const payloadLength = buffer.readUInt32BE(offset + 1);
@@ -33,8 +36,11 @@ function connectFrameDecode(buffer) {
             break;
         }
         const payload = buffer.subarray(payloadStart, payloadEnd);
-        decoded.push(flags === FLAG_COMPRESSED ? (0, node_zlib_1.gunzipSync)(payload) : Buffer.from(payload));
+        decoded.push((flags & exports.FLAG_COMPRESSED) !== 0 ? (0, node_zlib_1.gunzipSync)(payload) : Buffer.from(payload));
+        if ((flags & exports.FLAG_END_STREAM) !== 0) {
+            isEndStream = true;
+        }
         offset = payloadEnd;
     }
-    return decoded;
+    return { payloads: decoded, isEndStream };
 }

package/dist/cjs/protocol/connect-frame.test.js

@@ -1,5 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+const node_zlib_1 = require("node:zlib");
 const node_fs_1 = require("node:fs");
 const node_path_1 = require("node:path");
 const vitest_1 = require("vitest");
@@ -15,22 +16,94 @@ const payloadWorld = Buffer.from('CONNECT\x02world', 'utf8');
         (0, vitest_1.expect)(encoded.readUInt8(0)).toBe(0);
         (0, vitest_1.expect)(encoded.readUInt32BE(1)).toBe(payloadHello.length);
         (0, vitest_1.expect)(encoded.subarray(5)).toEqual(payloadHello);
-        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(encoded)).toEqual([payloadHello]);
+        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(encoded)).toEqual({ payloads: [payloadHello], isEndStream: false });
     });
     (0, vitest_1.it)('gzip roundtrip uses compressed frame and decodes back to source payload', () => {
         const encoded = (0, connect_frame_js_1.connectFrameEncode)(payloadWorld, true);
         (0, vitest_1.expect)(encoded.readUInt8(0)).toBe(1);
-        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(encoded)).toEqual([payloadWorld]);
+        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(encoded)).toEqual({ payloads: [payloadWorld], isEndStream: false });
     });
 });
 (0, vitest_1.describe)('connectFrameDecode fixtures', () => {
     (0, vitest_1.it)('decodes frame-simple fixture payload', () => {
-        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(fixtureSimpleFrame)).toEqual([payloadHello]);
+        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(fixtureSimpleFrame)).toEqual({ payloads: [payloadHello], isEndStream: false });
     });
     (0, vitest_1.it)('decodes gzip fixture payload', () => {
-        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(fixtureGzipFrame)).toEqual([payloadWorld]);
+        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(fixtureGzipFrame)).toEqual({ payloads: [payloadWorld], isEndStream: false });
     });
     (0, vitest_1.it)('decodes multiple concatenated frames in order', () => {
-        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(fixtureMultiFrame)).toEqual([payloadHello, payloadHello]);
+        (0, vitest_1.expect)((0, connect_frame_js_1.connectFrameDecode)(fixtureMultiFrame)).toEqual({ payloads: [payloadHello, payloadHello], isEndStream: false });
+    });
+});
+(0, vitest_1.describe)('connect frame flags', () => {
+    (0, vitest_1.it)('exports FLAG_COMPRESSED constant with value 1', () => {
+        (0, vitest_1.expect)(connect_frame_js_1.FLAG_COMPRESSED).toBe(1);
+    });
+    (0, vitest_1.it)('exports FLAG_END_STREAM constant with value 2', () => {
+        (0, vitest_1.expect)(connect_frame_js_1.FLAG_END_STREAM).toBe(2);
+    });
+    (0, vitest_1.it)('supports bit combination: FLAG_COMPRESSED | FLAG_END_STREAM === 3', () => {
+        (0, vitest_1.expect)(connect_frame_js_1.FLAG_COMPRESSED | connect_frame_js_1.FLAG_END_STREAM).toBe(3);
+    });
+});
+(0, vitest_1.describe)('connectFrameDecode isEndStream', () => {
+    /**
+     * TDD RED PHASE: Tests for isEndStream flag detection.
+     * These tests MUST FAIL because connectFrameDecode currently returns Buffer[],
+     * not { payloads: Buffer[], isEndStream: boolean }.
+     *
+     * Connect-RPC flag bits:
+     * - Bit 0 (value 1): FLAG_COMPRESSED
+     * - Bit 1 (value 2): FLAG_END_STREAM
+     */
+    (0, vitest_1.it)('returns isEndStream: false for flags=0 (no flags)', () => {
+        // flags=0: neither compressed nor end stream
+        const frame = Buffer.allocUnsafe(5 + payloadHello.length);
+        frame.writeUInt8(0, 0); // flags = 0
+        frame.writeUInt32BE(payloadHello.length, 1);
+        payloadHello.copy(frame, 5);
+        const result = (0, connect_frame_js_1.connectFrameDecode)(frame);
+        (0, vitest_1.expect)(result).toHaveProperty('payloads');
+        (0, vitest_1.expect)(result).toHaveProperty('isEndStream');
+        (0, vitest_1.expect)(result.payloads).toEqual([payloadHello]);
+        (0, vitest_1.expect)(result.isEndStream).toBe(false);
+    });
+    (0, vitest_1.it)('returns isEndStream: false for flags=1 (compressed only)', () => {
+        // flags=1: compressed but not end stream
+        const gzipped = (0, node_zlib_1.gzipSync)(payloadHello);
+        const frame = Buffer.allocUnsafe(5 + gzipped.length);
+        frame.writeUInt8(connect_frame_js_1.FLAG_COMPRESSED, 0); // flags = 1
+        frame.writeUInt32BE(gzipped.length, 1);
+        gzipped.copy(frame, 5);
+        const result = (0, connect_frame_js_1.connectFrameDecode)(frame);
+        (0, vitest_1.expect)(result).toHaveProperty('payloads');
+        (0, vitest_1.expect)(result).toHaveProperty('isEndStream');
+        (0, vitest_1.expect)(result.payloads).toEqual([payloadHello]);
+        (0, vitest_1.expect)(result.isEndStream).toBe(false);
+    });
+    (0, vitest_1.it)('returns isEndStream: true for flags=2 (end stream only)', () => {
+        // flags=2: end stream but not compressed
+        const frame = Buffer.allocUnsafe(5 + payloadHello.length);
+        frame.writeUInt8(connect_frame_js_1.FLAG_END_STREAM, 0); // flags = 2
+        frame.writeUInt32BE(payloadHello.length, 1);
+        payloadHello.copy(frame, 5);
+        const result = (0, connect_frame_js_1.connectFrameDecode)(frame);
+        (0, vitest_1.expect)(result).toHaveProperty('payloads');
+        (0, vitest_1.expect)(result).toHaveProperty('isEndStream');
+        (0, vitest_1.expect)(result.payloads).toEqual([payloadHello]);
+        (0, vitest_1.expect)(result.isEndStream).toBe(true);
+    });
+    (0, vitest_1.it)('returns isEndStream: true for flags=3 (compressed + end stream)', () => {
+        // flags=3: both compressed and end stream
+        const gzipped = (0, node_zlib_1.gzipSync)(payloadHello);
+        const frame = Buffer.allocUnsafe(5 + gzipped.length);
+        frame.writeUInt8(connect_frame_js_1.FLAG_COMPRESSED | connect_frame_js_1.FLAG_END_STREAM, 0); // flags = 3
+        frame.writeUInt32BE(gzipped.length, 1);
+        gzipped.copy(frame, 5);
+        const result = (0, connect_frame_js_1.connectFrameDecode)(frame);
+        (0, vitest_1.expect)(result).toHaveProperty('payloads');
+        (0, vitest_1.expect)(result).toHaveProperty('isEndStream');
+        (0, vitest_1.expect)(result.payloads).toEqual([payloadHello]);
+        (0, vitest_1.expect)(result.isEndStream).toBe(true);
     });
 });

package/dist/esm/model/devstral-language-model.js

@@ -53,7 +53,7 @@ export class DevstralLanguageModel {
         const requestFrame = connectFrameEncode(requestPayload);
         const headers = createConnectHeaders(this.headers);
         const responseFrame = await this.transport.postUnary(`${this.baseURL}${API_SERVICE_PATH}${DEVSTRAL_STREAM_PATH}`, requestFrame, headers);
-        const responsePayloads = connectFrameDecode(responseFrame);
+        const { payloads: responsePayloads } = connectFrameDecode(responseFrame);
         const payloads = responsePayloads.length > 0 ? responsePayloads : [responseFrame];
         const content = payloads.flatMap((payload) => toV3Content(convertResponse(payload)));
         const unified = content.some((part) => part.type === 'tool-call')
@@ -107,7 +107,7 @@ export class DevstralLanguageModel {
                     try {
                         safeEnqueue(controller, { type: 'stream-start', warnings: [] });
                         safeEnqueue(controller, { type: 'response-metadata', modelId: this.modelId });
-                        while (!isAborted(options.abortSignal)) {
+                        outerLoop: while (!isAborted(options.abortSignal)) {
                             const next = await reader.read();
                             if (next.done) {
                                 break;
@@ -159,6 +159,9 @@ export class DevstralLanguageModel {
                                     });
                                     safeEnqueue(controller, part);
                                 }
+                                if (frameResult.isEndStream) {
+                                    break outerLoop;
+                                }
                             }
                         }
                         if (isAborted(options.abortSignal)) {
@@ -215,10 +218,11 @@ function readNextConnectFrame(buffer) {
         return null;
     }
     const frame = buffer.subarray(0, frameLength);
-    const decoded = connectFrameDecode(frame);
+    const { payloads, isEndStream } = connectFrameDecode(frame);
     return {
-        payload: decoded[0] ?? Buffer.alloc(0),
+        payload: payloads[0] ?? Buffer.alloc(0),
         rest: buffer.subarray(frameLength),
+        isEndStream,
     };
 }
 function safeEnqueue(controller, part) {

package/dist/esm/model/devstral-language-model.test.js

@@ -64,8 +64,8 @@ function waitFor(ms) {
 }
 function decodeRequestPayload(body) {
     // Body is now a connect frame directly (gzip is inside the frame, not outside)
-    const decodedFrames = connectFrameDecode(body);
-    return decodedFrames[0] ?? Buffer.alloc(0);
+    const { payloads } = connectFrameDecode(body);
+    return payloads[0] ?? Buffer.alloc(0);
 }
 function extractToolsPayload(strings) {
     return strings.find((value) => value.startsWith('[{"type":"function"'));
@@ -465,6 +465,76 @@ describe('DevstralLanguageModel doStream', () => {
             },
         });
     });
+    it('stream terminates immediately on EndStreamResponse frame (flags=2)', async () => {
+        const jwt = makeJwt(4_300_000_003, 'end-stream-response');
+        const textFrame = connectFrameEncode(Buffer.from('hello world', 'utf8'));
+        const endStreamPayload = Buffer.alloc(0);
+        const endStreamFrame = Buffer.allocUnsafe(5 + endStreamPayload.length);
+        endStreamFrame.writeUInt8(2, 0);
+        endStreamFrame.writeUInt32BE(endStreamPayload.length, 1);
+        endStreamPayload.copy(endStreamFrame, 5);
+        let releaseHttpClose;
+        const waitForHttpClose = new Promise((resolve) => {
+            releaseHttpClose = resolve;
+        });
+        const fakeFetch = async (input) => {
+            const url = String(input);
+            if (url.endsWith('/GetUserJwt')) {
+                return new Response(Uint8Array.from(Buffer.from(jwt, 'utf8')), { status: 200 });
+            }
+            let sentTextFrame = false;
+            let sentEndStreamFrame = false;
+            return new Response(new ReadableStream({
+                async pull(controller) {
+                    if (!sentTextFrame) {
+                        controller.enqueue(Uint8Array.from(textFrame));
+                        sentTextFrame = true;
+                        return;
+                    }
+                    if (!sentEndStreamFrame) {
+                        controller.enqueue(Uint8Array.from(endStreamFrame));
+                        sentEndStreamFrame = true;
+                        return;
+                    }
+                    await waitForHttpClose;
+                    controller.close();
+                },
+            }), { status: 200 });
+        };
+        const model = new DevstralLanguageModel({
+            apiKey: 'end-stream-key',
+            fetch: fakeFetch,
+            baseURL: 'https://windsurf.test',
+        });
+        const result = await model.doStream({
+            prompt: [{ role: 'user', content: [{ type: 'text', text: 'Stream with end.' }] }],
+        });
+        const reader = result.stream.getReader();
+        const parts = [];
+        const resultType = await Promise.race([
+            (async () => {
+                while (true) {
+                    const next = await reader.read();
+                    if (next.done) {
+                        break;
+                    }
+                    parts.push(next.value);
+                }
+                return 'completed';
+            })(),
+            waitFor(100).then(() => 'timed-out'),
+        ]);
+        expect(resultType).toBe('completed');
+        expect(parts.map((p) => p.type)).toEqual([
+            'stream-start',
+            'response-metadata',
+            'text-start',
+            'text-delta',
+            'text-end',
+            'finish',
+        ]);
+        releaseHttpClose?.();
+    });
     it('abort stops stream mid-response', async () => {
         const controller = new AbortController();
         const jwt = makeJwt(4_300_000_002, 'abort');

package/dist/esm/protocol/connect-frame.js

@@ -1,6 +1,7 @@
 import { gunzipSync, gzipSync } from 'node:zlib';
 const FRAME_HEADER_SIZE = 5;
-const FLAG_COMPRESSED = 1;
+export const FLAG_COMPRESSED = 1;
+export const FLAG_END_STREAM = 2;
 /**
  * Encode a protobuf payload into a Connect-RPC frame.
  *
@@ -20,6 +21,7 @@ export function connectFrameEncode(payload, compress = true) {
 export function connectFrameDecode(buffer) {
     const decoded = [];
     let offset = 0;
+    let isEndStream = false;
     while (offset + FRAME_HEADER_SIZE <= buffer.length) {
         const flags = buffer.readUInt8(offset);
         const payloadLength = buffer.readUInt32BE(offset + 1);
@@ -29,8 +31,11 @@ export function connectFrameDecode(buffer) {
             break;
         }
         const payload = buffer.subarray(payloadStart, payloadEnd);
-        decoded.push(flags === FLAG_COMPRESSED ? gunzipSync(payload) : Buffer.from(payload));
+        decoded.push((flags & FLAG_COMPRESSED) !== 0 ? gunzipSync(payload) : Buffer.from(payload));
+        if ((flags & FLAG_END_STREAM) !== 0) {
+            isEndStream = true;
+        }
         offset = payloadEnd;
     }
-    return decoded;
+    return { payloads: decoded, isEndStream };
 }

package/dist/esm/protocol/connect-frame.test.js

@@ -1,7 +1,8 @@
+import { gzipSync } from 'node:zlib';
 import { readFileSync } from 'node:fs';
 import { join } from 'node:path';
 import { describe, expect, it } from 'vitest';
-import { connectFrameDecode, connectFrameEncode } from './connect-frame.js';
+import { connectFrameDecode, connectFrameEncode, FLAG_COMPRESSED, FLAG_END_STREAM } from './connect-frame.js';
 const fixtureSimpleFrame = readFileSync(join(process.cwd(), 'test/fixtures/connect/frame-simple.bin'));
 const fixtureGzipFrame = readFileSync(join(process.cwd(), 'test/fixtures/connect/frame-gzip.bin'));
 const fixtureMultiFrame = readFileSync(join(process.cwd(), 'test/fixtures/connect/frame-multi.bin'));
@@ -13,22 +14,94 @@ describe('connectFrameEncode', () => {
         expect(encoded.readUInt8(0)).toBe(0);
         expect(encoded.readUInt32BE(1)).toBe(payloadHello.length);
         expect(encoded.subarray(5)).toEqual(payloadHello);
-        expect(connectFrameDecode(encoded)).toEqual([payloadHello]);
+        expect(connectFrameDecode(encoded)).toEqual({ payloads: [payloadHello], isEndStream: false });
     });
     it('gzip roundtrip uses compressed frame and decodes back to source payload', () => {
         const encoded = connectFrameEncode(payloadWorld, true);
         expect(encoded.readUInt8(0)).toBe(1);
-        expect(connectFrameDecode(encoded)).toEqual([payloadWorld]);
+        expect(connectFrameDecode(encoded)).toEqual({ payloads: [payloadWorld], isEndStream: false });
     });
 });
 describe('connectFrameDecode fixtures', () => {
     it('decodes frame-simple fixture payload', () => {
-        expect(connectFrameDecode(fixtureSimpleFrame)).toEqual([payloadHello]);
+        expect(connectFrameDecode(fixtureSimpleFrame)).toEqual({ payloads: [payloadHello], isEndStream: false });
     });
     it('decodes gzip fixture payload', () => {
-        expect(connectFrameDecode(fixtureGzipFrame)).toEqual([payloadWorld]);
+        expect(connectFrameDecode(fixtureGzipFrame)).toEqual({ payloads: [payloadWorld], isEndStream: false });
     });
     it('decodes multiple concatenated frames in order', () => {
-        expect(connectFrameDecode(fixtureMultiFrame)).toEqual([payloadHello, payloadHello]);
+        expect(connectFrameDecode(fixtureMultiFrame)).toEqual({ payloads: [payloadHello, payloadHello], isEndStream: false });
+    });
+});
+describe('connect frame flags', () => {
+    it('exports FLAG_COMPRESSED constant with value 1', () => {
+        expect(FLAG_COMPRESSED).toBe(1);
+    });
+    it('exports FLAG_END_STREAM constant with value 2', () => {
+        expect(FLAG_END_STREAM).toBe(2);
+    });
+    it('supports bit combination: FLAG_COMPRESSED | FLAG_END_STREAM === 3', () => {
+        expect(FLAG_COMPRESSED | FLAG_END_STREAM).toBe(3);
+    });
+});
+describe('connectFrameDecode isEndStream', () => {
+    /**
+     * TDD RED PHASE: Tests for isEndStream flag detection.
+     * These tests MUST FAIL because connectFrameDecode currently returns Buffer[],
+     * not { payloads: Buffer[], isEndStream: boolean }.
+     *
+     * Connect-RPC flag bits:
+     * - Bit 0 (value 1): FLAG_COMPRESSED
+     * - Bit 1 (value 2): FLAG_END_STREAM
+     */
+    it('returns isEndStream: false for flags=0 (no flags)', () => {
+        // flags=0: neither compressed nor end stream
+        const frame = Buffer.allocUnsafe(5 + payloadHello.length);
+        frame.writeUInt8(0, 0); // flags = 0
+        frame.writeUInt32BE(payloadHello.length, 1);
+        payloadHello.copy(frame, 5);
+        const result = connectFrameDecode(frame);
+        expect(result).toHaveProperty('payloads');
+        expect(result).toHaveProperty('isEndStream');
+        expect(result.payloads).toEqual([payloadHello]);
+        expect(result.isEndStream).toBe(false);
+    });
+    it('returns isEndStream: false for flags=1 (compressed only)', () => {
+        // flags=1: compressed but not end stream
+        const gzipped = gzipSync(payloadHello);
+        const frame = Buffer.allocUnsafe(5 + gzipped.length);
+        frame.writeUInt8(FLAG_COMPRESSED, 0); // flags = 1
+        frame.writeUInt32BE(gzipped.length, 1);
+        gzipped.copy(frame, 5);
+        const result = connectFrameDecode(frame);
+        expect(result).toHaveProperty('payloads');
+        expect(result).toHaveProperty('isEndStream');
+        expect(result.payloads).toEqual([payloadHello]);
+        expect(result.isEndStream).toBe(false);
+    });
+    it('returns isEndStream: true for flags=2 (end stream only)', () => {
+        // flags=2: end stream but not compressed
+        const frame = Buffer.allocUnsafe(5 + payloadHello.length);
+        frame.writeUInt8(FLAG_END_STREAM, 0); // flags = 2
+        frame.writeUInt32BE(payloadHello.length, 1);
+        payloadHello.copy(frame, 5);
+        const result = connectFrameDecode(frame);
+        expect(result).toHaveProperty('payloads');
+        expect(result).toHaveProperty('isEndStream');
+        expect(result.payloads).toEqual([payloadHello]);
+        expect(result.isEndStream).toBe(true);
+    });
+    it('returns isEndStream: true for flags=3 (compressed + end stream)', () => {
+        // flags=3: both compressed and end stream
+        const gzipped = gzipSync(payloadHello);
+        const frame = Buffer.allocUnsafe(5 + gzipped.length);
+        frame.writeUInt8(FLAG_COMPRESSED | FLAG_END_STREAM, 0); // flags = 3
+        frame.writeUInt32BE(gzipped.length, 1);
+        gzipped.copy(frame, 5);
+        const result = connectFrameDecode(frame);
+        expect(result).toHaveProperty('payloads');
+        expect(result).toHaveProperty('isEndStream');
+        expect(result.payloads).toEqual([payloadHello]);
+        expect(result.isEndStream).toBe(true);
     });
 });

package/dist/protocol/connect-frame.d.ts

@@ -1,3 +1,5 @@
+export declare const FLAG_COMPRESSED = 1;
+export declare const FLAG_END_STREAM = 2;
 /**
  * Encode a protobuf payload into a Connect-RPC frame.
  *
@@ -7,4 +9,7 @@
  * @returns Buffer containing: [flags byte][4-byte length][payload (gzipped if compress=true)]
  */
 export declare function connectFrameEncode(payload: Buffer, compress?: boolean): Buffer;
-export declare function connectFrameDecode(buffer: Buffer): Buffer[];
+export declare function connectFrameDecode(buffer: Buffer): {
+    payloads: Buffer[];
+    isEndStream: boolean;
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bxb1337/windsurf-fast-context",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "AI SDK V3 provider for Windsurf's Devstral code search API",
   "type": "module",
   "scripts": {