a dg:@sdZddlmZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd lmZgdZgdZGdddZGddde Z dS)zImplements Program-Aided Language Models. This module implements the Program-Aided Language Models (PAL) for generating code solutions. PAL is a technique described in the paper "Program-Aided Language Models" (https://arxiv.org/pdf/2211.10435.pdf). ) annotationsN)AnyDictListOptional)Chain)LLMChain)CallbackManagerForChainRun)BaseLanguageModel) ConfigDictFieldmodel_validator)Self)COLORED_OBJECT_PROMPT) MATH_PROMPT) PythonREPL)systemexecZexecfileeval __import__compile) r__subclasses__ __builtins__ __globals____getattribute____code__ __bases____mro____base__c@s2eZdZdZejZejZd ddddddd Z dS) PALValidationz"Validation for PAL generated code.NFz Optional[str]zOptional[type]bool)solution_expression_namesolution_expression_type allow_importsallow_command_execcCs||_||_|dur4t|jts4tdt|j|durt|j|jurt|j|jurttd|jd|jd|j|dur|durtd|dur|durtd||_ ||_ dS)aInitialize a PALValidation instance. Args: solution_expression_name (str): Name of the expected solution expression. If passed, solution_expression_type must be passed as well. solution_expression_type (str): AST type of the expected solution expression. If passed, solution_expression_name must be passed as well. Must be one of PALValidation.SOLUTION_EXPRESSION_TYPE_FUNCTION, PALValidation.SOLUTION_EXPRESSION_TYPE_VARIABLE. allow_imports (bool): Allow import statements. allow_command_exec (bool): Allow using known command execution functions. Nz;Expected solution_expression_name to be str, instead found z0Expected solution_expression_type to be one of (,z),instead found zOsolution_expression_name requires solution_expression_type to be passed as wellzOsolution_expression_type requires solution_expression_name to be passed as well) r!r" isinstancestr ValueErrortype!SOLUTION_EXPRESSION_TYPE_FUNCTION!SOLUTION_EXPRESSION_TYPE_VARIABLE TypeErrorr#r$)selfr!r"r#r$r.s/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/pal_chain/base.py__init__3sJ zPALValidation.__init__)NNFF) __name__ __module__ __qualname____doc__ast FunctionDefr*Namer+r0r.r.r.r/r-src@s<eZdZUdZded<dZded<dZded<d Zd ed <d Zd ed <d Z ded<dZ ded<e e dZ ded<dZded<dZded<eddddddZedd d!Zed"dd#d$Zed"dd%d&Zd:d'd(d)d*d+d,Zeddd-d.d/d0Zed1d2dd3d4d5Zed1d2dd3d6d7Zeddd8d9Zd S);PALChainaChain that implements Program-Aided Language Models (PAL). This class implements the Program-Aided Language Models (PAL) for generating code solutions. PAL is a technique described in the paper "Program-Aided Language Models" (https://arxiv.org/pdf/2211.10435.pdf). *Security note*: This class implements an AI technique that generates and evaluates Python code, which can be dangerous and requires a specially sandboxed environment to be safely used. While this class implements some basic guardrails by limiting available locals/globals and by parsing and inspecting the generated Python AST using `PALValidation`, those guardrails will not deter sophisticated attackers and are not a replacement for a proper sandbox. Do not use this class on untrusted inputs, with elevated permissions, or without consulting your security team about proper sandboxing! r llm_chain r'stopprint(solution())get_answer_exprNzOptional[Dict[str, Any]]python_globals python_localsresult output_keyFr return_intermediate_steps)default_factoryrcode_validations z Optional[int]timeoutallow_dangerous_codeafter)moder)returncCs|jstd|S)NaThis chain relies on the execution of generated code, which can be dangerous. Please read the security notice for this class, and only use it if you understand the security implications. If you want to proceed, you will need to opt-in, by setting `allow_dangerous_code` to `True`.)rGr(r-r.r.r/ post_inits  zPALChain.post_initTZforbid)Zarbitrary_types_allowedextraz List[str]cCs |jjjS)z?Return the singular input key. :meta private: )r9promptZinput_variablesrKr.r.r/ input_keysszPALChain.input_keyscCs|js|jgS|jdgSdS)z@Return the singular output key. :meta private: intermediate_stepsN)rBrArKr.r.r/ output_keysszPALChain.output_keyszDict[str, Any]z$Optional[CallbackManagerForChainRun]zDict[str, str])inputs run_managerrJcCs|p t}|jjf|jg|d|}|j|dd|jdt ||j t |j |j d}|j|d|j|jd}|j|i}|jr||d<|S)N)r; callbacksgreen )colorendverbose)Z_globalsZ_locals)rFrP)r Zget_noop_managerr9Zpredictr;Z get_childZon_textrYr8 validate_coderDrr>r?runr=rFrAstriprB)r-rRrSZ _run_managercodereplresoutputr.r.r/_calls"  zPALChain._callNone)r]rDrJc Cszt|}Wndttfy2td|YnBtyTtdt|Yn tyrtd|Yn0d}|jdurd}d}t t |}|D]}|jdur|j durt ||j rt |dr|j|jkrd}t |tjr|jD]*}t ||j rt |dr|j|jkrd}qt |tjs8t |tjrd}q|s\td|jd |j |jsx|rxtd ||jr|jst|D]}|jst |tjr|jtvrtd |jd ||jsNt |tjrNt |jdr|jjtvrtd |jjd |t |jtjrN|jjtvrNtd |jjd ||jst |tjsrt |tjrtd |qdS) Nz)Generated code is not valid python code: z9Generated code is expected to be a string, instead found z7Generated code too long / complex to be parsed by ast: FTnameidz3Generated code is missing the solution expression: z of type: z'Generated code has disallowed imports: z)Found illegal command execution function z in code )r5parse SyntaxErrorUnicodeDecodeErrorr(r,r) OverflowErrorr!listiter_child_nodesr"r&hasattrrcAssigntargetsrdImport ImportFromr#r$walk AttributeattrCOMMAND_EXECUTION_ATTRIBUTESCallfuncCOMMAND_EXECUTION_FUNCTIONS) clsr]rDZ code_treeZfound_solution_exprZ has_importsZtop_level_nodesnodeZ target_noder.r.r/rZs               zPALChain.validate_coder r)llmkwargsrJcKs2t|td}tdtjd}|f|dd|d|S)zLoad PAL from math prompt. Args: llm (BaseLanguageModel): The language model to use for generating code. Returns: PALChain: An instance of PALChain. ryrNZsolutionr!r"r:r<r9r;r=rD)rrrr*rwryrzr9rDr.r.r/from_math_prompt@s zPALChain.from_math_promptcKs2t|td}tdtjd}|f|dd|d|S)zLoad PAL from colored object prompt. Args: llm (BaseLanguageModel): The language model to use for generating code. Returns: PALChain: An instance of PALChain. r{Zanswerr|z z print(answer)r})rrrr+r~r.r.r/from_colored_object_promptXs z#PALChain.from_colored_object_promptcCsdS)NZ pal_chainr.rKr.r.r/ _chain_typeqszPALChain._chain_type)N)r1r2r3r4__annotations__r;r=r>r?rArBr rrDrFrGr rLr Z model_configpropertyrOrQra classmethodrZrrrr.r.r.r/r8ls>           cr8)!r4 __future__rr5typingrrrrZlangchain.chains.baserZlangchain.chains.llmrZ langchain_core.callbacks.managerr Zlangchain_core.language_modelsr Zpydanticr r r Ztyping_extensionsrZ6langchain_experimental.pal_chain.colored_object_promptrZ,langchain_experimental.pal_chain.math_promptrZ langchain_experimental.utilitiesrrvrsrr8r.r.r.r/s           ?