
    ~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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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gZ* G d de          Z+ G d de          Z, G d d          Z- G d d          Z. G d d          Z/ G d d          Z0dS )    )annotations)DictListUnionIterableOptionalN)Literal   )BodyQueryHeaders)with_sts_tokenasync_with_sts_token)make_request_options)SyncAPIResourceAsyncAPIResource)cached_property)to_raw_response_wrapperasync_to_raw_response_wrapperto_streamed_response_wrapper"async_to_streamed_response_wrapper)StreamAsyncStream)BotChatCompletionBotChatCompletionChunk)ChatCompletionMessageParamcompletion_create_params ChatCompletionStreamOptionsParamChatCompletionToolParam#ChatCompletionToolChoiceOptionParamCompletionsAsyncCompletionsc                      e Zd Zed:d            Zed;d            Z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<d9            ZdS )=r!   returnCompletionsWithRawResponsec                     t          |           S N)r%   selfs    z/root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/resources/bot/completions.pywith_raw_responsezCompletions.with_raw_response0   s    )$///     CompletionsWithStreamingResponsec                     t          |           S r'   )r-   r(   s    r*   with_streaming_responsez#Completions.with_streaming_response4   s    /555r,   NF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usermetadataextra_headersextra_query
extra_bodytimeoutmessages$Iterable[ChatCompletionMessageParam]modelstrr1   Optional[float] | Noner2   ,completion_create_params.FunctionCall | Noner3   Optional[Dict[str, int]] | Noner4   Optional[bool] | Noner5   Optional[int] | Noner6   r7   &Union[Optional[str], List[str]] | Noner8   (Optional[Literal[False]] | Literal[True]r9   1Optional[ChatCompletionStreamOptionsParam] | Noner:   r;   (Iterable[ChatCompletionToolParam] | Noner<   r=   r>   r?   r@   rA   +Optional[Literal['auto', 'default']] | NonerB   *ChatCompletionToolChoiceOptionParam | NonerC   .completion_create_params.ResponseFormat | NonerD   
str | NonerE   Dict[str, object] | NonerF   Headers | NonerG   Query | NonerH   Body | NonerI   float | httpx.Timeout | None2BotChatCompletion | Stream[BotChatCompletionChunk]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/bots/chat/completionsrJ   rL   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   rD   r>   )r?   r@   rA   rB   rC   rE   )rF   rG   rH   rI   F)bodyoptionscast_tor8   
stream_cls)_postr   r   r   r   r)   rJ   rL   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   s                               r*   createzCompletions.create8   sL   @ zz$H $%6  	
 j H j #$4  & !. {     !" %&8#$ ': ,*#2$/  2 )+'%	   &?U45G  $
 $
 $	
r,   )r$   r%   )r$   r-   )8rJ   rK   rL   rM   r1   rN   r2   rO   r3   rP   r4   rQ   r5   rR   r6   rN   r7   rS   r8   rT   r9   rU   r:   rN   r;   rV   r<   rR   r=   rN   r>   rN   r?   rR   r@   rQ   rA   rW   rB   rX   rC   rY   rD   rZ   rE   r[   rF   r\   rG   r]   rH   r^   rI   r_   r$   r`   )__name__
__module____qualname__r   r+   r/   r   ri    r,   r*   r!   r!   /   s        0 0 0 _0 6 6 6 _6  59FJ6:*.+/377;;@LP.2:>-1(,59"&59DHBFJN-1(,$("&04;C
 C
 C
 C
 C
 ^C
 C
 C
r,   c                      e Zd Zed:d            Zed;d            Z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<d9            ZdS )=r"   r$   AsyncCompletionsWithRawResponsec                     t          |           S r'   )ro   r(   s    r*   r+   z"AsyncCompletions.with_raw_response   s    .t444r,   %AsyncCompletionsWithStreamingResponsec                     t          |           S r'   )rq   r(   s    r*   r/   z(AsyncCompletions.with_streaming_response   s    4T:::r,   NFr0   rJ   rK   rL   rM   r1   rN   r2   rO   r3   rP   r4   rQ   r5   rR   r6   r7   rS   r8   rT   r9   rU   r:   r;   rV   r<   r=   r>   r?   r@   rA   rW   rB   rX   rC   rY   rD   rZ   rE   r[   rF   r\   rG   r]   rH   r^   rI   r_   7BotChatCompletion | AsyncStream[BotChatCompletionChunk]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 rb   )rg   r   r   r   r   rh   s                               r*   ri   zAsyncCompletions.create   sn     @ ZZ$H $%6  	
 j H j #$4  & !. {     !" %&8#$ ': ,*#2$/  2 )+'%	   &?U"#9:G   $
 $
 $
 $
 $
 $
 $
 $
 $	
r,   )r$   ro   )r$   rq   )8rJ   rK   rL   rM   r1   rN   r2   rO   r3   rP   r4   rQ   r5   rR   r6   rN   r7   rS   r8   rT   r9   rU   r:   rN   r;   rV   r<   rR   r=   rN   r>   rN   r?   rR   r@   rQ   rA   rW   rB   rX   rC   rY   rD   rZ   rE   r[   rF   r\   rG   r]   rH   r^   rI   r_   r$   rs   )rj   rk   rl   r   r+   r/   r   ri   rm   r,   r*   r"   r"      s        5 5 5 _5 ; ; ; _;  59FJ6:*.+/377;;@LP.2:>-1(,59"&59DHBFJN-1(,$("&04;C
 C
 C
 C
 C
 C
 C
 C
r,   c                      e Zd ZddZdS )r%   completionsr!   r$   Nonec                F    || _         t          |j                  | _        d S r'   )_completionsr   ri   r)   rv   s     r*   __init__z#CompletionsWithRawResponse.__init__   s%    '-
 
r,   Nrv   r!   r$   rw   rj   rk   rl   r{   rm   r,   r*   r%   r%      (        
 
 
 
 
 
r,   r%   c                      e Zd ZddZdS )ro   rv   r"   r$   rw   c                F    || _         t          |j                  | _        d S r'   )ry   r   ri   rz   s     r*   r{   z(AsyncCompletionsWithRawResponse.__init__   s%    '3
 
r,   Nrv   r"   r$   rw   r}   rm   r,   r*   ro   ro      r~   r,   ro   c                      e Zd ZddZdS )r-   rv   r!   r$   rw   c                F    || _         t          |j                  | _        d S r'   )ry   r   ri   rz   s     r*   r{   z)CompletionsWithStreamingResponse.__init__   s%    '2
 
r,   Nr|   r}   rm   r,   r*   r-   r-      r~   r,   r-   c                      e Zd ZddZdS )rq   rv   r"   r$   rw   c                F    || _         t          |j                  | _        d S r'   )ry   r   ri   rz   s     r*   r{   z.AsyncCompletionsWithStreamingResponse.__init__   s%    '8
 
r,   Nr   r}   rm   r,   r*   rq   rq      r~   r,   rq   )1
__future__r   typingr   r   r   r   r   httpxtyping_extensionsr	   _typesr   r   r   _utils._utilsr   r   _base_clientr   	_resourcer   r   _compatr   	_responser   r   r   r   
_streamingr   r   types.bot_chatr   r   
types.chatr   r   r   r   r    __all__r!   r"   r%   ro   r-   rq   rm   r,   r*   <module>r      s   # " " " " " 8 8 8 8 8 8 8 8 8 8 8 8 8 8  % % % % % % * * * * * * * * * * A A A A A A A A 0 0 0 0 0 0 : : : : : : : : & & & & & &            . - - - - - - -                     ,
-M
 M
 M
 M
 M
/ M
 M
 M
`M
 M
 M
 M
 M
' M
 M
 M
`
 
 
 
 
 
 
 

 
 
 
 
 
 
 

 
 
 
 
 
 
 

 
 
 
 
 
 
 
 
 
r,   