amaye15 commited on
Commit
f811398
·
1 Parent(s): a965d9d

debug - read endpoint

Browse files
Files changed (1) hide show
  1. src/main.py +22 -22
src/main.py CHANGED
@@ -139,28 +139,28 @@ async def read_embeddings(
139
 
140
 
141
  # Endpoint to update embeddings
142
- @app.post("/update_embeddings")
143
- async def update_embeddings(
144
- request: UpdateEmbeddingRequest,
145
- huggingface_service: HuggingFaceService = Depends(get_huggingface_service),
146
- ):
147
- """
148
- Update embeddings in a Hugging Face dataset.
149
- """
150
- try:
151
- df = await huggingface_service.update_dataset(
152
- request.dataset_name, request.updates
153
- )
154
- return {
155
- "message": "Embeddings updated successfully.",
156
- "dataset_name": request.dataset_name,
157
- }
158
- except DatasetPushError as e:
159
- logger.error(f"Failed to update dataset: {e}")
160
- raise HTTPException(status_code=500, detail=f"Failed to update dataset: {e}")
161
- except Exception as e:
162
- logger.error(f"An error occurred: {e}")
163
- raise HTTPException(status_code=500, detail=f"An error occurred: {e}")
164
 
165
 
166
  # Endpoint to delete embeddings
 
139
 
140
 
141
  # Endpoint to update embeddings
142
+ # @app.post("/update_embeddings")
143
+ # async def update_embeddings(
144
+ # request: UpdateEmbeddingRequest,
145
+ # huggingface_service: HuggingFaceService = Depends(get_huggingface_service),
146
+ # ):
147
+ # """
148
+ # Update embeddings in a Hugging Face dataset.
149
+ # """
150
+ # try:
151
+ # df = await huggingface_service.update_dataset(
152
+ # request.dataset_name, request.updates
153
+ # )
154
+ # return {
155
+ # "message": "Embeddings updated successfully.",
156
+ # "dataset_name": request.dataset_name,
157
+ # }
158
+ # except DatasetPushError as e:
159
+ # logger.error(f"Failed to update dataset: {e}")
160
+ # raise HTTPException(status_code=500, detail=f"Failed to update dataset: {e}")
161
+ # except Exception as e:
162
+ # logger.error(f"An error occurred: {e}")
163
+ # raise HTTPException(status_code=500, detail=f"An error occurred: {e}")
164
 
165
 
166
  # Endpoint to delete embeddings