o ii@s ddlmZddlZddlZddlmZmZddlm Z ej dkr(ddl m Z ne Z er2ddl mZd ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGddde eZGdddeZGdddeZdS) ) annotationsN) TYPE_CHECKINGAny) core_schema) )ExceptionGroup)RetryPromptPart) ModelRetry CallDeferredApprovalRequired UserError AgentRunErrorUnexpectedModelBehaviorUsageLimitExceededModelHTTPErrorIncompleteToolCallFallbackExceptionGroupcsPeZdZUdZded< dfdd Zdd d Zdd dZedddZ Z S)r zException to raise when a tool function should be retried. The agent will return the message to the model and ask it to try calling the function/tool again. strmessagec||_t|dSNrsuper__init__selfr __class__\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/exceptions.pyr(zModelRetry.__init__otherrreturnboolcCst||jo |j|jkSr) isinstancerr)rr#r r r!__eq__,szModelRetry.__eq__intcCst|j|jfSr)hashrrrr r r!__hash__/szModelRetry.__hash_____core_schema.CoreSchemacCsJtttttdgd}tjdd|tjdd|ddS)z@Pydantic core schema to allow `ModelRetry` to be (de)serialized. model-retryrkindcSs t|dS)Nr)r )dctr r r!< z9ModelRetry.__get_pydantic_core_schema__..cSs |jddS)Nr/r0r)xr r r!r3?r4) return_schema) serialization)rtyped_dict_schematyped_dict_field str_schemaliteral_schema no_info_after_validator_function$plain_serializer_function_ser_schema)clsr,r-schemar r r!__get_pydantic_core_schema__2s z'ModelRetry.__get_pydantic_core_schema__rr)r#rr$r%)r$r()r,rr-rr$r.) __name__ __module__ __qualname____doc____annotations__rr'r+ classmethodrA __classcell__r r rr!r s   r c@eZdZdZdS)r zException to raise when a tool call should be deferred. See [tools docs](../deferred-tools.md#deferred-tools) for more information. NrCrDrErFr r r r!r Er c@rJ)r zException to raise when a tool call requires human-in-the-loop approval. See [tools docs](../deferred-tools.md#human-in-the-loop-tool-approval) for more information. NrKr r r r!r NrLr cs.eZdZUdZded< dfdd ZZS)ruEError caused by a usage mistake by the application developer — You!rrcrrrrrr r!r]r"zUserError.__init__rBrCrDrErFrGrrIr r rr!rWs rcs8eZdZUdZded< d fdd Zd ddZZS) rz4Base class for errors occurring during an agent run.rrcrrrrrr r!rhr"zAgentRunError.__init__r$cCs|jSrr5r*r r r!__str__lszAgentRunError.__str__rBr$rrCrDrErFrGrrNrIr r rr!rbs rc@rJ)rz?Error raised when a Model's usage exceeds the specified limits.NrKr r r r!rprcsDeZdZUdZded< ded< d d fdd Zdd d ZZS)rzLError caused by unexpected Model behavior, e.g. an unexpected response code.rr str | NonebodyNcsZ||_|dur d|_nz tjt|dd|_Wn ty$||_Ynwt|dS)N)indent)rrSjsondumpsloads ValueErrorrr)rrrSrr r!r|s  z UnexpectedModelBehavior.__init__r$cCs|jr |jd|jS|jS)Nz, body: )rSrr*r r r!rNszUnexpectedModelBehavior.__str__r)rrrSrRrOrPr r rr!rts  rcsNeZdZUdZded< ded< ded< ded< d d fd d ZZS)rzGRaised when an model provider response has a status code of 4xx or 5xx.r( status_coder model_name object | NonerSrNcs8||_||_||_d|d|d|}t|dS)Nz status_code: z, model_name: z, body: )rZr[rSrr)rrZr[rSrrr r!rs zModelHTTPError.__init__r)rZr(r[rrSr\rMr r rr!rs rc@rJ)rzGA group of exceptions that can be raised when all fallback models fail.NrKr r r r!rrQrcs"eZdZdZdfdd ZZS)ToolRetryErrorzMException used to signal a `ToolRetry` message should be returned to the LLM. tool_retryr cs||_tdSr)r^rr)rr^rr r!rszToolRetryError.__init__)r^r )rCrDrErFrrIr r rr!r]sr]c@rJ)rzNError raised when a model stops due to token limit while emitting a tool call.NrKr r r r!rrQr) __future__r _annotationsrVsystypingrr pydantic_corer version_infoexceptiongrouprmessagesr __all__ Exceptionr r r RuntimeErrorrrrrrrr]rr r r r!s,    &