Debug - Read Endpoint

src/main.py

@@ -40,7 +40,7 @@ app = FastAPI(
     version="1.0.0",
 )
 
-# app.add_middleware(GZipMiddleware, minimum_size=1000)
+app.add_middleware(GZipMiddleware, minimum_size=1000)
 
 
 # Root endpoint redirects to /docs
@@ -119,23 +119,23 @@ async def create_embedding(
 
 # Endpoint to read embeddings
 # @app.get("/read_embeddings/{dataset_name}")
-# @app.post("/read_embeddings")
-# async def read_embeddings(
-#     request: ReadEmbeddingRequest,
-#     huggingface_service: HuggingFaceService = Depends(get_huggingface_service),
-# ):
-#     """
-#     Read embeddings from a Hugging Face dataset.
-#     """
-#     try:
-#         df = await huggingface_service.read_dataset(request.dataset_name)
-#         return df
-#     except DatasetNotFoundError as e:
-#         logger.error(f"Dataset not found: {e}")
-#         raise HTTPException(status_code=404, detail=f"Dataset not found: {e}")
-#     except Exception as e:
-#         logger.error(f"An error occurred: {e}")
-#         raise HTTPException(status_code=500, detail=f"An error occurred: {e}")
+@app.post("/read_embeddings")
+async def read_embeddings(
+    request: ReadEmbeddingRequest,
+    huggingface_service: HuggingFaceService = Depends(get_huggingface_service),
+):
+    """
+    Read embeddings from a Hugging Face dataset.
+    """
+    try:
+        df = await huggingface_service.read_dataset(request.dataset_name)
+        return df
+    except DatasetNotFoundError as e:
+        logger.error(f"Dataset not found: {e}")
+        raise HTTPException(status_code=404, detail=f"Dataset not found: {e}")
+    except Exception as e:
+        logger.error(f"An error occurred: {e}")
+        raise HTTPException(status_code=500, detail=f"An error occurred: {e}")
 
 
 # Endpoint to update embeddings