Skip to content

Commit 5e6ee6c

Browse files
committed
fix sse/streamable compatible issue by change mcp_router.py and update config.yaml.example
1 parent e470dae commit 5e6ee6c

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

config.yaml.example

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ tool:
4141
url: #mcp sse url
4242
dataset_id: #dataset name
4343
mcp_router:
44-
url: #mcp sse url
44+
url: #mcp sse/streamable-http url
4545

4646

4747
observability:

veadk/tools/builtin_tools/mcp_router.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,11 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, SseConnectionParams
15+
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
1616

1717
from veadk.config import getenv
18+
from veadk.utils.mcp_utils import get_mcp_params
1819

1920
url = getenv("TOOL_MCP_ROUTER_URL")
2021

21-
mcp_router = MCPToolset(connection_params=SseConnectionParams(url=url))
22+
mcp_router = MCPToolset(connection_params=get_mcp_params(url=url))

0 commit comments

Comments
 (0)