Skip to content

Commit 2ab4a93

Browse files
committed
Suppress failed web search
1 parent 40c9c67 commit 2ab4a93

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

apex/validator/generate_query.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,13 @@
1919
async def generate_query(llm: LLMBase, websearch: WebSearchBase) -> str:
2020
random_words = " ".join(random.sample(get_english_words(), 3))
2121
# Perform a lightweight search and pick a single result as context.
22-
search_results = await websearch.search(random_words, max_results=5)
23-
search_website = random.choice(search_results)
24-
search_content = search_website.content
22+
try:
23+
search_results = await websearch.search(random_words, max_results=5)
24+
search_website = random.choice(search_results)
25+
search_content = search_website.content
26+
except BaseException as exc:
27+
logger.error(f"Error during web search: {exc}")
28+
search_content = ""
2529
query = QUERY_PROMPT_TEMPLATE.format(context=search_content)
2630
query_response, _ = await llm.invoke([{"role": "user", "content": query}])
2731
logger.debug(f"Generated query.\nPrompt: '{query}'\nResponse: '{query_response}'")

0 commit comments

Comments
 (0)