a bg,@sddlmZmZmZmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZGddde Zed kreZeed d S) )AnyDictListOptionalSequence)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDict SecretStr)enforce_stop_tokensc@seZdZUdZdZeed<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<e7d6d7Z8e9eed8d9d:Z:e;ee efd;dd?Z=e;e d;d@dAZ>dEe eee ee?ee dBdCdDZ@dS)F 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)extra)valuesreturncCsxtt|dd|d<zDddlm}||d|d|d|d|d|d d |d <Wntyrtd Yn0|S) 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?rDr/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/aleph_alpha.pyvalidate_environments$     zAlephAlpha.validate_environment)r>c Cs|j|j|j|j|j|j|j|j|j|j |j |j |j |j |j|j|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/r0r1r2r3rGselfrDrDrE_default_paramss@zAlephAlpha._default_paramscCsid|ji|jS)zGet the identifying parameters.r)rrJrHrDrDrE_identifying_paramsszAlephAlpha._identifying_paramscCsdS)zReturn type of llm.Z aleph_alpharDrHrDrDrE _llm_typeszAlephAlpha._llm_type)promptstop run_managerkwargsr>c Ksddlm}m}|j}|jdur2|dur2tdn|jdurH|j|d<n||d<i||}|fd||i|}|jj|j |d} | j dj } |dus|jdurt | |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.r4rM)rrequest) rArQrRrJr4 ValueErrorZ from_textrZcompleterZ completions completionr) rIrMrNrOrPrQrRparamsrSresponsetextrDrDrE_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 rFpropertyrJrKrLrrYrDrDrDrEr sz                     $ r__main__z How are you?N)typingrrrrrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r r Zpydanticr r Zlangchain_community.llms.utilsrrrZZaaprintZinvokerDrDrDrEs