Skip to content

Commit 6e6301e

Browse files
authored
Add embed_documents for ParallelRetriever (#7025)
1 parent 405b64c commit 6e6301e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pipelines/pipelines/nodes/retriever/parallel_retriever.py

+3
Original file line numberDiff line numberDiff line change
@@ -337,3 +337,6 @@ def run_indexing(self, documents: List[dict], **kwargs):
337337

338338
def embed_queries(self, texts: List[str], **kwargs) -> List[np.ndarray]:
339339
return run_main_query(texts, self.url)
340+
341+
def embed_documents(self, docs: List[Document], **kwargs):
342+
return run_main_query([d.content for d in docs], self.url)

0 commit comments

Comments
 (0)