refactor(tests): improve formatting and structure in SERPEX integration tests

Author: divyeshradadiya

llama-index-integrations/tools/llama-index-tools-serpex/examples/serpex_example.py

@@ -1,23 +1,27 @@
 """Example usage of SERPEX tool with LlamaIndex."""
 
 import os
+
 from llama_index.tools.serpex import SerpexToolSpec
 
 # Set your API key (or use environment variable SERPEX_API_KEY)
 os.environ["SERPEX_API_KEY"] = "your_api_key_here"
 
+
 def basic_search_example():
     """Basic search example."""
     print("=" * 60)
     print("Basic Search Example")
     print("=" * 60)
-    
+
     # Initialize tool
     tool = SerpexToolSpec()
-    
+
     # Perform search
-    results = tool.search("latest developments in artificial intelligence", num_results=5)
-    
+    results = tool.search(
+        "latest developments in artificial intelligence", num_results=5
+    )
+
     for doc in results:
         print(doc.text)
     print()
@@ -28,17 +32,15 @@ def location_search_example():
     print("=" * 60)
     print("Location-Based Search Example")
     print("=" * 60)
-    
+
     # Initialize tool
     tool = SerpexToolSpec()
-    
+
     # Search with location
     results = tool.search_with_location(
-        query="best Italian restaurants",
-        location="San Francisco, CA",
-        num_results=5
+        query="best Italian restaurants", location="San Francisco, CA", num_results=5
     )
-    
+
     for doc in results:
         print(doc.text)
     print()
@@ -49,18 +51,18 @@ def international_search_example():
     print("=" * 60)
     print("International Search Example")
     print("=" * 60)
-    
+
     # Initialize tool
     tool = SerpexToolSpec()
-    
+
     # Search with country and language
     results = tool.search(
         query="noticias de tecnología",
         num_results=5,
         gl="es",  # Spain
-        hl="es"   # Spanish
+        hl="es",  # Spanish
     )
-    
+
     for doc in results:
         print(doc.text)
     print()
@@ -71,29 +73,27 @@ def agent_example():
     print("=" * 60)
     print("Agent Example")
     print("=" * 60)
-    
+
     try:
         from llama_index.agent.openai import OpenAIAgent
         from llama_index.llms.openai import OpenAI
-        
+
         # Initialize SERPEX tool
         serpex_tool = SerpexToolSpec()
-        
+
         # Create agent
         llm = OpenAI(model="gpt-4")
         agent = OpenAIAgent.from_tools(
-            serpex_tool.to_tool_list(),
-            llm=llm,
-            verbose=True
+            serpex_tool.to_tool_list(), llm=llm, verbose=True
         )
-        
+
         # Ask question that requires web search
         response = agent.chat(
             "What are the latest features announced for LlamaIndex? "
             "Search the web for recent news."
         )
         print(response)
-        
+
     except ImportError:
         print("OpenAI dependencies not installed. Install with:")
         print("pip install llama-index-agent-openai llama-index-llms-openai")
@@ -107,7 +107,7 @@ def main():
         print("Please set SERPEX_API_KEY environment variable")
         print("Get your API key at: https://serpex.io/dashboard")
         return
-    
+
     # Run examples
     basic_search_example()
     location_search_example()

llama-index-integrations/tools/llama-index-tools-serpex/test_local.py

@@ -2,6 +2,7 @@
 """Local test script for SERPEX tool integration."""
 
 import os
+
 from llama_index.tools.serpex import SerpexToolSpec
 
 # Test 1: Check initialization
@@ -18,27 +19,28 @@
     else:
         tool = SerpexToolSpec(api_key=api_key)
         print("✅ Tool initialized with real API key!")
-        
+
         # Test 2: Basic search
         print("\nTest 2: Testing basic search...")
         results = tool.search("LlamaIndex tutorial", num_results=3)
         print(f"✅ Search returned {len(results)} results (as Document objects):")
         for i, result in enumerate(results, 1):
             print(f"\nResult {i}:")
             print(result.text[:500])  # Print first 500 chars
-        
+
         # Test 3: Check tool list conversion
-        print("\n" + "="*60)
+        print("\n" + "=" * 60)
         print("Test 4: Testing tool list conversion...")
         tool_list = tool.to_tool_list()
         print(f"✅ Tool list has {len(tool_list)} tools:")
         for t in tool_list:
             print(f"   - {t.metadata.name}: {t.metadata.description[:60]}...")
-        
-        print("\n" + "="*60)
+
+        print("\n" + "=" * 60)
         print("🎉 All tests passed! SERPEX integration is working!")
-        
+
 except Exception as e:
     print(f"❌ Test failed: {e}")
     import traceback
+
     traceback.print_exc()