Skip to content

Commit 896a66f

Browse files
committed
make extended_card_modifier async
1 parent ffe31e2 commit 896a66f

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

src/a2a/server/apps/rest/rest_adapter.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def __init__( # noqa: PLR0913
6060
context_builder: CallContextBuilder | None = None,
6161
card_modifier: Callable[[AgentCard], AgentCard] | None = None,
6262
extended_card_modifier: Callable[
63-
[AgentCard, ServerCallContext], AgentCard
63+
[AgentCard, ServerCallContext], Awaitable[AgentCard]
6464
]
6565
| None = None,
6666
):
@@ -77,9 +77,9 @@ def __init__( # noqa: PLR0913
7777
ServerCallContext is passed.
7878
card_modifier: An optional callback to dynamically modify the public
7979
agent card before it is served.
80-
extended_card_modifier: An optional callback to dynamically modify
81-
the extended agent card before it is served. It receives the
82-
call context.
80+
extended_card_modifier: An optional async callback to dynamically
81+
modify the extended agent card before it is served. It receives
82+
the call context.
8383
"""
8484
if not _package_starlette_installed:
8585
raise ImportError(
@@ -182,7 +182,9 @@ async def handle_authenticated_agent_card(
182182

183183
if self.extended_card_modifier:
184184
context = self._context_builder.build(request)
185-
card_to_serve = self.extended_card_modifier(card_to_serve, context)
185+
card_to_serve = await self.extended_card_modifier(
186+
card_to_serve, context
187+
)
186188
elif self.card_modifier:
187189
card_to_serve = self.card_modifier(card_to_serve)
188190

0 commit comments

Comments
 (0)