From b6f74fb2751050effcea29d29b51daeda416d8bf Mon Sep 17 00:00:00 2001 From: Christophe Bornet Date: Tue, 22 Oct 2024 18:55:40 +0200 Subject: [PATCH] ref: Move get_settings_service out of Depends() (#4239) Move get_settings_service out of Depends() --- src/backend/base/langflow/api/v1/store.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/src/backend/base/langflow/api/v1/store.py b/src/backend/base/langflow/api/v1/store.py index 188c69eaf..33b11b7fd 100644 --- a/src/backend/base/langflow/api/v1/store.py +++ b/src/backend/base/langflow/api/v1/store.py @@ -20,26 +20,20 @@ from langflow.services.store.schema import ( router = APIRouter(prefix="/store", tags=["Components Store"]) -def get_user_store_api_key( - user: CurrentActiveUser, - settings_service=Depends(get_settings_service), -): +def get_user_store_api_key(user: CurrentActiveUser): if not user.store_api_key: raise HTTPException(status_code=400, detail="You must have a store API key set.") try: - return auth_utils.decrypt_api_key(user.store_api_key, settings_service) + return auth_utils.decrypt_api_key(user.store_api_key, get_settings_service()) except Exception as e: raise HTTPException(status_code=500, detail="Failed to decrypt API key. Please set a new one.") from e -def get_optional_user_store_api_key( - user: CurrentActiveUser, - settings_service=Depends(get_settings_service), -): +def get_optional_user_store_api_key(user: CurrentActiveUser): if not user.store_api_key: return None try: - return auth_utils.decrypt_api_key(user.store_api_key, settings_service) + return auth_utils.decrypt_api_key(user.store_api_key, get_settings_service()) except Exception: # noqa: BLE001 logger.exception("Failed to decrypt API key") return user.store_api_key