diff --git a/rag_system/clear_index.py b/rag_system/clear_index.py index 3c38a83..1efa0fc 100644 --- a/rag_system/clear_index.py +++ b/rag_system/clear_index.py @@ -1,9 +1,16 @@ -from rag_system.vectordb.azure_search import delete_all_documents +import os, logging +import coloredlogs + +from rag_system.vectordb.azure_search import delete_documents + + +logger: logging.Logger = logging.getLogger("clear_index") +coloredlogs.install(level="DEBUG") def main(): - print("Deleting documents...") - delete_all_documents() + company = os.getenv("CRAWLER_COMPANY_NAME") + delete_documents(filters=f"search.in(company, '{company}')") if __name__ == "__main__":