Update agent.py

agent.py

@@ -25,7 +25,7 @@ from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.document_loaders import ArxivLoader
 #from langchain_community.vectorstores import SupabaseVectorStore
-#from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 #from langchain.tools.retriever import create_retriever_tool
 #from supabase.client import Client, create_client
 #from code_interpreter import CodeInterpreter
@@ -703,7 +703,10 @@ def build_graph(provider: str = "huggingface"):
             ),
             #verbose=True,
         )
-        
+
+    elif provider == "google":
+        # Google Gemini
+        llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     else:
         raise ValueError("Invalid provider. Choose 'huggingface'.")