chore: Update toolkit_listen decorator make it get the docstring from the parent class (#452)

This commit is contained in:
Wendong-Fan 2025-10-04 22:22:45 +08:00 committed by GitHub
commit edd44129e2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -24,7 +24,7 @@ def listen_toolkit(
if iscoroutinefunction(func):
# async function wrapper
@wraps(func)
@wraps(wrap)
async def async_wrapper(*args, **kwargs):
toolkit: AbstractToolkit = args[0]
task_lock = get_task_lock(toolkit.api_task_id)
@ -93,7 +93,7 @@ def listen_toolkit(
else:
# sync function wrapper
@wraps(func)
@wraps(wrap)
def sync_wrapper(*args, **kwargs):
toolkit: AbstractToolkit = args[0]
task_lock = get_task_lock(toolkit.api_task_id)