Squashed 'packages/tools/' content from commit 78317b9c

git-subtree-dir: packages/tools
git-subtree-split: 78317b9c127f18bd040c1d77e3c0840cdc9a5b38

crewai_tools/tools/serpapi_tool/serpapi_google_search_tool.py

@@ -0,0 +1,60 @@
+from typing import Any, Optional, Type
+
+import re
+from pydantic import BaseModel, Field, ConfigDict
+from .serpapi_base_tool import SerpApiBaseTool
+
+try:
+    from serpapi import HTTPError
+except ImportError:
+    HTTPError = Any
+
+
+class SerpApiGoogleSearchToolSchema(BaseModel):
+    """Input for Google Search."""
+
+    search_query: str = Field(
+        ..., description="Mandatory search query you want to use to Google search."
+    )
+    location: Optional[str] = Field(
+        None, description="Location you want the search to be performed in."
+    )
+
+
+class SerpApiGoogleSearchTool(SerpApiBaseTool):
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True, validate_assignment=True, frozen=False
+    )
+    name: str = "Google Search"
+    description: str = (
+        "A tool to perform to perform a Google search with a search_query."
+    )
+    args_schema: Type[BaseModel] = SerpApiGoogleSearchToolSchema
+
+    def _run(
+        self,
+        **kwargs: Any,
+    ) -> Any:
+        try:
+            results = self.client.search(
+                {
+                    "q": kwargs.get("search_query"),
+                    "location": kwargs.get("location"),
+                }
+            ).as_dict()
+
+            self._omit_fields(
+                results,
+                [
+                    r"search_metadata",
+                    r"search_parameters",
+                    r"serpapi_.+",
+                    r".+_token",
+                    r"displayed_link",
+                    r"pagination",
+                ],
+            )
+
+            return results
+        except HTTPError as e:
+            return f"An error occurred: {str(e)}. Some parameters may be invalid."