25 lines
1.0 KiB
TypeScript
25 lines
1.0 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { detectFormatFromEndpoint } from "@omniroute/open-sse/services/provider.ts";
|
|
|
|
const ENVELOPE = {
|
|
model: "gemini-2.5-pro",
|
|
project: "projects/test",
|
|
request: { contents: [{ role: "user", parts: [{ text: "oi" }] }] },
|
|
};
|
|
|
|
test("detectFormatFromEndpoint classifies the /v1/antigravity path as antigravity", () => {
|
|
assert.equal(detectFormatFromEndpoint(ENVELOPE, "/v1/antigravity"), "antigravity");
|
|
assert.equal(detectFormatFromEndpoint(ENVELOPE, "/api/v1/antigravity"), "antigravity");
|
|
assert.equal(
|
|
detectFormatFromEndpoint(ENVELOPE, "/v1/antigravity:streamGenerateContent"),
|
|
"antigravity"
|
|
);
|
|
});
|
|
|
|
test("the antigravity carve-out does not disturb the other endpoint formats", () => {
|
|
assert.equal(detectFormatFromEndpoint({ messages: [] }, "/v1/messages"), "claude");
|
|
assert.equal(detectFormatFromEndpoint({ messages: [] }, "/v1/chat/completions"), "openai");
|
|
assert.equal(detectFormatFromEndpoint({ input: "x" }, "/v1/responses"), "openai-responses");
|
|
});
|