7|hYddlmZmZmZmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZddlmZmZGdd e Zy ) )AnyDictListOptional)CallbackManagerForRetrieverRun)Document) BaseRetriever)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDict SecretStr) ArceeWrapper DALMFilterc eZdZUdZdZeeed< eed< e ed< dZ e ed< dZ e ed < d Z e ed < dZ eee efed < ed Zdeddffd ZededefdZde dededeefdZxZS)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_urlv2arcee_api_versionzhttps://app.arcee.ai arcee_app_url model_kwargsforbid)extradatareturnc t|di|t|jj |j |j |j|j|_ |jjy)zInitializes private fields.)rrrr model_nameN) super__init__rrget_secret_valuerrrrrvalidate_model_training_status)selfr __class__s c/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/retrievers/arcee.pyr#zArceeRetriever.__init__8sg  4 #,,==?,,"44**zz    335valuesctt|dd|d<t|dd|d<t|dd|d<t|dd|d<|d r|d }|jd |jd d k\s td |jd Kt |jd t s td|jd D] }t di||S)z%Validate Arcee environment variables.r ARCEE_API_KEYr ARCEE_API_URLr ARCEE_APP_URLrARCEE_API_VERSIONrsizerz`size` must not be negative.filtersz`filters` must be a list.r!)r r get ValueError isinstancerr)clsr*kwfs r(validate_environmentsz$ArceeRetriever.validate_environmentsGs  #8  # #7   #  #7   #  ';   ' "# . !'Bvvf~)vvf~*$%CDDvvi ,!"&&"3T:$%@AA *$AOO$ r)query run_managerkwargsc |js td|jjdd|i|S#t$r}td||d}~wwxYw)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.r9z"Error while retrieving documents: Nr!)rr3retrieve Exception)r&r9r:r;es r(_get_relevant_documentsz&ArceeRetriever._get_relevant_documentsxsa N<< !=>>(4<<((?u?? ? NA!EFA M Ns47 AAA)__name__ __module__ __qualname____doc__rrr__annotations__rstrrrrrrrr model_configr#r r8rrrr@ __classcell__)r's@r(rr s$'+GXl #* J/M3/!s!/M3/-1L(4S>*11L 6s 6t 6.4.D..`NN'ENQTN hNr)rN)typingrrrrlangchain_core.callbacksrlangchain_core.documentsrlangchain_core.retrieversr langchain_core.utilsr r r pydanticr r#langchain_community.utilities.arceerrrr!r)r(rPs0,,C-3VV*H}N]}Nr)