Add google chat

requirements.txt

@@ -14,4 +14,5 @@ google-search-results
 langchain-huggingface
 langchain-community
 supabase
-sentence-transformers
+sentence-transformers
+langchain-google-genai

tools.py

@@ -4,6 +4,7 @@ from langchain_core.messages import SystemMessage, HumanMessage, AnyMessage
 from langgraph.graph.message import add_messages
 from langchain_community.vectorstores import SupabaseVectorStore
 from supabase.client import  create_client
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_huggingface import HuggingFaceEmbeddings, ChatHuggingFace, HuggingFaceEndpoint
 from serpapi import GoogleSearch
 from dotenv import load_dotenv
@@ -34,7 +35,7 @@ def web_search(query: str) -> str:
     params = {
         "engine": "google",
         "q": query,
-        "api_key": os.getenv("SERPAPI_API_KEY"),
+        "api_key": os.getenv("SERPAPI_KEY"),
         "num": 5
     }
     search = GoogleSearch(params)
@@ -46,11 +47,14 @@ def web_search(query: str) -> str:
     )
     return context if context else "No results found."
 
-llm = ChatHuggingFace(llm = HuggingFaceEndpoint(
-                repo_id = "meta-llama/Llama-2-7b-chat-hf",
-                temperature=0,
-                huggingfacehub_api_token=os.environ.get("HUGGING_FACE_API_KEY")))
+# llm = ChatHuggingFace(llm = HuggingFaceEndpoint(
+#                 repo_id = "meta-llama/Llama-2-7b-chat-hf",
+#                 temperature=0,
+#                 huggingfacehub_api_token=os.environ.get("HUGGING_FACE_API_KEY")))
+
+
 tools = [add, subtract, divide, web_search]
+llm =ChatGoogleGenerativeAI(model = "gemini-2.0-flash")
 llm_with_tools = llm.bind_tools(tools)
 
 def retriever(state: AgentState) -> Dict: