delete docs

main.py

@@ -91,6 +91,7 @@ async def upload_file(file: UploadFile, enterprise_data: Json[EnterpriseData]):
                 "file_name":file.filename,
                 "enterprise_id": enterprise_data.id,
                 "number_of_chunks": len(text_chunks),
+                "filename_id":vector_store["filename_id"]
             }
         else:
             raise HTTPException(status_code=500, detail="Could not create vector store")
@@ -114,6 +115,15 @@ def get_documents(enterprise_id: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
     
+@app.delete("/documents/{enterprise_id}/{filename_id}")
+def delete_document(enterprise_id: str, filename_id: str):
+    try:
+        for ids in index.list(prefix=f"{filename_id}_", namespace=enterprise_id):
+            index.delete(ids=ids, namespace=enterprise_id)
+        return {"message": "Document deleted", "chunks_deleted": ids}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
+    
 @app.delete("/documents/all/{enterprise_id}")
 def delete_all_documents(enterprise_id: str):
     try:

rag.py

@@ -7,6 +7,7 @@ from langchain_core.documents import Document
 from langchain_openai import ChatOpenAI
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import PromptTemplate
+from uuid import uuid4
 
 import unicodedata
 
@@ -47,18 +48,20 @@ def get_vectorstore(text_chunks,filename, file_type,namespace,index):
         uuids = []
 
         for i, chunk in enumerate(text_chunks):
+            clean_filename = remove_non_standard_ascii(file_name)
+
             document = Document(
             page_content=chunk,
-            metadata={"filename":filename,"file_type":file_type},
+            metadata={"filename":filename,"file_type":file_type, "filename_id":clean_filename},
             )
-            clean_filename = remove_non_standard_ascii(file_name)
+    
             uuid = f"{clean_filename}_{i}"
             uuids.append(uuid)
             documents.append(document)
         
         vector_store.add_documents(documents=documents, ids=uuids)
 
-        return True
+        return {"filename_id":clean_filename}
     
     except Exception as e:
         print(e)