toolcount

app.py

@@ -111,6 +111,9 @@ def tool_node(state: AgentState) -> AgentState:
       - excel_path → parse_excel_tool
       - audio_path → audio_transcriber_tool
     """
+    tool_counter = state.get("tool_counter", 0)
+    if tool_counter > 5:
+        return {}
     if state.get("wiki_query"):
         return wikipedia_search_tool(state)
     if state.get("web_search_query"):
@@ -155,7 +158,7 @@ def inspect_node(state: AgentState) -> AgentState:
             question = msg.content
             break
     messages_for_llm.append(SystemMessage(content=f"USER_QUESTION: {question}"))
-
+    
     # 2) Add any tool results
     if sr := state.get("web_search_result"):
         messages_for_llm.append(SystemMessage(content=f"WEB_SEARCH_RESULT: {sr}"))
@@ -304,7 +307,7 @@ def respond_to_input(user_input: str, task_id) -> str:
     )
     human_msg = HumanMessage(content=user_input)
 
-    initial_state: AgentState = {"messages": [system_msg, human_msg], "task_id": task_id}
+    initial_state: AgentState = {"messages": [system_msg, human_msg], "task_id": task_id, "tool_counter": 0} 
     final_state = compiled_graph.invoke(initial_state)
     return final_state.get("final_answer", "Error: No final answer generated.")
 

state.py

@@ -18,4 +18,5 @@ class AgentState(TypedDict, total=False):
     audio_transcript: str
     wiki_query: str
     wiki_result: str
-    task_id: str
+    task_id: str
+    tool_counter: int