fix: update database query and model definitions (#14415)

2025-02-26 18:45:12 +08:00 · 2025-02-26 18:45:12 +08:00 · 9a417bfc5e
commit 9a417bfc5e
parent 90bc51ed2e
2 changed files with 7 additions and 7 deletions
--- a/api/controllers/service_api/wraps.py
+++ b/api/controllers/service_api/wraps.py
@ -154,7 +154,7 @@ def validate_dataset_token(view=None):
            )  # TODO: only owner information is required, so only one is returned.
            if tenant_account_join:
                tenant, ta = tenant_account_join
-                account = Account.query.filter_by(id=ta.account_id).first()
+                account = db.session.query(Account).filter(Account.id == ta.account_id).first()
                # Login admin
                if account:
                    account.current_tenant = tenant