o 0Xxi@sdZddlmZmZmZmZddlZerddlmZGddde Z Gdd d e Z Gd d d e Z Gd d d e ZGddde ZGddde ZGdddeZdS)z Error classes for the GenAI SDK.)AnyOptional TYPE_CHECKINGUnionN)ReplayResponsecseZdZUdZeed<ejed<dZe e ed<dZ e e ed<dZ e e ed<dedeejdfffdd Zd d Zd d ZddZddZedeejdffddZZS)APIErrorz'General errors raised by the GenAI API.coderesponseNstatusmessagercs||_t|tjr"z|}Wntjjy!|j|jd}Yn w|j d di}||_ | ||_ |||_|r>|n|||_t|jd|jd|j dS)N)r r rerror z. )r isinstancerequestsResponsejson exceptionsJSONDecodeErrortextreason body_segmentsgetdetails _get_messager _get_statusr _get_coder super__init__)selfr r response_json __class__R/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/errors.pyr$s     &zAPIError.__init__cC|d|diddS)Nr r rrr r#r#r$r=zAPIError._get_statuscCr%)Nr r r&r'r#r#r$rBr(zAPIError._get_messagecCr%)Nr r r&r'r#r#r$rGr(zAPIError._get_codecCsd|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 r r r))rr#r#r$_to_replay_recordLs zAPIError._to_replay_recordcCsb|jdkrdS|j}d|krdkrnnt||d|kr%dkr,nnt|||||)zPRaises an error with detailed error message if the response has an error status.NiiiX) status_code ClientError ServerError)clsr r,r#r#r$raise_for_responseZs    zAPIError.raise_for_response)__name__ __module__ __qualname____doc__int__annotations__rrr rstrr r rrrrrrr* classmethodr0 __classcell__r#r#r!r$rs(    rc@eZdZdZdS)r-z%Client error raised by the GenAI API.Nr1r2r3r4r#r#r#r$r-kr-c@r:)r.z%Server error raised by the GenAI API.Nr;r#r#r#r$r.pr<r.c@r:) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.Nr;r#r#r#r$r=ur=c@r:)UnsupportedFunctionErrorz*Raised when the function is not supported.Nr;r#r#r#r$r?{r?c@r:)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.Nr;r#r#r#r$rAr>rAc@r:)ExperimentalWarningz"Warning for experimental features.Nr;r#r#r#r$rBr@rB)r4typingrrrrrreplay_api_clientr Exceptionrr-r. ValueErrorr=r?rAWarningrBr#r#r#r$s P