+#h ddlmZddlmZmZmZddlZddlmZ ddl Z ddl m Z GddejZy#e $r e dwxYw) ) defaultdict)ListOptionalUnionN)dotdict) QdrantClientz]The 'qdrant' extra is required to use QdrantRM. Install it with `pip install dspy-ai[qdrant]`cteZdZdZ d dededeffd Zd deee efde ede jfdZ xZS) QdrantRMaX A retrieval module that uses Qdrant to return the top passages for a given query. Assumes that a Qdrant collection has been created and populated with the following payload: - document: The text of the passage Args: qdrant_collection_name (str): The name of the Qdrant collection. qdrant_client (QdrantClient): A QdrantClient instance. k (int, optional): The default number of top passages to retrieve. Defaults to 3. Examples: Below is a code snippet that shows how to use Qdrant as the default retriver: ```python from qdrant_client import QdrantClient llm = dspy.OpenAI(model="gpt-3.5-turbo") qdrant_client = QdrantClient() retriever_model = QdrantRM("my_collection_name", qdrant_client=qdrant_client) dspy.settings.configure(lm=llm, rm=retriever_model) ``` Below is a code snippet that shows how to use Qdrant in the forward() function of a module ```python self.retrieve = QdrantRM("my_collection_name", qdrant_client=qdrant_client, k=num_passages) ``` qdrant_collection_name qdrant_clientkcB||_||_t| |y)N)r )_qdrant_collection_name_qdrant_clientsuper__init__)selfr r r __class__s T/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dspy/retrieve/qdrant_rm.pyrzQdrantRM.__init__-s% (>$+ 1query_or_queriesreturnc t|tr|gn|}|Dcgc]}|s| }}||n |j}|jj|j f||d|}t t}|D]*}|D]#} || jxx| jz cc<%,t|jddd|} | D cgc]\} } td| i} } } | Scc}wcc} } w)a\Search with Qdrant 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. N) query_textslimitc |dS)N)xs rz"QdrantRM.forward..Ts 1Q4rT)keyreverse long_text) isinstancestrr r query_batchrrfloatdocumentscoresorteditemsr)rrr kwargsqueriesq batch_resultspassages_scoresbatchresultsorted_passagespassage_passagess rforwardzQdrantRM.forward8s*C0 !  &+1++ADFF7++77  ( (Q6=QQIOQ &e," AE A0FLL@0 A A !  ! ! #GGIKGVV G['23VV',"WsC!C!C&))N)__name__ __module__ __qualname____doc__r%rintrrrrdspy Predictionr7 __classcell__)rs@rr r sa@  # $   !c49n(=!(3-!dhdsds!rr ) collectionsrtypingrrrr> dsp.utilsr fastembedr r ImportErrorRetriever rrrrGsR#(( *It}}I  g s <A