Update app.py

app.py

@@ -240,22 +240,34 @@ defined_tools = [
 class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
-# --- Custom Conditional Edge ---
 def should_continue(state: AgentState):
-    """Custom logic to decide whether to continue or end."""
+    """
+    Custom logic to decide whether to continue or end.
+    Now allows for a "reasoning loop".
+    """
     last_message = state['messages'][-1]
+
     if isinstance(last_message, AIMessage):
         if last_message.tool_calls:
+            # Check for the final answer tool
             if last_message.tool_calls[0].get("name") == "final_answer_tool":
                 print("--- Condition: Saw final_answer_tool, ending graph. ---")
                 return END
             else:
+                # Any other tool call goes to the tools node
                 print("--- Condition: Saw other tools, calling tools node. ---")
                 return "tools"
-    # This path should ideally not be taken if the prompt is followed
-    print("--- Condition: No tool call detected, ending graph (forcing agent to use final_answer_tool). ---")
-    return END
+    
+    # --- THIS IS THE NEW LOGIC ---
+    # If the last message is from the AI and has NO tool calls (i.e., it's a "thought"),
+    # loop back to the agent node to let it "think" again.
+    print("--- Condition: No tool call detected. Looping back to agent. ---")
+    return "agent"
+
+    # The old "END" path is removed. The only way to END
+    # is to explicitly call final_answer_tool.
 
+    
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -286,6 +298,8 @@ Your primary goal is to provide **only the concise, factual, and direct answer**
 * **THE ONLY WAY** to provide a final answer is by calling the `final_answer_tool`.
 * **DO NOT** include conversational filler (e.g., "The answer is...").
 * **DO NOT** explain your reasoning unless it's inside a `code_interpreter` comment.
+* **DO NOT** mix plain text and tool-call JSON in the same response.
+* **DO NOT** use XML formats like `<function=...>` or `<code_interpreter>`. **THIS WILL FAIL.**
 
 You have access to the following tools:
 {tool_descriptions}
@@ -302,30 +316,24 @@ You have access to the following tools:
 * Do not add any text before or after the JSON block.
 
 **REASONING PROCESS & STOPPING CONDITION:**
-1.  **Analyze:** Read the user's question carefully. Identify the goal.
-2.  **Plan:** Break down the problem into a sequence of logical steps. 
-    * **For logic puzzles (like riddles):** You *must* use the `code_interpreter`. Your *first* step is to write out your step-by-step logical reasoning as Python comments *before* writing any solving code.
-3.  **Execute:** Choose the *one* best tool for the *next* step in your plan.
-4.  **Call Tool:** Call the chosen tool using the exact JSON format. Do not add any other text.
-5.  **Analyze Output:** Receive the tool's output (either a success message or an error).
-6.  **DECISION POINT:**
-    a. **Answer Found?** If the tool output directly contains the final answer (or makes it trivial to calculate), your *very next* action MUST be to call `final_answer_tool` with that answer.
-    b. **Error Occurred?** If the tool returned an error message:
-        * **STOP.** Do not try the *exact same tool call* again.
-        * **Rethink:** Analyze the error message. What went wrong?
-        * **Correct:** Try a *different* tool, *different* arguments, or a *different approach* entirely.
-        * If you are completely stuck after an error, call `final_answer_tool` with a concise error message (e.g., "Error: Could not find user").
-    c. **More Steps Needed?** If the answer is not yet found and there was no error, go back to step 2 (Plan) to determine the next logical step.
+1.  **Analyze:** Read the question. Break it down into logical steps.
+2.  **DECIDE:** Do you have enough information to call a tool, or do you need to write down your plan first?
+3.  **ACT (Two Options):**
+    a. **Write Plan (Chain of Thought):** If you are not ready to call a tool, or if the problem is a complex logic puzzle, respond with your step-by-step reasoning plan as **plain text**. This allows you to "think" and add your plan to memory before your next step.
+    b. **Call Tool:** If you are ready, call **one** tool using the JSON format.
+4.  **Analyze Output:** After a tool is called, you will receive its output.
+5.  **GOTO 1:** Repeat the process. Analyze the new information and decide your next step (think, or call another tool).
+6.  **STOPPING:** The *only* way to provide the final answer is by calling `final_answer_tool`.
 7.  **FINAL OUTPUT:** The graph will stop *only* when you call `final_answer_tool`. Do not provide the answer in any other way."""
         # =============================================================
 
         print("Initializing Groq LLM Endpoint...")
         try:
-        chat_llm = ChatGroq(
-                    temperature=0.01,
-                    groq_api_key=GROQ_API_KEY,
-                    model_name="qwen/qwen3-32b"  # <-- Your change is here
-                )
+            chat_llm = ChatGroq(
+                temperature=0.01,
+                groq_api_key=GROQ_API_KEY,
+                model_name="qwen/qwen3-32b"  # <-- Your change is here
+            )
             print("✅ Groq LLM Endpoint initialized with llama-3.1-8b-instant.")
         except Exception as e: print(f"Error initializing Groq: {e}"); raise