fix: replace all dataset.Model.query to db.session.query(Model) (#19509)

2025-05-12 13:52:33 +08:00 · 2025-05-12 13:52:33 +08:00 · b00f94df64
commit b00f94df64
parent 49af07f444
21 changed files with 430 additions and 265 deletions
--- a/api/tasks/deal_dataset_vector_index_task.py
+++ b/api/tasks/deal_dataset_vector_index_task.py
@ -24,7 +24,7 @@ def deal_dataset_vector_index_task(dataset_id: str, action: str):
    start_at = time.perf_counter()

    try:
-        dataset = Dataset.query.filter_by(id=dataset_id).first()
+        dataset = db.session.query(Dataset).filter_by(id=dataset_id).first()

        if not dataset:
            raise Exception("Dataset not found")