wehub-resource-sync
35 lines
1.1 KiB
TypeScript
35 lines
1.1 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { AntigravityExecutor } from "../../open-sse/executors/antigravity.ts";
|
|
|
|
type ChatCompletionPayload = {
|
|
choices: Array<{
|
|
message: { content: string };
|
|
finish_reason: string;
|
|
}>;
|
|
};
|
|
|
|
test("AntigravityExecutor.collectStreamToResponse normalizes prohibited content finish reasons", async () => {
|
|
const executor = new AntigravityExecutor();
|
|
const response = new Response(
|
|
'data: {"response":{"candidates":[{"content":{"parts":[{"text":"partial text"}]},"finishReason":"PROHIBITED_CONTENT"}]}}\n\n',
|
|
{
|
|
status: 200,
|
|
headers: { "Content-Type": "text/event-stream" },
|
|
}
|
|
);
|
|
|
|
const result = await executor.collectStreamToResponse(
|
|
response,
|
|
"gemini-2.5-flash",
|
|
"https://example.com",
|
|
{ Authorization: "Bearer ag-token" },
|
|
{ request: {} }
|
|
);
|
|
const payload = (await result.response.json()) as ChatCompletionPayload;
|
|
|
|
assert.equal(payload.choices[0].message.content, "partial text");
|
|
assert.equal(payload.choices[0].finish_reason, "content_filter");
|
|
});
|