Skip to content

Commit 6ebc46e

Browse files
committed
Revert a few things
1 parent 184f8d5 commit 6ebc46e

2 files changed

Lines changed: 4 additions & 23 deletions

File tree

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ async def event_generator(
159159
yield json.dumps(item)
160160

161161
return EventSourceResponse(
162-
event_generator(await method(request, call_context))
162+
event_generator(method(request, call_context))
163163
)
164164

165165
async def handle_get_agent_card(

src/a2a/utils/helpers.py

Lines changed: 3 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -389,27 +389,8 @@ def async_gen_wrapper(
389389

390390
return cast('F', async_gen_wrapper)
391391

392-
if inspect.iscoroutinefunction(inspect.unwrap(func)):
393-
394-
@functools.wraps(func)
395-
async def async_wrapper(self: Any, *args: Any, **kwargs: Any) -> Any:
396-
actual_version = _get_actual_version(args, kwargs)
397-
if not _is_version_compatible(actual_version):
398-
logger.warning(
399-
"Version mismatch: actual='%s', expected='%s'",
400-
actual_version,
401-
expected_version,
402-
)
403-
raise VersionNotSupportedError(
404-
message=f"A2A version '{actual_version}' is not supported by this handler. "
405-
f"Expected version '{expected_version}'."
406-
)
407-
return await func(self, *args, **kwargs)
408-
409-
return cast('F', async_wrapper)
410-
411392
@functools.wraps(func)
412-
def sync_wrapper(self: Any, *args: Any, **kwargs: Any) -> Any:
393+
async def async_wrapper(self: Any, *args: Any, **kwargs: Any) -> Any:
413394
actual_version = _get_actual_version(args, kwargs)
414395
if not _is_version_compatible(actual_version):
415396
logger.warning(
@@ -421,8 +402,8 @@ def sync_wrapper(self: Any, *args: Any, **kwargs: Any) -> Any:
421402
message=f"A2A version '{actual_version}' is not supported by this handler. "
422403
f"Expected version '{expected_version}'."
423404
)
424-
return func(self, *args, **kwargs)
405+
return await func(self, *args, **kwargs)
425406

426-
return cast('F', sync_wrapper)
407+
return cast('F', async_wrapper)
427408

428409
return decorator

0 commit comments

Comments
 (0)