mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-05-10 04:00:24 +00:00
chore: Update toolkit_listen decorator make it get the docstring from the parent class (#452)
This commit is contained in:
commit
edd44129e2
1 changed files with 2 additions and 2 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue