o Zh,@sddlmZmZmZmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZGddde Zed krJeZeed d Sd S) )AnyDictListOptionalSequence)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDict SecretStr)enforce_stop_tokensc @s*eZdZUdZdZeed<dZee ed< dZ e ed< dZ e ed < d Ze ed < dZe ed < dZe ed < dZe ed< dZeeed< dZeeed< dZee ed< dZeee ed< dZeeed< dZee ed< dZe ed< dZeee e fed< dZee ed< dZeeed< dZeeed<d Z ee ed< dZ!eed< dZ"eed<dZ#e ed<d Z$e ed!<dZ%eed"<dZ&ee'e ed#<dZ(eed$<dZ)ee'e ed%<dZ*eed&< dZ+ee ed'< d(Z,eeed)< d(Z-eed*< dZ.eed+< dZ/eee ed,< dZ0ee1ed-< d.Z2e ed/< dZ3ee ed0< d1Z4e ed2< d3Z5e ed4< dZ6eed5< e7d6d7Z8e9d8ed9efd:d;Z:e;d9ee effdd?Z=e;d9e fd@dAZ>  dHdBe dCeee dDee?dEed9e f dFdGZ@dS)I AlephAlphaamAleph Alpha large language models. To use, you should have the ``aleph_alpha_client`` python package installed, and the environment variable ``ALEPH_ALPHA_API_KEY`` set with your API key, or pass it as a named parameter to the constructor. Parameters are explained more in depth here: https://github.com/Aleph-Alpha/aleph-alpha-client/blob/c14b7dd2b4325c7da0d6a119f6e76385800e097b/aleph_alpha_client/completion.py#L10 Example: .. code-block:: python from langchain_community.llms import AlephAlpha aleph_alpha = AlephAlpha(aleph_alpha_api_key="my-api-key") Nclientz luminous-basemodel@maximum_tokensg temperaturertop_ktop_ppresence_penaltyfrequency_penaltyF#repetition_penalties_include_prompt#use_multiplicative_presence_penalty penalty_biaspenalty_exceptions)penalty_exceptions_include_stop_sequencesbest_ofn logit_bias log_probstokensdisable_optimizationsminimum_tokensecho$use_multiplicative_frequency_penaltysequence_penaltysequence_penalty_min_length#use_multiplicative_sequence_penaltycompletion_bias_inclusion*completion_bias_inclusion_first_token_onlycompletion_bias_exclusion*completion_bias_exclusion_first_token_onlycontextual_control_thresholdTcontrol_log_additive'repetition_penalties_include_completionraw_completionstop_sequencesaleph_alpha_api_keyzhttps://api.aleph-alpha.comhosthostingi1request_timeout_seconds total_retriesniceZforbid)extravaluesreturncCsrtt|dd|d<z#ddlm}||d|d|d|d|d|d d |d <W|Sty8td w) z?Validate that api key and python package exists in environment.r5ZALEPH_ALPHA_API_KEYr)Clientr6r7r8r:r;)tokenr6r7r8r:r;rzlCould not import aleph_alpha_client python package. Please install it with `pip install aleph_alpha_client`.)r r aleph_alpha_clientr?Zget_secret_value ImportError)clsr=r?rD[/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/aleph_alpha.pyvalidate_environments&     zAlephAlpha.validate_environmentcCsid|jd|jd|jd|jd|jd|jd|jd|jd |jd |j d |j d |j d |j d|j d|jd|jd|j|j|j|j|j|j|j|j|j|j|j|j|j|j|jdS)z;Get the default parameters for calling the Aleph Alpha API.rrrrrrr rrrrrrr!r"r#r$)r%r&r'r(r*r+r,r-r.r/r0r1r2r3)rrrrrrr rrrrrrr!r"r#r$r%r&r'r(r*r+r,r-r.r/r0r1r2r3selfrDrDrE_default_paramssd     zAlephAlpha._default_paramscCsid|ji|jS)zGet the identifying parameters.r)rrIrGrDrDrE_identifying_paramsszAlephAlpha._identifying_paramscCsdS)zReturn type of llm.Z aleph_alpharDrGrDrDrE _llm_typeszAlephAlpha._llm_typepromptstop run_managerkwargsc Ksddlm}m}|j}|jdur|durtd|jdur#|j|d<n||d<i||}|dd||i|}|jj|j |d} | j dj } |dusQ|jdurXt | |d} | S) atCall out to Aleph Alpha's completion endpoint. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: The string generated by the model. Example: .. code-block:: python response = aleph_alpha("Tell me a joke.") r)CompletionRequestPromptNz:stop sequences found in both the input and default params.r4rL)rrequestrD) rArPrQrIr4 ValueErrorZ from_textrZcompleterZ completions completionr) rHrLrMrNrOrPrQparamsrRresponsetextrDrDrE_calls     zAlephAlpha._call)NN)A__name__ __module__ __qualname____doc__rr__annotations__rrstrrintrfloatrrrrrboolrrrrrrr r!rr"r#r$r%r&r'r(r*r+r,rr-r.r/r0r1r2r3r4r5r r6r7r8r:r;r Z model_configr rFpropertyrIrJrKrrXrDrDrDrEr s                      $ r__main__z How are you?N)typingrrrrrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r r Zpydanticr r Zlangchain_community.llms.utilsrrrYZaaprintZinvokerDrDrDrEs