fix(huggingface_restapi.py): async implementation

This commit is contained in:
Krrish Dholakia 2023-11-15 16:54:08 -08:00
parent 1bd7d13653
commit e28b9534e4
2 changed files with 31 additions and 34 deletions

View file

@ -4,7 +4,9 @@
import sys, os
import pytest
import traceback
import asyncio
import asyncio, logging
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
sys.path.insert(
0, os.path.abspath("../..")
@ -99,5 +101,3 @@ def test_get_response_non_openai_streaming():
pytest.fail(f"An exception occurred: {e}")
return response
asyncio.run(test_async_call())
test_get_response_non_openai_streaming()