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

🐛 fix(schema.py): handle empty list case in tags_to_list method to prevent potential errors

Browse source 
🐛 fix(service.py): add validation for missing API key when filter_by_user is True to avoid unauthorized access
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-30 19:50:39 -03:00
commit 127721e986
2 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/services/store/schema.py
View file

@ -53,6 +53,8 @@ class ListComponentResponse(BaseModel):
def tags_to_list(cls, v): def tags_to_list(cls, v):
# Check if all values are have id and name # Check if all values are have id and name
# if so, return v else transform to TagResponse # if so, return v else transform to TagResponse
if not v:
return v
if all(["id" in tag and "name" in tag for tag in v]): if all(["id" in tag and "name" in tag for tag in v]):
return v return v
else: else:

4
src/backend/langflow/services/store/service.py
View file

@ -168,8 +168,10 @@ class StoreService(Service):
# Only public components or the ones created by the user # Only public components or the ones created by the user
# check for "public" or "Public" # check for "public" or "Public"
if filter_by_user and not api_key:
raise ValueError("No API key provided")
if filter_by_user: if filter_by_user and api_key:
user_data = self._get( user_data = self._get(
f"{self.base_url}/users/me", api_key, params={"fields": "id"} f"{self.base_url}/users/me", api_key, params={"fields": "id"}
) )