Update app.py

app.py

@@ -201,12 +201,12 @@ def inspect_node(state: AgentState) -> AgentState:
 
         # Now ask for JSON ONLY (no reasoning, no extra text)
         final_prompt = (
-            "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
+            "Finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
             "Using only the information above—including the USER_QUESTION, "
             "any TOOL_RESULT, and the INTERIM_ANSWER—produce a concise final answer. "
             "Return exactly one JSON object and nothing else, in this format:\n\n"
             "{\"final_answer\":\"<your final answer>\"}\n"
-            "Do not include any other words or punctuation outside that JSON."
+            "Do not include any other words or punctuation outside that JSON. if its numbers, dont show the units"
         )
         messages_for_llm.append(SystemMessage(content=final_prompt))
 
@@ -368,6 +368,7 @@ def respond_to_input(user_input: str, task_id) -> str:
     system_msg = SystemMessage(
         content=(
             "You are an agent orchestrator. Decide whether to use a tool or answer directly.\n"
+            "Try not to use tools so many times. If you think you can answer the question without using a tool, do it Please.\n"
             "Tools available:\n"
             "  • Wikipedia: set {\"wiki_query\":\"<search terms>\"}\n"
             "  • OCR: set {\"ocr_path\":\"<image path or task_id>\"}\n"