update you_rm to return an iterable of predictions, in line with othe…

dsp/primitives/search.py

@@ -1,4 +1,7 @@
+from collections.abc import Iterable
+
 import numpy as np
+
 import dsp
 
 
@@ -7,6 +10,10 @@ def retrieve(query: str, k: int, **kwargs) -> list[str]:
     if not dsp.settings.rm:
         raise AssertionError("No RM is loaded.")
     passages = dsp.settings.rm(query, k=k, **kwargs)
+    if not isinstance(passages, Iterable):
+        # it's not an iterable yet; make it one.
+        # TODO: we should unify the type signatures of dspy.Retriever
+        passages = [passages]
     passages = [psg.long_text for psg in passages]
 
     if dsp.settings.reranker:

dspy/retrieve/you_rm.py

@@ -1,6 +1,7 @@
 import dspy
 import os
 import requests
+from dsp.utils import dotdict
 
 from typing import Union, List, Optional
 
@@ -43,4 +44,4 @@ def forward(self, query_or_queries: Union[str, List[str]], k: Optional[int] = No
             for hit in results["hits"][:k]:
                 for snippet in hit["snippets"]:
                     docs.append(snippet)
-        return dspy.Prediction(passages=docs)
+        return [dotdict({"long_text": document}) for document in docs]