add server and client name to policy get

This commit is contained in:
knielsen404
2025-05-09 18:43:08 +02:00
parent 8988f9a7dd
commit 8f9c28158d
2 changed files with 9 additions and 3 deletions
+6 -2
View File
@@ -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:
+3 -1
View File
@@ -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,
)