Tai Truong
fix readme
d202ada
raw
history blame contribute delete
532 Bytes
from langflow.custom import CustomComponent
from langflow.field_typing import Embeddings
from langflow.schema import Data
class EmbedComponent(CustomComponent):
display_name = "Embed Texts"
name = "Embed"
def build_config(self):
return {"texts": {"display_name": "Texts"}, "embbedings": {"display_name": "Embeddings"}}
def build(self, texts: list[str], embbedings: Embeddings) -> Data:
vectors = Data(vector=embbedings.embed_documents(texts))
self.status = vectors
return vectors