fix: remove hard-coded 6s timeout from tavily request

This commit is contained in:
Soulter
2026-02-22 16:58:49 +08:00
parent 3678688433
commit d0998a9dfb
+2 -4
View File
@@ -70,7 +70,7 @@ class Main(star.Star):
header = HEADERS header = HEADERS
header.update({"User-Agent": random.choice(USER_AGENTS)}) header.update({"User-Agent": random.choice(USER_AGENTS)})
async with aiohttp.ClientSession(trust_env=True) as session: async with aiohttp.ClientSession(trust_env=True) as session:
async with session.get(url, headers=header, timeout=6) as response: async with session.get(url, headers=header) as response:
html = await response.text(encoding="utf-8") html = await response.text(encoding="utf-8")
doc = Document(html) doc = Document(html)
ret = doc.summary(html_partial=True) ret = doc.summary(html_partial=True)
@@ -151,7 +151,6 @@ class Main(star.Star):
url, url,
json=payload, json=payload,
headers=header, headers=header,
timeout=6,
) as response: ) as response:
if response.status != 200: if response.status != 200:
reason = await response.text() reason = await response.text()
@@ -183,7 +182,6 @@ class Main(star.Star):
url, url,
json=payload, json=payload,
headers=header, headers=header,
timeout=6,
) as response: ) as response:
if response.status != 200: if response.status != 200:
reason = await response.text() reason = await response.text()
@@ -265,7 +263,7 @@ class Main(star.Star):
"transport": "sse", "transport": "sse",
"url": f"http://appbuilder.baidu.com/v2/ai_search/mcp/sse?api_key={key}", "url": f"http://appbuilder.baidu.com/v2/ai_search/mcp/sse?api_key={key}",
"headers": {}, "headers": {},
"timeout": 30, "timeout": 600,
}, },
) )
self.baidu_initialized = True self.baidu_initialized = True