chore: fix legacy API usages of Query.get() by Session.get() in SqlAlchemy 2 (#6340)

2024-07-17 13:54:35 +08:00
parent 7c397f5722
commit 7943f7f697
6 changed files with 28 additions and 28 deletions
--- a/api/controllers/console/datasets/datasets_segments.py
+++ b/api/controllers/console/datasets/datasets_segments.py
@@ -75,7 +75,7 @@ class DatasetDocumentSegmentListApi(Resource):
        )

        if last_id is not None:
-            last_segment = DocumentSegment.query.get(str(last_id))
+            last_segment = db.session.get(DocumentSegment, str(last_id))
            if last_segment:
                query = query.filter(
                    DocumentSegment.position > last_segment.position)