
    ~in&                       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mZmZ ddlmZmZ dd	lmZmZ dd
lmZ ddlmZmZ ddlmZ ddlmZ ddlmZm Z m!Z!m"Z" ddl#m$Z$m%Z% ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- ddgZ. G d de          Z/ G d de          Z0 G d d          Z1 G d d          Z2 G d d          Z3 G d d          Z4dS )    )annotations)DictListUnionIterableOptionalN)Literal   )BodyQueryHeaders)with_sts_tokenasync_with_sts_token   )with_e2e_encryptionasync_with_e2e_encryption)make_request_options)SyncAPIResourceAsyncAPIResource)cached_property)ReasoningEffort)to_raw_response_wrapperasync_to_raw_response_wrapperto_streamed_response_wrapper"async_to_streamed_response_wrapper)StreamAsyncStream)ChatCompletionChatCompletionChunkChatCompletionMessageParamcompletion_create_params ChatCompletionStreamOptionsParamChatCompletionToolParam#ChatCompletionToolChoiceOptionParamCompletionsAsyncCompletionsc                      e Zd Zed<d            Zed=d            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dddd>d;                        ZdS )?r%   returnCompletionsWithRawResponsec                     t          |           S N)r)   selfs    {/root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/resources/chat/completions.pywith_raw_responsezCompletions.with_raw_response5   s    )$///     CompletionsWithStreamingResponsec                     t          |           S r+   )r1   r,   s    r.   with_streaming_responsez#Completions.with_streaming_response9   s    /555r0   N)frequency_penaltyfunction_call
logit_biaslogprobs
max_tokenspresence_penaltystopstreamstream_optionstemperaturetoolstop_logprobstop_prepetition_penaltynparallel_tool_callsservice_tiertool_choiceresponse_formatthinkingmax_completion_tokensreasoning_effortuserextra_headersextra_query
extra_bodytimeoutmessages$Iterable[ChatCompletionMessageParam]modelstrr4   Optional[float] | Noner5   ,completion_create_params.FunctionCall | Noner6   Optional[Dict[str, int]] | Noner7   Optional[bool] | Noner8   Optional[int] | Noner9   r:   &Union[Optional[str], List[str]] | Noner;   /Optional[Literal[False]] | Literal[True] | Noner<   1Optional[ChatCompletionStreamOptionsParam] | Noner=   r>   (Iterable[ChatCompletionToolParam] | Noner?   r@   rA   rB   rC   rD   +Optional[Literal['auto', 'default']] | NonerE   *ChatCompletionToolChoiceOptionParam | NonerF   .completion_create_params.ResponseFormat | NonerG   (completion_create_params.Thinking | NonerH   rI    Optional[ReasoningEffort] | NonerJ   
str | NonerK   Headers | NonerL   Query | NonerM   Body | NonerN   float | httpx.Timeout | None,ChatCompletion | Stream[ChatCompletionChunk]c                  |                      di d|d|d|d|d|d|d|d	|d
|	d|
d|d|d|d|d|d|d|||||||||dt          ||||          t          |
pdt          t                             }|S Nz/chat/completionsrO   rQ   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rJ   rA   )rB   rC   rD   rE   rF   rG   rH   rI   )rK   rL   rM   rN   F)bodyoptionscast_tor;   
stream_cls)_postr   r   r   r   )r-   rO   rQ   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   resps                                  r.   createzCompletions.create=   sW   F zzH $%6  	
 j H j #$4  & !. {     !" %&8#$ ': ,*#2$)>$43  6 )+'%	   #?U12K  &
 &
