
    ~i                       d dl mZ d dl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 ddlmZmZ dd	lmZmZmZ dd
lmZmZ ddlmZmZ ddlmZmZmZm Z m!Z! ddl"m#Z#m$Z$ ddlm%Z%m&Z&m'Z' ddgZ( G d de          Z) G d de          Z*dS )    )annotations)DictIterableListOptionalUnionN)Literal   )make_request_options)cached_property)AsyncAPIResourceSyncAPIResource)BodyHeadersQuery)async_with_sts_tokenwith_sts_token   )with_e2e_encryptionasync_with_e2e_encryption)ChatCompletionChatCompletionMessageParam#ChatCompletionToolChoiceOptionParamChatCompletionToolParamcompletion_create_params)AsyncCompletionsWithRawResponseCompletionsWithRawResponse   )async_with_batch_retryget_request_last_timewith_batch_retryCompletionsAsyncCompletionsc                      e Zd Zed4d            Zeedddddddddddddddddddddddddd5d3                        ZdS )6r"   returnr   c                     t          |           S N)r   selfs    /root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/resources/batch/chat/completions.pywith_raw_responsezCompletions.with_raw_response   s    )$///    N)frequency_penaltyfunction_call
logit_biaslogprobs
max_tokenspresence_penaltystoptemperaturetoolstop_logprobstop_prepetition_penaltynparallel_tool_callsservice_tiertool_choiceresponse_formatthinkingmax_completion_tokensuserextra_headersextra_query
extra_bodytimeoutmessages$Iterable[ChatCompletionMessageParam]modelstrr-   Optional[float] | Noner.   ,completion_create_params.FunctionCall | Noner/   Optional[Dict[str, int]] | Noner0   Optional[bool] | Noner1   Optional[int] | Noner2   r3   &Union[Optional[str], List[str]] | Noner4   r5   (Iterable[ChatCompletionToolParam] | Noner6   r7   r8   r9   r:   r;   +Optional[Literal['auto', 'default']] | Noner<   *ChatCompletionToolChoiceOptionParam | Noner=   .completion_create_params.ResponseFormat | Noner>   (completion_create_params.Thinking | Noner?   r@   
str | NonerA   Headers | NonerB   Query | NonerC   Body | NonerD   float | httpx.Timeout | Noner   c               <   t          | j        |          }| j                            |          }t          ||| j        di d|d|d|d|d|d|d|d	|d
|	d|
d|d|d|d|d|d|d||||||dt          ||||          t                    }|S Nz/batch/chat/completionsrE   rG   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r@   r8   r9   r:   )r;   r<   r=   r>   r?   )rA   rB   rC   rD   )bodyoptionscast_to)r    _clientget_model_breakerr!   _post_without_retryr   r   )r)   rE   rG   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   deadlinebreakerresps                                 r*   createzCompletions.create#   sm   @ )w??,0077$%H $%6  	
 j H j #$4  {     %&8  Q!" &':#$ !-*#2$)>-  0 )+'%	   #G$
 $
 $
J r,   )r%   r   )6rE   rF   rG   rH   r-   rI   r.   rJ   r/   rK   r0   rL   r1   rM   r2   rI   r3   rN   r4   rI   r5   rO   r6   rM   r7   rI   r8   rI   r9   rM   r:   rL   r;   rP   r<   rQ   r=   rR   r>   rS   r?   rM   r@   rT   rA   rU   rB   rV   rC   rW   rD   rX   r%   r   )__name__
__module____qualname__r   r+   r   r   rd    r,   r*   r"   r"      s        0 0 0 _0  59FJ6:*.+/377;.2:>-1(,59"&59DHBFJN=A6:(,$("&049F F F F F  ^F F Fr,   c                      e Zd Zed4d            Zeedddddddddddddddddddddddddd5d3                        ZdS )6r#   r%   r   c                     t          |           S r'   )r   r(   s    r*   r+   z"AsyncCompletions.with_raw_responseo   s    .t444r,   N)r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r@   r8   r9   r:   r;   r<   r=   r>   r?   rA   rB   rC   rD   rE   rF   rG   rH   r-   rI   r.   rJ   r/   rK   r0   rL   r1   rM   r2   r3   rN   r4   r5   rO   r6   r7   r@   rT   r8   r9   r:   r;   rP   r<   rQ   r=   rR   r>   rS   r?   rA   rU   rB   rV   rC   rW   rD   rX   r   c               X  K   t          | j        |          }| j                            |           d {V }t          ||| j        di d|d|d|d|d|d|d|d	|d
|	d|
d|d|d|d|d|d|d||||||dt          ||||          t                     d {V }|S rZ   )r    r^   r_   r   r`   r   r   )r)   rE   rG   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r@   r8   r9   r:   r;   r<   r=   r>   r?   rA   rB   rC   rD   ra   rb   rc   s                                 r*   rd   zAsyncCompletions.creates   s     @ )w??66u========+$%H $%6  	
 j H j #$4  {     %&8  Q!" &':#$ !-*#2$)>-  0 )+'%	   #G$
 $
 $
 $
 $
 $
 $
 $
 $
L r,   )r%   r   )6rE   rF   rG   rH   r-   rI   r.   rJ   r/   rK   r0   rL   r1   rM   r2   rI   r3   rN   r4   rI   r5   rO   r6   rM   r7   rI   r@   rT   r8   rI   r9   rM   r:   rL   r;   rP   r<   rQ   r=   rR   r>   rS   r?   rM   rA   rU   rB   rV   rC   rW   rD   rX   r%   r   )re   rf   rg   r   r+   r   r   rd   rh   r,   r*   r#   r#   n   s        5 5 5 _5  59FJ6:*.+/377;.2:>-1(,59"&59DHBFJN=A6:(,$("&049G G G G G  G G Gr,   )+
__future__r   typingr   r   r   r   r   httpxtyping_extensionsr	   _base_clientr   _compatr   	_resourcer   r   _typesr   r   r   _utilsr   r   
encryptionr   r   
types.chatr   r   r   r   r   chat.completionsr   r   r   r    r!   __all__r"   r#   rh   r,   r*   <module>ry      s   " " " " " " 8 8 8 8 8 8 8 8 8 8 8 8 8 8  % % % % % % 1 1 1 1 1 1 ' ' ' ' ' ' ; ; ; ; ; ; ; ; + + + + + + + + + + ; ; ; ; ; ; ; ; H H H H H H H H                     U T T T T T T T T T,
-M M M M M/ M M M`N N N N N' N N N N Nr,   