Update app.py

app.py

@@ -39,59 +39,87 @@ except Exception as e:
 prompt_templates = {
     "system_prompt": (
         "You are a helpful assistant that responds in the same language as the user's query. "
-        "Detect the language of the query and respond accordingly, ensuring clarity and accuracy. "
-        "For time-related questions (e.g., 'What time is it in [place]?'), use the `get_current_time_in_timezone` tool "
+        "Detect the language of the query and respond with the result only, without extra explanations. "
+        "For time-related questions (e.g., 'What time is it in [place]?') or city names (e.g., 'Birsk'), use the `get_current_time_in_timezone` tool "
         "with the appropriate timezone (e.g., 'Europe/Moscow' for Moscow, 'Asia/Yekaterinburg' for Birsk). "
         "NEVER redefine tools—use only the provided ones. "
-        "Always return the response via `final_answer()`, not through print or variables. "
-        "If the query is unclear, ask for clarification in the same language as the query."
+        "Always return the response via `final_answer()` in a code block like this:\n"
+        "```py\n"
+        "final_answer('YOUR ANSWER HERE')\n"
+        "```\n"
+        "Do NOT include tool calls (e.g., get_current_time_in_timezone()) inside final_answer(). "
+        "Execute the tool first, then pass the result as a string to final_answer(). "
+        "If the query is unclear, ask for clarification in the same language via final_answer()."
     ),
     "default": "Response: {{question}}",
     "planning": {
         "initial_plan": (
             "Analyze the query: {{question}}. Determine if a tool is needed. "
-            "If the query is about time, use `get_current_time_in_timezone` with the correct timezone (e.g., 'Asia/Yekaterinburg' for Birsk). "
-            "Describe the steps and write code. Always return the result via `final_answer()`."
+            "If the query is about time or mentions a city (e.g., 'Birsk'), use `get_current_time_in_timezone` with the correct timezone (e.g., 'Asia/Yekaterinburg' for Birsk). "
+            "Write code to execute the tool and return the result via final_answer() in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR ANSWER HERE')\n"
+            "```"
         ),
         "update_plan_pre_messages": (
             "Review the query: {{question}} and previous steps. "
-            "If the plan needs adjustment (e.g., wrong timezone or tool), update it. "
-            "Write corrected code and return the result via `final_answer()`."
+            "If the plan needs adjustment (e.g., wrong timezone), update it. "
+            "Write corrected code in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR ANSWER HERE')\n"
+            "```"
         ),
         "update_plan_post_messages": (
             "Review the query: {{question}} and execution results. "
             "If the result is incorrect or incomplete, adjust the plan and code. "
-            "Return the final result via `final_answer()`."
+            "Return the result via final_answer() in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR ANSWER HERE')\n"
+            "```"
         )
     },
     "managed_agent": {
         "execute": (
             "Execute the task: {{question}}. Use tools as specified in the plan. "
-            "Ensure the code is correct and uses only provided tools. "
-            "Return the result via `final_answer()`."
+            "Write code in a code block:\n"
+            "```py\n"
+            "result = get_current_time_in_timezone('TIMEZONE')\n"
+            "final_answer(result)\n"
+            "```"
         ),
         "report": (
             "Summarize the execution of the task: {{question}}. "
-            "Report the result or any errors encountered. "
-            "Return the summary via `final_answer()`."
+            "Return the result via final_answer() in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR ANSWER HERE')\n"
+            "```"
         ),
         "task": (
             "Define the task based on the query: {{question}}. "
-            "Identify the required tools and steps to complete it. "
-            "Return the task definition via `final_answer()`."
+            "Identify the required tools and steps. "
+            "Return the task definition via final_answer() in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR TASK DEFINITION HERE')\n"
+            "```"
         )
     },
     "final_answer": {
         "pre_messages": (
             "Prepare the final response for the query: {{question}}. "
             "Ensure the answer is clear and in the same language as the query. "
-            "Format the response using `final_answer()`."
+            "Return the response via final_answer() in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR ANSWER HERE')\n"
+            "```"
         ),
         "template": "Final response: {{answer}}",
         "post_messages": (
             "Review the formatted response for the query: {{question}}. "
             "Ensure the response is accurate and complete. "
-            "Return the final response via `final_answer()`."
+            "Return the final response via final_answer() in a code block:\n"
+            "```py\n"
+            "final_answer('YOUR ANSWER HERE')\n"
+            "```"
         )
     }
 }