N r0   )r(   r)   )r(   r1   )<rO   rP   rQ   rR   r4   rS   r5   rT   r6   rU   r7   rV   r8   rW   r9   rS   r:   rX   r;   rY   r<   rZ   r=   rS   r>   r[   r?   rW   r@   rS   rA   rS   rB   rW   rC   rV   rD   r\   rE   r]   rF   r^   rG   r_   rH   rW   rI   r`   rJ   ra   rK   rb   rL   rc   rM   rd   rN   re   r(   rf   )	__name__
__module____qualname__r   r/   r3   r   r   ro    r0   r.   r%   r%   4   s        0 0 0 _0 6 6 6 _6  59FJ6:*.+/377;BFLP.2:>-1(,59"&59DHBFJN=A6:=A(,$("&04?H H H H H  ^H H Hr0   c                      e Zd Zed<d            Zed=d            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dddd>d;                        ZdS )?r&   r(   AsyncCompletionsWithRawResponsec                     t          |           S r+   )ru   r,   s    r.   r/   z"AsyncCompletions.with_raw_response   s    .t444r0   %AsyncCompletionsWithStreamingResponsec                     t          |           S r+   )rw   r,   s    r.   r3   z(AsyncCompletions.with_streaming_response   s    4T:::r0   N)r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rJ   rA   rB   rC   rD   rE   rF   rG   rH   rI   rK   rL   rM   rN   rO   rP   rQ   rR   r4   rS   r5   rT   r6   rU   r7   rV   r8   rW   r9   r:   rX   r;   rY   r<   rZ   r=   r>   r[   r?   r@   rJ   ra   rA   rB   rC   rD   r\   rE   r]   rF   r^   rG   r_   rH   rI   r`   rK   rb   rL   rc   rM   rd   rN   re   1ChatCompletion | AsyncStream[ChatCompletionChunk]c                 K   |                      di d|d|d|d|d|d|d|d	|d
|	d|
d|d|d|d|d|d|d|||||||||dt          ||||          t          |
pdt          t                              d {V }|S rh   )rm   r   r   r   r   )r-   rO   rQ   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rJ   rA   rB   rC   rD   rE   rF   rG   rH   rI   rK   rL   rM   rN   rn   s                                  r.   ro   zAsyncCompletions.create   sy     F ZZH $%6  	
 j H j #$4  & !. {     !" %&8#$ ': ,*#2$)>$43  6 )+'%	   #?U"#67K   &
 &
 &
 &
 &
 &
 &
 &
P r0   )r(   ru   )r(   rw   )<rO   rP   rQ   rR   r4   rS   r5   rT   r6   rU   r7   rV   r8   rW   r9   rS   r:   rX   r;   rY   r<   rZ   r=   rS   r>   r[   r?   rW   r@   rS   rJ   ra   rA   rS   rB   rW   rC   rV   rD   r\   rE   r]   rF   r^   rG   r_   rH   rW   rI   r`   rK   rb   rL   rc   rM   rd   rN   re   r(   ry   )	rp   rq   rr   r   r/   r3   r   r   ro   rs   r0   r.   r&   r&      s        5 5 5 _5 ; ; ; _;  59FJ6:*.+/377;BFLP.2:>-1(,59"&59DHBFJN=A6:=A(,$("&04?I I I I I  I I Ir0   c                      e Zd ZddZdS )r)   completionsr%   r(   Nonec                F    || _         t          |j                  | _        d S r+   )_completionsr   ro   r-   r|   s     r.   __init__z#CompletionsWithRawResponse.__init__   s%    '-
 
r0   Nr|   r%   r(   r}   rp   rq   rr   r   rs   r0   r.   r)   r)      (        
 
 
 
 
 
r0   r)   c                      e Zd ZddZdS )ru   r|   r&   r(   r}   c                F    || _         t          |j                  | _        d S r+   )r   r   ro   r   s     r.   r   z(AsyncCompletionsWithRawResponse.__init__   s%    '3
 
r0   Nr|   r&   r(   r}   r   rs   r0   r.   ru   ru      r   r0   ru   c                      e Zd ZddZdS )r1   r|   r%   r(   r}   c                F    || _         t          |j                  | _        d S r+   )r   r   ro   r   s     r.   r   z)CompletionsWithStreamingResponse.__init__   s%    '2
 
r0   Nr   r   rs   r0   r.   r1   r1      r   r0   r1   c                      e Zd ZddZdS )rw   r|   r&   r(   r}   c                F    || _         t          |j                  | _        d S r+   )r   r   ro   r   s     r.   r   z.AsyncCompletionsWithStreamingResponse.__init__   s%    '8
 
r0   Nr   r   rs   r0   r.   rw   rw      r   r0   rw   )5
__future__r   typingr   r   r   r   r   httpxtyping_extensionsr	   _typesr   r   r   _utils._utilsr   r   
encryptionr   r   _base_clientr   	_resourcer   r   _compatr   types.shared.reasoning_effortr   	_responser   r   r   r   
_streamingr   r   
types.chatr   r   r    r!   r"   r#   r$   __all__r%   r&   r)   ru   r1   rw   rs   r0   r.   <module>r      s   # " " " " "               % % % % % % * * * * * * * * * * A A A A A A A A G G G G G G G G 0 0 0 0 0 0 : : : : : : : : & & & & & & < < < < < <            . - - - - - - -                  ,
-S S S S S/ S S SlT T T T T' T T Tn
 
 
 
 
 
 
 

 
 
 
 
 
 
 

 
 
 
 
 
 
 

 
 
 
 
 
 
 
 
 
r0   