chore: fix the mcp_sse_agent

seanzhougoogle · copybara-github · commit 7f8dc8927aaa · 2025-06-18T18:32:06.000-07:00
PiperOrigin-RevId: 773140021
diff --git 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 @@
     """,
     tools=[
         MCPToolset(
-            connection_params=SseServerParams(
+            connection_params=SseConnectionParams(
                 url='http://localhost:3000/sse',
                 headers={'Accept': 'text/event-stream'},
             ),
