Ë
    µŒj•  ã                   ó¾   — d dl Z d dlmZmZmZmZmZmZ d dlZd dl	m
Z
 d dlmZ d dlmZmZmZ d dlmZmZmZmZ d dlmZ  e j0                  e«      Z G d„ d	e«      Zy)
é    N)ÚAnyÚDictÚListÚMappingÚOptionalÚcast)ÚCallbackManagerForLLMRun)ÚLLM)Úconvert_to_secret_strÚget_from_dict_or_envÚpre_init)Ú
ConfigDictÚFieldÚ	SecretStrÚmodel_validator)Úenforce_stop_tokensc                   ó0  — e Zd ZU dZdZeed<   	  ee¬«      Z	e
eef   ed<   	 dZee   ed<    ed¬	«      Z ed
¬«      ede
eef   defd„«       «       Zede
de
fd„«       Zedeeef   fd„«       Zedefd„«       Z	 	 ddedeee      dee   dedef
d„Zy)ÚCerebriumAIaH  CerebriumAI large language models.

    To use, you should have the ``cerebrium`` python package installed.
    You should also have the environment variable ``CEREBRIUMAI_API_KEY``
    set with your API key or pass it as a named argument in the constructor.

    Any parameters that are valid to be passed to the call can be passed
    in, even if not explicitly saved on this class.

    Example:
        .. code-block:: python

            from langchain_community.llms import CerebriumAI
            cerebrium = CerebriumAI(endpoint_url="", cerebriumai_api_key="my-api-key")

    Ú Úendpoint_url)Údefault_factoryÚmodel_kwargsNÚcerebriumai_api_keyÚforbid)ÚextraÚbefore)ÚmodeÚvaluesÚreturnc                 ó:  — t        t        | j                  j                  «       «      «      }|j	                  di «      }t        |«      D ]I  }||vsŒ||v rt        d|› d�«      ‚t        j                  |› d|› d�«       |j                  |«      ||<   ŒK ||d<   |S )z>Build extra kwargs from additional params that were passed in.r   zFound z supplied twice.zJ was transferred to model_kwargs.
                    Please confirm that z is what you intended.)	ÚsetÚlistÚmodel_fieldsÚkeysÚgetÚ
ValueErrorÚloggerÚwarningÚpop)Úclsr   Úall_required_field_namesr   Ú
field_names        ún/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/llms/cerebriumai.pyÚbuild_extrazCerebriumAI.build_extra.   s±   € ô $'¤t¨C×,<Ñ,<×,AÑ,AÓ,CÓ'DÓ#EÐ à—
‘
˜>¨2Ó.ˆÜ˜vž,ˆJØÐ!9Ò9Ø Ñ&Ü$ v¨j¨\Ð9IÐ%JÓKÐKÜ—‘Ø#˜ð %)Ø)3¨Ð4JðNôð %+§J¡J¨zÓ$:��jÒ!ð 'ð "'ˆˆ~ÑØˆó    c                 ó<   — t        t        |dd«      «      }||d<   |S )z?Validate that api key and python package exists in environment.r   ÚCEREBRIUMAI_API_KEY)r   r   )r*   r   r   s      r-   Úvalidate_environmentz CerebriumAI.validate_environmentA   s/   € ô 4Ü  Ð)>Ð@UÓVó
Ðð )<ˆÐ$Ñ%Øˆr/   c                 ó>   — i d| j                   i¥d| j                  i¥S )zGet the identifying parameters.r   r   )r   r   ©Úselfs    r-   Ú_identifying_paramszCerebriumAI._identifying_paramsJ   s2   € ð
Ø˜t×0Ñ0Ð1ð
à˜t×0Ñ0Ð1ð
ð 	
r/   c                  ó   — y)zReturn type of llm.Úcerebriumai© r4   s    r-   Ú	_llm_typezCerebriumAI._llm_typeR   s   € ð r/   ÚpromptÚstopÚrun_managerÚkwargsc                 óV  — t        t        | j                  «      j                  «       ddœ}| j                  xs i }d|i|¥|¥}t        j                  | j                  ||¬«      }|j                  dk(  r%|j                  «       }	|	d   }
|�t        |
|«      }
|
S |j                  «        y)Nzapplication/json)ÚAuthorizationzContent-Typer;   )ÚjsonÚheaderséÈ   Úresultr   )r   r   r   Úget_secret_valuer   ÚrequestsÚpostr   Ústatus_coderA   r   Úraise_for_status)r5   r;   r<   r=   r>   rB   ÚparamsÚpayloadÚresponseÚdataÚtexts              r-   Ú_callzCerebriumAI._callW   s²   € ô "Ü˜4×3Ñ3óçÑÓ Ø.ñ	
ˆð ×"Ñ"Ò( bˆØ˜VÐ8 vÐ8°Ð8ˆÜ—=‘= ×!2Ñ!2¸È'ÔRˆØ×Ñ 3Ò&Ø—=‘=“?ˆDØ˜‘>ˆDØÐô +¨4°Ó6�ØˆKà×%Ñ%Ô'Ør/   )NN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚstrÚ__annotations__r   Údictr   r   r   r   r   r   r   Úmodel_configr   Úclassmethodr.   r   r2   Úpropertyr   r6   r:   r   r	   rO   r9   r/   r-   r   r      s5  … ñð" €L�#ÓØá#(¸Ô#>€L�$�s˜C�x‘.Ó>ðð 04Ð˜ )Ñ,Ó3áØô€Lñ ˜(Ô#Øð  c¨3 h¡ð °Cò ó ó $ðð" ð¨$ð °4ò ó ðð ð
 W¨S°#¨XÑ%6ò 
ó ð
ð ð˜3ò ó ðð %)Ø:>ñ	àðð �t˜C‘yÑ!ðð Ð6Ñ7ð	ð
 ðð 
ôr/   r   )ÚloggingÚtypingr   r   r   r   r   r   rF   Úlangchain_core.callbacksr	   Ú#langchain_core.language_models.llmsr
   Úlangchain_core.utilsr   r   r   Úpydanticr   r   r   r   Úlangchain_community.llms.utilsr   Ú	getLoggerrP   r'   r   r9   r/   r-   Ú<module>rb      sF   ðÛ ß ;× ;ã Ý =Ý 3ß VÑ Vß BÓ Bå >à	ˆ×	Ñ	˜8Ó	$€ôb�#õ br/   