romtuck/openclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge ec8869d87f into da71eaebd2

Browse Source
This commit is contained in:
zxz0622 2026-01-30 20:00:37 +08:00 committed by GitHub
commit 1608da6eaa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/agents/tools/web-search.ts
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";
@ -32,6 +33,18 @@ const SEARCH_CACHE = new Map<string, CacheEntry<Record<string, unknown>>>();
const BRAVE_FRESHNESS_SHORTCUTS = new Set(["pd", "pw", "pm", "py"]); const BRAVE_FRESHNESS_SHORTCUTS = new Set(["pd", "pw", "pm", "py"]);
const BRAVE_FRESHNESS_RANGE = /^(\d{4}-\d{2}-\d{2})to(\d{4}-\d{2}-\d{2})$/; const BRAVE_FRESHNESS_RANGE = /^(\d{4}-\d{2}-\d{2})to(\d{4}-\d{2}-\d{2})$/;
/**
* Get proxy dispatcher from environment variables for HTTP requests.
* Respects HTTP_PROXY, HTTPS_PROXY environment variables.
*/
function getProxyDispatcher(): ProxyAgent | undefined {
const proxyUrl = process.env.HTTPS_PROXY || process.env.HTTP_PROXY;
if (proxyUrl) {
return new ProxyAgent(proxyUrl);
}
return undefined;
}
const WebSearchSchema = Type.Object({ const WebSearchSchema = Type.Object({
query: Type.String({ description: "Search query string." }), query: Type.String({ description: "Search query string." }),
count: Type.Optional( count: Type.Optional(
@ -273,6 +286,7 @@ async function runPerplexitySearch(params: {
timeoutSeconds: number; timeoutSeconds: number;
}): 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 proxyDispatcher = getProxyDispatcher();
const res = await fetch(endpoint, { const res = await fetch(endpoint, {
method: "POST", method: "POST",
@ -292,6 +306,7 @@ async function runPerplexitySearch(params: {
], ],
}), }),
signal: withTimeout(undefined, params.timeoutSeconds * 1000), signal: withTimeout(undefined, params.timeoutSeconds * 1000),
...(proxyDispatcher && { dispatcher: proxyDispatcher }),
}); });
if (!res.ok) { if (!res.ok) {
@ -371,6 +386,7 @@ async function runWebSearch(params: {
url.searchParams.set("freshness", params.freshness); url.searchParams.set("freshness", params.freshness);
} }
const proxyDispatcher = getProxyDispatcher();
const res = await fetch(url.toString(), { const res = await fetch(url.toString(), {
method: "GET", method: "GET",
headers: { headers: {
@ -378,6 +394,7 @@ 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),
...(proxyDispatcher && { dispatcher: proxyDispatcher }),
}); });
if (!res.ok) { if (!res.ok) {