Debug - read endpoint

src/main.py

@@ -119,50 +119,50 @@ 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}")
-
-
-# Endpoint to update embeddings
-# @app.post("/update_embeddings")
-# async def update_embeddings(
-#     request: UpdateEmbeddingRequest,
+# @app.post("/read_embeddings")
+# async def read_embeddings(
+#     request: ReadEmbeddingRequest,
 #     huggingface_service: HuggingFaceService = Depends(get_huggingface_service),
 # ):
 #     """
-#     Update embeddings in a Hugging Face dataset.
+#     Read embeddings from a Hugging Face dataset.
 #     """
 #     try:
-#         df = await huggingface_service.update_dataset(
-#             request.dataset_name, request.updates
-#         )
-#         return {
-#             "message": "Embeddings updated successfully.",
-#             "dataset_name": request.dataset_name,
-#         }
-#     except DatasetPushError as e:
-#         logger.error(f"Failed to update dataset: {e}")
-#         raise HTTPException(status_code=500, detail=f"Failed to update dataset: {e}")
+#         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
+@app.post("/update_embeddings")
+async def update_embeddings(
+    request: UpdateEmbeddingRequest,
+    huggingface_service: HuggingFaceService = Depends(get_huggingface_service),
+):
+    """
+    Update embeddings in a Hugging Face dataset.
+    """
+    try:
+        df = await huggingface_service.update_dataset(
+            request.dataset_name, request.updates
+        )
+        return {
+            "message": "Embeddings updated successfully.",
+            "dataset_name": request.dataset_name,
+        }
+    except DatasetPushError as e:
+        logger.error(f"Failed to update dataset: {e}")
+        raise HTTPException(status_code=500, detail=f"Failed to update dataset: {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 delete embeddings
 @app.post("/delete_embeddings")
 async def delete_embeddings(