U h@s|ddlmZmZmZmZmZmZddlmZddl m Z ddl m Z m Z ddlmZmZddlmZmZGddde Zd S) )AnyDictListOptionalUnioncast)CallbackManagerForLLMRun)LLM) SecretStrroot_validator)convert_to_secret_strget_from_dict_or_env) ArceeWrapper DALMFiltercseZdZUdZdZeeed<dZe e e dfed<e ed<dZ e ed<dZ e ed <d Ze ed <d Ze ed <dZeee efed<GdddZee dddZeddfdd ZeddeedddZde eee eeee dddZZS) ArceeaArcee's Domain Adapted Language Models (DALMs). 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.llms import Arcee arcee = Arcee( model="DALM-PubMed", arcee_api_key="ARCEE-API-KEY" ) response = arcee("AI-driven music therapy") N_client arcee_api_keymodelzhttps://api.arcee.ai arcee_api_urlZv2arcee_api_versionzhttps://app.arcee.ai arcee_app_urlmodel_id model_kwargsc@seZdZdZdZdS)z Arcee.ConfigZforbidTN)__name__ __module__ __qualname__extraZunderscore_attrs_are_privaterrB/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/arcee.pyConfig6sr )returncCsdS)zReturn type of llm.Zarceer)selfrrr _llm_type:szArcee._llm_type)datar!c s:tjf|tt|j}t||j|j|j|j d|_ dS)zInitializes private fields.)rrrrZ model_nameN) super__init__rr rrrrrrr)r"r$Zapi_key __class__rrr&?s zArcee.__init__T)pre)valuesr!cCstt|dd|d<t|dd|d<t|dd|d<t|dd|d<|d r|d }|d d k rz|d d ksztd |dd k rt|dtstd|dD]}tf|q|S)z%Validate Arcee environment variables.rZ ARCEE_API_KEYrZ ARCEE_API_URLrZ ARCEE_APP_URLrZARCEE_API_VERSIONrsizeNrz`size` must be positivefiltersz`filters` must be a list)r r get ValueError isinstancerr)clsr*kwfrrrvalidate_environmentsLsB  zArcee.validate_environments)promptstop run_managerkwargsr!c Ks^z&|jstd|jjfd|i|WStk rX}ztd||W5d}~XYnXdS)a)Generate text from Arcee DALM. Args: prompt: Prompt to generate text from. 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.r4zFailed to generate text: N)rr.generate Exception)r"r4r5r6r7errr_call|s z Arcee._call)NN)rrr__doc__rrr__annotations__rrr strrrrrrrrr propertyr#r&r r3rrr; __classcell__rrr'rr s0      2 rN)typingrrrrrrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsr Zlangchain_core.pydantic_v1r r Zlangchain_core.utilsr r Z#langchain_community.utilities.arceerrrrrrrs