Update ai_service.py

ai_service.py

@@ -1,10 +1,11 @@
-# ai_service.py (Final Version with 'Part' import fix)
+# ai_service.py (Definitive fix for the ImportError)
 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
+
+# [修正] 移除 'Part' 的 import，因為它導致了錯誤
+# from google.generativeai.types import Part
 
 # 從設定檔匯入金鑰和 URL
 from config import GEMINI_API_KEY, MCP_SERVER_URL
@@ -109,9 +110,9 @@ def generate_ai_text(user_prompt: str) -> str:
         tool_result = tool_function(**dict(function_call.args))
         print("--- 將工具結果回傳給 Gemini ---")
         
-        # [修正] 將 genai.Part 改為 Part (因為我們已經從 types 模組匯入)
+        # [修正] 直接傳送包含 function_response 的字典，不再使用 Part 類別
         response = chat.send_message(
-            Part(function_response={"name": function_call.name, "response": {"result": tool_result}}),
+            {"function_response": {"name": function_call.name, "response": {"result": tool_result}}}
         )
         
         print("--- Gemini 根據工具結果生成最終回覆 ---")