mirror of
https://github.com/invariantlabs-ai/invariant-gateway.git
synced 2026-06-05 04:38:08 +02:00
add server and client name to policy get
This commit is contained in:
@@ -145,7 +145,10 @@ async def push_trace(
|
||||
|
||||
|
||||
async def fetch_guardrails_from_explorer(
|
||||
dataset_name: str, invariant_authorization: str
|
||||
dataset_name: str,
|
||||
invariant_authorization: str,
|
||||
client_name: str | None = None,
|
||||
server_name: str | None = None,
|
||||
) -> GuardrailRuleSet:
|
||||
"""Get the guardrails for the dataset.
|
||||
|
||||
@@ -179,7 +182,8 @@ async def fetch_guardrails_from_explorer(
|
||||
|
||||
# Get the dataset policies.
|
||||
policies_response = await client.get(
|
||||
f"/api/v1/dataset/byuser/{username}/{dataset_name}/policy"
|
||||
f"/api/v1/dataset/byuser/{username}/{dataset_name}/policy",
|
||||
params={"client_name": client_name, "server_name": server_name},
|
||||
)
|
||||
if policies_response.status_code != 200:
|
||||
if policies_response.status_code == 404:
|
||||
|
||||
@@ -105,5 +105,7 @@ class McpContext:
|
||||
async def load_guardrails(self):
|
||||
"""Run async setup logic (e.g. fetching guardrails)."""
|
||||
self.guardrails = await fetch_guardrails_from_explorer(
|
||||
self.explorer_dataset, "Bearer " + os.getenv("INVARIANT_API_KEY")
|
||||
self.explorer_dataset, "Bearer " + os.getenv("INVARIANT_API_KEY"),
|
||||
self.mcp_client_name,
|
||||
self.mcp_server_name,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user