$ik!4dZddlmZmZmZmZmZddlZddlZddl Z ddl m Z er ddl m Z ddlZGddeZGd d eZGd d eZGd deZGddeZGddeZGddeZe jZdS)z Error classes for the GenAI SDK.)AnyCallableOptional TYPE_CHECKINGUnionN)_common)ReplayResponsec eZdZUdZeed<edejdfed<dZ e e ed<dZ e e ed< dded e de edejdfffd Zd eed eee e fffd Zedee e fd dfdZd e d e fdZd e d e fdZd e d e fdZd ejfdZededejfd dfdZeded e de edejdfd dfdZededejdfd dfdZeded e de edejdfd dfdZxZ S)APIErrorz'General errors raised by the GenAI API.coder zaiohttp.ClientResponseresponseNstatusmessage response_jsonct|trt|dkr|d}||_||_|||_|||_|r|n| ||_ t |j d|jd|jdS)Nrr . ) isinstancelistlenrdetails _get_messager _get_statusr _get_coder super__init__)selfr rr __class__s ]/var/www/html/jarvis.com/web/backend/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)rstates r __reduce__zAPIError.__reduce__9s) 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 Exceptionrr rr)r'objs r r&zAPIError._rebuild@sd   8 $ $CL ssxEE#*EE EEFFF Jr!c |d|didd}|S#t$rYdSwxYw)NrerrorgetAttributeError)rrrs r rzAPIError._get_statusHsl  M%%gr2266xFFfm TT?A AAc |d|didd}|S#t$rYdSwxYw)Nrr/r0)rrrs r rzAPIError._get_messageSsl!! ]&&w3377 4HHgn TTr3c~|d|diddS)Nr r/)r1)rrs r rzAPIError._get_code^s?    !!'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. r/r rrr7)rs r _to_replay_recordzAPIError._to_replay_recordcs* I|k   r!cx|jdkrdSt|tjrU ||}nL#t jj$r|j}||j d}Yn%wxYw|j d di}| |j||dS)PRaises an error with detailed error message if the response has an error status.Nrrrr/) status_coderhttpxResponsereadjsondecoderJSONDecodeErrortext reason_phrase body_segmentsr1 raise_error)clsrrrs r raise_for_responsezAPIError.raise_for_responseqs s"" f(EN++ A    \ )   -,   ,Q/33GR@@mOOH(-BBBBBs(A%A:9A:r=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 ServerErrorrHr=rrs r rGzAPIError.raise_errors, kC  ]H = == ! ! ! !c ! ! ! ! !  ]H = == C ]H 5 55r!cKd}d}t|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||jrj|jdkrdS |d{V}n9#|jj$r'|d{V}||jd}YnwxYw|j}nt'dt)|n-#t*$r t'dt)|wxYw||||d{VdS) r:rNr;r<rFr=r/zUnsupported response type: )rr>r?r=areadrArBrCrDrEhasattrrFr1aiohttpClientResponserclient_exceptionsContentTypeErrorreason ValueErrortype ImportErrorraise_error_async)rHrr=rrrSs r raise_for_async_responsez!APIError.raise_for_async_responsestKM(EN++'I   $ $ nn  \ )   -,   (kk ? + +I-11I   $ $,Q/33GR@@m(kkI h 6 7 7 K _ # # F "*--//111111MM*;   $MMOO++++++G""/MMM  !++IhIIJJ J IIIGtH~~GGHHHI    ]H E EEEEEEEEEEsA.A%BB)$F D+*F +3E!F E!!*F *F6cKd|cxkrdkrnnt|||d|cxkrdkrnnt|||||||rKrLrOs r r[zAPIError.raise_error_asyncs& kC  ]H = == ! ! ! !c ! ! ! ! !  ]H = == C ]H 5 55r!)N)!__name__ __module__ __qualname____doc__int__annotations__rr>r?rrstrrrrtuplerdictr( staticmethodr&rrrr StringDictr8 classmethodrIrGr\r[ __classcell__)rs@r r r s// ))) "EN4LL MMMM&(3-'8C=  CC CC  %.2JJ K CCCCCC&/ Xo &d38n(= =>////d38n< s s          SS !3    C+U^;<C CCC;C,666  %.2JJ K 6 666;682F EN,D D2F  2F2F2F;2Fh66,/6;C  %.2JJ K<6 666;66666r!r ceZdZdZdS)rMz%Client error raised by the GenAI API.Nr^r_r`rar!r rMrM--$r!rMceZdZdZdS)rNz%Server error raised by the GenAI API.Nrlrmr!r rNrNrnr!rNceZdZdZdS) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.Nrlrmr!r rqrqs__$r!rqceZdZdZdS)UnsupportedFunctionErrorz*Raised when the function is not supported.Nrlrmr!r rsrss22$r!rsceZdZdZdS)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.Nrlrmr!r ruru sLL$r!ruceZdZdZdS)UnknownApiResponseErrorz?Raised when the response from the API cannot be parsed as JSON.Nrlrmr!r rwrwsGG$r!rw)ratypingrrrrrr>rA websocketsr replay_api_clientr rSr,r rMrNrXrqrsrurwExperimentalWarningrmr!r r}s '&@@@@@@@@@@@@@@ //////...T6T6T6T6T6yT6T6T6n( ( z z j j1r!