From 1935029f664607bc8c238e454a3f867bd28dd6f6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:23:36 -0300 Subject: [PATCH] refactor: Update EmbedComponent build method return type to Data --- src/backend/base/langflow/components/experimental/Embed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/experimental/Embed.py b/src/backend/base/langflow/components/experimental/Embed.py index e99ab0d03..a67c6c4cf 100644 --- a/src/backend/base/langflow/components/experimental/Embed.py +++ b/src/backend/base/langflow/components/experimental/Embed.py @@ -9,7 +9,7 @@ class EmbedComponent(CustomComponent): def build_config(self): return {"texts": {"display_name": "Texts"}, "embbedings": {"display_name": "Embeddings"}} - def build(self, texts: list[str], embbedings: Embeddings) -> Embeddings: + def build(self, texts: list[str], embbedings: Embeddings) -> Data: vectors = Data(vector=embbedings.embed_documents(texts)) self.status = vectors return vectors