File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 1919async 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.\n Prompt: '{ query } '\n Response: '{ query_response } '" )
You can’t perform that action at this time.
0 commit comments