diff --git a/contributing/samples/mcp_sse_agent/agent.py b/contributing/samples/mcp_sse_agent/agent.py index 888a88b2..5423bfc6 100755 --- a/contributing/samples/mcp_sse_agent/agent.py +++ b/contributing/samples/mcp_sse_agent/agent.py @@ -16,8 +16,8 @@ import os from google.adk.agents.llm_agent import LlmAgent +from google.adk.tools.mcp_tool.mcp_session_manager import SseConnectionParams from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset -from google.adk.tools.mcp_tool.mcp_toolset import SseServerParams _allowed_path = os.path.dirname(os.path.abspath(__file__)) @@ -31,7 +31,7 @@ Allowed directory: {_allowed_path} """, tools=[ MCPToolset( - connection_params=SseServerParams( + connection_params=SseConnectionParams( url='http://localhost:3000/sse', headers={'Accept': 'text/event-stream'}, ),