
    ~i                       d dl mZ d dlmZmZ d dlmZ d dlZerddlm	Z	 g dZ
 G d d	e          Z G d
 de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d  d!e          Z G d" d#e          Z G d$ d%e          Z G d& d'e          ZdS )(    )annotations)TYPE_CHECKINGOptional)LiteralN   )ChatCompletion)ArkBadRequestErrorArkAuthenticationErrorArkPermissionDeniedErrorArkNotFoundErrorArkConflictErrorArkUnprocessableEntityErrorArkRateLimitErrorArkInternalServerErrorc                      e Zd ZdS )ArkErrorN__name__
__module____qualname__     l/root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/_exceptions.pyr   r   "           Dr   r   c                       e Zd ZU ded<   dZded<   dZded<   dZded	<   	 dZded
<   dZded<   ded<   	 ddddd fdZ	d Z
 xZS )ArkAPIErrorstrmessageNOptional[httpx.Request]requestOptional[object]bodyOptional[str]
request_idcodeparamtyper"   r$   returnNonec               v   t                                          |           || _        || _        || _        || _        t          |t                    rP|                    d          | _	        |                    d          | _
        |                    d          | _        d S d | _	        d | _
        d | _        d S )Nr%   r&   r'   )super__init__r    r   r"   r$   
isinstancedictgetr%   r&   r'   )selfr   r    r"   r$   	__class__s        r   r-   zArkAPIError.__init__9   s     	!!!	$dD!! 	((DI'**DJ((DIIIDIDJDIIIr   c                $    | j          d| j         S )Nz, request_id: )r   r$   )r1   s    r   __str__zArkAPIError.__str__P   s    ,??do???r   )N)
r   r   r    r   r"   r!   r$   r#   r)   r*   )r   r   r   __annotations__r    r"   r$   r%   r&   r-   r4   __classcell__r2   s   @r   r   r   &   s         LLL'+G++++!D!!!! $J$$$$ DE
 ,0
 "&$(       .@ @ @ @ @ @ @r   r   c                  <     e Zd ZU ded<   ded<   ddd fdZ xZS )ArkAPIResponseValidationErrorhttpx.Responseresponseintstatus_codeNr   r"   object | Noner   
str | Noner$   r   r)   r*   c                   t                                          |pd|j        |           || _        |j        | _        || _        d S )Nz1Data returned by API invalid for expected schema.)r"   r,   r-   r    r;   r=   r$   )r1   r;   r"   r   r$   r2   s        r   r-   z&ArkAPIResponseValidationError.__init__X   sW     	JJ 	 	
 	
 	

 !#/$r   )
r;   r:   r"   r?   r   r@   r$   r   r)   r*   r   r   r   r5   r-   r6   r7   s   @r   r9   r9   T   sf          #% % % % % % % % % % % %r   r9   c                  :     e Zd ZU dZded<   ded<   d fdZ xZS )ArkAPIStatusErrorz<Raised when an API response has a status code of 4xx or 5xx.r:   r;   r<   r=   r   r   r"   r?   r$   r)   r*   c                   t                                          ||j        ||           || _        |j        | _        || _        d S Nr(   rB   )r1   r   r;   r"   r$   r2   s        r   r-   zArkAPIStatusError.__init__p   sF     	("2*UUU #/$r   )
r   r   r;   r:   r"   r?   r$   r   r)   r*   )r   r   r   __doc__r5   r-   r6   r7   s   @r   rE   rE   j   s]         FF% % % % % % % % % %r   rE   c                  &     e Zd Zddd fd
Z xZS )ArkAPIConnectionErrorzConnection error.r>   r   r   r    httpx.Requestr$   r)   r*   c               R    t                                          ||d |           d S rG   r,   r-   )r1   r   r    r$   r2   s       r   r-   zArkAPIConnectionError.__init__   s,     	'LLLLLr   )r   r   r    rK   r$   r   r)   r*   r   r   r   r-   r6   r7   s   @r   rJ   rJ   ~   sX         +M M M M M M M M M M M Mr   rJ   c                        e Zd Zd fdZ xZS )	ArkAPITimeoutErrorr    rK   r$   r   r)   r*   c                P    t                                          d||           d S )NzRequest timed out.)r   r    r$   rM   )r1   r    r$   r2   s      r   r-   zArkAPITimeoutError.__init__   s5    ('j 	 	
 	
 	
 	
 	
r   )r    rK   r$   r   r)   r*   rN   r7   s   @r   rP   rP      =        
 
 
 
 
 
 
 
 
 
r   rP   c                      e Zd ZU dZded<   dS )r	   i  zLiteral[400]r=   Nr   r   r   r=   r5   r   r   r   r	   r	      #          #K######r   r	   c                      e Zd ZU dZded<   dS )r
   i  zLiteral[401]r=   NrT   r   r   r   r
   r
      rU   r   r
   c                      e Zd ZU dZded<   dS )r   i  zLiteral[403]r=   NrT   r   r   r   r   r      rU   r   r   c                      e Zd ZU dZded<   dS )r   i  zLiteral[404]r=   NrT   r   r   r   r   r      rU   r   r   c                      e Zd ZU dZded<   dS )r   i  zLiteral[409]r=   NrT   r   r   r   r   r      rU   r   r   c                      e Zd ZU dZded<   dS )r   i  zLiteral[422]r=   NrT   r   r   r   r   r      rU   r   r   c                      e Zd ZU dZded<   dS )r   i  zLiteral[429]r=   NrT   r   r   r   r   r      rU   r   r   c                      e Zd ZdS )r   Nr   r   r   r   r   r      r   r   r   c                  .     e Zd ZU ded<   	 d fdZ xZS )ArkLengthFinishReasonErrorr   
completionr)   r*   c                   d}|j         r|d|j          z  }t                                          |           || _        d S )Nz@Could not parse response content as the length limit was reachedz - )usager,   r-   r_   )r1   r_   msgr2   s      r   r-   z#ArkLengthFinishReasonError.__init__   sL    P 	,+)+++C$r   )r_   r   r)   r*   rC   r7   s   @r   r^   r^      sP         % % % % % % % % % %r   r^   c                        e Zd Zd fdZ xZS )!ArkContentFilterFinishReasonErrorr)   r*   c                J    t                                          d           d S )NzRCould not parse response content as the request was rejected by the content filterrM   )r1   r2   s    r   r-   z*ArkContentFilterFinishReasonError.__init__   s,    `	
 	
 	
 	
 	
r   )r)   r*   rN   r7   s   @r   rd   rd      rR   r   rd   )
__future__r   typingr   r   typing_extensionsr   httpx
types.chatr   __all__	Exceptionr   r   r9   rE   rJ   rP   r	   r
   r   r   r   r   r   r   r^   rd   r   r   r   <module>rm      s;   # " " " " " * * * * * * * * % % % % % %  +******	 	 		 	 	 	 	y 	 	 	+@ +@ +@ +@ +@( +@ +@ +@\% % % % %K % % %,% % % % % % % %(M M M M MK M M M
 
 
 
 
. 
 
 
$ $ $ $ $* $ $ $$ $ $ $ $. $ $ $$ $ $ $ $0 $ $ $$ $ $ $ $( $ $ $$ $ $ $ $( $ $ $$ $ $ $ $"3 $ $ $$ $ $ $ $) $ $ $	 	 	 	 	. 	 	 	% % % % % % % %"
 
 
 
 
 
 
 
 
 
r   