From 5921e644468c0b11fcf02f11a854959752334c33 Mon Sep 17 00:00:00 2001 From: yucheng Date: Thu, 10 Sep 2026 08:14:01 +0000 Subject: [PATCH] refactor(guardrails): read key_alias from the typed UserAPIKeyAuth in Agent 365 payload Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/proxy/guardrails/guardrail_hooks/agent_365/agent_365.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/agent_365/agent_365.py b/litellm/proxy/guardrails/guardrail_hooks/agent_365/agent_365.py index 34c4d67a829..8d0957ff2be 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/agent_365/agent_365.py +++ b/litellm/proxy/guardrails/guardrail_hooks/agent_365/agent_365.py @@ -345,7 +345,7 @@ class Agent365Guardrail(CustomGuardrail): tool_name: Final = str(data.get("mcp_tool_name") or "") arguments: Final = data.get("mcp_arguments") server_name: Final = str(data.get("mcp_server_name") or "litellm") - agent_id: Final = self.agent_id or getattr(user_api_key_dict, "key_alias", None) + agent_id: Final = self.agent_id or user_api_key_dict.key_alias tool_reference: Final[_ToolReference] = {"name": tool_name} payload: Final[dict[str, object]] = { # mutable-ok: JSON body with optional fields added below "tool": tool_reference,