feat: add SerperSearch tool and suppress Pydantic warnings

- Add SerperSearch tool for Google search via Serper API
- Auto-register tool on import with @register_tool
- Suppress Pydantic serialization warnings from litellm
- Set version to 0.1.2

Author: SNHuan

easyagent/model/litellm_model.py

@@ -1,8 +1,12 @@
 import json
+import warnings
 from typing import Any
 
 import litellm
 
+# Suppress Pydantic serialization warnings from litellm
+warnings.filterwarnings("ignore", message="Pydantic serializer warnings")
+
 from easyagent.config.base import is_debug
 from easyagent.debug.log import Color, Logger
 from easyagent.model.base import BaseLLM

easyagent/test/test_serper.py

@@ -0,0 +1,25 @@
+import asyncio
+
+from easyagent.agent import ReactAgent
+from easyagent.config.base import ModelConfig
+from easyagent.model.litellm_model import LiteLLMModel
+from easyagent.tool import SerperSearch  # 导入即注册
+
+
+async def main():
+    config = ModelConfig.load()
+    model = LiteLLMModel(**config.get_model("claude-4-5-haiku"))
+
+    agent = ReactAgent(
+        model=model,
+        tools=["serper_search"],
+        system_prompt="You are a helpful assistant. Use the search tool to find information.",
+        max_iterations=5,
+    )
+
+    result = await agent.run("What is the latest version of Python?")
+    print(f"Result: {result}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

easyagent/tool/__init__.py

@@ -1,5 +1,6 @@
 from easyagent.tool.base import Tool
 from easyagent.tool.manager import ToolManager, register_tool
+from easyagent.tool.web import SerperSearch
 
-__all__ = ["Tool", "ToolManager", "register_tool"]
+__all__ = ["Tool", "ToolManager", "register_tool", "SerperSearch"]
 

easyagent/tool/web/__init__.py

@@ -0,0 +1,3 @@
+from easyagent.tool.web.serper import SerperSearch
+
+__all__ = ["SerperSearch"]

easyagent/tool/web/serper.py

@@ -0,0 +1,92 @@
+"""Serper Google Search Tool - https://serper.dev"""
+
+import os
+from typing import Any
+
+import httpx
+
+from easyagent.tool.manager import register_tool
+
+
+@register_tool
+class SerperSearch:
+    """Google Search via Serper API"""
+
+    name = "serper_search"
+    type = "function"
+    description = "Search the web using Google via Serper API. Returns top search results."
+    parameters = {
+        "type": "object",
+        "properties": {
+            "query": {
+                "type": "string",
+                "description": "The search query",
+            },
+            "num_results": {
+                "type": "integer",
+                "description": "Number of results to return (default: 5, max: 10)",
+            },
+        },
+        "required": ["query"],
+    }
+
+    _api_key: str | None = None
+
+    def init(self) -> None:
+        self._api_key = os.environ.get("SERPER_API_KEY")
+
+    def execute(self, query: str, num_results: int = 5, **kwargs: Any) -> str:
+        if not self._api_key:
+            return "Error: SERPER_API_KEY environment variable not set"
+
+        num_results = min(max(1, num_results), 10)
+
+        try:
+            response = httpx.post(
+                "https://google.serper.dev/search",
+                headers={
+                    "X-API-KEY": self._api_key,
+                    "Content-Type": "application/json",
+                },
+                json={"q": query, "num": num_results},
+                timeout=30,
+            )
+            response.raise_for_status()
+            data = response.json()
+            return _format_results(data, num_results)
+        except httpx.HTTPStatusError as e:
+            return f"Error: HTTP {e.response.status_code} - {e.response.text}"
+        except Exception as e:
+            return f"Error: {e}"
+
+
+def _format_results(data: dict[str, Any], limit: int) -> str:
+    """Format search results into readable text"""
+    lines: list[str] = []
+
+    # Knowledge Graph (if present)
+    if kg := data.get("knowledgeGraph"):
+        lines.append(f"[Knowledge Graph] {kg.get('title', '')}")
+        if desc := kg.get("description"):
+            lines.append(f"  {desc}")
+        lines.append("")
+
+    # Answer Box (if present)
+    if answer := data.get("answerBox"):
+        if snippet := answer.get("snippet") or answer.get("answer"):
+            lines.append(f"[Answer] {snippet}")
+            lines.append("")
+
+    # Organic Results
+    organic = data.get("organic", [])[:limit]
+    for i, result in enumerate(organic, 1):
+        title = result.get("title", "No title")
+        link = result.get("link", "")
+        snippet = result.get("snippet", "")
+        lines.append(f"{i}. {title}")
+        lines.append(f"   URL: {link}")
+        if snippet:
+            lines.append(f"   {snippet}")
+        lines.append("")
+
+    return "\n".join(lines).strip() or "No results found"

pyproject.toml

@@ -9,10 +9,10 @@ description = "A lightweight AI Agent framework built on LiteLLM with ReAct reas
 readme = "README.md"
 license = {text = "MIT"}
 authors = [
-    {name = "Yiran Peng", email = "pyr@pyr.sh"}
+    {name = "Yiran Peng", email = "amagipeng@gmail.com"}
 ]
 maintainers = [
-    {name = "Yiran Peng", email = "pyr@pyr.sh"}
+    {name = "Yiran Peng", email = "amagipeng@gmail.com"}
 ]
 requires-python = ">=3.12"
 classifiers = [