Skip to content

Commit 0cc8adc

Browse files
committed
Few minor fixes
1 parent 76b949d commit 0cc8adc

4 files changed

Lines changed: 11 additions & 17 deletions

File tree

src/a2a/server/request_handlers/default_request_handler.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,6 @@ def __init__( # noqa: PLR0913
100100
push_sender: PushNotificationSender | None = None,
101101
request_context_builder: RequestContextBuilder | None = None,
102102
extended_agent_card: AgentCard | None = None,
103-
card_modifier: Callable[[AgentCard], Awaitable[AgentCard] | AgentCard]
104-
| None = None,
105103
extended_card_modifier: Callable[
106104
[AgentCard, ServerCallContext], Awaitable[AgentCard] | AgentCard
107105
]
@@ -119,7 +117,6 @@ def __init__( # noqa: PLR0913
119117
request_context_builder: The `RequestContextBuilder` instance used
120118
to build request contexts. Defaults to `SimpleRequestContextBuilder`.
121119
extended_agent_card: An optional, distinct `AgentCard` to be served at the extended card endpoint.
122-
card_modifier: An optional callback to dynamically modify the public `AgentCard` before it is served.
123120
extended_card_modifier: An optional callback to dynamically modify the extended `AgentCard` before it is served.
124121
"""
125122
self.agent_executor = agent_executor
@@ -129,7 +126,6 @@ def __init__( # noqa: PLR0913
129126
self._push_config_store = push_config_store
130127
self._push_sender = push_sender
131128
self.extended_agent_card = extended_agent_card
132-
self.card_modifier = card_modifier
133129
self.extended_card_modifier = extended_card_modifier
134130
self._request_context_builder = (
135131
request_context_builder

src/a2a/server/request_handlers/request_handler.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -373,12 +373,10 @@ def decorator(function: Callable) -> Callable:
373373
async def async_gen_wrapper(self: Any, *args, **kwargs) -> Any:
374374
if not expression(self):
375375
final_message = error_message or str(expression)
376-
logger.error('Validation failure: %s', final_message)
377-
raise (
378-
error_type(final_message)
379-
if final_message
380-
else error_type
376+
logging.getLogger(__name__).error(
377+
'Validation failure: %s', final_message
381378
)
379+
raise error_type(final_message)
382380
inner = function(self, *args, **kwargs)
383381
try:
384382
async for item in inner:
@@ -394,12 +392,10 @@ async def async_gen_wrapper(self: Any, *args, **kwargs) -> Any:
394392
async def async_wrapper(self: Any, *args, **kwargs) -> Any:
395393
if not expression(self):
396394
final_message = error_message or str(expression)
397-
logger.error('Validation failure: %s', final_message)
398-
raise (
399-
error_type(final_message)
400-
if final_message
401-
else error_type
395+
logging.getLogger(__name__).error(
396+
'Validation failure: %s', final_message
402397
)
398+
raise error_type(final_message)
403399
return await function(self, *args, **kwargs)
404400

405401
return async_wrapper
@@ -408,7 +404,9 @@ async def async_wrapper(self: Any, *args, **kwargs) -> Any:
408404
def sync_wrapper(self: Any, *args, **kwargs) -> Any:
409405
if not expression(self):
410406
final_message = error_message or str(expression)
411-
logger.error('Validation failure: %s', final_message)
407+
logging.getLogger(__name__).error(
408+
'Validation failure: %s', final_message
409+
)
412410
raise error_type(final_message)
413411
return function(self, *args, **kwargs)
414412

tests/integration/test_agent_card.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ async def test_agent_card_integration(header_val: str | None) -> None:
6666
handler = DefaultRequestHandler(
6767
agent_executor=DummyAgentExecutor(),
6868
task_store=task_store,
69-
agent_card=AgentCard(name='test_agent', version='1.0'),
69+
agent_card=agent_card,
7070
queue_manager=InMemoryQueueManager(),
7171
push_config_store=InMemoryPushNotificationConfigStore(),
7272
)

tests/integration/test_end_to_end.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,8 +171,8 @@ def base_e2e_setup(agent_card):
171171
handler = DefaultRequestHandler(
172172
agent_executor=MockAgentExecutor(),
173173
task_store=task_store,
174-
queue_manager=InMemoryQueueManager(),
175174
agent_card=agent_card,
175+
queue_manager=InMemoryQueueManager(),
176176
)
177177
return task_store, handler
178178

0 commit comments

Comments
 (0)