a bg@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 DALMFiltercseZdZUdZdZeeed<dZe e e dfed<e ed<dZ e ed<dZ e ed <d Ze ed <d Ze ed <dZeee efed<eddZee dddZeddfdd ZeddeeedddZde 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_kwargsZforbid)extra)returncCsdS)zReturn type of llm.Zarcee)selfrrl/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/arcee.py _llm_type:szArcee._llm_type)datarc s>tjfi|tt|j}t||j|j|j|j d|_ dS)zInitializes private fields.)rrrrZ model_nameN) super__init__rr rrrrrrr)rr!Zapi_key __class__rrr#?s zArcee.__init__before)mode)valuesrcCstt|dd|d<t|dd|d<t|dd|d<t|dd|d<|d r|d }|d d urz|d d ksztd |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 be positivefiltersz`filters` must be a list)r r get ValueError isinstancerr)clsr(kwfrrrvalidate_environmentsLsB zArcee.validate_environments)promptstop run_managerkwargsrc Ks`z&|jstd|jjfd|i|WStyZ}ztd||WYd}~n d}~00dS)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.r2zFailed to generate text: N)rr,generate Exception)rr2r3r4r5errr_call}s z Arcee._call)NN)__name__ __module__ __qualname____doc__rrr__annotations__rrr strrrrrrrrr Z model_configpropertyr r#r classmethodr1rrr9 __classcell__rrr$rr s6      2 rN)typingrrrrrrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsr Zlangchain_core.utilsr r Zpydanticr r rZ#langchain_community.utilities.arceerrrrrrrs