Spaces:
Sleeping
Sleeping
Daniel Amendoeira
commited on
Update agent.py
Browse files
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":
|
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:
|