
    ~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)ContextChatCompletionContextChatCompletionChunk)ChatCompletionMessageParamcompletion_create_params ChatCompletionStreamOptionsParamChatCompletionToolParam#ChatCompletionToolChoiceOptionParamCompletionsAsyncCompletionsc                      e Zd Zed5d            Zed6d            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d7d4            ZdS )8r!   returnCompletionsWithRawResponsec                     t          |           S N)r%   selfs    ~/root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/resources/context/completions.pywith_raw_responsezCompletions.with_raw_response-   s    )$///     CompletionsWithStreamingResponsec                     t          |           S r'   )r-   r(   s    r*   with_streaming_responsez#Completions.with_streaming_response1   s    /555r,   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tool_choiceresponse_formatuserextra_headersextra_query
extra_bodytimeout
context_idstrmessages$Iterable[ChatCompletionMessageParam]modelr0   Optional[float] | Noner1   ,completion_create_params.FunctionCall | Noner2   Optional[Dict[str, int]] | Noner3   Optional[bool] | Noner4   Optional[int] | Noner5   r6   &Union[Optional[str], List[str]] | Noner7   /Optional[Literal[False]] | Literal[True] | Noner8   1Optional[ChatCompletionStreamOptionsParam] | Noner9   r:   (Iterable[ChatCompletionToolParam] | Noner;   r<   r=   r>   r?   *ChatCompletionToolChoiceOptionParam | Noner@   .completion_create_params.ResponseFormat | NonerA   
str | NonerB   Headers | NonerC   Query | NonerD   Body | NonerE   float | httpx.Timeout | None:ContextChatCompletion | Stream[ContextChatCompletionChunk]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/context/chat/completionsrF   rH   rJ   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   rA   )r=   r>   r?   r@   )rB   rC   rD   rE   F)bodyoptionscast_tor7   
stream_cls)_postr   r   r   r   )r)   rF   rH   rJ   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   s                             r*   createzCompletions.create5   sD   < zz'jH  $%6	
   j H j #$4  & !. {    !" #$ '9*#2+  . )+'%	   *?U89C  "
 "
 "	
r,   )r$   r%   )r$   r-   )4rF   rG   rH   rI   rJ   rG   r0   rK   r1   rL   r2   rM   r3   rN   r4   rO   r5   rK   r6   rP   r7   rQ   r8   rR   r9   rK   r:   rS   r;   rO   r<   rK   r=   rK   r>   rO   r?   rT   r@   rU   rA   rV   rB   rW   rC   rX   rD   rY   rE   rZ   r$   r[   )__name__
__module____qualname__r   r+   r/   r   rc    r,   r*   r!   r!   ,   s        0 0 0 _0 6 6 6 _6  59FJ6:*.+/377;BFLP.2:>-1(,59"&BFJN(,$("&047?
 ?
 ?
 ?
 ?
 ^?
 ?
 ?
r,   c                      e Zd Zed5d            Zed6d            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d7d4            ZdS )8r"   r$   AsyncCompletionsWithRawResponsec                     t          |           S r'   )ri   r(   s    r*   r+   z"AsyncCompletions.with_raw_responsey   s    .t444r,   %AsyncCompletionsWithStreamingResponsec                     t          |           S r'   )rk   r(   s    r*   r/   z(AsyncCompletions.with_streaming_response}   s    4T:::r,   N)r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   rA   r=   r>   r?   r@   rB   rC   rD   rE   rF   rG   rH   rI   rJ   r0   rK   r1   rL   r2   rM   r3   rN   r4   rO   r5   r6   rP   r7   rQ   r8   rR   r9   r:   rS   r;   r<   rA   rV   r=   r>   r?   rT   r@   rU   rB   rW   rC   rX   rD   rY   rE   rZ   ?ContextChatCompletion | AsyncStream[ContextChatCompletionChunk]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 r]   )rb   r   r   r   r   )r)   rF   rH   rJ   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   rA   r=   r>   r?   r@   rB   rC   rD   rE   s                             r*   rc   zAsyncCompletions.create   sf     < ZZ'jH  $%6	
   j H j #$4  & !. {    !" #$ '9*#2+  . )+'%	   *?U"#=>C   "
 "
 "
 "
 "
 "
 "
 "
 "	
r,   )r$   ri   )r$   rk   )4rF   rG   rH   rI   rJ   rG   r0   rK   r1   rL   r2   rM   r3   rN   r4   rO   r5   rK   r6   rP   r7   rQ   r8   rR   r9   rK   r:   rS   r;   rO   r<   rK   rA   rV   r=   rK   r>   rO   r?   rT   r@   rU   rB   rW   rC   rX   rD   rY   rE   rZ   r$   rm   )rd   re   rf   r   r+   r/   r   rc   rg   r,   r*   r"   r"   x   s        5 5 5 _5 ; ; ; _;  59FJ6:*.+/377;BFLP.2:>-1(,59"&BFJN(,$("&047?
 ?
 ?
 ?
 ?
 ?
 ?
 ?
r,   c                      e Zd ZddZdS )r%   completionsr!   r$   Nonec                F    || _         t          |j                  | _        d S r'   )_completionsr   rc   r)   rp   s     r*   __init__z#CompletionsWithRawResponse.__init__   s%    '-
 
r,   Nrp   r!   r$   rq   rd   re   rf   ru   rg   r,   r*   r%   r%      (        
 
 
 
 
 
r,   r%   c                      e Zd ZddZdS )ri   rp   r"   r$   rq   c                F    || _         t          |j                  | _        d S r'   )rs   r   rc   rt   s     r*   ru   z(AsyncCompletionsWithRawResponse.__init__   s%    '3
 
r,   Nrp   r"   r$   rq   rw   rg   r,   r*   ri   ri      rx   r,   ri   c                      e Zd ZddZdS )r-   rp   r!   r$   rq   c                F    || _         t          |j                  | _        d S r'   )rs   r   rc   rt   s     r*   ru   z)CompletionsWithStreamingResponse.__init__   s%    '2
 
r,   Nrv   rw   rg   r,   r*   r-   r-      rx   r,   r-   c                      e Zd ZddZdS )rk   rp   r"   r$   rq   c                F    || _         t          |j                  | _        d S r'   )rs   r   rc   rt   s     r*   ru   z.AsyncCompletionsWithStreamingResponse.__init__   s%    '8
 
r,   Nr{   rw   rg   r,   r*   rk   rk      rx   r,   rk   )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.contextr   r   
types.chatr   r   r   r   r    __all__r!   r"   r%   ri   r-   rk   rg   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 : : : : : : : : & & & & & &            . - - - - - - - N N N N N N N N              ,
-I
 I
 I
 I
 I
/ I
 I
 I
XI
 I
 I
 I
 I
' I
 I
 I
X
 
 
 
 
 
 
 

 
 
 
 
 
 
 

 
 
 
 
 
 
 

 
 
 
 
 
 
 
 
 
r,   