Skip to content

Commit dd28898

Browse files
committed
fix: change sse to mcp
1 parent 40b749a commit dd28898

2 files changed

Lines changed: 6 additions & 3 deletions

File tree

agents/matmaster_agent/sub_agents/structure_search_agent/agent.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from dp.agent.adapter.adk import CalculationMCPToolset
22
from 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

55
from agents.matmaster_agent.constant import LOCAL_EXECUTOR, BohriumStorge
66
from agents.matmaster_agent.core_agents.public_agents.sync_agent import (
@@ -12,8 +12,11 @@
1212
STRUCTURE_SEARCH_URL,
1313
)
1414

15+
mcp_params = StreamableHTTPServerParams(
16+
url=STRUCTURE_SEARCH_URL,
17+
)
1518
structure_search_toolset = CalculationMCPToolset(
16-
connection_params=SseServerParams(url=STRUCTURE_SEARCH_URL),
19+
connection_params=mcp_params,
1720
storage=BohriumStorge,
1821
executor=LOCAL_EXECUTOR,
1922
)

agents/matmaster_agent/sub_agents/structure_search_agent/constant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@
55
if CURRENT_ENV in ['test', 'uat']:
66
STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp'
77
else:
8-
STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50001/mcp'
8+
STRUCTURE_SEARCH_URL = 'http://chvz1424099.bohrium.tech:50002/mcp'

0 commit comments

Comments
 (0)