{hS,ddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ejRdZ*edeZ+edZ,e(dZ-edZ. d ddZ/ddZ0d dZ1dddejddf d!dZ3dddejddf d"dZ4y)#) annotationsN)JSONDecodeError)AnyCallableTypeVar)InstructorRetryException)Hooks)Mode)handle_reask_kwargs)process_responseprocess_response_async)update_total_usage)AsyncValidationError)ChatCompletion)CompletionUsageCompletionTokensDetailsPromptTokensDetails) BaseModelValidationError) AsyncRetrying RetryErrorRetryingstop_after_attemptstop_after_delay) ParamSpec instructorT_Model)boundT_Retval T_ParamSpecTcXt|trutjd|d|t |g}||j t ||d}|ddD]}||z} |rt|}|St|}|St|ttfsddl m }|d|S) a Initialize the retrying mechanism based on the type (synchronous or asynchronous). Args: max_retries (int | Retrying | AsyncRetrying): Maximum number of retries or a retrying object. is_async (bool): Flag indicating if the retrying is asynchronous. timeout (float | None): Optional timeout in seconds to limit total retry duration. Returns: Retrying | AsyncRetrying: Configured retrying object. z max_retries: z , timeout: Nr)stop)ConfigurationErrorzSmax_retries must be an int or a `tenacity.Retrying`/`tenacity.AsyncRetrying` object) isinstanceintloggerdebugrappendrrrinstructor.exceptionsr%) max_retriesis_asynctimeoutstop_conditionsstop_condition conditionr%s Q/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/instructor/retry.pyinitialize_retryingr3,s +s# }[MWIFG.k:;    " "#3G#< =)+(, 8I+i7N 8 '^ ?< a   c tdddtddtdd}|tjtj hvrddlm}|dddd}|S)z Initialize the total usage based on the mode. Args: mode (Mode): The mode of operation. Returns: CompletionUsage | Any: Initialized usage object. r) audio_tokensreasoning_tokens)r6 cached_tokens)completion_tokens prompt_tokens total_tokenscompletion_tokens_detailsprompt_tokens_details)Usage) input_tokens output_tokenscache_read_input_tokenscache_creation_input_tokens)rrrr ANTHROPIC_TOOLSANTHROPIC_JSONanthropic.typesr>)mode total_usageAnthropicUsages r2initialize_usagerIWsm""9Q# 2qPQRK $$d&9&9::;$$%()  r4c<d|vr|dSd|vr|dSd|vr|dSgS)z Extract messages from kwargs, helps handles the cohere and gemini chat history cases Args: kwargs (Dict[str, Any]): Keyword arguments containing message data. Returns: Any: Extracted messages. messagescontents chat_history)kwargss r2extract_messagesrPvsBVj!!Vj!!n%% Ir4r#c D|xs t}t|} |jd} t|d| }|jdd} d} |D]} | 5tj d| j j |j|i|||i|} |j| t| | } t| ||||| cdddcSy#ttf$r?}tj d ||j|t||| | }|d}~wwxYw#1swYxYw#t $r`}tj d |t#|j$j&  j jt)||| |d}~wwxYw) a Retry a synchronous function upon specified exceptions. Args: func (Callable[T_ParamSpec, T_Retval]): The function to retry. response_model (Optional[type[T_Model]]): The model to validate the response against. args (Any): Positional arguments for the function. kwargs (Any): Keyword arguments for the function. context (Optional[Dict[str, Any]], optional): Additional context for validation. Defaults to None. max_retries (int | Retrying, optional): Maximum number of retries or a retrying object. Defaults to 1. strict (Optional[bool], optional): Strict mode flag. Defaults to None. mode (Mode, optional): The mode of operation. Defaults to Mode.TOOLS. hooks (Optional[Hooks], optional): Hooks for emitting events. Defaults to None. Returns: T_Model | None: The processed response model or None. Raises: InstructorRetryException: If all retry attempts fail. r.Fr-r.streamNRetrying, attempt: responserGrVresponse_modelvalidation_contextstrictrFrS Parse error: rOrFrV exception Retry error: last_completion n_attemptsrK create_kwargsrG)r rIgetr3r(r) retry_stateattempt_numberemit_completion_argumentsemit_completion_responserr rremit_parse_errorr rr last_attempt _exceptionrPfuncrXargsrOcontextr,rZrFhooksrGr.rSrVattemptes r2 retry_syncrrs>  UWE"4(Kjj#G%kE7SKZZ% (F+" G  273F3F3U3U2VWX3E33TDVD#T4V4H228<1!){ H,!)'5+2%!%    &(9 LL=!45**1-0%!!)"# F G %  8   }QC()& NN % %$**99%!#    sV D6-D*AC D6D6D'(:D""D''D**D3 /D66 F?AFFc K|xs t}t|} |jd} t|d| }|jdd} d} |23d{} tj d| j j| 5 |j|i|||i|d{} |j| t| | } t| ||||| d{cdddcS77H7#tttf$r?}tj d ||j|t!||| | }|d}~wwxYw#1swYxYw6y#t"$r`}tj d |t%|j&j(  j jt+||| |d}~wwxYww)a Retry an asynchronous function upon specified exceptions. Args: func (Callable[T_ParamSpec, T_Retval]): The asynchronous function to retry. response_model (Optional[type[T_Model]]): The model to validate the response against. context (Optional[Dict[str, Any]]): Additional context for validation. args (Any): Positional arguments for the function. kwargs (Any): Keyword arguments for the function. max_retries (int | AsyncRetrying, optional): Maximum number of retries or an async retrying object. Defaults to 1. strict (Optional[bool], optional): Strict mode flag. Defaults to None. mode (Mode, optional): The mode of operation. Defaults to Mode.TOOLS. hooks (Optional[Hooks], optional): Hooks for emitting events. Defaults to None. Returns: T_Model | None: The processed response model or None. Raises: InstructorRetryException: If all retry attempts fail. r.TrRrSFNrTrUrWr[r\r^r_)r rIrcr3r(r)rdrerfrgrr rrrrhr rrrirjrPrks r2 retry_asyncrts>  UWE"4(Kjj#G%kD'RKZZ% (F/(  ' LL.w/B/B/Q/Q.RS T 3E33TDVD5945J65J/JH228<1!){ H"8!)'5+2%!% "    0K $#(  LL=!45**1-0%!!)"# F G #  )B   }QC()& NN % %$**99%!#    sA GEEC/E/E E  C5(C1)6C5C3 C5# E.G/E1C53C55E :EEE  E  EG G#AF>>GG)N)r,zint | Retrying | AsyncRetryingr-boolr.z float | None)rFr returnzCompletionUsage | Any)rOzdict[str, Any]rvr)rlCallable[T_ParamSpec, T_Retval]rXtype[T_Model] | NonermrrOrrndict[str, Any] | Noner,zint | RetryingrZ bool | NonerFr ro Hooks | NonervT_Model | None)rlrwrXrxrmrrOrrnryr,zint | AsyncRetryingrZrzrFr ror{rvr|)5 __future__rloggingjsonrtypingrrrr+rinstructor.hooksr instructor.moder instructor.reaskr instructor.process_responser r instructor.utilsrinstructor.validatorsropenai.types.chatropenai.types.completion_usagerrrpydanticrrtenacityrrrrrtyping_extensionsr getLoggerr(rrr r!r3rIrPTOOLSrrrtrNr4r2rs# )):" 006, 0(   < ( )9 - :  &  CL !(/(((V>4&*"#S )S(S S  S # S  S S S SSv&*'(W )W(W W  W # W % W W W WWr4