o Zh@sddlmZmZmZmZmZmZddlmZddl m Z ddl m Z m Z ddlmZmZmZddlmZmZGddde Zd S) )AnyDictListOptionalUnioncast)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_env) ConfigDict SecretStrmodel_validator) ArceeWrapper DALMFilterc seZdZUdZdZeeed< dZe e e dfed< e ed< dZ e ed< dZ e ed < d Ze ed < d Ze ed < dZeee efed< eddZede fddZdeddffdd ZeddededefddZ  d"de deee deedede f d d!ZZS)#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_urlv2arcee_api_versionzhttps://app.arcee.ai arcee_app_urlmodel_id model_kwargsZforbid)extrareturncCsdS)zReturn type of llm.Zarcee)selfrrU/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/arcee.py _llm_type:szArcee._llm_typedatac s>tjdi|tt|j}t||j|j|j|j d|_ dS)zInitializes private fields.)rrrrZ model_nameNr) super__init__rr rrrrrrr)rr"Zapi_key __class__rr r$?s  zArcee.__init__before)modevaluescCstt|dd|d<t|dd|d<t|dd|d<t|dd|d<|d r_|d }|d d ur=|d d ks=td |dd ur_t|dtsPtd|dD] }tdi|qU|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 listr)r r get ValueError isinstancerr)clsr)kwfrrr validate_environmentsLsB zArcee.validate_environmentspromptstop run_managerkwargsc KsPz|jstd|jjdd|i|WSty'}ztd||d}~ww)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.r3zFailed to generate text: Nr)rr-generate Exception)rr3r4r5r6errr _call}sz Arcee._call)NN)__name__ __module__ __qualname____doc__rrr__annotations__rrr strrrrrrrrr Z model_configpropertyr!r$r classmethodr2rrr: __classcell__rrr%r r sP      2 rN)typingrrrrrrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsr Zlangchain_core.utilsr r Zpydanticr r rZ#langchain_community.utilities.arceerrrrrrr s