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,