estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Bowen Liang
2024-07-17 13:54:35 +08:00
committed by GitHub
parent 7c397f5722
commit 7943f7f697
6 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/models/dataset.py
View File

@@ -68,7 +68,7 @@ class Dataset(db.Model):
@property
def created_by_account(self):
return Account.query.get(self.created_by)
return db.session.get(Account, self.created_by)
@property
def latest_process_rule(self):
@@ -336,7 +336,7 @@ class Document(db.Model):
@property
def dataset_process_rule(self):
if self.dataset_process_rule_id:
return DatasetProcessRule.query.get(self.dataset_process_rule_id)
return db.session.get(DatasetProcessRule, self.dataset_process_rule_id)
return None
@property
@@ -560,7 +560,7 @@ class AppDatasetJoin(db.Model):
@property
def app(self):
return App.query.get(self.app_id)
return db.session.get(App, self.app_id)
class DatasetQuery(db.Model):