cwadayi commited on
Commit
c5df44d
·
verified ·
1 Parent(s): bf1c5ab

Update ai_service.py

Browse files
Files changed (1) hide show
  1. ai_service.py +3 -4
ai_service.py CHANGED
@@ -1,4 +1,4 @@
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,8 +46,8 @@ def call_mcp_pws_search() -> str:
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,4 +122,3 @@ def generate_ai_text(user_prompt: str) -> str:
122
  except Exception as e:
123
  print(f"與 Gemini AI 互動時發生錯誤: {e}")
124
  return f"🤖 AI 服務發生錯誤: {e}"
125
-
 
1
+ # ai_service.py (Final version with leading slash for api_name)
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}"