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

Refactor store.py to check if api_key is valid

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-16 17:19:14 -03:00
commit 7c27cceb98
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/api/v1/store.py
View file

@ -4,7 +4,6 @@ from uuid import UUID
from fastapi import APIRouter, Depends, HTTPException, Query from fastapi import APIRouter, Depends, HTTPException, Query
from httpx import HTTPStatusError from httpx import HTTPStatusError
from langflow.services.auth import utils as auth_utils from langflow.services.auth import utils as auth_utils
from langflow.services.database.models.user.user import User from langflow.services.database.models.user.user import User
from langflow.services.deps import get_settings_service, get_store_service from langflow.services.deps import get_settings_service, get_store_service
@ -57,14 +56,14 @@ async def check_if_store_has_api_key(
store_service: StoreService = Depends(get_store_service), store_service: StoreService = Depends(get_store_service),
): ):
if api_key is None: if api_key is None:
return {"has_api_key": False} return {"has_api_key": False, "is_valid": False}
try: try:
is_valid = await store_service.check_api_key(api_key) is_valid = await store_service.check_api_key(api_key)
except Exception as e: except Exception as e:
raise HTTPException(status_code=400, detail=str(e)) raise HTTPException(status_code=400, detail=str(e))
return {"has_api_key": is_valid} return {"has_api_key": api_key is not None, "is_valid": is_valid}
@router.post("/components/", response_model=CreateComponentResponse, status_code=201) @router.post("/components/", response_model=CreateComponentResponse, status_code=201)