Skip to content

Commit a59353d

Browse files
authored
support header authorization in mcp_router (#267)
1 parent 37c3702 commit a59353d

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

config.yaml.full

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ tool:
5757
dataset_id: #dataset name
5858
mcp_router:
5959
url: #mcp sse/streamable-http url
60+
api_key: #mcp api key
6061
code_sandbox:
6162
url: #mcp sse/streamable-http url
6263
api_key: #mcp api key

veadk/tools/builtin_tools/mcp_router.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,15 @@
1515
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
18+
from google.adk.tools.mcp_tool.mcp_session_manager import (
19+
StreamableHTTPConnectionParams,
20+
)
1921

2022
url = getenv("TOOL_MCP_ROUTER_URL")
23+
api_key = getenv("TOOL_MCP_ROUTER_API_KEY")
2124

22-
mcp_router = MCPToolset(connection_params=get_mcp_params(url=url))
25+
mcp_router = MCPToolset(
26+
connection_params=StreamableHTTPConnectionParams(
27+
url=url, headers={"Authorization": f"Bearer {api_key}"}
28+
),
29+
)

0 commit comments

Comments
 (0)