From 4a214661be5b4a35a5e694108135de6e28e950a1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 16 Nov 2023 19:21:19 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(schema.py):=20remove=20unuse?= =?UTF-8?q?d=20downloads=5Fcount=20field=20from=20DownloadComponentRespons?= =?UTF-8?q?e=20class=20=F0=9F=94=A7=20fix(service.py):=20handle=20case=20w?= =?UTF-8?q?hen=20component=20is=20returned=20as=20a=20list=20instead=20of?= =?UTF-8?q?=20a=20dictionary=20in=20download=20method?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/store/schema.py | 3 +-- src/backend/langflow/services/store/service.py | 4 +++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/services/store/schema.py b/src/backend/langflow/services/store/schema.py index e07e7d384..89eb7a094 100644 --- a/src/backend/langflow/services/store/schema.py +++ b/src/backend/langflow/services/store/schema.py @@ -60,8 +60,7 @@ class DownloadComponentResponse(BaseModel): description: Optional[str] data: Optional[dict] is_component: Optional[bool] - metadata: Optional[dict] - downloads_count: Optional[int] + metadata: Optional[dict] = {} class StoreComponentCreate(BaseModel): diff --git a/src/backend/langflow/services/store/service.py b/src/backend/langflow/services/store/service.py index acf6efcd0..b79731cd2 100644 --- a/src/backend/langflow/services/store/service.py +++ b/src/backend/langflow/services/store/service.py @@ -298,11 +298,13 @@ class StoreService(Service): async def download(self, api_key: str, component_id: UUID) -> DownloadComponentResponse: url = f"{self.components_url}/{component_id}" - params = {"fields": ",".join(["id", "name", "description", "data", "is_component"])} + params = {"fields": ",".join(["id", "name", "description", "data", "is_component", "metadata"])} if not self.download_webhook_url: raise ValueError("DOWNLOAD_WEBHOOK_URL is not set") component, _ = await self._get(url, api_key, params) await self.call_webhook(api_key, self.download_webhook_url, component_id) + if isinstance(component, list): + component = component[0] return DownloadComponentResponse(**component)