Skip to content

Commit ce86765

Browse files
Corrected documentation URL
1 parent 2576453 commit ce86765

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

src/google/adk/a2a/utils/agent_card_builder.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ def __init__(
5959
agent: BaseAgent,
6060
rpc_url: Optional[str] = None,
6161
capabilities: Optional[AgentCapabilities] = None,
62+
documentation_url: Optional[str] = None,
6263
provider: Optional[AgentProvider] = None,
6364
agent_version: Optional[str] = None,
6465
security_schemes: Optional[Dict[str, SecurityScheme]] = None,
@@ -69,6 +70,7 @@ def __init__(
6970
self._agent = agent
7071
self._rpc_url = rpc_url or 'http://localhost:80/a2a'
7172
self._capabilities = capabilities or AgentCapabilities()
73+
self._documentation_url = documentation_url or ""
7274
self._provider = provider
7375
self._security_schemes = security_schemes
7476
self._agent_version = agent_version or '0.0.1'
@@ -86,6 +88,7 @@ async def build(self) -> AgentCard:
8688
url=f"{self._rpc_url.rstrip('/')}",
8789
version=self._agent_version,
8890
capabilities=self._capabilities,
91+
documentation_url=self._documentation_url,
8992
skills=all_skills,
9093
default_input_modes=['text/plain'],
9194
default_output_modes=['text/plain'],

tests/unittests/a2a/utils/test_agent_card_builder.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)