+#hy ddlmZmZddlZddlmZ ddlmZGddejZ y#e$r e dwxYw))OptionalUnionNdotdict)RAGPretrainedModelzXYou need to install RAGAtouille library, Please use the command: pip install ragatouillecpeZdZdZd dededeffd Zdeeeefde ede jfdZ xZ S) RAGatouilleRMaA retrieval model that uses RAGatouille library to return the top passages for a given query. Assumes that you already have an index created with RAGatouille. Reference: https://github.com/bclavie/RAGatouille Args: index_root (str): Folder path where you index is stored. index_name (str): Name of the index you want to retrieve from. k (int, optional): The default number of passages to retrieve. Defaults to 3. Examples: Below is a code snippet that shows how to use RAGatouille index as the default retriver: ```python llm = dspy.OpenAI(model="gpt-3.5-turbo") rm = RAGatouilleRM(index_root="ragatouille/colbert/indexes", index_name="my_index") dspy.settings.configure(lm=llm, rm=rm) ``` index_root index_namekct|||d||_ tj|j|_y#t $r}t d|d}~wwxYw)N)r /) index_pathzIndex not found: )super__init__rr from_indexmodelFileNotFoundError)selfr r r e __class__s Y/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dspy/retrieve/ragatouille_rm.pyrzRAGatouilleRM.__init__"sc 1'L*6 =+66$//RDJ  =#&7s$;< < =s%A A!AA!query_or_queriesreturnc"||n |j}t|tr|gn|}|Dcgc]}|s| }}g}|D]7}|jj ||}|j d|D9t j|Scc}w)amSearch with RAGAtouille based index for self.k top passages for query Args: query_or_queries (Union[str, List[str]]): The query or queries to search for. k (Optional[int]): The number of top passages to retrieve. Defaults to self.k. Returns: dspy.Prediction: An object containing the retrieved passages. )queryr c3<K|]}td|diyw) long_textcontentNr).0ds r z(RAGatouilleRM.forward..=sRQG[!I,$?@Rs)passages)r isinstancestrrsearchextenddspy Prediction)rrr queriesqr#rresultss rforwardzRAGatouilleRM.forward*sADFF*C0 !  &+1++ SEjj''eq'9G OOR'R R S11 ,s B B ))__name__ __module__ __qualname____doc__r%intrrlistrr(r)r- __classcell__)rs@rr r sT&=3=C=C=2c49n(=2(3-2TXTcTc2r ) typingrrr(dsp.utils.utilsr ragatouiller ImportError ExceptionRetriever r6rr>sG" #.02DMM02  b s0>