Update ai_service.py

ai_service.py

@@ -1,4 +1,4 @@
-# ai_service.py (Implements API Key rotation)
+# ai_service.py (Improved tool description for date inference)
 import json
 import os
 import random
@@ -7,23 +7,17 @@ import google.generativeai as genai
 from gradio_client import Client
 
 # 從設定檔匯入金鑰和 URL
-from config import MCP_SERVER_URL #不再直接從config匯入GEMINI_API_KEY
+from config import MCP_SERVER_URL
 
 # --- 1. 設定 Gemini API 金鑰 (金鑰輪替邏輯) ---
-
-# [核心修改] 讀取並輪流使用兩組 API 金鑰
-# 從環境變數讀取您設定的兩組金鑰
 key1 = os.getenv("GEMINI_API_KEY")
 key2 = os.getenv("GEMINI_API_KEY2")
-
-# 建立一個包含所有有效金鑰的列表
 valid_keys = []
 if key1 and "YOUR_GEMINI_API_KEY" not in key1:
     valid_keys.append(key1)
 if key2 and "YOUR_GEMINI_API_KEY" not in key2:
     valid_keys.append(key2)
 
-# 如果列表中有可用的金鑰，就隨機選取一個來設定
 selected_key = None
 if valid_keys:
     selected_key = random.choice(valid_keys)
@@ -73,18 +67,35 @@ def call_mcp_pws_search() -> str:
 
 # --- 3. 向 Gemini 定義工具 (Tool Declarations) ---
 
+# [*** 核心修正 ***]
+# 強化參數描述，引導 AI 自動從對話中推斷日期
 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": "搜尋的結束日期 (格式 'YYYY-MM-DD')。"},
-        }, "required": ["start_date", "end_date"]
+    "name": "call_earthquake_search_tool",
+    "description": "從台灣中央氣象署的資料庫中搜尋地震事件。",
+    "parameters": {
+        "type": "OBJECT",
+        "properties": {
+            "start_date": {
+                "type": "STRING",
+                "description": "搜尋的開始日期 (格式 'YYYY-MM-DD')。模型應根據使用者問題中的相對時間（例如：昨天、上個月、去年）或絕對時間（例如：2024年）來主動推斷此日期。"
+            },
+            "end_date": {
+                "type": "STRING",
+                "description": "搜尋的結束日期 (格式 'YYYY-MM-DD')。模型應根據使用者問題中的相對時間或絕對時間來主動推斷此日期。"
+            },
+        },
+        "required": ["start_date", "end_date"]
     }
 }
 
+
 pws_search_tool_declaration = {
-    "name": "call_mcp_pws_search", "description": "查詢最新的 PWS (Public Weather Service) 公共天氣服務發布情形。",
-    "parameters": {"type": "OBJECT", "properties": {}}
+    "name": "call_mcp_pws_search",
+    "description": "查詢最新的 PWS (Public Weather Service) 公共天氣服務發布情形。",
+    "parameters": {
+        "type": "OBJECT",
+        "properties": {}
+    }
 }
 
 available_tools = {
@@ -94,7 +105,6 @@ available_tools = {
 
 # --- 4. 建立 Gemini 模型 ---
 model = None
-# [核心修改] 只有在成功選取到key之後才建立模型
 if selected_key:
     try:
         system_instruction = (