Merge pull request #1 from pietz/feat/brave-retry

retries, v bump

Author: pietz

mcp_web_tools/search.py

@@ -22,30 +22,42 @@ async def brave_search(query: str, limit: int = 10) -> dict | None:
     headers = {"X-Subscription-Token": api_key}
     params = {"q": query, "count": limit}
 
-    try:
-        async with httpx.AsyncClient() as client:
-            r = await client.get(url, headers=headers, params=params, timeout=10)
-            r.raise_for_status()
-            results = r.json()["web"]["results"]
-            return {
-                "provider": "brave",
-                "results": [
-                    {
-                        "title": x["title"],
-                        "url": x["url"],
-                        "description": x.get("description", ""),
-                    }
-                    for x in results
-                ],
-            }
-    except httpx.HTTPStatusError as e:
-        logger.warning(
-            f"Brave Search API returned status code {e.response.status_code}"
-        )
-    except httpx.TimeoutException:
-        logger.warning("Brave Search API request timed out")
-    except Exception as e:
-        logger.warning(f"Error using Brave Search: {str(e)}")
+    import asyncio
+
+    attempts = 3
+    for attempt in range(1, attempts + 1):
+        try:
+            async with httpx.AsyncClient() as client:
+                r = await client.get(url, headers=headers, params=params, timeout=10)
+                r.raise_for_status()
+                results = r.json()["web"]["results"]
+                return {
+                    "provider": "brave",
+                    "results": [
+                        {
+                            "title": x["title"],
+                            "url": x["url"],
+                            "description": x.get("description", ""),
+                        }
+                        for x in results
+                    ],
+                }
+        except httpx.HTTPStatusError as e:
+            status = e.response.status_code if e.response is not None else None
+            if status == 429 and attempt < attempts:
+                logger.info("Brave rate limited (429); retrying in 1s (%d/%d)", attempt, attempts)
+                await asyncio.sleep(1)
+                continue
+            logger.warning(
+                f"Brave Search API returned status code {status}"
+            )
+            break
+        except httpx.TimeoutException:
+            logger.warning("Brave Search API request timed out")
+            break
+        except Exception as e:
+            logger.warning(f"Error using Brave Search: {str(e)}")
+            break
 
     return None
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcp-web-tools"
-version = "0.7.7"
+version = "0.7.8"
 description = "A powerful MCP server to equip LLMs with web access, search, and content extraction capabilities"
 readme = "README.md"
 authors = [