o
    Æ°iú  ã                   @   s   d dl mZ G dd„ dƒZdS )é    )ÚOptionalc                   @   s*   e Zd Z	ddedee defdd„ZdS )ÚSnowflakeBaseNÚheadersÚJWTÚreturnc                 C   s,   |du rt dƒ‚| ddd| ddœ¡ |S )a¿  
        Return headers to use for Snowflake completion request

        Snowflake REST API Ref: https://docs.snowflake.com/en/user-guide/snowflake-cortex/cortex-llm-rest-api#api-reference
        Expected headers:
        {
            "Content-Type": "application/json",
            "Accept": "application/json",
            "Authorization": "Bearer " + <JWT>,
            "X-Snowflake-Authorization-Token-Type": "KEYPAIR_JWT"
        }
        NzMissing Snowflake JWT keyzapplication/jsonzBearer ZKEYPAIR_JWT)zContent-TypeÚAcceptÚAuthorizationz$X-Snowflake-Authorization-Token-Type)Ú
ValueErrorÚupdate)Úselfr   r   © r   úZ/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/snowflake/common_utils.pyÚvalidate_environment   s   üÿz"SnowflakeBase.validate_environment)N)Ú__name__Ú
__module__Ú__qualname__Údictr   Ústrr   r   r   r   r   r      s    ýþýür   N)Útypingr   r   r   r   r   r   Ú<module>   s    