Update ai_service.py

ai_service.py

@@ -1,8 +1,10 @@
-# ai_service.py (Final Version with Improved Tool Definition)
+# ai_service.py (Final Version with 'Part' import fix)
 import json
 from datetime import datetime
 import google.generativeai as genai
 from gradio_client import Client
+# [修正] 從 google.generativeai.types 匯入 Part 類別
+from google.generativeai.types import Part
 
 # 從設定檔匯入金鑰和 URL
 from config import GEMINI_API_KEY, MCP_SERVER_URL
@@ -48,32 +50,17 @@ 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')。模型應根據使用者問題推斷此日期，例如從『去年』或『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"]
+        "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。"},
+        }, "required": ["start_date", "end_date"]
     }
 }
 
@@ -113,17 +100,23 @@ def generate_ai_text(user_prompt: str) -> str:
         if not function_call:
             print("--- Gemini 直接回覆文字 ---")
             return response.text
+        
         print(f"--- Gemini 要求呼叫工具: {function_call.name} ---")
         tool_function = available_tools.get(function_call.name)
         if not tool_function:
             return f"錯誤：模型嘗試呼叫一個不存在的工具 '{function_call.name}'。"
+        
         tool_result = tool_function(**dict(function_call.args))
         print("--- 將工具結果回傳給 Gemini ---")
+        
+        # [修正] 將 genai.Part 改為 Part (因為我們已經從 types 模組匯入)
         response = chat.send_message(
-            genai.Part(function_response={"name": function_call.name, "response": {"result": tool_result}}),
+            Part(function_response={"name": function_call.name, "response": {"result": tool_result}}),
         )
+        
         print("--- Gemini 根據工具結果生成最終回覆 ---")
         return response.text
     except Exception as e:
         print(f"與 Gemini AI 互動時發生錯誤: {e}")
         return f"🤖 AI 服務發生錯誤: {e}"
+