jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source 
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
commit b6f74fb275
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/base/langflow/api/v1/store.py
View file

@ -20,26 +20,20 @@ from langflow.services.store.schema import (
router = APIRouter(prefix="/store", tags=["Components Store"]) router = APIRouter(prefix="/store", tags=["Components Store"])
def get_user_store_api_key( def get_user_store_api_key(user: CurrentActiveUser):
user: CurrentActiveUser,
settings_service=Depends(get_settings_service),
):
if not user.store_api_key: if not user.store_api_key:
raise HTTPException(status_code=400, detail="You must have a store API key set.") raise HTTPException(status_code=400, detail="You must have a store API key set.")
try: 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: except Exception as e:
raise HTTPException(status_code=500, detail="Failed to decrypt API key. Please set a new one.") from 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( def get_optional_user_store_api_key(user: CurrentActiveUser):
user: CurrentActiveUser,
settings_service=Depends(get_settings_service),
):
if not user.store_api_key: if not user.store_api_key:
return None return None
try: 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 except Exception: # noqa: BLE001
logger.exception("Failed to decrypt API key") logger.exception("Failed to decrypt API key")
return user.store_api_key return user.store_api_key