diff --git a/api/controllers/console/app/statistic.py b/api/controllers/console/app/statistic.py index 37c3500448..742e940873 100644 --- a/api/controllers/console/app/statistic.py +++ b/api/controllers/console/app/statistic.py @@ -59,7 +59,9 @@ class DailyConversationStatistic(Resource): arg_dict['end'] = end_datetime_utc sql_query += ' GROUP BY date order by date' - rs = db.session.execute(sql_query, arg_dict) + + with db.engine.begin() as conn: + rs = conn.execute(db.text(sql_query), arg_dict) response_date = [] @@ -119,7 +121,9 @@ class DailyTerminalsStatistic(Resource): arg_dict['end'] = end_datetime_utc sql_query += ' GROUP BY date order by date' - rs = db.session.execute(sql_query, arg_dict) + + with db.engine.begin() as conn: + rs = conn.execute(db.text(sql_query), arg_dict) response_date = [] @@ -180,7 +184,9 @@ class DailyTokenCostStatistic(Resource): arg_dict['end'] = end_datetime_utc sql_query += ' GROUP BY date order by date' - rs = db.session.execute(sql_query, arg_dict) + + with db.engine.begin() as conn: + rs = conn.execute(db.text(sql_query), arg_dict) response_date = [] diff --git a/api/controllers/web/wraps.py b/api/controllers/web/wraps.py index c68b8f1cf2..9321c427c2 100644 --- a/api/controllers/web/wraps.py +++ b/api/controllers/web/wraps.py @@ -16,7 +16,7 @@ def validate_token(view=None): def decorated(*args, **kwargs): site = validate_and_get_site() - app_model = db.session.query(App).get(site.app_id) + app_model = db.session.query(App).filter(App.id == site.app_id).first() if not app_model: raise NotFound()