U 9%e'@sddlZddlZddlmZddlmZmZmZGdddeZ de ee efddd Z ej ee efee efd d d Z d dZddZddZddZddZddZddZdS)N)Mapping)AnyCallableDictc@seZdZdZdS)InterpretorErrorz An error raised when the interpretor cannot evaluate a Python expression, due to syntax error or unsupported operations. N)__name__ __module__ __qualname____doc__r r d/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/tools/python_interpreter.pyrsrF)codetoolsc Cszt|}Wn2tk r@}ztd|WYdSd}~XYnX|dkrNi}d}t|jD]\}}zt|||} Wndtk r}zFd|d} |r| d|d7} n| d|7} t| WYqW5d}~XYnX| dk r\| }q\|S)a@ Evaluate a python expression using the content of the variables stored in a state and only evaluating a given set of functions. This function will recurse through the nodes of the tree provided. Args: code (`str`): The code to evaluate. tools (`Dict[str, Callable]`): The functions that may be called during the evaluation. Any call to another function will fail with an `InterpretorError`. state (`Dict[str, Any]`): A dictionary mapping variable names to values. The `state` should contain the initial inputs but will be updated by this function to contain all variables as they are evaluated. chat_mode (`bool`, *optional*, defaults to `False`): Whether or not the function is called from `Agent.chat`. z.The code generated by the agent is not valid. Nz'Evaluation of the code stopped at line z. before the end because of the following errorzY. Copy paste the following error message and send it back to the agent: I get an error: ''z: )astparse SyntaxErrorprint enumeratebody evaluate_astr) r rstateZ chat_mode expressioneresultidxnode line_resultmsgr r r evaluate s.   r)rrrcst|tjrt|St|tjr0t|St|tjrB|jSt|tjrfdd|j D}fdd|j D}t t ||St|tj rt|jSt|tjrt|St|tjrt|jSt|tjrt|Sttdrt|tjrt|jSt|tjr@dfdd|j DSt|tjrdfdd|jDSt|tjr~t|St|tjrt|St|jj dd S) a Evaluate an absract syntax tree using the content of the variables stored in a state and only evaluating a given set of functions. This function will recurse trough the nodes of the tree provided. Args: expression (`ast.AST`): The code to evaluate, as an abastract syntax tree. state (`Dict[str, Any]`): A dictionary mapping variable names to values. The `state` is updated if need be when the evaluation encounters assignements. tools (`Dict[str, Callable]`): The functions that may be called during the evaluation. Any call to another function will fail with an `InterpretorError`. csg|]}t|qSr r).0krrr r ksz evaluate_ast..csg|]}t|qSr r r!vr#r r r$lsIndexcsg|]}tt|qSr )strrr%r#r r r$}scsg|]}t|qSr r )r!eltr#r r r$sz is not supported.N)! isinstancerAssignevaluate_assignCall evaluate_callConstantvaluerkeysvaluesdictzipExprrFor evaluate_forFormattedValueIf evaluate_ifhasattrr' JoinedStrjoinListeltsName evaluate_name Subscriptevaluate_subscriptr __class__r)rrrr2r3r r#r rNs:             rcCs|j}t|j||}t|dkr0|||dj<nLt|t|kr^tdt|dt|dt||D]\}}|||j<qh|S)Nrz Expected z values but got .)targetsrr1lenidrr5)ZassignrrZ var_namesrvar_namerr r r r-s  r-cst|jtjs,td|jdt|jd|jj}|krPtd|jjd|}fdd|jD}fdd|jD}|||S) NzZIt is not permitted to evaluate other functions than the provided tools (tried to execute z of type rGz).csg|]}t|qSr r )r!argr#r r r$sz!evaluate_call..csi|]}|jt|jqSr )rMrr1)r!keywordr#r r sz!evaluate_call..) r+funcrrArtyperJargskeywords)callrr func_namerPrRkwargsr r#r r/sr/cCst|j||}t|j||}t|ttfr6|t|S||krF||St|trt|trt |t| }t |dkr||dSt d|d|ddS)NrzCould not index z with 'z'.)rslicer1r+listtupleintr)rdifflibget_close_matchesr2rIr)Z subscriptrrindexr1 close_matchesr r r rDs   rDcCsX|j|kr||jSt|jt|}t|dkrB||dStd|jddS)NrzThe variable `z` is not defined.)rJr[r\rXr2rIr)namerrr^r r r rBs     rBcCst|jdkrtdt|j||}|jd}t|jd||}t|tjrT||kSt|tj rh||kSt|tj r|||kSt|tj r||kSt|tj r||kSt|tj r||kSt|tjr||kSt|tjr||k St|tjr||kSt|tjr ||kStd|dS)NrFz2Cannot evaluate conditions with multiple operatorsrzOperator not supported: )rIZopsrrleftZ comparatorsr+rEqNotEqLtLtEGtGtEIsIsNotInNotIn) conditionrrr`Z comparatorrightr r r evaluate_conditions4         rmcCs`d}t|j||r8|jD]}t|||}|dk r|}qn$|jD]}t|||}|dk r>|}q>|SN)rmtestrrZorelse)Z if_statementrrrlinerr r r r;s    r;cCsPd}t|j||}|D]4}|||jj<|jD]}t|||}|dk r,|}q,q|Srn)ritertargetrJr)Zfor_looprrriteratorcounterrrr r r r8s   r8)NF)rr[collections.abcrtypingrrr ValueErrorrr)rASTrr-r/rDrBrmr;r8r r r r s  .$>