github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-20 08:59:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent): Support MCP authentication

Browse Source
This commit is contained in:
Fangyin Cheng
2025-04-03 09:21:05 +08:00
parent 0b6633703d
commit 2e0e140b51
7 changed files with 143 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
packages/dbgpt-serve/src/dbgpt_serve/agent/resource/mcp.py
View File

@@ -49,7 +49,27 @@ class MCPPackResourceParameters(PackResourceParameters):
class MCPSSEToolPack(MCPToolPack):
def __init__(self, mcp_servers: Union[str, List[str]], **kwargs):
super().__init__(mcp_servers=mcp_servers, **kwargs)
"""Initialize the MCPSSEToolPack with the given MCP servers."""
headers = {}
if "token" in kwargs and kwargs["token"]:
# token is not supported in sse mode
servers = (
mcp_servers.split(";") if isinstance(mcp_servers, str) else mcp_servers
)
tokens = (
kwargs["token"].split(";")
if isinstance(kwargs["token"], str)
else kwargs["token"]
)
if len(servers) == len(tokens):
for i, token in enumerate(tokens):
headers[servers[i]] = {"Authorization": f"Bearer {token}"}
else:
token = tokens[0]
for server in servers:
headers[server] = {"Authorization": f"Bearer {token}"}
kwargs.pop("token")
super().__init__(mcp_servers=mcp_servers, headers=headers, **kwargs)
@classmethod
def type_alias(cls) -> str:
@@ -64,7 +84,17 @@ class MCPSSEToolPack(MCPToolPack):
mcp_servers: str = dataclasses.field(
default="http://127.0.0.1:8000/sse",
metadata={
"help": _("MCP SSE Server URL, split by ':'"),
"help": _("MCP SSE Server URL, split by ';'"),
},
)
token: Optional[str] = dataclasses.field(
default=None,
metadata={
"help": _(
'MCP SSE Server token, split by ";", It will be '
'added to the header({"Authorization": "Bearer your_token"}'
),
"tags": "privacy",
},
)