
    c
i                    *   d dl mZ d dlZd dlZd dlmZmZmZ d dlZd dl	Z	 ej
        e          Z G d de          Z G d dej        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/d%Zd0d(Zd0d)Zd1d+Z d2d-Z!d2d.Z"dS )3    )annotationsN)AnyLiteralcastc                      e Zd ZdS )LangGraphErrorN__name__
__module____qualname__     C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph_sdk/errors.pyr   r              Dr   r   c                  T    e Zd ZU ded<   ded<   ded<   ded<   ded	<   ded
<   ddZdS )APIErrorstrmessagehttpx.Requestrequestobject | Nonebody
str | Nonecodeparamtyperesponse_or_requesthttpx.Response | httpx.RequestreturnNonec                  t          |t          j                  r
|j        }|}n|}d }t          j                            | |||           t                              | |           || _        || _        || _        t          |t                    rt          d|          }|                    d          }t          |t                    r|nd | _        |                    d          }t          |t                    r|nd | _        |                    d          }	t          |	t                    r|	nd | _        d S d | _        d | _        d | _        d S )N)r   responsedict[str, Any]r   r   r   )
isinstancehttpxResponser   HTTPStatusError__init__r   r   r   dictr   getr   r   r   r   )
selfr   r   r   reqr"   bcode_val	param_valts
             r   r(   zAPIError.__init__   s7    )5>:: 	%-C*HH%CH&&tWcH&UUUg...	dD!! 	%t,,AuuV}}H$.x$=$=G4DIgI&0C&@&@JdDJfA'3//9TDIIIDIDJDIIIr   N)r   r   r   r   r   r   r   r    )r
   r   r   __annotations__r(   r   r   r   r   r      so         LLL! ! ! ! ! !r   r   c                  <     e Zd ZU ded<   ded<   ddd fdZ xZS )APIResponseValidationErrorhttpx.Responser"   intstatus_codeNr   r   r   r   r   r   r    c               z    t                                          |pd||           || _        |j        | _        d S )Nz1Data returned by API invalid for expected schema.r   )superr(   r"   r6   )r+   r"   r   r   	__class__s       r   r(   z#APIResponseValidationError.__init__B   sO     	JJ 	 	
 	
 	

 !#/r   )r"   r4   r   r   r   r   r   r    r
   r   r   r1   r(   __classcell__r;   s   @r   r3   r3   >   sf          #0 0 0 0 0 0 0 0 0 0 0 0r   r3   c                  @     e Zd ZU ded<   ded<   ded<   d fdZ xZS )APIStatusErrorr4   r"   r5   r6   r   
request_idr   r   r   r   r   r    c                   t                                          |||           || _        |j        | _        |j                            d          | _        d S )Nr9   zx-request-id)r:   r(   r"   r6   headersr*   rA   )r+   r   r"   r   r;   s       r   r(   zAPIStatusError.__init__W   sR     	(666 #/"*..~>>r   )r   r   r"   r4   r   r   r   r    r<   r>   s   @r   r@   r@   R   sc         ? ? ? ? ? ? ? ? ? ?r   r@   c                  &     e Zd Zddd
 fd	Z xZS )APIConnectionErrorzConnection error.r7   r   r   r   r   r   r    c               P    t                                          ||d            d S )N)r   r   r:   r(   )r+   r   r   r;   s      r   r(   zAPIConnectionError.__init__a   s*     	gDIIIIIr   )r   r   r   r   r   r    r
   r   r   r(   r=   r>   s   @r   rE   rE   `   sV         3J J J J J J J J J J J Jr   rE   c                        e Zd Zd fdZ xZS )APITimeoutErrorr   r   r   r    c                N    t                                          d|           d S )NzRequest timed out.)r   r   rG   )r+   r   r;   s     r   r(   zAPITimeoutError.__init__h   s'    !5wGGGGGr   )r   r   r   r    rH   r>   s   @r   rJ   rJ   g   sG        H H H H H H H H H Hr   rJ   c                      e Zd ZU dZded<   dS )BadRequestError  zLiteral[400]r6   Nr
   r   r   r6   r1   r   r   r   rM   rM   l   #          #K######r   rM   c                      e Zd ZU dZded<   dS )AuthenticationError  zLiteral[401]r6   NrO   r   r   r   rR   rR   p   rP   r   rR   c                      e Zd ZU dZded<   dS )PermissionDeniedError  zLiteral[403]r6   NrO   r   r   r   rU   rU   t   rP   r   rU   c                      e Zd ZU dZded<   dS )NotFoundError  zLiteral[404]r6   NrO   r   r   r   rX   rX   x   rP   r   rX   c                      e Zd ZU dZded<   dS )ConflictError  zLiteral[409]r6   NrO   r   r   r   r[   r[   |   rP   r   r[   c                      e Zd ZU dZded<   dS )UnprocessableEntityError  zLiteral[422]r6   NrO   r   r   r   r^   r^      rP   r   r^   c                      e Zd ZU dZded<   dS )RateLimitError  zLiteral[429]r6   NrO   r   r   r   ra   ra      rP   r   ra   c                      e Zd ZdS )InternalServerErrorNr	   r   r   r   rd   rd      r   r   rd   r   r   fallbackr   r   c                   t          | t                    rt          d|           }dD ]2}|                    |          }t          |t                    r|r|c S 3|                    d          }t          |t                    rEt          d|          }dD ]2}|                    |          }t          |t                    r|r|c S 3|S )Nr#   )r   detailerrorrh   )r   rg   )r$   r)   r   r*   r   )r   re   r-   keyvalerres          r   _extract_error_messagerm      s    $ !4((1 	 	C%%**C#s##  


eeGnnc4   	%s++A,  eeCjjc3'' C JJJOr   rr4   c                   K   	 |                                   d {V }n# t          $ r Y d S w xY w|sd S 	 t          j        |          S # t          $ r* 	 |                                cY S # t          $ r Y Y d S w xY ww xY wN)aread	Exceptionorjsonloadsdecodern   datas     r   _adecode_error_bodyrx      s      WWYY   tt t|D!!!   	;;==    	 	 	444	s;    
--A	 	
A=A*'A=*
A94A=8A99A=c                    	 |                                  }n# t          $ r Y d S w xY w|sd S 	 t          j        |          S # t          $ r* 	 |                                cY S # t          $ r Y Y d S w xY ww xY wrp   )readrr   rs   rt   ru   rv   s     r   _decode_error_bodyr{      s    vvxx   tt t|D!!!   	;;==    	 	 	444	s;    
%%A 
A5A"A5"
A1,A50A11A5r"   c                   | j         }| j        pd}t          || d|           }|dk    rt          || |          S |dk    rt	          || |          S |dk    rt          || |          S |dk    rt          || |          S |dk    rt          || |          S |d	k    rt          || |          S |d
k    rt          || |          S |dk    rt          || |          S t          || |          S )Nz
HTTP Error rN   )r"   r   rS   rV   rY   r\   r_   rb   i  )r6   reason_phraserm   rM   rR   rU   rX   r[   r^   ra   rd   r@   )r"   r   statusreasonr   s        r   _map_status_errorr      s9   !F#3|F$Tf+?+?v+?+?@@G}}wEEEE}}"7XDIIII}}$WxdKKKK}}WxdCCCC}}WxdCCCC}}'(NNNN}}gtDDDD}}"7XDIIII'H4@@@@r   r    c                   K   | j         dk     rd S t          |            d {V }t          | |          }t          j        dk    s,t
                              dt          |dd                      |NrN   )      zError from langgraph-api: r    )r6   rx   r   sysversion_infologgerrh   getattrrn   r   rk   s      r   _araise_for_status_typedr      s      }s$Q''''''''D
At
$
$C''O'#y"2M2MOOPPP
Ir   c                    | j         dk     rd S t          |           }t          | |          }t          j        dk    s,t
                              dt          |dd                      |r   )r6   r{   r   r   r   r   rh   r   r   s      r   _raise_for_status_typedr      sj    }sa  D
At
$
$C''O'#y"2M2MOOPPP
Ir   )r   r   re   r   r   r   )rn   r4   r   r   )r"   r4   r   r   r   r@   )rn   r4   r   r    )#
__future__r   loggingr   typingr   r   r   r%   rs   	getLoggerr
   r   rr   r   r'   r   r3   r@   rE   rJ   rM   rR   rU   rX   r[   r^   ra   rd   rm   rx   r{   r   r   r   r   r   r   <module>r      sT   " " " " " "  



 % % % % % % % % % %  		8	$	$	 	 	 	 	Y 	 	 	* * * * *u$n * * *Z0 0 0 0 0 0 0 0(? ? ? ? ?X ? ? ?J J J J J J J JH H H H H( H H H
$ $ $ $ $n $ $ $$ $ $ $ $. $ $ $$ $ $ $ $N $ $ $$ $ $ $ $N $ $ $$ $ $ $ $N $ $ $$ $ $ $ $~ $ $ $$ $ $ $ $^ $ $ $	 	 	 	 	. 	 	 	   $        A A A A.        r   