Spaces:
Build error
Build error
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
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 |
-
|
25 |
-
logger.info(f"Web search
|
26 |
|
27 |
# Extract URLs from the web search result
|
28 |
-
urls = re.findall(r'https?://[^\s)]+',
|
29 |
if not urls:
|
30 |
logger.info("No URLs found in web search result")
|
31 |
-
return f"Web search
|
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
|
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:
|