diff --git a/backend/main.py b/backend/main.py index bd24c369b..744c84b6d 100644 --- a/backend/main.py +++ b/backend/main.py @@ -309,7 +309,10 @@ async def get_function_call_response( "__id__": tool_id, } - function_result = function(**params) + if inspect.iscoroutinefunction(function): + function_result = await function(**params) + else: + function_result = function(**params) if hasattr(toolkit_module, "citation") and toolkit_module.citation: citation = {