dh bSrSSKJrJrJr SSKJr SSKJr SSK J r SSK J r "SS\ 5r g ) zWrapper around Dria Retriever.)AnyListOptional)CallbackManagerForRetrieverRun)Document) BaseRetriever)DriaAPIWrapperc ^\rSrSr%Sr\\S'SS\S\\S\ 4U4Sjjjr SS \S \S \S \S \4 Sjjr S\ S S4Sjr S\S\S \ \4SjrSrU=r$) DriaRetriever z*`Dria` retriever using the DriaAPIWrapper. api_wrapperNapi_key contract_idkwargsc <>[XS9n[TU]"SSU0UD6 g)z Initialize the DriaRetriever with a DriaAPIWrapper instance. Args: api_key: The API key for Dria. contract_id: The contract ID of the knowledge base to interact with. )rrr N)r super__init__)selfrrrr __class__s a/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/retrievers/dria_index.pyrDriaRetriever.__init__s$%WN  ;[;F;name descriptioncategory embeddingreturnc>URRXX45nU$)a7Create a new knowledge base in Dria. Args: name: The name of the knowledge base. description: The description of the knowledge base. category: The category of the knowledge base. embedding: The embedding model to use for the knowledge base. Returns: The ID of the created knowledge base. )r create_knowledge_base)rrrrrresponses rr #DriaRetriever.create_knowledge_bases&&##99 x rtextscvUVs/sH o"SUSS.PM nnURRU5 gs snf)zAdd texts to the Dria knowledge base. Args: texts: An iterable of texts and metadatas to add to the knowledge base. Returns: List of IDs representing the added texts. textmetadata)r%r&N)r insert_data)rr#r%datas r add_textsDriaRetriever.add_texts4sBRWWQVf4 3CDQVW $$T*Xs6query run_managerc URRU5nUVs/sHn[USUSUSS.S9PM nnU$s snf)zRetrieve relevant documents from Dria based on a query. Args: query: The query string to search for in the knowledge base. run_manager: Callback manager for the retriever run. Returns: A list of Documents containing the search results. r&idscore)r.r/) page_contentr&)r searchr)rr+r,resultsresultdocss r_get_relevant_documents%DriaRetriever._get_relevant_documentsCsi""))%0 "  " #J/ &t vgG "    sAr)N) Unspecifiedjina)__name__ __module__ __qualname____firstlineno____doc__r __annotations__strrrrr rr)rrr5__static_attributes__ __classcell__)rs@rr r s4 < <(3- rGs($&&C-38KMKr