From 7f8dc8927aaa401fe7079e430c941597c14237a3 Mon Sep 17 00:00:00 2001 From: "Xiang (Sean) Zhou" Date: Wed, 18 Jun 2025 18:31:24 -0700 Subject: [PATCH] chore: fix the mcp_sse_agent PiperOrigin-RevId: 773140021 --- contributing/samples/mcp_sse_agent/agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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'}, ),