
    h                     $    d dl mZ  G d d      Zy)    )Optionalc                   *    e Zd Z	 ddedee   defdZy)SnowflakeBaseNheadersJWTreturnc                 R    |t        d      |j                  ddd|z   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"
        }
        zMissing Snowflake JWT keyzapplication/jsonzBearer KEYPAIR_JWT)zContent-TypeAcceptAuthorizationz$X-Snowflake-Authorization-Token-Type)
ValueErrorupdate)selfr   r   s      _/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/snowflake/common_utils.pyvalidate_environmentz"SnowflakeBase.validate_environment   s=    $ ;899 2,!*S8E		
     )N)__name__
__module____qualname__dictr   strr    r   r   r   r      s,     " c] 
	r   r   N)typingr   r   r   r   r   <module>r      s     r   