Update tools.py

tools.py

@@ -111,9 +111,9 @@ def simple_search(query: str, max_results: int = 5) -> List[str]:
             print(f"Search error: {str(e)}")
             return []
 
-    # Retry logic with exponential backoff and longer delays
+    # Retry logic with rate limit handling
     max_attempts = 4
-    base_delay = 30  # increased base delay to 30 seconds
+    rate_limit_delay = 20  # seconds to wait on rate limit
     
     # Clean the input query
     query = query.strip()
@@ -128,14 +128,20 @@ def simple_search(query: str, max_results: int = 5) -> List[str]:
                 return results
             print(f"Attempt {attempt + 1}/{max_attempts}: No results found")
         except Exception as e:
-            print(f"Attempt {attempt + 1}/{max_attempts} failed: {str(e)}")
-            if attempt < max_attempts - 1:
-                # Longer delays between attempts
-                delay = base_delay * (2 ** attempt)  # exponential backoff
+            error_msg = str(e)
+            print(f"Attempt {attempt + 1}/{max_attempts} failed: {error_msg}")
+            
+            # Check if it's a rate limit error
+            if "Ratelimit" in error_msg or "202" in error_msg:
+                print(f"Rate limit detected. Waiting {rate_limit_delay} seconds...")
+                time.sleep(rate_limit_delay)
+            elif attempt < max_attempts - 1:
+                # For other errors, use exponential backoff
+                delay = 30 * (2 ** attempt)
                 print(f"Retrying in {delay}s...")
                 time.sleep(delay)
             else:
-                print(f"All {max_attempts} attempts failed. Last exception: {str(e)}")
+                print(f"All {max_attempts} attempts failed. Last exception: {error_msg}")
                 return []
 
     return []