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

🔍 refactor(store.py): add support for filtering components based on is_component flag in list_components function

Browse source 
🔍 refactor(service.py): add support for filtering components based on is_component flag in get_components function
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-02 09:34:02 -03:00
commit 0c90c12c94
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -75,6 +75,7 @@ def list_components(
filter_by_user: bool = Query(False), filter_by_user: bool = Query(False),
page: int = 1, page: int = 1,
limit: int = 10, limit: int = 10,
is_component: Optional[bool] = Query(None),
store_service: StoreService = Depends(get_store_service), store_service: StoreService = Depends(get_store_service),
store_api_Key: Optional[str] = Depends(get_optional_user_store_api_key), store_api_Key: Optional[str] = Depends(get_optional_user_store_api_key),
): ):
@ -85,6 +86,7 @@ def list_components(
page=page, page=page,
limit=limit, limit=limit,
filter_by_user=filter_by_user, filter_by_user=filter_by_user,
is_component=is_component,
) )
if not store_api_Key: if not store_api_Key:

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

@ -188,15 +188,18 @@ class StoreService(Service):
page: int = 1, page: int = 1,
limit: int = 15, limit: int = 15,
fields: Optional[List[str]] = None, fields: Optional[List[str]] = None,
is_component: Optional[bool] = None,
filter_by_user: bool = False, filter_by_user: bool = False,
) -> Union[List[ListComponentResponse], List[Dict[str, int]]]: ) -> Union[List[ListComponentResponse], List[Dict[str, int]]]:
params = {"page": page, "limit": limit} params = {"page": page, "limit": limit}
# ?aggregate[count]=likes # ?aggregate[count]=likes
params["fields"] = ",".join(fields) if fields else ",".join(self.default_fields) params["fields"] = ",".join(fields) if fields else ",".join(self.default_fields)
if is_component:
params["filter[is_component][_eq]"] = is_component
# 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: if filter_by_user and not api_key:
raise ValueError("No API key provided") raise ValueError("No API key provided")