bstraehle commited on
Commit
a9795d3
1 Parent(s): 4802652

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +2 -2
rag_langgraph.py CHANGED
@@ -2,7 +2,7 @@ from typing import Annotated, List, Tuple, Union
2
 
3
  from langchain_community.tools.tavily_search import TavilySearchResults
4
  from langchain_core.tools import tool
5
- from langchain_experimental.utilities import PythonREPL
6
 
7
  from langchain.agents import AgentExecutor, create_openai_tools_agent
8
  from langchain_core.messages import BaseMessage, HumanMessage
@@ -48,7 +48,7 @@ class AgentState(TypedDict):
48
 
49
  def run_multi_agent(prompt):
50
  tavily_tool = TavilySearchResults(max_results=5)
51
- repl = PythonREPL()
52
 
53
  members = ["Researcher", "Coder"]
54
  system_prompt = (
 
2
 
3
  from langchain_community.tools.tavily_search import TavilySearchResults
4
  from langchain_core.tools import tool
5
+ from langchain_experimental.tools import PythonREPLTool
6
 
7
  from langchain.agents import AgentExecutor, create_openai_tools_agent
8
  from langchain_core.messages import BaseMessage, HumanMessage
 
48
 
49
  def run_multi_agent(prompt):
50
  tavily_tool = TavilySearchResults(max_results=5)
51
+ python_repl_tool = PythonREPLTool()
52
 
53
  members = ["Researcher", "Coder"]
54
  system_prompt = (