typo and better error handling

1 files changed, 5 insertions, 1 deletions

diff --git a/src/collector/main.py b/src/collector/main.py
index beaf2ea..72c4ef7 100755
--- a/src/collector/main.py
+++ b/src/collector/main.py
@@ -164,4 +164,8 @@ async def info() -> JSONResponse:
     """
 
     count = await db.estimated_document_count()
-    return JSONResponse(content={"status": "success", "estimated document count": count})
+
+    if count is None:
+        return JSONResponse(content={"status": "error", "message": "DB error"}, status_code=400)
+
+    return JSONResponse(content={"status": "success", "Estimated document count": count})