diff --git a/dspy/retrieve/retrieve.py b/dspy/retrieve/retrieve.py index 71b4f57349..f3a03e6ac8 100644 --- a/dspy/retrieve/retrieve.py +++ b/dspy/retrieve/retrieve.py @@ -36,7 +36,8 @@ def forward(self, query_or_queries: Union[str, List[str]], k: Optional[int] = No # print(queries) # TODO: Consider removing any quote-like markers that surround the query too. k = k if k is not None else self.k - passages = dsp.retrieveEnsemble(queries, k=k) + passages = dsp.settings.rm(queries=queries, k=k) + #passages = dsp.retrieveEnsemble(queries, k=k) return Prediction(passages=passages) # TODO: Consider doing Prediction.from_completions with the individual sets of passages (per query) too.