Update ai_service.py

ai_service.py

@@ -1,4 +1,4 @@
-# ai_service.py (Final Version with System Prompt)
+# ai_service.py (Final Version with Improved Tool Definition)
 import json
 from datetime import datetime
 import google.generativeai as genai
@@ -48,27 +48,41 @@ def call_mcp_earthquake_search(
         print(f"呼叫 MCP 伺服器失敗: {e}")
         return f"工具執行失敗，錯誤訊息: {e}"
 
-# --- 3. 向 Gemini 定義工具 ---
+# --- 3. [修正] 向 Gemini 定義更清晰的工具 ---
 earthquake_search_tool_declaration = {
     "name": "call_earthquake_search_tool",
     "description": "根據指定的條件（時間、地點、規模等）從台灣中央氣象署的資料庫中搜尋地震事件。預設搜尋台灣周邊地區。",
     "parameters": {
-        "type": "OBJECT", "properties": {
-            "start_date": {"type": "STRING", "description": "搜尋的開始日期，格式為 'YYYY-MM-DD'。"},
-            "end_date": {"type": "STRING", "description": f"搜尋的結束日期，格式為 'YYYY-MM-DD'。預設為今天: {datetime.now().strftime('%Y-%m-%d')}。"},
-            "min_magnitude": {"type": "NUMBER", "description": "要搜尋的最小地震規模。預設為 4.5。"},
-            "max_magnitude": {"type": "NUMBER", "description": "要搜尋的最大地震規模。預設為 8.0。"},
-        }, "required": ["start_date", "end_date", "min_magnitude"]
+        "type": "OBJECT",
+        "properties": {
+            "start_date": {
+                "type": "STRING",
+                "description": "搜尋的開始日期 (格式 'YYYY-MM-DD')。模型應根據使用者問題推斷此日期，例如從『去年』或『2024年』推斷出 '2024-01-01'。"
+            },
+            "end_date": {
+                "type": "STRING",
+                "description": "搜尋的結束日期 (格式 'YYYY-MM-DD')。模型應根據使用者問題推斷此日期，例如從『昨天』或『2024年』推斷出 '2024-12-31'。"
+            },
+            "min_magnitude": {
+                "type": "NUMBER",
+                "description": "要搜尋的最小地震規模。如果使用者未指定，請使用預設值 4.5。"
+            },
+            "max_magnitude": {
+                "type": "NUMBER",
+                "description": "要搜尋的最大地震規模。預設為 8.0。"
+            },
+        },
+        # 將 min_magnitude 改為非必填，讓 AI 更靈活
+        "required": ["start_date", "end_date"]
     }
 }
 
 available_tools = {"call_earthquake_search_tool": call_mcp_earthquake_search}
 
-# --- 4. 建立 Gemini 模型 (Singleton 模式) ---
+# --- 4. 建立 Gemini 模型 ---
 model = None
 if GEMINI_API_KEY and "YOUR_GEMINI_API_KEY" not in GEMINI_API_KEY:
     try:
-        # [修正] 加入 system_instruction 賦予 AI 分析能力和指定語言
         system_instruction = (
             "You are a helpful AI assistant. You must answer in Traditional Chinese."
             "You have access to tools. When a tool returns data in JSON format, "