bstraehle commited on
Commit
2dc6790
·
verified ·
1 Parent(s): 26e5bf9

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +2 -2
rag_langgraph.py CHANGED
@@ -48,7 +48,7 @@ class AgentState(TypedDict):
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 = (
@@ -133,7 +133,7 @@ def run_multi_agent(prompt):
133
  for s in graph.stream(
134
  {
135
  "messages": [
136
- HumanMessage(content=str(prompt))
137
  ]
138
  }
139
  ):
 
48
 
49
  def run_multi_agent(prompt):
50
  tavily_tool = TavilySearchResults(max_results=5)
51
+ python_repl_tool = PythonREPLTool(return_direct=true)
52
 
53
  members = ["Researcher", "Coder"]
54
  system_prompt = (
 
133
  for s in graph.stream(
134
  {
135
  "messages": [
136
+ HumanMessage(content="Code hello world and print it to the terminal")
137
  ]
138
  }
139
  ):