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

Handle 401 Unauthorized error in get_components

Browse source 
function
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-16 10:57:02 -03:00
commit 0547dd5d76
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -142,6 +142,8 @@ async def get_components(
except HTTPStatusError as exc: except HTTPStatusError as exc:
if exc.response.status_code == 403: if exc.response.status_code == 403:
raise ValueError("You are not authorized to access this public resource") raise ValueError("You are not authorized to access this public resource")
elif exc.response.status_code == 401:
raise ValueError("You are not authorized to access this resource. Please check your API key.")
if store_api_Key and result: if store_api_Key and result:
# Now, from the result, we need to get the components # Now, from the result, we need to get the components
@ -161,6 +163,8 @@ async def get_components(
if exc.response.status_code == 403: if exc.response.status_code == 403:
raise HTTPException(status_code=403, detail="Forbidden") raise HTTPException(status_code=403, detail="Forbidden")
elif isinstance(exc, ValueError): elif isinstance(exc, ValueError):
if "Check your API key" in str(exc):
raise HTTPException(status_code=401, detail=str(exc))
raise HTTPException(status_code=403, detail=str(exc)) raise HTTPException(status_code=403, detail=str(exc))
raise HTTPException(status_code=500, detail=str(exc)) raise HTTPException(status_code=500, detail=str(exc))