Add main function to tool.py for direct execution of SmartSearchTool. Implement logging and command-line argument handling for search queries, enhancing usability and testing capabilities.

tools/smart_search/tool.py

@@ -55,3 +55,43 @@ class SmartSearchTool(Tool):
         # If no Wikipedia link was found or Wikipedia search failed, return the web search result
         logger.info("Returning web search result only")
         return f"Web search result:\n{web_result}"
+
+
+def main(query: str) -> str:
+    """
+    Test function to run the SmartSearchTool directly.
+    
+    Args:
+        query: The search query to test
+        
+    Returns:
+        The search results
+    """
+    # Configure logging
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
+    
+    # Create and run the tool
+    tool = SmartSearchTool()
+    result = tool.forward(query)
+    
+    # Print the result
+    print("\nSearch Results:")
+    print("-" * 80)
+    print(result)
+    print("-" * 80)
+    
+    return result
+
+
+if __name__ == "__main__":
+    import sys
+    
+    if len(sys.argv) > 1:
+        query = " ".join(sys.argv[1:])
+        main(query)
+    else:
+        print("Usage: python tool.py <search query>")
+        print("Example: python tool.py 'Mercedes Sosa discography'")