fix: infer perplexity baseUrl from api key
This commit is contained in:
59
src/agents/tools/web-search.test.ts
Normal file
59
src/agents/tools/web-search.test.ts
Normal file
@@ -0,0 +1,59 @@
|
||||
import { describe, expect, it } from "vitest";
|
||||
|
||||
import { __testing } from "./web-search.js";
|
||||
|
||||
const { inferPerplexityBaseUrlFromApiKey, resolvePerplexityBaseUrl } = __testing;
|
||||
|
||||
describe("web_search perplexity baseUrl defaults", () => {
|
||||
it("detects a Perplexity key prefix", () => {
|
||||
expect(inferPerplexityBaseUrlFromApiKey("pplx-123")).toBe("direct");
|
||||
});
|
||||
|
||||
it("detects an OpenRouter key prefix", () => {
|
||||
expect(inferPerplexityBaseUrlFromApiKey("sk-or-v1-123")).toBe("openrouter");
|
||||
});
|
||||
|
||||
it("returns undefined for unknown key formats", () => {
|
||||
expect(inferPerplexityBaseUrlFromApiKey("unknown-key")).toBeUndefined();
|
||||
});
|
||||
|
||||
it("prefers explicit baseUrl over key-based defaults", () => {
|
||||
expect(
|
||||
resolvePerplexityBaseUrl(
|
||||
{ baseUrl: "https://example.com" },
|
||||
"config",
|
||||
"pplx-123",
|
||||
),
|
||||
).toBe("https://example.com");
|
||||
});
|
||||
|
||||
it("defaults to direct when using PERPLEXITY_API_KEY", () => {
|
||||
expect(resolvePerplexityBaseUrl(undefined, "perplexity_env")).toBe(
|
||||
"https://api.perplexity.ai",
|
||||
);
|
||||
});
|
||||
|
||||
it("defaults to OpenRouter when using OPENROUTER_API_KEY", () => {
|
||||
expect(resolvePerplexityBaseUrl(undefined, "openrouter_env")).toBe(
|
||||
"https://openrouter.ai/api/v1",
|
||||
);
|
||||
});
|
||||
|
||||
it("defaults to direct when config key looks like Perplexity", () => {
|
||||
expect(resolvePerplexityBaseUrl(undefined, "config", "pplx-123")).toBe(
|
||||
"https://api.perplexity.ai",
|
||||
);
|
||||
});
|
||||
|
||||
it("defaults to OpenRouter when config key looks like OpenRouter", () => {
|
||||
expect(resolvePerplexityBaseUrl(undefined, "config", "sk-or-v1-123")).toBe(
|
||||
"https://openrouter.ai/api/v1",
|
||||
);
|
||||
});
|
||||
|
||||
it("defaults to OpenRouter for unknown config key formats", () => {
|
||||
expect(resolvePerplexityBaseUrl(undefined, "config", "weird-key")).toBe(
|
||||
"https://openrouter.ai/api/v1",
|
||||
);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user