+jI%@dZddlmZmZmZmZmZddlZddlZddl Z ddl m Z erddl m Z ddlZddlmZGdd eZGd d eZGd d eZGddeZGddeZGddeZGddeZe jZdS)z Error classes for the GenAI SDK.)AnyCallableOptional TYPE_CHECKINGUnionN)_common)ReplayResponse)Responsec eZdZUdZeed<eejde jdfed<dZ e e ed<dZ e e ed< dded ede eejde jdfffd Zd eed eee efffd Zedee efd dfdZd ed efdZd ed efdZd ed efdZd ejfdZedede jejfd dfdZeded ede ede jejfd dfdZedede jddfd dfdZeded ede ede jdfd dfdZ xZ!S)APIErrorz'General errors raised by the GenAI API.coder AsyncAuthorizedSessionResponseresponseNstatusmessage response_jsonct|trt|dkr|d}||_||_|||_|||_|r|n| ||_ t |j d|jd|jdS)Nrr . ) isinstancelistlenrdetails _get_messager _get_statusr _get_codersuper__init__)selfrrr __class__s W/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/errors.pyrzAPIError.__init__,s-&&'3}+=+=+B+B#A&mDM DL$$]33DL""=11DK?$.."?"?DI GG AADKAA4<AABBBBBreturn).r cR|j}|jj|ffS)zGReturns a tuple that can be used to reconstruct the error for pickling.)__dict__copyr!_rebuild)r states r" __reduce__zAPIError.__reduce__Ds) M   E N #eX ..r#r)ctt}|j|t||jd|jd|j|S)z"Rebuilds the error from the state.rr) r __new__r&update Exceptionrrrr)r)objs r"r(zAPIError._rebuildKsd   8 $ $CL ssxEE#*EE EEFFF Jr#c |d|didd}|S#t$rYdSwxYw)NrerrorgetAttributeError)r rrs r"rzAPIError._get_statusSsl  M%%gr2266xFFfm TT?A AAc |d|didd}|S#t$rYdSwxYw)Nrr1r2)r rrs r"rzAPIError._get_message^sl!! ]&&w3377 4HHgn TTr5c~|d|diddS)Nrr1)r3)r rs r"rzAPIError._get_codeis?    !!'2..2264@@  r#c0d|j|j|jdiS)zReturns a dictionary representation of the error for replay recording. details is not included since it may expose internal information in the replay file. r1rrrr9)r s r"_to_replay_recordzAPIError._to_replay_recordns* I|k   r#c*|jdkrdSt|tjrU ||}n#t jj$r|j}||j d}Yn~wxYwt|tjr? |}nJ#tj j$r|j|j d}Yn%wxYw|j ddi}||j||dS)PRaises an error with detailed error message if the response has an error status.Nrrrr1) status_coderhttpxr readjsondecoderJSONDecodeErrortext reason_phraserequests exceptionsreason body_segmentsr3 raise_error)clsrrrs r"raise_for_responsezAPIError.raise_for_response|s4 s"" f(EN++A    \ )   -,   Hh/ 0 0 A !   0   }o   ,Q/33GR@@mOOH(-BBBBBs#(A%A:9A:B--#CCr?cd|cxkrdkrnnt|||d|cxkrdkrnnt|||||||aRaises an appropriate APIError subclass based on the status code. Args: status_code: The HTTP status code of the response. response_json: The JSON body of the response, or a dict containing error details. response: The original response object. Raises: ClientError: If the status code is in the 4xx range. ServerError: If the status code is in the 5xx range. APIError: For other error status codes. iiiX ClientError ServerErrorrLr?rrs r"rKzAPIError.raise_errors4 kC  ]H = == ! ! ! !c ! ! ! ! !  ]H = == C ]H 5 55r#zaiohttp.ClientResponsecKt|tjrq|jdkrdS |d{V|}n+#t jj$r|j}||j d}YnwxYw|j}n@t|drFt|dr6|jdkrdS|j d di}|j}n ddl }|}t|dr|j}t||jrj|jdkrdS |d{V}n9#|jj$r'|d{V}||jd}YnwxYw|j}nt)d t+|n-#t,$r t)d t+|wxYw||||d{VdS) r<r=Nr>rJr?rr1 _responsezUnsupported response type: )rr@r r?areadrBrCrDrErFhasattrrJr3aiohttprUClientResponserclient_exceptionsContentTypeErrorrI ValueErrortype ImportErrorraise_error_async)rLrrrr?rXunwrapped_responses r"raise_for_async_responsez!APIError.raise_for_async_responses(EN++,I   $ $ nn  \ )   -,   (kk ? + +I-11I   $ $,Q/33GR@@m(kkI#+ %{ 3 3 <1;  ('*@ A A K  &# - - F "4"9"9";";;;;;;;MM*;   .3355555555G",3MMM  +1++IhIIJJ J IIIGtH~~GGHHHI    ]H E EEEEEEEEEEsA.A%BB%=F!%E?F!3E63F!5E66*F!!*G cKd|cxkrdkrnnt|||d|cxkrdkrnnt|||||||rOrPrSs r"r_zAPIError.raise_error_asyncs& kC  ]H = == ! ! ! !c ! ! ! ! !  ]H = == C ]H 5 55r#)N)"__name__ __module__ __qualname____doc__int__annotations__rrGr r@rrstrrrrtuplerdictr* staticmethodr(rrrr StringDictr: classmethodrMrKrar_ __classcell__)r!s@r"r r sE// )))  n&( &(3-'8C= CC CC n.0  CCCCCC0/ Xo &d38n(= =>////d38n< s s          SS !3    C+U^X=NNOC CCC;C@666 n!  6 666;6@8F  . " * ,8F 8F8F8F;8Ft66,/6;C  %.2JJ K<6 666;66666r#r ceZdZdZdS)rQz%Client error raised by the GenAI API.Nrcrdrerfr#r"rQrQ--$r#rQceZdZdZdS)rRz%Server error raised by the GenAI API.Nrqrrr#r"rRrRrsr#rRceZdZdZdS) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.Nrqrrr#r"rvrvs__$r#rvceZdZdZdS)UnsupportedFunctionErrorz*Raised when the function is not supported.Nrqrrr#r"rxrx#s22$r#rxceZdZdZdS)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.Nrqrrr#r"rzrz(sLL$r#rzceZdZdZdS)UnknownApiResponseErrorz?Raised when the response from the API cannot be parsed as JSON.Nrqrrr#r"r|r|-sGG$r#r|)rftypingrrrrrr@rBrGr replay_api_clientr rX!google.auth.aio.transport.aiohttpr rr.r rQrRr\rvrxrzr|ExperimentalWarningrrr#r"rs '&@@@@@@@@@@@@@@ [//////...ZZZZZZr6r6r6r6r6yr6r6r6j( ( z z j j1r#