Create google_search_agent.py

google_search_agent.py

@@ -0,0 +1,20 @@
+import pandas as pd
+from langchain_community.utilities import GoogleSerperAPIWrapper
+
+
+class GoogleSearchAgent:
+    def __init__(self) -> None:
+        self._api = GoogleSerperAPIWrapper()
+
+    def run(self, query: str, source: str | None = None, limit: int = 3) -> pd.DataFrame:
+        if source is not None:
+            query += f" site:{source}"
+
+        self._api.k = limit
+
+        results = self._api.results(query)["organic"]
+        articles = [{"title": article["title"], "url": article["link"], "date": article.get("date", "")} for article in results]
+
+        df = pd.DataFrame(articles)
+
+        return df