From dd2889839c1ba4f62b86a03e3a2f2e642b40e5e7 Mon Sep 17 00:00:00 2001 From: styxhuang Date: Tue, 10 Feb 2026 13:32:36 +0800 Subject: [PATCH 1/2] fix: change sse to mcp --- .../sub_agents/structure_search_agent/agent.py | 7 +++++-- .../sub_agents/structure_search_agent/constant.py | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py b/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py index 0d98ae55..c4db306f 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py @@ -1,6 +1,6 @@ from dp.agent.adapter.adk import CalculationMCPToolset from google.adk.agents import BaseAgent -from google.adk.tools.mcp_tool.mcp_session_manager import SseServerParams +from google.adk.tools.mcp_tool.mcp_session_manager import StreamableHTTPServerParams from agents.matmaster_agent.constant import LOCAL_EXECUTOR, BohriumStorge from agents.matmaster_agent.core_agents.public_agents.sync_agent import ( @@ -12,8 +12,11 @@ STRUCTURE_SEARCH_URL, ) +mcp_params = StreamableHTTPServerParams( + url=STRUCTURE_SEARCH_URL, +) structure_search_toolset = CalculationMCPToolset( - connection_params=SseServerParams(url=STRUCTURE_SEARCH_URL), + connection_params=mcp_params, storage=BohriumStorge, executor=LOCAL_EXECUTOR, ) diff --git a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py index 57e3887d..38d5cd1a 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py @@ -5,4 +5,4 @@ if CURRENT_ENV in ['test', 'uat']: STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp' else: - STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp' + STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp' From bb168055dd3d3ba086296df665e9fc1ded536331 Mon Sep 17 00:00:00 2001 From: styxhuang Date: Tue, 10 Feb 2026 14:01:19 +0800 Subject: [PATCH 2/2] fix: add token for prod env --- .../sub_agents/structure_search_agent/constant.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py index 38d5cd1a..d547888a 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py @@ -5,4 +5,4 @@ if CURRENT_ENV in ['test', 'uat']: STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp' else: - STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp' + STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp?token=eGdk3puy52InRnuOuEbJdGACkos34rSOluFbQqL1HDQ'