From c5a415aade68df33104f44acb48a087834d584c8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=9A=D0=B8=D1=80=D0=B8=D0=BB=D0=BB=20=D0=A0=D0=BE=D0=BC?= =?UTF-8?q?=D0=B0=D0=BD=D0=BE=D0=B2?= Date: Thu, 4 Jun 2026 16:21:09 +0000 Subject: [PATCH] Update agent.py --- agent.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/agent.py b/agent.py index db5c40b..8e2c086 100644 --- a/agent.py +++ b/agent.py @@ -39,7 +39,7 @@ def create_agent() -> Any: """ try: from langgraph.graph import StateGraph - from langgraph.prebuilt import create_react_agent + from langgraph.prebuilt import create_react_agent as create_agent from langchain_ollama import ChatOllama from langchain_core.prompts import ChatPromptTemplate from langchain_core.output_parsers import StrOutputParser @@ -60,8 +60,8 @@ def create_agent() -> Any: ) parser = StrOutputParser() - # Build a simple react agent using LangGraph prebuilt - react_agent = create_react_agent( + # Build a simple agent using LangGraph prebuilt + react_agent = create_agent( llm=llm, tools=[web_search_tool, create_file_tool, export_files_tool], prompt=prompt, @@ -85,7 +85,7 @@ def create_agent_executor() -> Any: """ try: from langchain.agents import AgentExecutor - from langgraph.prebuilt import create_react_agent + from langgraph.prebuilt import create_react_agent as create_agent from langchain_ollama import ChatOllama from langchain_core.prompts import ChatPromptTemplate from langchain_core.output_parsers import StrOutputParser @@ -103,8 +103,8 @@ def create_agent_executor() -> Any: ) parser = StrOutputParser() - # Build the react agent - react_agent = create_react_agent( + # Build the agent + react_agent = create_agent( llm=llm, tools=[web_search_tool, create_file_tool, export_files_tool], prompt=prompt,