Update agent.py

agent.py

@@ -2,6 +2,7 @@
 import os
 from supabase import create_client
 from sentence_transformers import SentenceTransformers
+from serpapi import GoogleSearch
 from langgraph import Graph, LLM, tool #or other graph library
 
 # ----Supabase setup----
@@ -9,6 +10,7 @@ SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_SERVICE_KEY")
 EMBED_MODEL_ID = os.getenv("HF_EMBEDDING_MODEL")
 
+
 sb_client = create_client(SUPABASE_URL, SUPABASE_KEY)
 embedder = SentenceTransformers(EMBED_MODEL_ID)
 
@@ -46,6 +48,22 @@ def retrieve_docs(query: str, k: int = 3) -> str:
     docs = [row["content"] for row in rows]
     return "\n\n---\n\n".join(docs)
     
+SERPAPI_KEY = os.getenv("SERPAPY_KEY")
+# ---- web_search tool
+@tool
+def web_search(query: str, num_results: int = 5) -> str:
+    """ Return top-5 snippets from Google search via SerpAPI."""
+    params = {
+        "engine": "google",
+        "q": query,
+        "num": num_results,
+        "api_key": SERPAPI_KEY,
+    }
+    search = GoogleSearch(params)
+    results = search.get_dict().get("organic_results", [])
+    snippets = [r.get("snippet","")for r in results]
+    return "\n".join(f"- {s}" for s in snippets)
+    
     
 # 2) Build your graph
 def build_graph(provider: str = "huggingface") -> Graph: