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

refactor: Update EmbedComponent build method return type to Data

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:23:36 -03:00
commit 1935029f66
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/components/experimental/Embed.py
View file

@ -9,7 +9,7 @@ class EmbedComponent(CustomComponent):
def build_config(self): def build_config(self):
return {"texts": {"display_name": "Texts"}, "embbedings": {"display_name": "Embeddings"}} 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)) vectors = Data(vector=embbedings.embed_documents(texts))
self.status = vectors self.status = vectors
return vectors return vectors