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

Update wellbe_agent.py

Browse files
Files changed (1) hide show
  1. wellbe_agent.py +2 -2
wellbe_agent.py CHANGED
@@ -14,7 +14,7 @@ import os
14
  import openai
15
  from agents import Agent, Runner
16
  from agents.extensions.models.litellm_model import LitellmModel
17
- from agents.mcp.server import MCPServerStdio, MCPServerStreamableHttp
18
 
19
  from config import PROMPT_TEMPLATE, MCP_CONFIGS
20
 
@@ -42,7 +42,7 @@ 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 = MCPServerStreamableHttp(
46
  params={
47
  "url": "https://agents-mcp-hackathon-whoop-mcp-server.hf.space/gradio_api/mcp/sse"
48
  },
 
14
  import openai
15
  from agents import Agent, Runner
16
  from agents.extensions.models.litellm_model import LitellmModel
17
+ from agents.mcp.server import MCPServerStdio, MCPServerStreamableHttp, MCPServerSse
18
 
19
  from config import PROMPT_TEMPLATE, MCP_CONFIGS
20
 
 
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 = MCPServerSse(
46
  params={
47
  "url": "https://agents-mcp-hackathon-whoop-mcp-server.hf.space/gradio_api/mcp/sse"
48
  },