From c39b23a5d53f1e7fb277d34944969283ca0c1ac7 Mon Sep 17 00:00:00 2001
From: AlphaRex-pixel <paulsa894@gmail.com>
Date: Fri, 11 Sep 2026 00:41:54 +0530
Subject: [PATCH] Apply ruff format

---
 litellm/a2a_protocol/main.py                  | 37 ++++++++++++-------
 .../proxy/agent_endpoints/a2a_endpoints.py    |  2 +-
 2 files changed, 24 insertions(+), 15 deletions(-)

diff --git a/litellm/a2a_protocol/main.py b/litellm/a2a_protocol/main.py
index a88cca0cda0..07ee9d1cbe4 100644
--- a/litellm/a2a_protocol/main.py
+++ b/litellm/a2a_protocol/main.py
@@ -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); "
diff --git a/litellm/proxy/agent_endpoints/a2a_endpoints.py b/litellm/proxy/agent_endpoints/a2a_endpoints.py
index 18ab3e6087d..9877042f034 100644
--- a/litellm/proxy/agent_endpoints/a2a_endpoints.py
+++ b/litellm/proxy/agent_endpoints/a2a_endpoints.py
@@ -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 (