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

🔧 refactor(store.py): remove duplicate fields list and use default_fields from StoreService class

Browse source 
🔧 refactor(service.py): move default_fields list to StoreService class and use it in query_components method
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-27 13:07:09 -03:00
commit 6f3172bc33
2 changed files with 17 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -92,23 +92,10 @@ def list_components(
store_api_Key: str = Depends(get_optional_user_store_api_key),
):
try:
fields = [
"id",
"name",
"description",
"user_created.first_name",
"is_component",
"tags.tags_id.name",
"tags.tags_id.id",
"count(liked_by)",
"count(downloads)",
"metadata",
]
result = store_service.query_components(
store_api_Key,
page,
limit,
fields=fields,
api_key=store_api_Key,
page=page,
limit=limit,
filter_by_user=filter_by_user,
)

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

@ -30,6 +30,18 @@ class StoreService(Service):
self.download_webhook_url = self.settings_service.settings.DOWNLOAD_WEBHOOK_URL
self.like_webhook_url = self.settings_service.settings.LIKE_WEBHOOK_URL
self.components_url = f"{self.base_url}/items/components"
self.default_fields = [
"id",
"name",
"description",
"user_created.first_name",
"is_component",
"tags.tags_id.name",
"tags.tags_id.id",
"count(liked_by)",
"count(downloads)",
"metadata",
]
def _get(
self, url: str, api_key: str, params: Dict[str, Any] = None
@ -149,24 +161,8 @@ class StoreService(Service):
) -> Union[List[ListComponentResponse], List[Dict[str, int]]]:
params = {"page": page, "limit": limit}
# ?aggregate[count]=likes
params["fields"] = (
",".join(fields)
if fields
else ",".join(
[
"id",
"name",
"description",
"user_created.first_name",
"is_component",
"tags.tags_id.name",
"tags.tags_id.id",
"count(liked_by)",
"count(downloads)",
"metadata",
]
)
)
params["fields"] = ",".join(fields) if fields else ",".join(self.default_fields)
# Only public components or the ones created by the user
# check for "public" or "Public"