7|hddlmZmZmZmZmZmZddlmZddl m Z ddl m Z m Z ddlmZmZmZddlmZmZGdde Zy ) )AnyDictListOptionalUnioncast)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_env) ConfigDict SecretStrmodel_validator) ArceeWrapper DALMFilterc ZeZdZUdZdZeeed< dZe e e dfed< e ed< dZ e ed< dZ e ed < d Ze ed < d Ze ed < dZeee efed< edZede fdZdeddffd ZedededefdZ dde deee deedede f dZxZS)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_kwargsforbid)extrareturncy)zReturn type of llm.arcee)selfs ]/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/llms/arcee.py _llm_typezArcee._llm_type:sdatac t|di|tt|j}t ||j |j|j|j|_ y)zInitializes private fields.)rrrr model_nameNr#) super__init__rrrrrrrrr)r$r(api_key __class__s r%r,zArcee.__init__?sX  4 y$"4"45#!,,"44**zz   r'before)modevaluesctt|dd|d<t|dd|d<t|dd|d<t|dd|d<|jd 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 be positivefiltersz`filters` must be a listr#)r r get ValueError isinstancerr)clsr1kwfs r%validate_environmentszArcee.validate_environmentsLs  #8  # #7   #  #7   #  ';   ' "# ::n %'Bvvf~)vvf~*$%>??vvi ,!"&&"3T:$%?@@ *$AOO$ r'promptstop run_managerkwargsc |js td|jjdd|i|S#t$r}td||d}~wwxYw)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.r@zFailed to generate text: Nr#)rr:generate Exception)r$r@rArBrCes r%_callz Arcee._call}sa  D<< !=>>(4<<((AA&A A D7s;*11L3  s  t  (#-4-C-$-d%):> DDtCy!D67 D  D Dr'rN)typingrrrrrrlangchain_core.callbacksr #langchain_core.language_models.llmsr langchain_core.utilsr r pydanticr rr#langchain_community.utilities.arceerrrr#r'r%rYs-99=3L;;HGDCGDr'