dh `SSKrSSKJrJrJrJrJr \R"\5r "SS5r g)N)AnyDictListOptionalUnionc \rSrSrSrSS\S\\S\4SjjrS\S \S \S \S \4 S jr S\ \ \\ 4S \4Sjr S\S \ \ \\ 44SjrS\ \S \ \ \\ 44SjrS\\\ \4S \\ \ \\ 44SjrSrg)DriaAPIWrapperaWrapper around Dria API. This wrapper facilitates interactions with Dria's vector search and retrieval services, including creating knowledge bases, inserting data, and fetching search results. Attributes: api_key: Your API key for accessing Dria. contract_id: The contract ID of the knowledge base to interact with. top_n: Number of top results to fetch for a search. Napi_key contract_idtop_nc(SSKJnJn XlXPlX lX0l U"UR S9Ul UR(a&URRUR5 gg![a [R S5 gf=f)Nr)DriaModelszDria is not installed. Please install Dria to use this wrapper. You can install Dria using the following command: pip install dria )r ) driarr ImportErrorloggererrorr modelsr r dria_client set_contract)selfr r r rrs `/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/utilities/dria_index.py__init__DriaAPIWrapper.__init__s  )  &  5       ) )$*:*: ;   LL   sA//BBname descriptioncategory embeddingreturncvURRXX2S9n[RSU35 XPlU$)zCreate a new knowledge base.)rrrrz Knowledge base created with ID: )rcreaterinfor )rrrrrr s rcreate_knowledge_base$DriaAPIWrapper.create_knowledge_base+sF&&--X.   6{mDE&dataclURRU5n[RSU35 U$)z$Insert data into the knowledge base.zData inserted: )r insert_textrr#)rr'responses r insert_dataDriaAPIWrapper.insert_data:s0##//5 ohZ01r&queryc|URRXRS9n[R SU35 U$)zPerform a text-based search.r zSearch results: )rsearchr rr#)rr-resultss rr0DriaAPIWrapper.search@s8""))%zz)B &wi01r&vectorc|URRXRS9n[R SU35 U$)zPerform a vector-based query.r/zVector query results: )rr-r rr#)rr3vector_query_resultss rquery_with_vector DriaAPIWrapper.query_with_vectorFs<#//55fJJ5O ,-A,BCD##r&c[U[5(aURU5$[U[5(a([ SU55(aUR U5$[ RS5 g)zMethod to handle both text-based searches and vector-based queries. Args: query: A string for text-based search or a list of floats for vector-based query. Returns: The search or query results from Dria. c3@# UHn[U[5v M g7f)N) isinstancefloat).0items r %DriaAPIWrapper.run..Xs,WQVZe-D-DQVszrInvalid query type. Please provide a string for text search or a list of floats for vector query.N)r:strr0listallr6rr)rr-s rrunDriaAPIWrapper.runLse eS ! !;;u% % t $ $,WQV,W)W)W))%0 0 LL4 r&)r r rrr )N )__name__ __module__ __qualname____firstlineno____doc__r@rintrr$rrrr+r0r;r6rrC__static_attributes__r&rr r s MO<<)1#>R5Sr&r ) loggingtypingrrrrr getLoggerrFrr rMr&rrQs*33   8 $XXr&