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