scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

Apply ruff format

Browse source
This commit is contained in:
AlphaRex-pixel 2026-09-11 00:41:54 +05:30
parent ce8b116396
commit c39b23a5d5
2 changed files with 24 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
litellm/a2a_protocol/main.py
View file

@ -387,7 +387,10 @@ async def asend_message(
api_base: str | None = None,
litellm_params: dict[str, object] | None = None,
agent_id: str | None = None,
agent_extra_headers: dict[str, str] | None = None, agent_card_params: dict[str, object] | None = None, **kwargs: object, ) -> LiteLLMSendMessageResponse:
agent_extra_headers: dict[str, str] | None = None,
agent_card_params: dict[str, object] | None = None,
**kwargs: object,
) -> LiteLLMSendMessageResponse:
"""
Async: Send a message to an A2A agent.
@ -472,7 +475,11 @@ async def asend_message(
# Overlay agent-level headers (agent headers take precedence over LiteLLM internal ones)
if agent_extra_headers:
extra_headers.update(agent_extra_headers)
a2a_client = await create_a2a_client( base_url=api_base, extra_headers=extra_headers, agent_card_params=agent_card_params, )
a2a_client = await create_a2a_client(
base_url=api_base,
extra_headers=extra_headers,
agent_card_params=agent_card_params,
)
# Type assertion: a2a_client is guaranteed to be non-None here
assert a2a_client is not None
@ -607,9 +614,10 @@ async def asend_message_streaming(
agent_id: str | None = None,
metadata: dict[str, object] | None = None,
proxy_server_request: dict[str, object] | None = None,
agent_extra_headers: dict[str, str] | None = None,
agent_card_params: dict[str, object] | None = None,
**kwargs: object, ) -> AsyncIterator[Any]:
agent_extra_headers: dict[str, str] | None = None,
agent_card_params: dict[str, object] | None = None,
**kwargs: object,
) -> AsyncIterator[Any]:
"""
Async: Send a streaming message to an A2A agent.
@ -693,7 +701,12 @@ async def asend_message_streaming(
extra_headers["X-LiteLLM-Agent-Id"] = agent_id
if agent_extra_headers:
extra_headers.update(agent_extra_headers)
a2a_client = await create_a2a_client( base_url=api_base, extra_headers=extra_headers, streaming=True, agent_card_params=agent_card_params, )
a2a_client = await create_a2a_client(
base_url=api_base,
extra_headers=extra_headers,
streaming=True,
agent_card_params=agent_card_params,
)
assert a2a_client is not None
@ -739,7 +752,7 @@ async def create_a2a_client(
timeout: float = DEFAULT_A2A_AGENT_TIMEOUT,
extra_headers: dict[str, str] | None = None,
streaming: bool = False,
agent_card_params: dict[str, object] | None = None,
agent_card_params: dict[str, object] | None = None,
) -> "A2AClientType":
"""
Create an A2A client for the given agent URL.
@ -783,19 +796,15 @@ async def create_a2a_client(
verbose_proxy_logger.debug("A2A client created with extra_headers=%s", list(extra_headers.keys()))
agent_card: "AgentCard | None" = None
if agent_card_params:
from pydantic import ValidationError as _ValidationError
from a2a.compat.v0_3.types import AgentCard as _AgentCard
try:
agent_card = normalize_agent_card_interfaces(
_AgentCard.model_validate(agent_card_params)
)
verbose_logger.info(
"Using pre-registered agent card for %s (skipping well-known discovery)", base_url
)
agent_card = normalize_agent_card_interfaces(_AgentCard.model_validate(agent_card_params))
verbose_logger.info("Using pre-registered agent card for %s (skipping well-known discovery)", base_url)
except _ValidationError as e:
verbose_logger.warning(
"Stored agent_card_params for %s failed AgentCard validation (%s); "

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

@ -475,7 +475,7 @@ async def _handle_stream_message(
metadata=metadata,
proxy_server_request=proxy_server_request,
agent_extra_headers=agent_extra_headers,
agent_card_params=agent_card_params,
agent_card_params=agent_card_params,
)
if (