Accept multiple file urls in delete endpoint (#43)

superagent-ai · Feb 16, 2024 · bd2d03a · bd2d03a
1 parent 68074e1
commit bd2d03a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/api/delete.py b/api/delete.py
@@ -17,5 +17,8 @@ async def delete(payload: RequestPayload):
         encoder=encoder,
         dimensions=encoder.dimensions,
     )
-    data = await vector_service.delete(file_url=payload.file_url)
+
+    for file in payload.files:
+        data = await vector_service.delete(file_url=file.url)
+
     return ResponsePayload(success=True, data=data)
diff --git a/models/delete.py b/models/delete.py
@@ -4,9 +4,13 @@
 from models.vector_database import VectorDatabase
 
 
+class File(BaseModel):
+    url: str
+
+
 class RequestPayload(BaseModel):
     index_name: str
-    file_url: str
+    files: list[File]
     vector_database: VectorDatabase
     encoder: Encoder