nezumi/9router
1
0
Fork 0
mirror of https://github.com/Nezumi-2711/9router.git synced 2026-09-22 13:38:31 +00:00
Code Issues 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
co-authored by Cursor
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
tests/translator/__snapshots__/golden-response-stream.test.js.snap
+143
View File
@@ -302,3 +302,146 @@ exports[`GOLDEN response stream: Gemini → OpenAI > text + thought(no-sig) + fu
},
]
`;
exports[`GOLDEN response stream: Kiro → OpenAI > text + reasoning + toolUse + usage + stop 1`] = `
[
{
"choices": [
{
"delta": {
"content": "Hello",
"role": "assistant",
},
"finish_reason": null,
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "kiro",
"object": "chat.completion.chunk",
},
{
"choices": [
{
"delta": {
"reasoning_content": "thinking",
},
"finish_reason": null,
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "kiro",
"object": "chat.completion.chunk",
},
{
"choices": [
{
"delta": {
"tool_calls": [
{
"function": {
"arguments": "{"city":"NYC"}",
"name": "get_weather",
},
"id": "tu_1",
"index": 0,
"type": "function",
},
],
},
"finish_reason": null,
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "kiro",
"object": "chat.completion.chunk",
},
{
"choices": [
{
"delta": {},
"finish_reason": "stop",
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "kiro",
"object": "chat.completion.chunk",
"usage": {
"completion_tokens": 5,
"prompt_tokens": 10,
"total_tokens": 15,
},
},
]
`;
exports[`GOLDEN response stream: Ollama → OpenAI > content + thinking + tool_calls + done usage 1`] = `
[
{
"choices": [
{
"delta": {
"content": "Hi",
"reasoning_content": "reason",
},
"finish_reason": null,
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "qwen3",
"object": "chat.completion.chunk",
},
{
"choices": [
{
"delta": {
"tool_calls": [
{
"function": {
"arguments": "{"q":"x"}",
"name": "search",
},
"id": "call_0_<TS>",
"index": 0,
"type": "function",
},
],
},
"finish_reason": null,
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "qwen3",
"object": "chat.completion.chunk",
},
{
"choices": [
{
"delta": {},
"finish_reason": "tool_calls",
"index": 0,
},
],
"created": 0,
"id": "chatcmpl-<TS>",
"model": "qwen3",
"object": "chat.completion.chunk",
"usage": {
"completion_tokens": 4,
"prompt_tokens": 8,
"total_tokens": 12,
},
},
]
`;
tests/translator/golden-response-stream.test.js
+31 -3
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();
});
});