mirror of https://github.com/openclaw/openclaw.git
109 lines
3.6 KiB
TypeScript
109 lines
3.6 KiB
TypeScript
import { beforeAll, describe, expect, it } from "vitest";
|
|
import type { OpenClawConfig } from "../config/config.js";
|
|
import { BUNDLED_WEB_SEARCH_PLUGIN_IDS } from "./bundled-web-search-ids.js";
|
|
import { loadPluginManifestRegistry } from "./manifest-registry.js";
|
|
|
|
let hasBundledWebSearchCredential: typeof import("./bundled-web-search-registry.js").hasBundledWebSearchCredential;
|
|
let listBundledWebSearchProviders: typeof import("./bundled-web-search.js").listBundledWebSearchProviders;
|
|
let resolveBundledWebSearchPluginIds: typeof import("./bundled-web-search.js").resolveBundledWebSearchPluginIds;
|
|
|
|
function resolveManifestBundledWebSearchPluginIds() {
|
|
return loadPluginManifestRegistry({})
|
|
.plugins.filter(
|
|
(plugin) =>
|
|
plugin.origin === "bundled" && (plugin.contracts?.webSearchProviders?.length ?? 0) > 0,
|
|
)
|
|
.map((plugin) => plugin.id)
|
|
.toSorted((left, right) => left.localeCompare(right));
|
|
}
|
|
|
|
async function resolveRegistryBundledWebSearchPluginIds() {
|
|
return listBundledWebSearchProviders()
|
|
.map(({ pluginId }) => pluginId)
|
|
.filter((value, index, values) => values.indexOf(value) === index)
|
|
.toSorted((left, right) => left.localeCompare(right));
|
|
}
|
|
|
|
beforeAll(async () => {
|
|
({ listBundledWebSearchProviders, resolveBundledWebSearchPluginIds } =
|
|
await import("./bundled-web-search.js"));
|
|
({ hasBundledWebSearchCredential } = await import("./bundled-web-search-registry.js"));
|
|
});
|
|
|
|
function expectBundledWebSearchIds(actual: readonly string[], expected: readonly string[]) {
|
|
expect(actual).toEqual(expected);
|
|
}
|
|
|
|
function expectBundledWebSearchAlignment(params: {
|
|
actual: readonly string[];
|
|
expected: readonly string[];
|
|
}) {
|
|
expectBundledWebSearchIds(params.actual, params.expected);
|
|
}
|
|
|
|
describe("bundled web search metadata", () => {
|
|
it("keeps bundled web search compat ids aligned with bundled manifests", async () => {
|
|
expectBundledWebSearchAlignment({
|
|
actual: resolveBundledWebSearchPluginIds({}),
|
|
expected: resolveManifestBundledWebSearchPluginIds(),
|
|
});
|
|
});
|
|
|
|
it("keeps bundled web search fast-path ids aligned with the registry", async () => {
|
|
expectBundledWebSearchAlignment({
|
|
actual: [...BUNDLED_WEB_SEARCH_PLUGIN_IDS],
|
|
expected: await resolveRegistryBundledWebSearchPluginIds(),
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("hasBundledWebSearchCredential", () => {
|
|
const baseCfg = {
|
|
agents: { defaults: { model: { primary: "ollama/mistral-8b" } } },
|
|
browser: { enabled: false },
|
|
tools: { web: { fetch: { enabled: false } } },
|
|
} satisfies OpenClawConfig;
|
|
|
|
it.each([
|
|
{
|
|
name: "detects google plugin web search credentials",
|
|
config: {
|
|
...baseCfg,
|
|
plugins: {
|
|
entries: {
|
|
google: { enabled: true, config: { webSearch: { apiKey: "AIza-test" } } },
|
|
},
|
|
},
|
|
} satisfies OpenClawConfig,
|
|
env: {},
|
|
},
|
|
{
|
|
name: "detects gemini env credentials",
|
|
config: baseCfg,
|
|
env: { GEMINI_API_KEY: "AIza-test" },
|
|
},
|
|
{
|
|
name: "detects xai env credentials",
|
|
config: baseCfg,
|
|
env: { XAI_API_KEY: "xai-test" },
|
|
},
|
|
{
|
|
name: "detects kimi env credentials",
|
|
config: baseCfg,
|
|
env: { KIMI_API_KEY: "sk-kimi-test" },
|
|
},
|
|
{
|
|
name: "detects moonshot env credentials",
|
|
config: baseCfg,
|
|
env: { MOONSHOT_API_KEY: "sk-moonshot-test" },
|
|
},
|
|
{
|
|
name: "detects openrouter env credentials through bundled web search providers",
|
|
config: baseCfg,
|
|
env: { OPENROUTER_API_KEY: "sk-or-v1-test" },
|
|
},
|
|
] as const)("$name", async ({ config, env }) => {
|
|
expect(hasBundledWebSearchCredential({ config, env })).toBe(true);
|
|
});
|
|
});
|