CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-02-12 14:32:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: use cli client and srver name

Browse Source
This commit is contained in:
Luca Beurer-Kellner
2025-11-05 18:07:06 +01:00
parent f6cb73cabe
commit a120e2c7c4
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
gateway/mcp/mcp_sessions_manager.py
View File

@@ -32,6 +32,7 @@ def user_and_host() -> str:
return f"{username}@{hostname}"
class McpAttributes(BaseModel):
"""
A Pydantic model to represent MCP attributes.
@@ -118,9 +119,9 @@ class McpAttributes(BaseModel):
for arg in extra_args:
assert "=" in arg, f"Invalid extra metadata argument: {arg}"
key, value = arg.split("=")
assert key.startswith(
"--metadata-"
), f"Invalid extra metadata argument: {arg}, must start with --metadata-"
assert key.startswith("--metadata-"), (
f"Invalid extra metadata argument: {arg}, must start with --metadata-"
)
key = key[len("--metadata-") :]
metadata[key] = value
@@ -185,8 +186,12 @@ class McpSession(BaseModel):
self.attributes.explorer_dataset,
self._get_invariant_authorization(),
# pylint: disable=no-member
self.attributes.metadata.get("mcp_client"),
self.attributes.metadata.get("mcp_server"),
self.attributes.metadata.get(
"client", self.attributes.metadata.get("mcp_client")
),
self.attributes.metadata.get(
"server", self.attributes.metadata.get("mcp_server")
),
)
@contextlib.asynccontextmanager