Spaces:
Running
Running
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 | |