Update agent.py

agent.py

@@ -12,6 +12,7 @@ from langchain_community.vectorstores import SupabaseVectorStore
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_community.retrievers import WikipediaRetriever
 from langchain.tools.retriever import create_retriever_tool
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_community.llms import YandexGPT
 from langchain_core.tools import tool
 from supabase.client import Client, create_client
@@ -102,13 +103,15 @@ def build_graph():
     #    ),
     #)
 
-    llm = YandexGPT(
-        api_key=os.environ["YANDEX_API_KEY"],
-        folder_id=os.environ["YANDEX_FOLDER_ID"],
-        model_uri=os.environ["YANDEX_MODEL_URI"],
-    )
+    #llm = YandexGPT(
+    #    api_key=os.environ["YANDEX_API_KEY"],
+    #    folder_id=os.environ["YANDEX_FOLDER_ID"],
+    #    model_uri=os.environ["YANDEX_MODEL_URI"],
+    #)
+
+    llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     
-    llm_with_tools = llm#.bind_tools(tools)
+    llm_with_tools = llm.bind_tools(tools)
 
     def assistant(state: MessagesState):
         """Assistant node"""