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

Update ai_service.py

Browse files
Files changed (1) hide show
  1. ai_service.py +3 -3
ai_service.py CHANGED
@@ -1,4 +1,4 @@
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,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:
 
1
+ # ai_service.py (Final version, added None placeholder for Gradio button input)
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
+ # [核心修正] 傳入一個 None 作為伺服器端按鈕事件所需的佔位符輸入
50
+ result = client.predict(None, 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: