@@ -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