Unverified Commit 6c872e7a authored by Marius van den Beek's avatar Marius van den Beek Committed by GitHub
Browse files

Merge pull request #14157 from mvdbeek/fix_streaming_response

[22.01] Adapt byte content for StreamingResponse
parents da4438ac 40c56059
Loading
Loading
Loading
Loading
+6 −1
Original line number Diff line number Diff line
@@ -2,7 +2,10 @@
API operations on the contents of a history dataset.
"""
import logging
from io import IOBase
from io import (
    BytesIO,
    IOBase,
)
from typing import (
    Any,
    cast,
@@ -259,6 +262,8 @@ class FastAPIDatasets:
                return FileResponse(file_name, headers=headers)
        elif isinstance(display_data, ZipstreamWrapper):
            return StreamingResponse(display_data.response(), headers=headers)
        elif isinstance(display_data, bytes):
            return StreamingResponse(BytesIO(display_data), headers=headers)
        return StreamingResponse(display_data, headers=headers)

    @router.get(