scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 69

refactor(guardrails): read key_alias from the typed UserAPIKeyAuth in Agent 365 payload

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-10 08:14:01 +00:00
parent f4033ab84a
commit 5921e64446
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/guardrails/guardrail_hooks/agent_365/agent_365.py
View file

@ -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,