From cf4a526e7f7321c965948ee71fd6ccdff07d18be Mon Sep 17 00:00:00 2001 From: Harry Date: Mon, 11 Aug 2025 20:35:46 +0800 Subject: [PATCH] refactor: replace db.session with session in DatasourceProviderService for consistency --- api/services/datasource_provider_service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/services/datasource_provider_service.py b/api/services/datasource_provider_service.py index e36d09110f..b0ef1f75bb 100644 --- a/api/services/datasource_provider_service.py +++ b/api/services/datasource_provider_service.py @@ -135,7 +135,7 @@ class DatasourceProviderService: datasource_provider=datasource_provider, ) datasource_provider.expires_at = refreshed_credentials.expires_at - db.session.commit() + session.commit() return self.decrypt_datasource_provider_credentials( tenant_id=tenant_id, @@ -580,8 +580,8 @@ class DatasourceProviderService: auth_type=CredentialType.API_KEY.value, encrypted_credentials=credentials, ) - db.session.add(datasource_provider) - db.session.commit() + session.add(datasource_provider) + session.commit() def extract_secret_variables(self, tenant_id: str, provider_id: str, credential_type: CredentialType) -> list[str]: """