switch to orjson

worldbank · Aug 29, 2024 · 66b3054 · 66b3054
1 parent 9ffe68c
commit 66b3054
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/space2stats_api/src/pyproject.toml b/space2stats_api/src/pyproject.toml
@@ -9,6 +9,7 @@ authors = [
 ]
 dynamic = ["version"]
 dependencies = [
+    "orjson",
     "fastapi>=0.112.0",
     "pandas",
     "shapely",

diff --git a/space2stats_api/src/space2stats/app.py b/space2stats_api/src/space2stats/app.py
@@ -2,6 +2,7 @@
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import ORJSONResponse
 
 from typing import Any, Dict, List
 
@@ -21,7 +22,10 @@ async def lifespan(app: FastAPI):
     await close_db_connection(app)
 
 
-app = FastAPI(lifespan=lifespan)
+app = FastAPI(
+    default_response_class=ORJSONResponse,
+    lifespan=lifespan,
+)
 
 app.add_middleware(
     CORSMiddleware,