Update ai_service.py

ai_service.py

@@ -1,4 +1,4 @@
-# ai_service.py (Gemini 最終版)
+# ai_service.py (Final Version with System Prompt)
 import json
 from datetime import datetime
 import google.generativeai as genai
@@ -68,39 +68,41 @@ available_tools = {"call_earthquake_search_tool": call_mcp_earthquake_search}
 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, "
+            "you must analyze the JSON data to fully answer the user's question. "
+            "For example, if the user asks for the largest earthquake, use the search tool for the relevant date range "
+            "and then find the entry with the highest magnitude from the JSON results before answering."
+        )
         model = genai.GenerativeModel(
             model_name="gemini-1.5-flash",
-            tools=[earthquake_search_tool_declaration]
+            tools=[earthquake_search_tool_declaration],
+            system_instruction=system_instruction
         )
     except Exception as e:
         print(f"建立 Gemini 模型失敗: {e}")
 
 # --- 5. 主要的 AI 文字生成函式 ---
 def generate_ai_text(user_prompt: str) -> str:
-    """使用 Gemini 模型生成回應，並在需要時觸發工具呼叫。"""
     if not model:
         return "🤖 AI (Gemini) 服務尚未設定 API 金鑰，或金鑰無效。"
     try:
         print(f"--- 開始 Gemini 對話，使用者輸入: '{user_prompt}' ---")
         chat = model.start_chat()
         response = chat.send_message(user_prompt)
-
-        # 檢查模型是否要求呼叫工具
         try:
             function_call = response.candidates[0].content.parts[0].function_call
         except (IndexError, AttributeError):
             function_call = None
-
         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 ---")
         response = chat.send_message(
@@ -108,7 +110,6 @@ def generate_ai_text(user_prompt: str) -> str:
         )
         print("--- Gemini 根據工具結果生成最終回覆 ---")
         return response.text
-
     except Exception as e:
         print(f"與 Gemini AI 互動時發生錯誤: {e}")
         return f"🤖 AI 服務發生錯誤: {e}"