File tree Expand file tree Collapse file tree
agents/matmaster_agent/sub_agents/structure_search_agent Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11from dp .agent .adapter .adk import CalculationMCPToolset
22from google .adk .agents import BaseAgent
3- from google .adk .tools .mcp_tool .mcp_session_manager import SseServerParams
3+ from google .adk .tools .mcp_tool .mcp_session_manager import StreamableHTTPServerParams
44
55from agents .matmaster_agent .constant import LOCAL_EXECUTOR , BohriumStorge
66from agents .matmaster_agent .core_agents .public_agents .sync_agent import (
1212 STRUCTURE_SEARCH_URL ,
1313)
1414
15+ mcp_params = StreamableHTTPServerParams (
16+ url = STRUCTURE_SEARCH_URL ,
17+ )
1518structure_search_toolset = CalculationMCPToolset (
16- connection_params = SseServerParams ( url = STRUCTURE_SEARCH_URL ) ,
19+ connection_params = mcp_params ,
1720 storage = BohriumStorge ,
1821 executor = LOCAL_EXECUTOR ,
1922)
Original file line number Diff line number Diff line change 55if CURRENT_ENV in ['test' , 'uat' ]:
66 STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp'
77else :
8- STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001 /mcp'
8+ STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002 /mcp'
You can’t perform that action at this time.
0 commit comments