luulam/9router
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(open-sse): extend golden response stream to kiro/ollama (T0.5)

Browse Source 
Lock chunk/usage/tool/thinking/finish behavior for kiro + ollama before
chunkBuilder refactor. Sanitize volatile stream/tool ids.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
decolua
2026-06-13 16:47:25 +07:00
parent b87cf0c96a
commit 0e34358e74
2 changed files with 174 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
tests/translator/golden-response-stream.test.js
View File

@@ -6,12 +6,16 @@ import "./registerAll.js";
import { translateResponse, initState } from "../../open-sse/translator/index.js";
import { FORMATS } from "../../open-sse/translator/formats.js";
// Chuẩn hoá field động (Date.now trong created + gemini tool id) để snapshot ổn định.
// Chuẩn hoá field động (Date.now trong created + id) để snapshot ổn định.
function stripVolatile(chunks) {
return JSON.parse(JSON.stringify(chunks), (key, val) => {
if (key === "created") return 0;
// gemini sinh id = `${name}-${Date.now()}-${idx}` → chuẩn hoá phần timestamp
if (key === "id" && typeof val === "string") return val.replace(/-\d{10,}-(\d+)$/, "-<TS>-$1");
if (key === "id" && typeof val === "string") {
return val
.replace(/-\d{10,}-(\d+)$/, "-<TS>-$1") // gemini: name-<ts>-idx
.replace(/^chatcmpl-\d{10,}$/, "chatcmpl-<TS>") // kiro/ollama stream id
.replace(/^call_(\d+)_\d{10,}$/, "call_$1_<TS>"); // ollama tool id
}
return val;
});
}
@@ -67,3 +71,27 @@ describe("GOLDEN response stream: Gemini → OpenAI", () => {
expect(runStream(FORMATS.GEMINI, FORMATS.OPENAI, events)).toMatchSnapshot();
});
});
describe("GOLDEN response stream: Kiro → OpenAI", () => {
it("text + reasoning + toolUse + usage + stop", () => {
const events = [
{ assistantResponseEvent: { content: "Hello" }, _eventType: "assistantResponseEvent" },
{ reasoningContentEvent: { text: "thinking" }, _eventType: "reasoningContentEvent" },
{ toolUseEvent: { toolUseId: "tu_1", name: "get_weather", input: { city: "NYC" } }, _eventType: "toolUseEvent" },
{ usageEvent: { inputTokens: 10, outputTokens: 5 }, _eventType: "usageEvent" },
{ _eventType: "messageStopEvent" },
];
expect(runStream(FORMATS.KIRO, FORMATS.OPENAI, events)).toMatchSnapshot();
});
});
describe("GOLDEN response stream: Ollama → OpenAI", () => {
it("content + thinking + tool_calls + done usage", () => {
const events = [
{ model: "qwen3", message: { role: "assistant", content: "Hi", thinking: "reason" } },
{ model: "qwen3", message: { role: "assistant", tool_calls: [{ function: { name: "search", arguments: { q: "x" } } }] } },
{ model: "qwen3", done: true, done_reason: "stop", prompt_eval_count: 8, eval_count: 4 },
];
expect(runStream(FORMATS.OLLAMA, FORMATS.OPENAI, events)).toMatchSnapshot();
});
});