§
    ÚÇÔi¾  ã                  ó:   — d dl mZ ddlmZ  G d„ de¦  «        ZdS )é    )Úannotationsé   )ÚBaseAdapterc                  óJ   — e Zd ZdZdddœdd„Zdd„Zdd„Zdd„Zdd„Zdd„Z	dS )ÚAdapterz
    OpenAI adapter
    N)Úbase_urlÚheadersÚauth_keyÚstrr   ú
str | Noner	   údict[str, str] | NoneÚmodelÚreturnÚNonec               óF   — || _         |pi | _        || _        |pd| _        dS )zÚ
        Args:
        ----
            auth_key: OpenAI API key.
            base_url: OpenAI API URL.
            headers: Additional headers to send with the request.
            model: OpenAI model to use.
        zhttps://api.openai.com/v1N)Ú	_auth_keyÚ_headersÚ_modelÚ_url)Úselfr
   r   r	   r   s        ú[/root/projects/qq-shell/venv/lib/python3.11/site-packages/inngest/experimental/ai/openai.pyÚ__init__zAdapter.__init__   s.   € ð" "ˆŒØ˜ 2ˆŒØˆŒØÐ;Ð ;ˆŒ	ˆ	ˆ	ó    c                ó   — | j         S )z@
        Return the authentication key for the adapter.
        )r   ©r   s    r   r
   zAdapter.auth_key!   s   € ð
 Œ~Ðr   c                ó   — dS )z4
        Return the format for the adapter.
        zopenai-chat© r   s    r   ÚformatzAdapter.format(   s	   € ð
 ˆ}r   údict[str, str]c                ó   — | j         S )z5
        Return the headers for the adapter.
        )r   r   s    r   r	   zAdapter.headers/   s   € ð
 Œ}Ðr   Úbodyúdict[str, object]c                óH   — |                      d¦  «        s| j        |d<   dS dS )z*
        Modify the request body.
        r   N)Úgetr   )r   r!   s     r   Úon_callzAdapter.on_call6   s1   € ð
 xŠx˜Ñ Ô ð 	(Ø œKˆD‰MˆMˆMð	(ð 	(r   c                ó<   — | j                              d¦  «        dz   S )z5
        Return the URL for generating text.
        ú/z/chat/completions)r   Úrstripr   s    r   Ú	url_inferzAdapter.url_infer>   s    € ð
 Œy×Ò Ñ$Ô$Ð':Ñ:Ð:r   )
r
   r   r   r   r	   r   r   r   r   r   )r   r   )r   r   )r!   r"   r   r   )
Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r
   r   r	   r%   r)   r   r   r   r   r      s¨   € € € € € ðð ð  $Ø)-ð<ð <ð <ð <ð <ð <ð,ð ð ð ðð ð ð ðð ð ð ð(ð (ð (ð (ð;ð ;ð ;ð ;ð ;ð ;r   r   N)Ú
__future__r   Úbaser   r   r   r   r   ú<module>r0      s]   ðØ "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ð Ð ð=;ð =;ð =;ð =;ð =;ˆkñ =;ô =;ð =;ð =;ð =;r   