Update agent.py

2026-06-04 16:21:09 +00:00
parent 06a6de1edc
commit c5a415aade
1 changed files with 6 additions and 6 deletions
@@ -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,