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,7 +309,10 @@ async def get_function_call_response(
"__id__": tool_id, "__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: if hasattr(toolkit_module, "citation") and toolkit_module.citation:
citation = { citation = {