Python: Add configurable timeout support to A2AAgent (#2432)

* a2a timeout config

* added default timeout info

---------

Co-authored-by: Dmytro Struk <13853051+dmytrostruk@users.noreply.github.com>

python/packages/a2a/tests/test_a2a_agent.py

@@ -5,6 +5,7 @@ from typing import Any
 from unittest.mock import AsyncMock, MagicMock, patch
 from uuid import uuid4
 
+import httpx
 from a2a.types import (
     AgentCard,
     Artifact,
@@ -554,3 +555,51 @@ def test_transport_negotiation_both_fail() -> None:
             name="test-agent",
             agent_card=mock_agent_card,
         )
+
+
+def test_create_timeout_config_httpx_timeout() -> None:
+    """Test _create_timeout_config with httpx.Timeout object returns it unchanged."""
+    agent = A2AAgent(name="Test Agent", client=MockA2AClient(), http_client=None)
+
+    custom_timeout = httpx.Timeout(connect=15.0, read=180.0, write=20.0, pool=8.0)
+    timeout_config = agent._create_timeout_config(custom_timeout)
+
+    assert timeout_config is custom_timeout  # Same object reference
+    assert timeout_config.connect == 15.0
+    assert timeout_config.read == 180.0
+    assert timeout_config.write == 20.0
+    assert timeout_config.pool == 8.0
+
+
+def test_create_timeout_config_invalid_type() -> None:
+    """Test _create_timeout_config with invalid type raises TypeError."""
+    agent = A2AAgent(name="Test Agent", client=MockA2AClient(), http_client=None)
+
+    with raises(TypeError, match="Invalid timeout type: <class 'str'>. Expected float, httpx.Timeout, or None."):
+        agent._create_timeout_config("invalid")
+
+
+def test_a2a_agent_initialization_with_timeout_parameter() -> None:
+    """Test A2AAgent initialization with timeout parameter."""
+    # Test with URL to trigger httpx client creation
+    with (
+        patch("agent_framework_a2a._agent.httpx.AsyncClient") as mock_async_client,
+        patch("agent_framework_a2a._agent.ClientFactory") as mock_factory,
+    ):
+        # Mock the factory and client creation
+        mock_client_instance = MagicMock()
+        mock_factory.return_value.create.return_value = mock_client_instance
+
+        # Create agent with custom timeout
+        A2AAgent(name="Test Agent", url="https://test-agent.example.com", timeout=120.0)
+
+        # Verify httpx.AsyncClient was called with the configured timeout
+        mock_async_client.assert_called_once()
+        call_args = mock_async_client.call_args
+
+        # Check that timeout parameter was passed
+        assert "timeout" in call_args.kwargs
+        timeout_arg = call_args.kwargs["timeout"]
+
+        # Verify it's an httpx.Timeout object with our custom timeout applied to all components
+        assert isinstance(timeout_arg, httpx.Timeout)