o
    ưi                     @   s@   d Z ddlmZmZ ddlmZ ddlmZ G dd deZd	S )
z
Support for OpenAI's `/v1/chat/completions` endpoint. 

Calls done in OpenAI/openai.py as Novita AI is openai-compatible.

Docs: https://novita.ai/docs/guides/llm-api
    )ListOptional   )AllMessageValues   )OpenAIGPTConfigc                   @   sH   e Zd Z		ddededee dededee dee d	efd
dZdS )NovitaConfigNheadersmodelmessagesoptional_paramslitellm_paramsapi_keyapi_basereturnc                 C   s2   |d u rt dd| |d< d|d< d|d< |S )Nz~Missing Novita AI API Key - A call is being made to novita but no key is set either in the environment variables or via paramszBearer Authorizationzapplication/jsonzContent-TypeZlitellmzX-Novita-Source)
ValueError)selfr	   r
   r   r   r   r   r    r   ^/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/novita/chat/transformation.pyvalidate_environment   s   
z!NovitaConfig.validate_environment)NN)	__name__
__module____qualname__dictstrr   r   r   r   r   r   r   r   r      s(    	r   N)	__doc__typingr   r   Ztypes.llms.openair   Zopenai.chat.gpt_transformationr   r   r   r   r   r   <module>   s
    