Skip to content

Commit 7f8dc89

Browse files
seanzhougooglecopybara-github
authored andcommitted
chore: fix the mcp_sse_agent
PiperOrigin-RevId: 773140021
1 parent 9a1115c commit 7f8dc89

File tree

1 file changed

+2
-2
lines changed
  • contributing/samples/mcp_sse_agent

1 file changed

+2
-2
lines changed

contributing/samples/mcp_sse_agent/agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
import os
1717

1818
from google.adk.agents.llm_agent import LlmAgent
19+
from google.adk.tools.mcp_tool.mcp_session_manager import SseConnectionParams
1920
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
20-
from google.adk.tools.mcp_tool.mcp_toolset import SseServerParams
2121

2222
_allowed_path = os.path.dirname(os.path.abspath(__file__))
2323

@@ -31,7 +31,7 @@
3131
""",
3232
tools=[
3333
MCPToolset(
34-
connection_params=SseServerParams(
34+
connection_params=SseConnectionParams(
3535
url='http://localhost:3000/sse',
3636
headers={'Accept': 'text/event-stream'},
3737
),

0 commit comments

Comments
 (0)