update clean embedding cache query logic (#6483)

2024-07-20 01:29:25 +08:00
parent 27e08a8e2e
commit 1e0e573165
4 changed files with 46 additions and 6 deletions
--- a/api/models/dataset.py
+++ b/api/models/dataset.py
@@ -630,7 +630,8 @@ class Embedding(db.Model):
    __tablename__ = 'embeddings'
    __table_args__ = (
        db.PrimaryKeyConstraint('id', name='embedding_pkey'),
-        db.UniqueConstraint('model_name', 'hash', 'provider_name', name='embedding_hash_idx')
+        db.UniqueConstraint('model_name', 'hash', 'provider_name', name='embedding_hash_idx'),
+        db.Index('created_at_idx', 'created_at')
    )

    id = db.Column(StringUUID, primary_key=True, server_default=db.text('uuid_generate_v4()'))