Update app.py

app.py

@@ -40,22 +40,21 @@ 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 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). "
+        "If the query contains a city name (e.g., 'Birsk', 'Tønsberg') or is explicitly about time (e.g., 'What time is it in Moscow?'), "
+        "ALWAYS use the `get_current_time_in_timezone` tool with the appropriate timezone (e.g., 'Europe/Moscow' for Moscow, 'Asia/Yekaterinburg' for Birsk, 'Europe/Oslo' for Tønsberg). "
         "NEVER redefine tools—use only the provided ones. "
         "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()."
+        "Do NOT include tool calls (e.g., get_current_time_in_timezone()) inside final_answer(). "
+        "If the query is unclear or does not specify a city, return a clarification request 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 or mentions a city (e.g., 'Birsk'), use `get_current_time_in_timezone` with the correct timezone (e.g., 'Asia/Yekaterinburg' for Birsk). "
+            "Analyze the query: {{question}}. If it contains a city name (e.g., 'Birsk', 'Tønsberg') or is about time, use `get_current_time_in_timezone` with the correct timezone (e.g., 'Asia/Yekaterinburg' for Birsk, 'Europe/Oslo' for Tønsberg). "
             "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"
@@ -80,7 +79,7 @@ prompt_templates = {
     },
     "managed_agent": {
         "execute": (
-            "Execute the task: {{question}}. Use tools as specified in the plan. "
+            "Execute the task: {{question}}. If it contains a city name or is about time, use `get_current_time_in_timezone`. "
             "Write code in a code block:\n"
             "```py\n"
             "result = get_current_time_in_timezone('TIMEZONE')\n"
@@ -96,7 +95,7 @@ prompt_templates = {
         ),
         "task": (
             "Define the task based on the query: {{question}}. "
-            "Identify the required tools and steps. "
+            "If it contains a city name, use `get_current_time_in_timezone`. "
             "Return the task definition via final_answer() in a code block:\n"
             "```py\n"
             "final_answer('YOUR TASK DEFINITION HERE')\n"