@bxb1337/windsurf-fast-context 1.0.2 → 1.0.4

package/dist/cjs/conversion/response-converter.js

@@ -5,6 +5,8 @@ const node_zlib_1 = require("node:zlib");
 const protobuf_js_1 = require("../protocol/protobuf.js");
 const TOOL_CALL_PREFIX = '[TOOL_CALLS]';
 const ARGS_PREFIX = '[ARGS]';
+const STOP_TOKEN = '</s>';
+const EMPTY_TOOL_CALLS_PATTERN = /TOOL_CALLS\d*(?:<\/s>)?\s*\{\s*\}/g;
 function pushText(parts, text) {
     if (text.length > 0) {
         parts.push({ type: 'text', text });
@@ -189,7 +191,9 @@ function decodeResponseText(buffer) {
     return pickBestExtractedText(extracted);
 }
 function convertResponse(buffer) {
-    const responseText = decodeResponseText(buffer);
+    let responseText = decodeResponseText(buffer);
+    responseText = responseText.replace(EMPTY_TOOL_CALLS_PATTERN, '');
+    responseText = responseText.replace(STOP_TOKEN, '');
     const parts = [];
     let cursor = 0;
     let toolCallCount = 0;

package/dist/cjs/conversion/response-converter.test.js

@@ -62,4 +62,29 @@ const response_converter_js_1 = require("./response-converter.js");
         const result = (0, response_converter_js_1.convertResponse)(compressed);
         (0, vitest_1.expect)(result).toEqual([{ type: 'text', text: 'hello from gzip' }]);
     });
+    (0, vitest_1.it)('strips empty TOOL_CALLS markers with stop token', () => {
+        const input = 'Hello world TOOL_CALLS0</s>{}';
+        const result = (0, response_converter_js_1.convertResponse)(Buffer.from(input, 'utf8'));
+        (0, vitest_1.expect)(result).toEqual([{ type: 'text', text: 'Hello world ' }]);
+    });
+    (0, vitest_1.it)('strips empty TOOL_CALLS markers without stop token', () => {
+        const input = 'Hello world TOOL_CALLS1{}';
+        const result = (0, response_converter_js_1.convertResponse)(Buffer.from(input, 'utf8'));
+        (0, vitest_1.expect)(result).toEqual([{ type: 'text', text: 'Hello world ' }]);
+    });
+    (0, vitest_1.it)('strips empty TOOL_CALLS markers with whitespace', () => {
+        const input = 'Text TOOL_CALLS2{   } more text';
+        const result = (0, response_converter_js_1.convertResponse)(Buffer.from(input, 'utf8'));
+        (0, vitest_1.expect)(result).toEqual([{ type: 'text', text: 'Text  more text' }]);
+    });
+    (0, vitest_1.it)('strips standalone stop token', () => {
+        const input = 'Hello world</s>';
+        const result = (0, response_converter_js_1.convertResponse)(Buffer.from(input, 'utf8'));
+        (0, vitest_1.expect)(result).toEqual([{ type: 'text', text: 'Hello world' }]);
+    });
+    (0, vitest_1.it)('handles TOOL_CALLS with number prefix before stop token', () => {
+        const input = 'Text before TOOL_CALLS1</s>{} text after';
+        const result = (0, response_converter_js_1.convertResponse)(Buffer.from(input, 'utf8'));
+        (0, vitest_1.expect)(result).toEqual([{ type: 'text', text: 'Text before  text after' }]);
+    });
 });

package/dist/esm/conversion/response-converter.js

@@ -2,6 +2,8 @@ import { gunzipSync } from 'node:zlib';
 import { extractStrings } from '../protocol/protobuf.js';
 const TOOL_CALL_PREFIX = '[TOOL_CALLS]';
 const ARGS_PREFIX = '[ARGS]';
+const STOP_TOKEN = '</s>';
+const EMPTY_TOOL_CALLS_PATTERN = /TOOL_CALLS\d*(?:<\/s>)?\s*\{\s*\}/g;
 function pushText(parts, text) {
     if (text.length > 0) {
         parts.push({ type: 'text', text });
@@ -186,7 +188,9 @@ function decodeResponseText(buffer) {
     return pickBestExtractedText(extracted);
 }
 export function convertResponse(buffer) {
-    const responseText = decodeResponseText(buffer);
+    let responseText = decodeResponseText(buffer);
+    responseText = responseText.replace(EMPTY_TOOL_CALLS_PATTERN, '');
+    responseText = responseText.replace(STOP_TOKEN, '');
     const parts = [];
     let cursor = 0;
     let toolCallCount = 0;

package/dist/esm/conversion/response-converter.test.js

@@ -60,4 +60,29 @@ describe('convertResponse', () => {
         const result = convertResponse(compressed);
         expect(result).toEqual([{ type: 'text', text: 'hello from gzip' }]);
     });
+    it('strips empty TOOL_CALLS markers with stop token', () => {
+        const input = 'Hello world TOOL_CALLS0</s>{}';
+        const result = convertResponse(Buffer.from(input, 'utf8'));
+        expect(result).toEqual([{ type: 'text', text: 'Hello world ' }]);
+    });
+    it('strips empty TOOL_CALLS markers without stop token', () => {
+        const input = 'Hello world TOOL_CALLS1{}';
+        const result = convertResponse(Buffer.from(input, 'utf8'));
+        expect(result).toEqual([{ type: 'text', text: 'Hello world ' }]);
+    });
+    it('strips empty TOOL_CALLS markers with whitespace', () => {
+        const input = 'Text TOOL_CALLS2{   } more text';
+        const result = convertResponse(Buffer.from(input, 'utf8'));
+        expect(result).toEqual([{ type: 'text', text: 'Text  more text' }]);
+    });
+    it('strips standalone stop token', () => {
+        const input = 'Hello world</s>';
+        const result = convertResponse(Buffer.from(input, 'utf8'));
+        expect(result).toEqual([{ type: 'text', text: 'Hello world' }]);
+    });
+    it('handles TOOL_CALLS with number prefix before stop token', () => {
+        const input = 'Text before TOOL_CALLS1</s>{} text after';
+        const result = convertResponse(Buffer.from(input, 'utf8'));
+        expect(result).toEqual([{ type: 'text', text: 'Text before  text after' }]);
+    });
 });

package/package.json

@@ -1,16 +1,8 @@
 {
   "name": "@bxb1337/windsurf-fast-context",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "AI SDK V3 provider for Windsurf's Devstral code search API",
   "type": "module",
-  "scripts": {
-    "test": "vitest --run",
-    "build:types": "tsc -p tsconfig.types.json",
-    "build:esm": "tsc -p tsconfig.json",
-    "build:cjs": "tsc -p tsconfig.cjs.json",
-    "build": "pnpm -s run build:types && pnpm -s run build:esm && pnpm -s run build:cjs && node ./scripts/postbuild.js",
-    "typecheck": "tsc -p tsconfig.json --noEmit"
-  },
   "keywords": [
     "windsurf",
     "devstral",
@@ -47,5 +39,13 @@
     "dist",
     "README.md",
     "LICENSE"
-  ]
-}
+  ],
+  "scripts": {
+    "test": "vitest --run",
+    "build:types": "tsc -p tsconfig.types.json",
+    "build:esm": "tsc -p tsconfig.json",
+    "build:cjs": "tsc -p tsconfig.cjs.json",
+    "build": "pnpm -s run build:types && pnpm -s run build:esm && pnpm -s run build:cjs && node ./scripts/postbuild.js",
+    "typecheck": "tsc -p tsconfig.json --noEmit"
+  }
+}