a bgY@sddlmZmZmZmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZddlmZmZGdd d e Zd S) )AnyDictListOptional)CallbackManagerForRetrieverRun)Document) BaseRetriever)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDict SecretStr) ArceeWrapper DALMFiltercseZdZUdZdZeeed<eed<e ed<dZ e ed<dZ e ed <d Z e ed <dZ eee efed <ed dZeddfdd ZeeedddZe eeeedddZZS)ArceeRetrieveraArcee Domain Adapted Language Models (DALMs) retriever. To use, set the ``ARCEE_API_KEY`` environment variable with your Arcee API key, or pass ``arcee_api_key`` as a named parameter. Example: .. code-block:: python from langchain_community.retrievers import ArceeRetriever retriever = ArceeRetriever( model="DALM-PubMed", arcee_api_key="ARCEE-API-KEY" ) documents = retriever.invoke("AI-driven music therapy") N_client arcee_api_keymodelzhttps://api.arcee.ai arcee_api_urlZv2arcee_api_versionzhttps://app.arcee.ai arcee_app_url model_kwargsZforbid)extra)datareturnc sBtjfi|t|j|j|j|j|jd|_ |j dS)zInitializes private fields.)rrrrZ model_nameN) super__init__rrZget_secret_valuerrrrrZvalidate_model_training_status)selfr __class__r/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/retrievers/arcee.pyr8szArceeRetriever.__init__)valuesrcCstt|dd|d<t|dd|d<t|dd|d<t|dd|d<|d r|d }|d d urx|d d ksxtd |dd urt|dtstd|dD]}tfi|q|S)z%Validate Arcee environment variables.rZ ARCEE_API_KEYrZ ARCEE_API_URLrZ ARCEE_APP_URLrZARCEE_API_VERSIONrsizeNrz`size` must not be negative.filtersz`filters` must be a list.)r r get ValueError isinstancerr)clsr"kwfr r r!validate_environmentsGsBz$ArceeRetriever.validate_environments)query run_managerkwargsrc Ks`z&|jstd|jjfd|i|WStyZ}ztd||WYd}~n d}~00dS)aGRetrieve {size} contexts with your retriever for a given query Args: query: Query to submit to the model size: The max number of context results to retrieve. Defaults to 3. (Can be less if filters are provided). filters: Filters to apply to the context dataset. zClient is not initialized.r,z"Error while retrieving documents: N)rr&retrieve Exception)rr,r-r.er r r!_get_relevant_documentsxs z&ArceeRetriever._get_relevant_documents)__name__ __module__ __qualname____doc__rrr__annotations__r strrrrrrrr Z model_configrr r+rrrr2 __classcell__r r rr!r s"    1rN)typingrrrrZlangchain_core.callbacksrZlangchain_core.documentsrZlangchain_core.retrieversrZlangchain_core.utilsr r r Zpydanticr r Z#langchain_community.utilities.arceerrrr r r r!s