Daniel Amendoeira commited on
Commit
245478b
·
verified ·
1 Parent(s): 45bfda2

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -104,7 +104,7 @@ class LangGraphAgent:
104
  def __call__(self, question: str) -> str:
105
  input_state = {"messages": [HumanMessage(content=question)]} # prepare the initial user message
106
  print(f"Running LangGraphAgent with input: {question[:150]}...")
107
- result = gaia_agent.invoke(input_state, {"recursion_limit": 100}) # prevents infinite looping when the LLM keeps calling tools over and over
108
  final_response = result["messages"][-1].content
109
 
110
  try:
 
104
  def __call__(self, question: str) -> str:
105
  input_state = {"messages": [HumanMessage(content=question)]} # prepare the initial user message
106
  print(f"Running LangGraphAgent with input: {question[:150]}...")
107
+ result = gaia_agent.invoke(input_state, {"recursion_limit": 30}) # prevents infinite looping when the LLM keeps calling tools over and over
108
  final_response = result["messages"][-1].content
109
 
110
  try: