natasha1704 commited on
Commit
9889694
·
verified ·
1 Parent(s): a61bafb

Update wellbe_agent.py

Browse files
Files changed (1) hide show
  1. wellbe_agent.py +9 -4
wellbe_agent.py CHANGED
@@ -42,7 +42,12 @@ def initialize_mcp_servers(whoop_email: str, whoop_password: str) -> None:
42
  params=cfg["healthcare-mcp-public"], name="Healthcare MCP Server"
43
  )
44
  # _whoop_server = MCPServerStdio(params=cfg["whoop"], name="Whoop MCP Server")
45
- _whoop_server = "Whoop Tool Not available now"
 
 
 
 
 
46
 
47
 
48
  def select_model() -> Union[str, LitellmModel]:
@@ -60,14 +65,14 @@ async def answer_question(
60
  if _healthcare_server is None:
61
  return "**Error:** MCP servers not initialized."
62
 
63
- async with _healthcare_server as hserver: # , _whoop_server as wserver:
64
  agent = Agent(
65
  name="WellBe+ Assistant",
66
  instructions=PROMPT_TEMPLATE,
67
  model=select_model(),
68
  mcp_servers=[
69
- hserver
70
- # , wserver
71
  ],
72
  )
73
  result = await Runner.run(agent, question)
 
42
  params=cfg["healthcare-mcp-public"], name="Healthcare MCP Server"
43
  )
44
  # _whoop_server = MCPServerStdio(params=cfg["whoop"], name="Whoop MCP Server")
45
+ _whoop_server = MCPServerStreamableHttp(
46
+ params={
47
+ "url": "https://agents-mcp-hackathon-whoop-mcp-server.hf.space/gradio_api/mcp/sse"
48
+ },
49
+ name="Whoop MCP Server"
50
+ )
51
 
52
 
53
  def select_model() -> Union[str, LitellmModel]:
 
65
  if _healthcare_server is None:
66
  return "**Error:** MCP servers not initialized."
67
 
68
+ async with _healthcare_server as hserver, _whoop_server as wserver:
69
  agent = Agent(
70
  name="WellBe+ Assistant",
71
  instructions=PROMPT_TEMPLATE,
72
  model=select_model(),
73
  mcp_servers=[
74
+ hserver,
75
+ wserver
76
  ],
77
  )
78
  result = await Runner.run(agent, question)