scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 76

fix(a2a_endpoints): merge agent litellm_params guardrails into data before post_call hooks

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-23 18:53:09 -07:00
parent f36f9c919e
commit 3064834f7e
No known key found for this signature in database
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
litellm/proxy/agent_endpoints/a2a_endpoints.py
View file

@ -6,7 +6,7 @@ The A2A SDK can point to LiteLLM's URL and invoke agents registered with LiteLLM
"""
import json
from typing import Any, Dict, Optional
from typing import Any, Dict, List, Optional
from fastapi import APIRouter, Depends, HTTPException, Request, Response
from fastapi.responses import JSONResponse, StreamingResponse
@ -445,6 +445,21 @@ async def invoke_agent_a2a( # noqa: PLR0915
static_headers=static_headers or None,
)
# Merge agent-level guardrails into data so post_call_success_hook and
# _handle_stream_message both pick them up. A2A agents use model
# a2a_agent/*, which is not an llm_router deployment, so
# _check_and_merge_model_level_guardrails() skips them.
_agent_guardrails = litellm_params.get("guardrails")
if _agent_guardrails:
if not isinstance(_agent_guardrails, list):
_agent_guardrails = [_agent_guardrails]
_existing_guardrails: List = data.get("guardrails") or []
if not isinstance(_existing_guardrails, list):
_existing_guardrails = [_existing_guardrails]
data["guardrails"] = _existing_guardrails + [
g for g in _agent_guardrails if g not in _existing_guardrails
]
# Route through SDK functions
if method == "message/send":
from a2a.types import MessageSendParams, SendMessageRequest