Skip to content

Commit cb52efb

Browse files
fix: avoid double evaluation of embeddings queryset
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
1 parent 6e5538d commit cb52efb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

scripts/cleanup_vespa_db.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,14 @@ def cleanup_stale_cache():
1919
vespa = get_vespa_app()
2020

2121
while True:
22-
stale_files = EmbeddedFile.objects.prefetch_related(
22+
stale_qs = EmbeddedFile.objects.prefetch_related(
2323
"embeddings_references"
2424
).filter(
2525
updated_at__lt=timezone.now() - timedelta(days=STALENESS_THRESHOLD_DAYS)
26-
)[:BATCH_SIZE]
26+
).order_by("updated_at")[:BATCH_SIZE]
27+
stale_files = list(stale_qs)
2728
if not stale_files:
2829
break
29-
3030
docs_to_delete = (
3131
{"id": ref.vespa_doc_id}
3232
for ef in stale_files

0 commit comments

Comments
 (0)