ref: Move get_settings_service out of Depends() (#4239)

Move get_settings_service out of Depends()
This commit is contained in:
Christophe Bornet
2024-10-22 18:55:40 +02:00
committed by GitHub
parent 6e39dedaae
commit b6f74fb275

View File

@ -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