o i@sdZddlmZmZmZmZddlZddlZddlm Z er(ddl m Z ddl Z Gddde ZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZe jZdS)z Error classes for the GenAI SDK.)AnyOptional TYPE_CHECKINGUnionN)_common)ReplayResponsec sdeZdZUdZeed<edejdfed<dZ e e ed<dZ e e ed< dded e de edejdfffd d Zd e d e fd dZd e d e fddZd e d e fddZd ejfddZededejfd dfddZeded e de edejdfd dfddZededejdfd dfddZeded e de edejdfd dfddZZS)APIErrorz'General errors raised by the GenAI API.coderzaiohttp.ClientResponseresponseNstatusmessage response_jsoncs|t|trt|dkr|d}||_||_|||_|||_|r%|n| ||_ t |j d|jd|jdS)Nrr z. ) isinstancelistlenr details _get_messager _get_statusr _get_coder super__init__)selfr rr  __class__Y/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/google/genai/errors.pyr%s  &zAPIError.__init__returncC|d|diddS)Nr errorgetrrrrrr8zAPIError._get_statuscCr)Nr r r!r#rrrr=r$zAPIError._get_messagecCr)Nr r r!r#rrrrBr$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%)rrrr_to_replay_recordGs zAPIError._to_replay_recordcCs~|jdkrdSt|tjr,z ||}Wntjjy+|j}||j d}Yn w|j d di}| |j||dS)PRaises an error with detailed error message if the response has an error status.Nr r rr ) status_coderhttpxResponsereadjsondecoderJSONDecodeErrortext reason_phrase body_segmentsr" raise_error)clsr rr rrrraise_for_responseUs    zAPIError.raise_for_responser*cCsTd|kr dkrnnt|||d|krdkr$nnt|||||||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 ServerErrorr5r*rr rrrr4ls    zAPIError.raise_errorcsld}d}t|tjr8|jdkrdSz |IdH|}Wntjjy3|j}||j d}Ynw|j}nrt |drVt |drV|jdkrIdS|j d di}|j}nTzCddl }t||jr|jdkriWdSz |IdH}Wn|jjy|IdH}||jd}Ynw|j}n tdt|Wntytdt|w||||IdHdS) r'rNr(r)r3r*r zUnsupported response type: )rr+r,r*areadr.r/r0r1r2hasattrr3r"aiohttpClientResponser client_exceptionsContentTypeErrorreason ValueErrortype ImportErrorraise_error_async)r5r r*rr r>rrrraise_for_async_responsesV          z!APIError.raise_for_async_responsecsVd|kr dkrnnt|||d|krdkr%nnt|||||||r7r8r;rrrrFs    zAPIError.raise_error_async)N)__name__ __module__ __qualname____doc__int__annotations__rr+r,r rstrr rrrrrr StringDictr& classmethodr6r4rGrF __classcell__rrrrr st      4r c@eZdZdZdS)r9z%Client error raised by the GenAI API.NrHrIrJrKrrrrr9r9c@rR)r:z%Server error raised by the GenAI API.NrSrrrrr:rTr:c@rR) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.NrSrrrrrUrTrUc@rR)UnsupportedFunctionErrorz*Raised when the function is not supported.NrSrrrrrVrTrVc@rR)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.NrSrrrrrWrTrWc@rR)UnknownApiResponseErrorz?Raised when the response from the API cannot be parsed as JSON.NrSrrrrrXrTrX)rKtypingrrrrr+r.rreplay_api_clientrr> Exceptionr r9r:rCrUrVrWrXExperimentalWarningrrrrs"  =