NgddlmZmZmZmZmZmZddlmZddl m Z ddl m Z m Z ddlmZmZmZddlmZmZGdde Zd S) )AnyDictListOptionalUnioncast)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_env) ConfigDict SecretStrmodel_validator) ArceeWrapper DALMFilterc eZdZUdZdZeeed< 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)extrareturncdS)zReturn type of llm.arcee)selfs Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/llms/arcee.py _llm_typezArcee._llm_type:s wdatac tjdi|tt|j}t ||j|j|j|j |_ dS)zInitializes private fields.)rrrr model_nameNr#) super__init__rrrrrrrrr)r$r(api_key __class__s r%r,zArcee.__init__?sh   4   y$"455#!,"4*z     r'before)modevaluescTtt|dd|d<t|dd|d<t|dd|d<t|dd|d<|d r|d }|d (|d d kstd |d[t |dt std|dD]}t di||S)z%Validate Arcee environment variables.r ARCEE_API_KEYr ARCEE_API_URLr ARCEE_APP_URLrARCEE_API_VERSIONrsizeNrz`size` must be positivefiltersz`filters` must be a listr#)r r get ValueError isinstancerr)clsr1kwfs r%validate_environmentszArcee.validate_environmentsLsU #8   # # #7   # #  #7   # #  ';   ' ' "# ::n % % $'Bvvf~~)vvf~~**$%>???vvi  ,!"&&"3"3T::A$%?@@@ **$$AOOOOOO r'promptstop run_managerkwargsc |jstd|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}s}  D< ? !=>>>(4<(AAA&AA A D D D;;;<*1111:L3X  s  t       _(###-4-C---[$#-d%):> DDDtCy!D67 D  D DDDDDDDDr'rN)typingrrrrrrlangchain_core.callbacksr #langchain_core.language_models.llmsr langchain_core.utilsr r pydanticr rr#langchain_community.utilities.arceerrrr#r'r%rYs9999999999999999======333333LLLLLLLL;;;;;;;;;;HHHHHHHHGDGDGDGDGDCGDGDGDGDGDr'