AgentCardResolutionError updates

Author: ishymko

src/a2a/client/card_resolver.py

@@ -91,14 +91,15 @@ async def get_agent_card(
         except httpx.HTTPStatusError as e:
             raise AgentCardResolutionError(
                 f'Failed to fetch agent card from {target_url} (HTTP {e.response.status_code}): {e}',
+                status_code=e.response.status_code,
             ) from e
         except json.JSONDecodeError as e:
             raise AgentCardResolutionError(
                 f'Failed to parse JSON for agent card from {target_url}: {e}'
             ) from e
         except httpx.RequestError as e:
             raise AgentCardResolutionError(
-                f'Network communication error fetching agent card from {target_url} (HTTP 503): {e}',
+                f'Network communication error fetching agent card from {target_url}: {e}',
             ) from e
         except ParseError as e:
             raise AgentCardResolutionError(

src/a2a/client/errors.py

@@ -9,3 +9,7 @@ class A2AClientError(A2AError):
 
 class AgentCardResolutionError(A2AClientError):
     """Exception raised when an agent card cannot be resolved."""
+
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message)
+        self.status_code = status_code

tests/client/test_card_resolver.py

@@ -221,6 +221,7 @@ async def test_get_agent_card_http_status_error(
         with pytest.raises(AgentCardResolutionError) as exc_info:
             await resolver.get_agent_card()
 
+        assert exc_info.value.status_code == status_code
         assert f'HTTP {status_code}' in str(exc_info.value)
         assert 'Failed to fetch agent card' in str(exc_info.value)
 
@@ -247,7 +248,6 @@ async def test_get_agent_card_request_error(
         )
         with pytest.raises(AgentCardResolutionError) as exc_info:
             await resolver.get_agent_card()
-        assert 'HTTP 503' in str(exc_info.value)
         assert 'Network communication error' in str(exc_info.value)
 
     @pytest.mark.asyncio