tj dddlZddlmZmZmZmZmZejeZ GddZ dS)N)AnyDictListOptionalUnionc 6eZdZdZ ddedeedefdZded ed ed ed ef d Zde e ee fd efdZ ded e e ee ffdZ de ed e e ee ffdZdeee efd ee e ee ffdZdS)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. N api_key contract_idtop_nc$ ddlm}m}n+#t$rtdYdSwxYw||_||_||_||_ ||j|_ |jr!|j |jdSdS)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 f/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/utilities/dria_index.py__init__zDriaAPIWrapper.__init__s  ) ) ) ) ) ) ) ) )    LL    FF   & 4 555   <   ) )$*: ; ; ; ; ; < z%DriaAPIWrapper.run..Xs,,W,WZe-D-D,W,W,W,W,W,Wr$zrInvalid query type. Please provide a string for text search or a list of floats for vector query.N)r5strr-listallr2rr)rr*s rrunzDriaAPIWrapper.runLs eS ! ! ;;u%% % t $ $ ,W,WQV,W,W,W)W)W ))%00 0 LL4   4r$)Nr )__name__ __module__ __qualname____doc__r:rintrr#rrrr)r-r6r2rr=r$rr r sq  MO<<<)1#>R5Sr$r ) loggingtypingrrrrr getLoggerr>rr rCr$rrGs33333333333333  8 $ $XXXXXXXXXXr$