dhYSSKJrJrJrJr SSKJr SSKJr SSK J r SSK J r J r Jr SSKJrJr SSKJrJr "SS \ 5rg ) )AnyDictListOptional)CallbackManagerForRetrieverRun)Document) BaseRetriever)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDict SecretStr) ArceeWrapper DALMFilterc ^\rSrSr%SrSr\\\S'\ \S'\ \S'Sr \ \S'S r \ \S 'S r \ \S 'Sr\\\ \4\S '\"SS9rS\SS4U4Sjjr\S\S\4Sj5rS\ S\S\S\\4SjrSrU=r$)ArceeRetriever aArcee 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 >[TU]"S0UD6 [URR 5UR UR URURS9Ul URR5 g)zInitializes private fields.)rrrr model_nameN) super__init__rrget_secret_valuerrrrrvalidate_model_training_status)selfr __class__s \/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/retrievers/arcee.pyr$ArceeRetriever.__init__8sg  4 #,,==?,,"44**zz    335valuesc[[USS55US'[USS5US'[USS5US'[USS5US'US (aUS nURS 5b URS 5S :d [S 5eURS 5bR[ URS 5[ 5(d [S5eURS 5Hn[ S0UD6 M U$)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_environments$ArceeRetriever.validate_environmentsGs  #8  # #7   #  #7   #  ';   ' "# . !'Bvvf~)vvf~*$%CDDvvi ,!"&&"3T::$%@AA *AOO+ r+query run_managerkwargsc UR(d [S5eURR"SSU0UD6$![an[SU35UeSnAff=f)aRetrieve {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: Nr")rr5retrieve Exception)r'r<r=r>es r)_get_relevant_documents&ArceeRetriever._get_relevant_documentsxs_ N<< !=>><<((?u?? ? NA!EFA M Ns9< AAA)r)__name__ __module__ __qualname____firstlineno____doc__rrr__annotations__rstrrrrrrrr model_configr$r r:rrrrC__static_attributes__ __classcell__)r(s@r)rr s$'+GXl #* J/M3/!s!/M3/-1L(4S>*11L 6s 6t 6.4.D..`NN'ENQTN hNNr+rN)typingrrrrlangchain_core.callbacksrlangchain_core.documentsrlangchain_core.retrieversr langchain_core.utilsr r r pydanticr r#langchain_community.utilities.arceerrrr"r+r)rVs0,,C-3VV*H}N]}Nr+