+#hfddlZddlmZmZmZddlZddlZddlmZGddejZ y)N)ListOptionalUnion)dotdictc`eZdZdfd Zddeeeefdeede jfdZ xZ S)YouRMct|||s*tjj ds t d|r||_ytjd|_y)N)k YDC_API_KEYzCYou must supply ydc_api_key or set environment variable YDC_API_KEY)super__init__osenvironget RuntimeError ydc_api_key)selfrr __class__s Q/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dspy/retrieve/you_rm.pyr zYouRM.__init__ sM 12::>>-#@de e *D !zz-8D query_or_queriesr returncZ||n |j}t|tr|gn|}g}|D]`}d|ji}t j d||j }|dd|D]}|dD]} |j| b|D cgc]} td| ic} Scc} w)aySearch with You.com for self.k top passages for query or queries Args: query_or_queries (Union[str, List[str]]): The query or queries to search for. k (Optional[int]): The number of context strings to return, if not already specified in self.k Returns: dspy.Prediction: An object containing the retrieved passages. Nz X-API-Keyz&https://api.ydc-index.io/search?query=)headershitssnippets long_text) r isinstancestrrrequestsrjsonappendr) rrr queriesdocsqueryrresultshitsnippetdocuments rforwardz YouRM.forwardsADFF*C0 !   )E"D$4$45Gll8@df vr* )":)GKK() )  )BFFXh/0FFFsB()N)N) __name__ __module__ __qualname__r rrrrintdspy Predictionr* __classcell__)rs@rrr s>9Gc49n(=G(3-G[_[j[jGrr) rtypingrrrr r0 dsp.utilsrRetrieverrrr7s) (( &GDMM&Gr