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_community.llms import YandexGPT
 from langchain_core.tools import tool
 from supabase.client import Client, create_client
 
@@ -95,11 +96,22 @@ tools = [
 ]
 
 def build_graph():
-    llm = ChatHuggingFace(
-        llm=HuggingFaceEndpoint(
-            repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
-        ),
+    #llm = ChatHuggingFace(
+    #    llm=HuggingFaceEndpoint(
+    #        repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
+    #    ),
+    #)
+
+    YANDEX_API_KEY: Final[str] = 
+YANDEX_FOLDER_ID: Final[str] =  
+    YANDEX_GEOCODER_API_KEY: Final[str] = 
+
+    llm = YandexGPT(
+        api_key=os.environ["YANDEX_API_KEY"],
+        folder_id=os.environ["YANDEX_FOLDER_ID"],
+        model_uri=os.environ["YANDEX_MODEL_URI"],
     )
+    
     llm_with_tools = llm.bind_tools(tools)
 
     def assistant(state: MessagesState):