Spaces:
Paused
Paused
| from typing import Any | |
| from starlette.responses import FileResponse as FileResponse # noqa | |
| from starlette.responses import HTMLResponse as HTMLResponse # noqa | |
| from starlette.responses import JSONResponse as JSONResponse # noqa | |
| from starlette.responses import PlainTextResponse as PlainTextResponse # noqa | |
| from starlette.responses import RedirectResponse as RedirectResponse # noqa | |
| from starlette.responses import Response as Response # noqa | |
| from starlette.responses import StreamingResponse as StreamingResponse # noqa | |
| try: | |
| import ujson | |
| except ImportError: # pragma: nocover | |
| ujson = None # type: ignore | |
| try: | |
| import orjson | |
| except ImportError: # pragma: nocover | |
| orjson = None # type: ignore | |
| class UJSONResponse(JSONResponse): | |
| """ | |
| JSON response using the high-performance ujson library to serialize data to JSON. | |
| Read more about it in the | |
| [FastAPI docs for Custom Response - HTML, Stream, File, others](https://fastapi.tiangolo.com/advanced/custom-response/). | |
| """ | |
| def render(self, content: Any) -> bytes: | |
| assert ujson is not None, "ujson must be installed to use UJSONResponse" | |
| return ujson.dumps(content, ensure_ascii=False).encode("utf-8") | |
| class ORJSONResponse(JSONResponse): | |
| """ | |
| JSON response using the high-performance orjson library to serialize data to JSON. | |
| Read more about it in the | |
| [FastAPI docs for Custom Response - HTML, Stream, File, others](https://fastapi.tiangolo.com/advanced/custom-response/). | |
| """ | |
| def render(self, content: Any) -> bytes: | |
| assert orjson is not None, "orjson must be installed to use ORJSONResponse" | |
| return orjson.dumps( | |
| content, option=orjson.OPT_NON_STR_KEYS | orjson.OPT_SERIALIZE_NUMPY | |
| ) | |