New_Final_Assignment

Sleeping

naman1102 commited on Jun 3

Commit

729b871

1 Parent(s): ad94802

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ def plan_node(state: AgentState) -> AgentState:
     try:
         parsed = json.loads(llm_out)
         if isinstance(parsed, dict):
-            partial: AgentState = {"messages": new_msgs}
             allowed_keys = {
                 "final_answer",
                 "wiki_query",

     try:
         parsed = json.loads(llm_out)
         if isinstance(parsed, dict):
+            partial: AgentState = {"messages": new_msgs, "tool_counter": state.get("tool_counter", 0)}
             allowed_keys = {
                 "final_answer",
                 "wiki_query",