From d0f121e6623a66da35caf078c943e4a2bdd44c6e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 26 Oct 2023 15:25:54 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(store.py):=20add=20"metadata?= =?UTF-8?q?"=20field=20to=20the=20list=20of=20requested=20fields=20in=20th?= =?UTF-8?q?e=20list=5Fcomponents=20function=20to=20include=20metadata=20in?= =?UTF-8?q?=20the=20response=20=F0=9F=94=A7=20fix(schema.py):=20add=20"met?= =?UTF-8?q?adata"=20field=20to=20the=20ComponentResponse,=20ListComponentR?= =?UTF-8?q?esponse,=20DownloadComponentResponse,=20and=20StoreComponentCre?= =?UTF-8?q?ate=20models=20to=20support=20metadata=20in=20the=20store=20com?= =?UTF-8?q?ponent=20data=20=F0=9F=94=A7=20fix(service.py):=20add=20"metada?= =?UTF-8?q?ta"=20field=20to=20the=20list=20of=20requested=20fields=20in=20?= =?UTF-8?q?the=20StoreService=20class=20to=20include=20metadata=20in=20the?= =?UTF-8?q?=20query=20parameters?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/store.py | 1 + src/backend/langflow/services/store/schema.py | 4 ++++ src/backend/langflow/services/store/service.py | 1 + 3 files changed, 6 insertions(+) diff --git a/src/backend/langflow/api/v1/store.py b/src/backend/langflow/api/v1/store.py index dd97a9e33..e2632911f 100644 --- a/src/backend/langflow/api/v1/store.py +++ b/src/backend/langflow/api/v1/store.py @@ -83,6 +83,7 @@ def list_components( "tags.tags_id.name", "tags.tags_id.id", "count(liked_by)", + "metadata", ] result = store_service.query_components( store_api_Key, diff --git a/src/backend/langflow/services/store/schema.py b/src/backend/langflow/services/store/schema.py index 6127eb8f7..ceef1c9a3 100644 --- a/src/backend/langflow/services/store/schema.py +++ b/src/backend/langflow/services/store/schema.py @@ -29,6 +29,7 @@ class ComponentResponse(BaseModel): tags: Optional[List[int]] liked_by_count: Optional[int] parent: Optional[UUID] + metadata: Optional[dict] class ListComponentResponse(BaseModel): @@ -37,6 +38,7 @@ class ListComponentResponse(BaseModel): description: Optional[str] liked_by_count: Optional[int] is_component: Optional[bool] + metadata: Optional[dict] class DownloadComponentResponse(BaseModel): @@ -45,6 +47,7 @@ class DownloadComponentResponse(BaseModel): description: Optional[str] data: Optional[dict] is_component: Optional[bool] + metadata: Optional[dict] class StoreComponentCreate(BaseModel): @@ -54,3 +57,4 @@ class StoreComponentCreate(BaseModel): tags: Optional[List[str]] parent: Optional[UUID] is_component: Optional[bool] + metadata: Optional[dict] diff --git a/src/backend/langflow/services/store/service.py b/src/backend/langflow/services/store/service.py index 3bedcb63e..e71f1def1 100644 --- a/src/backend/langflow/services/store/service.py +++ b/src/backend/langflow/services/store/service.py @@ -174,6 +174,7 @@ class StoreService(Service): "tags.tags_id.name", "tags.tags_id.id", "count(liked_by)", + "metadata", ] else: params["filter"] = params["filter"] = json.dumps(