@@ -107,6 +107,7 @@ def test_init_with_valid_agent(self):
107107 assert builder ._agent == mock_agent
108108 assert builder ._rpc_url == "http://localhost:80/a2a"
109109 assert isinstance (builder ._capabilities , AgentCapabilities )
110+ assert builder ._documentation_url == ""
110111 assert builder ._provider is None
111112 assert builder ._security_schemes is None
112113 assert builder ._agent_version == "0.0.1"
@@ -125,6 +126,7 @@ def test_init_with_custom_parameters(self):
125126 agent = mock_agent ,
126127 rpc_url = "https://example.com/a2a" ,
127128 capabilities = mock_capabilities ,
129+ documentation_url = "https://docs.example.com" ,
128130 provider = mock_provider ,
129131 agent_version = "1.2.3" ,
130132 security_schemes = mock_security_schemes ,
@@ -134,6 +136,7 @@ def test_init_with_custom_parameters(self):
134136 assert builder ._agent == mock_agent
135137 assert builder ._rpc_url == "https://example.com/a2a"
136138 assert builder ._capabilities == mock_capabilities
139+ assert builder ._documentation_url == "https://docs.example.com"
137140 assert builder ._provider == mock_provider
138141 assert builder ._security_schemes == mock_security_schemes
139142 assert builder ._agent_version == "1.2.3"
@@ -210,6 +213,7 @@ async def test_build_with_custom_parameters(
210213 agent = mock_agent ,
211214 rpc_url = "https://example.com/a2a/" ,
212215 provider = mock_provider ,
216+ documentation_url = "https://docs.example.com" ,
213217 agent_version = "2.0.0" ,
214218 security_schemes = mock_security_schemes ,
215219 )
@@ -223,6 +227,7 @@ async def test_build_with_custom_parameters(
223227 assert (
224228 result .url == "https://example.com/a2a"
225229 ) # Should strip trailing slash
230+ assert result .documentation_url == "https://docs.example.com"
226231 assert result .version == "2.0.0"
227232 assert result .provider == mock_provider
228233 assert result .security_schemes == mock_security_schemes
0 commit comments