Spaces:
Sleeping
Sleeping
Update ai_service.py
Browse files- ai_service.py +4 -3
ai_service.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
# ai_service.py (Final version
|
| 2 |
import json
|
| 3 |
from datetime import datetime
|
| 4 |
import google.generativeai as genai
|
|
@@ -46,8 +46,8 @@ def call_mcp_pws_search() -> str:
|
|
| 46 |
print(f"--- 正在呼叫遠端 PWS MCP 伺服器 (由 Gemini 觸發) ---")
|
| 47 |
client = Client("cwadayi/MCP-pws")
|
| 48 |
|
| 49 |
-
# [核心修正]
|
| 50 |
-
result = client.predict()
|
| 51 |
|
| 52 |
return result[0] if isinstance(result, tuple) and len(result) > 0 else str(result)
|
| 53 |
except Exception as e:
|
|
@@ -122,3 +122,4 @@ def generate_ai_text(user_prompt: str) -> str:
|
|
| 122 |
except Exception as e:
|
| 123 |
print(f"與 Gemini AI 互動時發生錯誤: {e}")
|
| 124 |
return f"🤖 AI 服務發生錯誤: {e}"
|
|
|
|
|
|
| 1 |
+
# ai_service.py (Final version with correct api_name from user's reference)
|
| 2 |
import json
|
| 3 |
from datetime import datetime
|
| 4 |
import google.generativeai as genai
|
|
|
|
| 46 |
print(f"--- 正在呼叫遠端 PWS MCP 伺服器 (由 Gemini 觸發) ---")
|
| 47 |
client = Client("cwadayi/MCP-pws")
|
| 48 |
|
| 49 |
+
# [核心修正] 根據您提供的參考資料,使用正確的 api_name
|
| 50 |
+
result = client.predict(api_name="get_disaster_warnings")
|
| 51 |
|
| 52 |
return result[0] if isinstance(result, tuple) and len(result) > 0 else str(result)
|
| 53 |
except Exception as e:
|
|
|
|
| 122 |
except Exception as e:
|
| 123 |
print(f"與 Gemini AI 互動時發生錯誤: {e}")
|
| 124 |
return f"🤖 AI 服務發生錯誤: {e}"
|
| 125 |
+
|