mjschock commited on
Commit
4e02cb8
·
unverified ·
1 Parent(s): 50aafe2

Refactor SmartSearchTool in tool.py to improve logging and variable naming for web search results. Update return statements to ensure consistency in terminology, enhancing clarity in the output messages.

Browse files
Files changed (1) hide show
  1. tools/smart_search/tool.py +5 -5
tools/smart_search/tool.py CHANGED
@@ -21,14 +21,14 @@ class SmartSearchTool(Tool):
21
  logger.info(f"Starting smart search for query: {query}")
22
 
23
  # Get web search results
24
- web_result = self.web_search_tool.forward(query)
25
- logger.info(f"Web search result: {web_result[:100]}...")
26
 
27
  # Extract URLs from the web search result
28
- urls = re.findall(r'https?://[^\s)]+', web_result)
29
  if not urls:
30
  logger.info("No URLs found in web search result")
31
- return f"Web search result:\n{web_result}"
32
 
33
  # Visit each URL and get its content
34
  contents = []
@@ -43,7 +43,7 @@ class SmartSearchTool(Tool):
43
  contents.append(f"\nError visiting {url}: {e}")
44
 
45
  # Combine all results
46
- return f"Web search result:\n{web_result}\n" + "\n".join(contents)
47
 
48
 
49
  def main(query: str) -> str:
 
21
  logger.info(f"Starting smart search for query: {query}")
22
 
23
  # Get web search results
24
+ web_search_results = self.web_search_tool.forward(query)
25
+ logger.info(f"Web search results: {web_search_results[:100]}...")
26
 
27
  # Extract URLs from the web search result
28
+ urls = re.findall(r'https?://[^\s)]+', web_search_results)
29
  if not urls:
30
  logger.info("No URLs found in web search result")
31
+ return f"Web search results:\n{web_search_results}"
32
 
33
  # Visit each URL and get its content
34
  contents = []
 
43
  contents.append(f"\nError visiting {url}: {e}")
44
 
45
  # Combine all results
46
+ return f"Web search results:\n{web_search_results}\n" + "\n".join(contents)
47
 
48
 
49
  def main(query: str) -> str: