WiC ddlmZddlZddlmZddlmZmZmZddl m Z m Z m Z m ZddlmZdd lmZdd lmZmZmZdd lmZdd lmZdd lmZmZmZddlm Z m Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ejRdZ*ede Z+edZ,e(dZ-edZ. d ddZ/ddZ0ddZ1dddejddf d dZ3dddejddf d!dZ4y)") annotationsN)JSONDecodeError)AnyCallableTypeVar)InstructorRetryExceptionAsyncValidationError FailedAttemptValidationError)Hooks)Mode)process_responseprocess_response_asynchandle_reask_kwargs)update_total_usage)ChatCompletion)CompletionUsageCompletionTokensDetailsPromptTokensDetails) BaseModelr ) 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: Nrr)stop)ConfigurationErrorzSmax_retries must be an int or a `tenacity.Retrying`/`tenacity.AsyncRetrying` object) isinstanceintloggerdebugrappendrrr exceptionsr') max_retriesis_asynctimeoutstop_conditionsstop_condition conditionr's a/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/instructor/core/retry.pyinitialize_retryingr50s +s# }[MWIFG.k:;    " "#3G#< =)+(, 8I+i7N 8 '^ ?2 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)r8 cached_tokens)completion_tokens prompt_tokens total_tokenscompletion_tokens_detailsprompt_tokens_details)Usage) input_tokens output_tokenscache_read_input_tokenscache_creation_input_tokens)rrrrANTHROPIC_TOOLSANTHROPIC_JSONanthropic.typesr@)mode total_usageAnthropicUsages r4initialize_usagerK[sm""9Q# 2qPQRK $$d&9&9::;$$%()  r6c<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 r4extract_messagesrRzsBVj!!Vj!!n%% Ir6c &|xs t}t|} |jd} t|d| }|jdd} g} d} |D]}|5tj d|j j |j|i|||i|} |j| t| | } t| ||||| cdddcSy#tttf$r}tj d ||j|| j!t#|j j|| t%|t&rt)|d r|j j*duxs!|j j*j, }| xs7|j jt/|j0d t3d k\}|r|j5|t7||| || }|d}~wt8$r}tj d||j;|| j!t#|j j|| t%|t&rt)|d r|j j*duxs!|j j*j, }| xs7|j jt/|j0d t3d k\}|r|j5||d}~wwxYw#1swYxYw#t<$ra}tj d|t?|j@jB j jtE||| | |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. r0Fr/r0streamNRetrying, attempt: responserIrXresponse_modelvalidation_contextstrictrHrU Parse error: attempt_number exception completionr&max_attempt_numberinfrQrHrXr`failed_attemptsCompletion error: Retry error: last_completion n_attemptsrM create_kwargsrIre)#r rKgetr5r*r+ retry_stater_emit_completion_argumentsemit_completion_responserrr rInstructorValidationErroremit_parse_errorr,r r(rhasattroutcomefailedgetattrr&floatemit_completion_last_attemptr Exceptionemit_completion_errorrr last_attempt _exceptionrRfuncrZargsrQcontextr.r\rHhooksrIr0rUrerXattempte will_retryis_last_attempts r4 retry_syncrs>  UWE"4(Kjj#G%kE7SKZZ% (F,.On"^ G]  273F3F3U3U2VWX[3E33TDVD#T4V4H228<1!){ H,!)'5+2%!% ] ] ^ ($#-+ LL=!45**1-$**%+2+>+>+M+M&''/"+x8W#V> $//774?F#*#6#6#>#>#E#EE# !+N&22AA& + 0 02Fe   (+!>>qA0%!!)"#(7 FG LL#5aS!9://2$**%+2+>+>+M+M&''/"+x8W#V>$//774?F#*#6#6#>#>#E#EE# !+N&22AA& + 0 02Fe   (+!>>qAG?}] ] |   }QC()& NN % %$**99%!#+    sd L&-LAC L&L&L0DH LDLLLL# L&& N/AN  Nc pK|xs t}t|} |jd} t|d| }|jdd} g} d} |23d{}tj d|j j|5 |j|i|||i|d{} |j| t| | } t| ||||| d{cdddcS77H7#ttttf$r}tj d ||j!|| j#t%|j j|| t'|t(rt+|d r|j j,duxs!|j j,j. }| xs7|j jt1|j2d t5dk\}|r|j7|t9||| || }|d}~wt:$r}tj d||j=|| j#t%|j j|| t'|t(rt+|d r|j j,duxs!|j j,j. }| xs7|j jt1|j2d t5dk\}|r|j7||d}~wwxYw#1swYxYw6y#t>$ra}tj d|tA|jBjD j jtG||| | |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. r0TrTrUFNrVrWrYr]r^r&rbrcrdrfrgrh)$r rKrlr5r*r+rmr_rnrorrr rr rprqr,r r(rrrrsrtrur&rvrwrrxryrr rzr{rRr|s r4 retry_asyncr+s>  UWE"4(Kjj#G%kD'RKZZ% (F,.Oo(_ _ ' LL.w/B/B/Q/Q.RS T] \3E33TDVD5945J65J/JH228<1!){ H"8!)'5+2%!% "] ] _  0K $#(- , LL=!45**1-$**%+2+>+>+M+M&''/"+}='#VC $//774?F#*#6#6#>#>#E#EE# !+N&22AA& + 0 02Fe   (+!>>qA0%!!)"#(7 FG LL#5aS!9://2$**%+2+>+>+M+M&''/"+}='#VC$//774?F#*#6#6#>#>#E#EE# !+N&22AA& + 0 02Fe   (+!>>qAG?}] ] )@   }QC()& NN % %$**99%!#+    sA N6M MC1M/M L: C7*C3+6C7!C5"C7% M 0N61M3C75C77L7DH## L70DL22L77L::M ? M N6 N3AN..N33N6)N)r.zint | Retrying | AsyncRetryingr/boolr0z float | None)rHrreturnzCompletionUsage | Any)rQzdict[str, Any]rr)r}Callable[T_ParamSpec, T_Retval]rZtype[T_Model] | Noner~rrQrrdict[str, Any] | Noner.zint | Retryingr\ bool | NonerHrr Hooks | NonerT_Model | None)r}rrZrr~rrQrrrr.zint | AsyncRetryingr\rrHrrrrr)5 __future__rloggingjsonrtypingrrrr-r r r r rprr rHrprocessing.responserrrutilsropenai.types.chatropenai.types.completion_usagerrrpydanticrtenacityrrrrrtyping_extensionsr getLoggerr*r r"r#r$r5rKrRTOOLSrrrPr6r4rs# ))  ', 0(   < ( )9 - :  &  CL !(/(((V>4&*"#Y )Y(Y Y  Y # Y  Y Y Y YYB&*'(Z )Z(Z Z  Z # Z % Z Z Z ZZr6