This commit is contained in:
kdwycz 2026-01-30 17:05:45 +05:30 committed by GitHub
commit 1711e1f878
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 23 additions and 3 deletions

View File

@ -107,6 +107,7 @@ crypto/prepaid).
search: { search: {
enabled: true, enabled: true,
provider: "perplexity", provider: "perplexity",
proxy: "http://127.0.0.1:7890", // optional
perplexity: { perplexity: {
// API key (optional if OPENROUTER_API_KEY or PERPLEXITY_API_KEY is set) // API key (optional if OPENROUTER_API_KEY or PERPLEXITY_API_KEY is set)
apiKey: "sk-or-v1-...", apiKey: "sk-or-v1-...",
@ -160,7 +161,8 @@ Search the web using your configured provider.
apiKey: "BRAVE_API_KEY_HERE", // optional if BRAVE_API_KEY is set apiKey: "BRAVE_API_KEY_HERE", // optional if BRAVE_API_KEY is set
maxResults: 5, maxResults: 5,
timeoutSeconds: 30, timeoutSeconds: 30,
cacheTtlMinutes: 15 cacheTtlMinutes: 15,
proxy: "http://127.0.0.1:7890" // optional
} }
} }
} }

View File

@ -1,4 +1,5 @@
import { Type } from "@sinclair/typebox"; import { Type } from "@sinclair/typebox";
import { ProxyAgent } from "undici";
import type { OpenClawConfig } from "../../config/config.js"; import type { OpenClawConfig } from "../../config/config.js";
import { formatCliCommand } from "../../cli/command-format.js"; import { formatCliCommand } from "../../cli/command-format.js";
@ -122,6 +123,12 @@ function resolveSearchApiKey(search?: WebSearchConfig): string | undefined {
return fromConfig || fromEnv || undefined; return fromConfig || fromEnv || undefined;
} }
function resolveProxy(search?: WebSearchConfig): string | undefined {
const fromConfig =
search && "proxy" in search && typeof search.proxy === "string" ? search.proxy.trim() : "";
return fromConfig || undefined;
}
function missingSearchKeyPayload(provider: (typeof SEARCH_PROVIDERS)[number]) { function missingSearchKeyPayload(provider: (typeof SEARCH_PROVIDERS)[number]) {
if (provider === "perplexity") { if (provider === "perplexity") {
return { return {
@ -271,9 +278,12 @@ async function runPerplexitySearch(params: {
baseUrl: string; baseUrl: string;
model: string; model: string;
timeoutSeconds: number; timeoutSeconds: number;
proxy?: string;
}): Promise<{ content: string; citations: string[] }> { }): Promise<{ content: string; citations: string[] }> {
const endpoint = `${params.baseUrl.replace(/\/$/, "")}/chat/completions`; const endpoint = `${params.baseUrl.replace(/\/$/, "")}/chat/completions`;
const dispatcher = params.proxy ? new ProxyAgent(params.proxy) : undefined;
const res = await fetch(endpoint, { const res = await fetch(endpoint, {
method: "POST", method: "POST",
headers: { headers: {
@ -292,7 +302,8 @@ async function runPerplexitySearch(params: {
], ],
}), }),
signal: withTimeout(undefined, params.timeoutSeconds * 1000), signal: withTimeout(undefined, params.timeoutSeconds * 1000),
}); dispatcher,
} as RequestInit);
if (!res.ok) { if (!res.ok) {
const detail = await readResponseText(res); const detail = await readResponseText(res);
@ -319,6 +330,7 @@ async function runWebSearch(params: {
freshness?: string; freshness?: string;
perplexityBaseUrl?: string; perplexityBaseUrl?: string;
perplexityModel?: string; perplexityModel?: string;
proxy?: string;
}): Promise<Record<string, unknown>> { }): Promise<Record<string, unknown>> {
const cacheKey = normalizeCacheKey( const cacheKey = normalizeCacheKey(
params.provider === "brave" params.provider === "brave"
@ -337,6 +349,7 @@ async function runWebSearch(params: {
baseUrl: params.perplexityBaseUrl ?? DEFAULT_PERPLEXITY_BASE_URL, baseUrl: params.perplexityBaseUrl ?? DEFAULT_PERPLEXITY_BASE_URL,
model: params.perplexityModel ?? DEFAULT_PERPLEXITY_MODEL, model: params.perplexityModel ?? DEFAULT_PERPLEXITY_MODEL,
timeoutSeconds: params.timeoutSeconds, timeoutSeconds: params.timeoutSeconds,
proxy: params.proxy,
}); });
const payload = { const payload = {
@ -371,6 +384,8 @@ async function runWebSearch(params: {
url.searchParams.set("freshness", params.freshness); url.searchParams.set("freshness", params.freshness);
} }
const dispatcher = params.proxy ? new ProxyAgent(params.proxy) : undefined;
const res = await fetch(url.toString(), { const res = await fetch(url.toString(), {
method: "GET", method: "GET",
headers: { headers: {
@ -378,7 +393,8 @@ async function runWebSearch(params: {
"X-Subscription-Token": params.apiKey, "X-Subscription-Token": params.apiKey,
}, },
signal: withTimeout(undefined, params.timeoutSeconds * 1000), signal: withTimeout(undefined, params.timeoutSeconds * 1000),
}); dispatcher,
} as RequestInit);
if (!res.ok) { if (!res.ok) {
const detail = await readResponseText(res); const detail = await readResponseText(res);
@ -476,6 +492,7 @@ export function createWebSearchTool(options?: {
perplexityAuth?.apiKey, perplexityAuth?.apiKey,
), ),
perplexityModel: resolvePerplexityModel(perplexityConfig), perplexityModel: resolvePerplexityModel(perplexityConfig),
proxy: resolveProxy(search),
}); });
return jsonResult(result); return jsonResult(result);
}, },

View File

@ -170,6 +170,7 @@ export const ToolsWebSearchSchema = z
maxResults: z.number().int().positive().optional(), maxResults: z.number().int().positive().optional(),
timeoutSeconds: z.number().int().positive().optional(), timeoutSeconds: z.number().int().positive().optional(),
cacheTtlMinutes: z.number().nonnegative().optional(), cacheTtlMinutes: z.number().nonnegative().optional(),
proxy: z.string().optional(),
perplexity: z perplexity: z
.object({ .object({
apiKey: z.string().optional(), apiKey: z.string().optional(),