From 07cda09aa82e3ce6295b93a253c6eb4d5cc13210 Mon Sep 17 00:00:00 2001 From: Julian Risch Date: Tue, 21 Nov 2023 11:27:10 +0100 Subject: [PATCH] docs: Include TextEmbedder in DocumentJoiner usage example (#6369) * docs: Include TextEmbedder in DocumentJoiner usage example * black --- haystack/preview/components/routers/document_joiner.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/haystack/preview/components/routers/document_joiner.py b/haystack/preview/components/routers/document_joiner.py index 1a966c209a..96b9b989b2 100644 --- a/haystack/preview/components/routers/document_joiner.py +++ b/haystack/preview/components/routers/document_joiner.py @@ -30,10 +30,18 @@ class DocumentJoiner: document_store = InMemoryDocumentStore() p = Pipeline() p.add_component(instance=InMemoryBM25Retriever(document_store=document_store), name="bm25_retriever") + p.add_component( + instance=SentenceTransformersTextEmbedder(model_name_or_path="sentence-transformers/all-MiniLM-L6-v2"), + name="text_embedder", + ) p.add_component(instance=InMemoryEmbeddingRetriever(document_store=document_store), name="embedding_retriever") p.add_component(instance=DocumentJoiner(), name="joiner") p.connect("bm25_retriever", "joiner") p.connect("embedding_retriever", "joiner") + p.connect("text_embedder", "embedding_retriever") + query = "What is the capital of France?" + p.run(data={"bm25_retriever": {"query": query}, + "text_embedder": {"text": query}}) ``` """