Update app.py

app.py

@@ -743,19 +743,14 @@ def should_continue(state: AgentState):
     
     # 3. If last message is AIMessage with tool calls
     if isinstance(last_message, AIMessage) and last_message.tool_calls:
-        # Check each tool call explicitly
-        for tc in last_message.tool_calls:
-            tool_name = tc.get("name", "")
-            print(f"🔧 Tool call detected: '{tool_name}'")
-            
-            # ONLY end if it's explicitly final_answer_tool
-            if tool_name == "final_answer_tool":
-                print(f"✅ final_answer_tool confirmed → ending")
-                return END
+        # Only check the FIRST tool call, not all of them
+        first_tool = last_message.tool_calls[0]
+        tool_name = first_tool.get("name", "")
         
-        # Not final answer, route to tools
-        print(f"🔄 Routing to tools node")
-        return "tools"
+        if tool_name == "final_answer_tool":
+            return END
+        else:
+            return "tools"
     
     # 4. If AIMessage but no tool calls (reasoning text)
     if isinstance(last_message, AIMessage) and not last_message.tool_calls:
@@ -898,18 +893,24 @@ REMEMBER: One tool per turn. No reasoning without tools. Exact answer format.
 
         print("Initializing Groq LLM...")
         try:
+        #. Initialize the LLM ()
+        self.llm_with_tools = HuggingFaceEndpoint(
+            repo_id="HuggingFaceH4/zephyr-7b-beta", # Changed model
+            huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+            max_new_tokens=2048, # Increased token limit for potentially longer reasoning/tool use
+            temperature=0.01, # Keep temperature low for factual tasks     
             # Use tool_choice="any" to FORCE tool usage
-            self.llm_with_tools = ChatGroq(
-                temperature=0,
-                groq_api_key=GROQ_API_KEY,
-                model_name="llama-3.3-70b-versatile",
-                max_tokens=4096,
-                timeout=60
-            ).bind_tools(self.tools, tool_choice="any")  # FORCE tool calls
+           # self.llm_with_tools = ChatGroq(
+            #    temperature=0,
+             #   groq_api_key=GROQ_API_KEY,
+              #  model_name="llama-3.3-70b-versatile",
+               # max_tokens=4096,
+                #timeout=60
+            ).bind_tools(self.tools, tool_choice="auto")  # FORCE tool calls
             print("✅ LLM initialized with FORCED tool usage.")
 
         except Exception as e:
-            print(f"❌ Error initializing Groq: {e}")
+            print(f"❌ Error initializing HuggingFace: {e}")
             raise
         
         # Agent Node with AGGRESSIVE tool forcing