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

Update ai_service.py

Browse files
Files changed (1) hide show
  1. ai_service.py +4 -3
ai_service.py CHANGED
@@ -1,4 +1,4 @@
1
- # ai_service.py (Final version, removed api_name for PWS tool)
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 參數,讓 gradio_client 自動尋找預設的 API 端點。
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
+