Update app.py

app.py

@@ -466,44 +466,57 @@ def get_time_tool2(query: str) -> datetime:
         tz_str = location_to_timezone(location)
         now = datetime.now(ZoneInfo(tz_str))
 
-        # Step 3: 用 FewShot + 格式限制 prompt 推出正確格式的時間
+        # Step 3: 動態 few-shot prompt（每次更新 based on now）
+        examples = [
+            ("five hours later", now + timedelta(hours=5)),
+            ("later", now + timedelta(hours=2)),
+            ("soon", now + timedelta(minutes=30)),
+            ("shortly", now + timedelta(minutes=15)),
+            ("after a while", now + timedelta(hours=1)),
+            ("tomorrow at 3pm", now.replace(hour=15, minute=0, second=0) + timedelta(days=1)),
+            ("the day after tomorrow at 10am", now.replace(hour=10, minute=0, second=0) + timedelta(days=2)),
+            ("last Monday 9am", (now - timedelta(days=(now.weekday() + 7))).replace(hour=9, minute=0, second=0)),
+            ("next Monday", (now + timedelta(days=(7 - now.weekday()))).replace(hour=12, minute=0, second=0)),
+            ("last Friday", (now - timedelta(days=(now.weekday() - 4 + 7) % 7)).replace(hour=12, minute=0, second=0)),
+            ("next Friday", (now + timedelta(days=(4 - now.weekday() + 7) % 7)).replace(hour=12, minute=0, second=0)),
+            ("in 10 hours", now + timedelta(hours=10)),
+            ("this weekend", (now + timedelta(days=(5 - now.weekday()) % 7)).replace(hour=10, minute=0, second=0)),
+            ("next weekend", (now + timedelta(days=((5 - now.weekday()) % 7) + 7)).replace(hour=10, minute=0, second=0)),
+            ("下週一下午三點", (now + timedelta(days=(7 - now.weekday() + 0) % 7)).replace(hour=15, minute=0, second=0)),
+            ("昨天下午五點", (now - timedelta(days=1)).replace(hour=17, minute=0, second=0)),
+            ("昨天早上八點", (now - timedelta(days=1)).replace(hour=8, minute=0, second=0)),
+            ("later this evening", now.replace(hour=20, minute=0, second=0)),
+            ("現在", now),
+            ("last month", (now - timedelta(days=30)).replace(hour=12, minute=0, second=0)),
+            ("early tomorrow morning", now.replace(hour=6, minute=0, second=0) + timedelta(days=1)),
+        ]
+        examples_str = "\n".join([f'User Query: "{q}" → {dt.strftime("%Y-%m-%d %H:%M:%S")}' for q, dt in examples])
+
+        # Step 4: 构建完整 prompt
         time_query_prompt = f"""
 You are a time interpreter. Based on the user's query, calculate the **exact target time** they are referring to.
 Today is: {now.strftime('%A')}, {now.strftime('%Y-%m-%d %H:%M:%S')} (local time)
 
-Please return the result in **exact format**: `YYYY-MM-DD HH:MM:SS` (24-hour clock, no timezone info). 
+Please return the result in this **exact format**: `YYYY-MM-DD HH:MM:SS` (24-hour clock, no timezone info).
 Only return the time string — no explanation, no extra words.
 
 ### Examples:
-
-User Query: "two hours later"
-→ 2025-04-02 17:30:00
-
-User Query: "tomorrow at 3pm"
-→ 2025-04-03 15:00:00
-
-User Query: "昨天下午五點"
-→ 2025-04-01 17:00:00
-
-User Query: "later this evening"
-→ 2025-04-02 20:00:00
-
-User Query: "現在"
-→ 2025-04-02 15:30:00
+{examples_str}
 
 ### Now process:
 User Query: "{query}"
 →
 """
+
         time_response = llm_gpt4.invoke(time_query_prompt)
         time_str = time_response.content.strip() if isinstance(time_response, AIMessage) else str(time_response).strip()
 
-        # Step 4: 嘗試解析時間
+        # Step 5: 嘗試解析時間
         try:
             target_time = datetime.strptime(time_str, "%Y-%m-%d %H:%M:%S")
             target_time = target_time.replace(tzinfo=ZoneInfo(tz_str))
             return target_time  # ✅ 回傳 datetime，不是 str
-        except Exception as e:
+        except Exception:
             return f"Failed to parse time string from LLM: '{time_str}'"
 
     except Exception as e: