fix: file delete from knowledge not working with bypass embedding

This commit is contained in:
Timothy Jaeryang Baek 2025-03-20 17:46:11 -07:00
parent a28436237c
commit 87a06a1976

View File

@ -437,14 +437,24 @@ def remove_file_from_knowledge_by_id(
) )
# Remove content from the vector database # Remove content from the vector database
VECTOR_DB_CLIENT.delete( try:
collection_name=knowledge.id, filter={"file_id": form_data.file_id} VECTOR_DB_CLIENT.delete(
) collection_name=knowledge.id, filter={"file_id": form_data.file_id}
)
except Exception as e:
log.debug("This was most likely caused by bypassing embedding processing")
log.debug(e)
pass
# Remove the file's collection from vector database try:
file_collection = f"file-{form_data.file_id}" # Remove the file's collection from vector database
if VECTOR_DB_CLIENT.has_collection(collection_name=file_collection): file_collection = f"file-{form_data.file_id}"
VECTOR_DB_CLIENT.delete_collection(collection_name=file_collection) if VECTOR_DB_CLIENT.has_collection(collection_name=file_collection):
VECTOR_DB_CLIENT.delete_collection(collection_name=file_collection)
except Exception as e:
log.debug("This was most likely caused by bypassing embedding processing")
log.debug(e)
pass
# Delete file from database # Delete file from database
Files.delete_file_by_id(form_data.file_id) Files.delete_file_by_id(form_data.file_id)