NgI)xddlmZddlmZmZmZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZddlmZdd lmZdd lmZmZd Zd ZejeZejeZed ZGddeZGddeZ Gdde eZ!Gdde eZ"dS)) annotations)AnyTypeVarUnion)OutputParserException)BaseLanguageModel)BaseOutputParserStrOutputParser) PromptValue)BasePromptTemplatePromptTemplate)RunnableSerializable)SkipValidation) Annotated TypedDictzPrompt: {prompt} Completion: {completion} Above, the Completion did not satisfy the constraints given in the Prompt. Please try again:zPrompt: {prompt} Completion: {completion} Above, the Completion did not satisfy the constraints given in the Prompt. Details: {error} Please try again:Tc$eZdZUded<ded<dS) RetryOutputParserRetryChainInputstrprompt completionN__name__ __module__ __qualname____annotations__Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/output_parsers/retry.pyrr's"KKKOOOOOrrc.eZdZUded<ded<ded<dS))RetryWithErrorOutputParserRetryChainInputrrrerrorNrrrrr!r!,s+KKKOOOJJJJJrr!ceZdZUdZded< ded< dZded< d Zd ed < eedfd!dZ d"dZ d"dZ d#dZ d$dZ ed$dZed%dZd S)&RetryOutputParserzWrap a parser and try to fix parsing errors. Does this by passing the original prompt and the completion to another LLM, and telling it the completion did not satisfy criteria in the prompt. 0Annotated[BaseOutputParser[T], SkipValidation()]parserzdAnnotated[Union[RunnableSerializable[RetryOutputParserRetryChainInput, str], Any], SkipValidation()] retry_chainint max_retriesTboollegacyllmrBaseOutputParser[T]rr returnRetryOutputParser[T]cF||ztz}||||S)aFCreate an RetryOutputParser from a language model and a parser. Args: llm: llm to use for fixing parser: parser to use for parsing prompt: prompt to use for fixing max_retries: Maximum number of retries to parse. Returns: RetryOutputParser r&r'r*r clsr-r&rr*chains rfrom_llmzRetryOutputParser.from_llmF/& 000s&eMMMMrrr prompt_valuer rcd}||jkr |j|S#t$r}||jkr||dz }|jrDt |jdr/|j||}n;|j t||}Yd}~nd}~wwxYw||jktd)Parse the output of an LLM call using a wrapped parser. Args: completion: The chain completion to parse. prompt_value: The prompt to use to parse the completion. Returns: The parsed completion. rr(runrrNFailed to parse) r*r&parserr,hasattrr'r< to_stringinvokedictselfrr9retrieses rparse_with_promptz#RetryOutputParser.parse_with_prompt\s ))) {((444(   d...GqLG{ wt/?'G'G %)%5%9%9#/#9#9#;#;'1&:&& &*%5%<%< '3'='='?'?+5&&  )))*$$5666s) CBCCc(Kd}||jkr |j|d{VS#t$r}||jkr||dz }|jrXt |jdrC|j||t|d{V}nA|j t||d{V}Yd}~nd}~wwxYw||jktd)r;rNr(arunrrr"r=r> r*r&aparserr,r@r'rJrAreprainvokerCrDs raparse_with_promptz$RetryOutputParser.aparse_with_prompts|))) ![// ;;;;;;;;;(   d...GqLG{ wt/?'H'H +/+;+@+@#/#9#9#;#;'1"&q'',A,,&&&&&& ,0+;+C+C '3'='='?'?+5,,&&&&&&  ))),$$5666s1 C7B2C22C7c tdNzGThis OutputParser can only be called by the `parse_with_prompt` method.NotImplementedErrorrErs rr?zRetryOutputParser.parse! U   rc4|jSNr&get_format_instructionsrEs rrZz)RetryOutputParser.get_format_instructions{22444rcdS)Nretryrr[s r_typezRetryOutputParser._typeswrtype[T]c|jjSrXr& OutputTyper[s rrczRetryOutputParser.OutputType {%%rN) r-rr&r.rr r*r)r/r0rrr9r r/rrrr/rr/rr/r`)rrr__doc__rr*r, classmethodNAIVE_RETRY_PROMPTr7rHrPr?rZpropertyr_rcrrrr$r$2s5 =<<<0VK9FC &8 NNNN[N*!7!7!7!7F"7"7"7"7H    5555X&&&X&&&rr$ceZdZUdZded< ded< dZded< d Zd ed < eedfd!dZ d"dZ d"dZ d#dZ d$dZ ed$dZed%dZd S)&RetryWithErrorOutputParseraWrap a parser and try to fix parsing errors. Does this by passing the original prompt, the completion, AND the error that was raised to another language model and telling it that the completion did not work, and raised the given error. Differs from RetryOutputParser in that this implementation provides the error that was raised back to the LLM, which in theory should give it more information on how to fix it. r%r&zmAnnotated[Union[RunnableSerializable[RetryWithErrorOutputParserRetryChainInput, str], Any], SkipValidation()]r'r(r)r*Tr+r,r-rr.rr r/RetryWithErrorOutputParser[T]cF||ztz}||||S)aCreate a RetryWithErrorOutputParser from an LLM. Args: llm: The LLM to use to retry the completion. parser: The parser to use to parse the output. prompt: The prompt to use to retry the completion. max_retries: The maximum number of times to retry the completion. Returns: A RetryWithErrorOutputParser. r2r3r4s rr7z#RetryWithErrorOutputParser.from_llmr8rrrr9r rc d}||jkr |j|S#t$r}||jkr||dz }|jrRt |jdr=|j||t|}nI|j t||t|}Yd}~nd}~wwxYw||jktd)Nrr(r<rK)rrr"r>) r*r&r?rr,r@r'r<rArNrBrCrDs rrHz,RetryWithErrorOutputParser.parse_with_prompts0))) {((444(   d...GqLG{ wt/?'G'G %)%5%9%9#/#9#9#;#;'1"&q''&:&& &*%5%<%< +5'3'='='?'?&*1gg&&  ))).$$5666s) C1B4C,,C1c HKd}||jkr |j|d{VS#t$r}||jkr||dz }|jrXt |jdrC|j||t|d{V}nO|j t||t|d{V}Yd}~nd}~wwxYw||jktd)Nrr(rJrKr>rLrDs rrPz-RetryWithErrorOutputParser.aparse_with_prompts))) ![// ;;;;;;;;;(   d...GqLG{ wt/?'H'H +/+;+@+@#/#9#9#;#;'1"&q'',A,,&&&&&& ,0+;+C+C '3'='='?'?+5&*1gg,,&&&&&&  ))).$$5666s2 DCDDc tdrRrSrUs rr?z RetryWithErrorOutputParser.parserVrc4|jSrXrYr[s rrZz2RetryWithErrorOutputParser.get_format_instructions r\rcdS)Nretry_with_errorrr[s rr_z RetryWithErrorOutputParser._type#s!!rr`c|jjSrXrbr[s rrcz%RetryWithErrorOutputParser.OutputType'rdrN) r-rr&r.rr r*r)r/rorerfrgrh)rrrrirr*r,rjNAIVE_RETRY_WITH_ERROR_PROMPTr7rHrPr?rZrlr_rcrrrrnrns4=<<<0 VK9FC &C NNNN[N*7777877778    5555"""X"&&&X&&&rrnN)# __future__rtypingrrrlangchain_core.exceptionsrlangchain_core.language_modelsrlangchain_core.output_parsersr r langchain_core.prompt_valuesr langchain_core.promptsr r langchain_core.runnablesrpydanticrtyping_extensionsrrNAIVE_COMPLETION_RETRY!NAIVE_COMPLETION_RETRY_WITH_ERROR from_templaterkrxrrr!r$rnrrrrs""""""&&&&&&&&&&;;;;;;<<<<<<KKKKKKKK444444EEEEEEEE999999######22222222%!2^12HII < <%!! GCLLy   &&&&&(+&&&Du&u&u&u&u&!1!!4u&u&u&u&u&r