wehub-resource-sync
84 lines
3.0 KiB
TypeScript
84 lines
3.0 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
const base = await import("../../open-sse/executors/base.ts");
|
|
|
|
test("isOpenAICompatibleEndpoint matches openai-compatible-* providers", () => {
|
|
assert.equal(base.isOpenAICompatibleEndpoint("openai-compatible-foo", "https://x/y"), true);
|
|
assert.equal(
|
|
base.isOpenAICompatibleEndpoint("claude", "https://api.anthropic.com/v1/messages"),
|
|
false
|
|
);
|
|
});
|
|
|
|
test("isOpenAICompatibleEndpoint matches chat/completions and responses URLs", () => {
|
|
assert.equal(
|
|
base.isOpenAICompatibleEndpoint("groq", "https://api.groq.com/openai/v1/chat/completions"),
|
|
true
|
|
);
|
|
assert.equal(base.isOpenAICompatibleEndpoint("groq", "https://x/v1/responses"), true);
|
|
assert.equal(base.isOpenAICompatibleEndpoint("groq", "https://x/v1/embeddings"), false);
|
|
});
|
|
|
|
test("strips X-Stainless-* headers on OpenAI-compatible passthrough", () => {
|
|
const headers: Record<string, string> = {
|
|
"Content-Type": "application/json",
|
|
Authorization: "Bearer sk-test",
|
|
"X-Stainless-Lang": "js",
|
|
"X-Stainless-OS": "Linux",
|
|
"x-stainless-runtime": "node",
|
|
};
|
|
const stripped = base.stripStainlessHeadersForOpenAICompat(
|
|
headers,
|
|
"openai-compatible-acme",
|
|
"https://acme.example/v1/chat/completions"
|
|
);
|
|
assert.deepEqual(stripped.sort(), ["X-Stainless-Lang", "X-Stainless-OS", "x-stainless-runtime"]);
|
|
assert.equal(headers["X-Stainless-Lang"], undefined);
|
|
assert.equal(headers["X-Stainless-OS"], undefined);
|
|
assert.equal(headers["x-stainless-runtime"], undefined);
|
|
// Non-stainless headers untouched.
|
|
assert.equal(headers["Authorization"], "Bearer sk-test");
|
|
assert.equal(headers["Content-Type"], "application/json");
|
|
});
|
|
|
|
test("does NOT strip X-Stainless-* for non-OpenAI-compatible endpoints", () => {
|
|
const headers: Record<string, string> = {
|
|
"X-Stainless-Lang": "js",
|
|
"X-Stainless-Package-Version": "1.0.0",
|
|
};
|
|
const stripped = base.stripStainlessHeadersForOpenAICompat(
|
|
headers,
|
|
"claude",
|
|
"https://api.anthropic.com/v1/messages"
|
|
);
|
|
assert.deepEqual(stripped, []);
|
|
// Claude-code-compat spoofing path keeps its X-Stainless-* headers intact.
|
|
assert.equal(headers["X-Stainless-Lang"], "js");
|
|
assert.equal(headers["X-Stainless-Package-Version"], "1.0.0");
|
|
});
|
|
|
|
test("normalizes SDK-derived User-Agent on OpenAI-compatible request", () => {
|
|
const headers: Record<string, string> = {
|
|
"User-Agent": "OpenAI/NodeJS 4.20.0 undici",
|
|
};
|
|
base.stripStainlessHeadersForOpenAICompat(
|
|
headers,
|
|
"openai-compatible-acme",
|
|
"https://acme.example/v1/chat/completions"
|
|
);
|
|
assert.equal(headers["User-Agent"], "Mozilla/5.0 (compatible; OpenAI Compatible)");
|
|
});
|
|
|
|
test("leaves a non-SDK User-Agent untouched", () => {
|
|
const headers: Record<string, string> = {
|
|
"User-Agent": "my-custom-agent/2.0",
|
|
};
|
|
base.stripStainlessHeadersForOpenAICompat(
|
|
headers,
|
|
"openai-compatible-acme",
|
|
"https://acme.example/v1/chat/completions"
|
|
);
|
|
assert.equal(headers["User-Agent"], "my-custom-agent/2.0");
|
|
});
|