diff --git a/api/services/datasource_provider_service.py b/api/services/datasource_provider_service.py index 12807a41f04dd0..cd796d2d7670a0 100644 --- a/api/services/datasource_provider_service.py +++ b/api/services/datasource_provider_service.py @@ -433,7 +433,7 @@ def is_system_oauth_params_exist(self, datasource_provider_id: DatasourceProvide """ check if system oauth params exist """ - with Session(db.engine).no_autoflush as session: + with Session(db.engine) as session: return ( session.scalar( select(DatasourceOauthParamConfig) @@ -510,7 +510,7 @@ def get_oauth_client(self, tenant_id: str, datasource_provider_id: DatasourcePro """ provider = datasource_provider_id.provider_name plugin_id = datasource_provider_id.plugin_id - with Session(db.engine).no_autoflush as session: + with Session(db.engine) as session: # get tenant oauth client params tenant_oauth_client_params = session.scalar( select(DatasourceOauthTenantParamConfig)