Merge pull request #740 from AnguseZhang/new_db_search

fix: fix db search connect

Author: Rasic2

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,
 )

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?token=eGdk3puy52InRnuOuEbJdGACkos34rSOluFbQqL1HDQ'