@@ -107,6 +107,7 @@ def test_init_with_valid_agent(self):
107
107
assert builder ._agent == mock_agent
108
108
assert builder ._rpc_url == "http://localhost:80/a2a"
109
109
assert isinstance (builder ._capabilities , AgentCapabilities )
110
+ assert builder ._documentation_url == ""
110
111
assert builder ._provider is None
111
112
assert builder ._security_schemes is None
112
113
assert builder ._agent_version == "0.0.1"
@@ -125,6 +126,7 @@ def test_init_with_custom_parameters(self):
125
126
agent = mock_agent ,
126
127
rpc_url = "https://example.com/a2a" ,
127
128
capabilities = mock_capabilities ,
129
+ documentation_url = "https://docs.example.com" ,
128
130
provider = mock_provider ,
129
131
agent_version = "1.2.3" ,
130
132
security_schemes = mock_security_schemes ,
@@ -134,6 +136,7 @@ def test_init_with_custom_parameters(self):
134
136
assert builder ._agent == mock_agent
135
137
assert builder ._rpc_url == "https://example.com/a2a"
136
138
assert builder ._capabilities == mock_capabilities
139
+ assert builder ._documentation_url == "https://docs.example.com"
137
140
assert builder ._provider == mock_provider
138
141
assert builder ._security_schemes == mock_security_schemes
139
142
assert builder ._agent_version == "1.2.3"
@@ -210,6 +213,7 @@ async def test_build_with_custom_parameters(
210
213
agent = mock_agent ,
211
214
rpc_url = "https://example.com/a2a/" ,
212
215
provider = mock_provider ,
216
+ documentation_url = "https://docs.example.com" ,
213
217
agent_version = "2.0.0" ,
214
218
security_schemes = mock_security_schemes ,
215
219
)
@@ -223,6 +227,7 @@ async def test_build_with_custom_parameters(
223
227
assert (
224
228
result .url == "https://example.com/a2a"
225
229
) # Should strip trailing slash
230
+ assert result .documentation_url == "https://docs.example.com"
226
231
assert result .version == "2.0.0"
227
232
assert result .provider == mock_provider
228
233
assert result .security_schemes == mock_security_schemes
0 commit comments