diff --git a/src/main.py b/src/main.py index 9c0e24c..d94ebde 100644 --- a/src/main.py +++ b/src/main.py @@ -44,7 +44,7 @@ class MinSizePerRequestError(Exception): @api.get('/') -def test_data(api_key: str, request: Request, size: str) -> StreamingResponse: +async def test_data(api_key: str, request: Request, size: str) -> StreamingResponse: try: if api_key not in AUTHORIZED_KEYS: raise HTTPException( diff --git a/src/utils.py b/src/utils.py index 521e23c..d703da9 100644 --- a/src/utils.py +++ b/src/utils.py @@ -20,7 +20,7 @@ def convert_to_bytes(size: int | str) -> int: raise ValueError -def generate_data(size: int, buffer_size: int = 4 * 1024) -> bytes: +async def generate_data(size: int, buffer_size: int = 4 * 1024) -> bytes: size_left = size while size_left > buffer_size: