clear_embeddings.py

from models.db.db_setup import get_db
from models.contextual_response.database import DocumentEmbedding

def clear_embeddings():
    with get_db() as db:
        db.query(DocumentEmbedding).delete()
        db.commit()
        print("Successfully cleared all document embeddings")

if __name__ == "__main__":
    clear_embeddings()