From 9ff116611ac9b3fac8cce86f1398ff5fdcf62fed Mon Sep 17 00:00:00 2001 From: styxhuang Date: Sat, 28 Feb 2026 18:51:24 +0800 Subject: [PATCH 1/3] feat: change executor to dflow --- .../sub_agents/structure_search_agent/agent.py | 6 ++++-- .../sub_agents/structure_search_agent/constant.py | 9 +++++++-- 2 files changed, 11 insertions(+), 4 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 c4db306f..117a56ad 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py @@ -10,6 +10,8 @@ from agents.matmaster_agent.sub_agents.structure_search_agent.constant import ( STRUCTURE_SEARCH_AGENT_NAME, STRUCTURE_SEARCH_URL, + STRUCTURE_SEARCH_EXECUTOR, + STRUCTURE_SEARCH_STORAGE ) mcp_params = StreamableHTTPServerParams( @@ -17,8 +19,8 @@ ) structure_search_toolset = CalculationMCPToolset( connection_params=mcp_params, - storage=BohriumStorge, - executor=LOCAL_EXECUTOR, + storage=STRUCTURE_SEARCH_STORAGE, + executor=STRUCTURE_SEARCH_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 d547888a..8643e994 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py @@ -1,8 +1,13 @@ -from agents.matmaster_agent.constant import CURRENT_ENV +import copy + +from agents.matmaster_agent.constant import CURRENT_ENV, DFlowExecutor, BohriumStorge STRUCTURE_SEARCH_AGENT_NAME = 'structure_search_agent' if CURRENT_ENV in ['test', 'uat']: - STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp' + STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp' else: STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp?token=eGdk3puy52InRnuOuEbJdGACkos34rSOluFbQqL1HDQ' + +STRUCTURE_SEARCH_EXECUTOR = copy.deepcopy(DFlowExecutor) +STRUCTURE_SEARCH_STORAGE = copy.deepcopy(BohriumStorge) \ No newline at end of file From e258547dce3a1f6186865b61c74b76f0ffd30e06 Mon Sep 17 00:00:00 2001 From: styxhuang Date: Sat, 28 Feb 2026 18:52:42 +0800 Subject: [PATCH 2/3] fix pre-commit --- .../sub_agents/structure_search_agent/agent.py | 5 ++--- .../sub_agents/structure_search_agent/constant.py | 4 ++-- 2 files changed, 4 insertions(+), 5 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 117a56ad..5b1cd7a7 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/agent.py @@ -2,16 +2,15 @@ from google.adk.agents import BaseAgent 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 ( BaseSyncAgentWithToolValidator, ) from agents.matmaster_agent.sub_agents.MrDice_agent.constant import MrDice_Agent_Name from agents.matmaster_agent.sub_agents.structure_search_agent.constant import ( STRUCTURE_SEARCH_AGENT_NAME, - STRUCTURE_SEARCH_URL, STRUCTURE_SEARCH_EXECUTOR, - STRUCTURE_SEARCH_STORAGE + STRUCTURE_SEARCH_STORAGE, + STRUCTURE_SEARCH_URL, ) mcp_params = StreamableHTTPServerParams( 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 8643e994..7041b210 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py @@ -1,6 +1,6 @@ import copy -from agents.matmaster_agent.constant import CURRENT_ENV, DFlowExecutor, BohriumStorge +from agents.matmaster_agent.constant import CURRENT_ENV, BohriumStorge, DFlowExecutor STRUCTURE_SEARCH_AGENT_NAME = 'structure_search_agent' @@ -10,4 +10,4 @@ STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp?token=eGdk3puy52InRnuOuEbJdGACkos34rSOluFbQqL1HDQ' STRUCTURE_SEARCH_EXECUTOR = copy.deepcopy(DFlowExecutor) -STRUCTURE_SEARCH_STORAGE = copy.deepcopy(BohriumStorge) \ No newline at end of file +STRUCTURE_SEARCH_STORAGE = copy.deepcopy(BohriumStorge) From be968174590b8bde128bce9b9ffeb036e4799311 Mon Sep 17 00:00:00 2001 From: styxhuang Date: Sat, 28 Feb 2026 20:55:33 +0800 Subject: [PATCH 3/3] fix: change prod url --- .../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 7041b210..3600b857 100644 --- a/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py +++ b/agents/matmaster_agent/sub_agents/structure_search_agent/constant.py @@ -7,7 +7,7 @@ if CURRENT_ENV in ['test', 'uat']: STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp' else: - STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp?token=eGdk3puy52InRnuOuEbJdGACkos34rSOluFbQqL1HDQ' + STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp' STRUCTURE_SEARCH_EXECUTOR = copy.deepcopy(DFlowExecutor) STRUCTURE_SEARCH_STORAGE = copy.deepcopy(BohriumStorge)