enh: tool async support

This commit is contained in:
Timothy J. Baek 2024-06-20 20:40:03 -07:00
parent 4370f233a1
commit a2ea6b1b5b

View File

@ -309,6 +309,9 @@ async def get_function_call_response(
"__id__": tool_id,
}
if inspect.iscoroutinefunction(function):
function_result = await function(**params)
else:
function_result = function(**params)
if hasattr(toolkit_module, "citation") and toolkit_module.citation: