U di4@s\ddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z dd lmZmZdd lmZmZmZd d Zeeeeeeeeeeee ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ee/e0e1e2e3e4e5e6e7e8e9e:e;ee?e@eAeBeCeDeEeFd 3ZGeGdddZHddgZIdddZJddZKGdddeZLGdddeZMdS)N) dataclass)sqrt)Dict)hf_hub_download list_spaces)is_offline_mode)LIST_SAFE_MODULESevaluate_python_code)TOOL_CONFIG_FILE TOOL_MAPPINGToolcGsdSN)argsrrE/tmp/pip-unpacked-wheel-bm_b0l5e/transformers/agents/default_tools.py custom_printsr)3print isinstancerangefloatintboolstrsetlistdicttupleroundceilfloorlogexpsincostanasinacosatanatan2degreesradianspowrlensummaxminabs enumeratezipreversedsortedallanymapfilterordchrnextiterdivmodcallablegetattrhasattrsetattr issubclasstypec@sFeZdZUeed<eeefed<eed<eed<eed<eed<dS)PreToolnameinputs output_typetask descriptionrepo_idN)__name__ __module__ __qualname__r__annotations__rrErrrrrFZs rFzimage-transformationz text-to-imagehuggingface-toolsc Cstr|diSt|d}i}|D]p}|j}t|tdd}t|dd}t|}W5QRX| dd} t | |d || |d |d d ||d <q&|S)Nz;You are in offline mode, so remote tools are not available.)authorspace)Z repo_typezutf-8)encoding/rKrHrI)rJrKrLrGrHrIrG) rinforidrr openjsonloadsplitrF) loggerZ organizationspacestoolsZ space_inforLZresolved_config_filereaderconfigrJrrrget_remote_toolsjs(   rbc Cs\i}td}|j}tD]:\}}t||}|}t|j|j|j ||j dd||j<q|S)NZ transformers)rGrHrIrJrKrL) importlib import_moduleZagentsr itemsrArFrGrHrIrK)r]Z default_toolsZ main_moduleZ tools_moduleZ task_nameZtool_class_nameZ tool_classZ tool_instancerrrsetup_default_toolss   rfcs6eZdZdZdZdZddfdd Zdd ZZS) PythonInterpreterToolpython_interpreterzRThis is a tool that evaluates python code. It can be used to perform calculations.stringN)authorized_importscsX|dkrttt|_ntttt|B|_ddd|ddi|_tj||dS)NcoderizThe code snippet to evaluate. All variables used in this snippet must be defined in this same snippet, else you will get an error. This code can only import the following python libraries: .rErK)rrr rjrHsuper__init__)selfrjrkwargs __class__rrros  zPythonInterpreterTool.__init__cCstt|t|jd}|S)N)Z static_toolsrj)rr BASE_PYTHON_TOOLSrj)rprkoutputrrrforwardszPythonInterpreterTool.forward) rMrNrOrGrKrIrorv __classcell__rrrrrrgs rgc@s.eZdZdZdZddddiZdZddZd S) FinalAnswerToolZ final_answerz-Provides a final answer to the given problem.answerr8zThe final answer to the problemrmcCs|Srr)rpryrrrrvszFinalAnswerTool.forwardN)rMrNrOrGrKrHrIrvrrrrrxs rx)rQ)Nimportlib.utilrcrZmathZ dataclassesrrtypingrZhuggingface_hubrrutilsrrhr r r_r r rrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErtrFZ"HUGGINGFACE_DEFAULT_TOOLS_FROM_HUBrbrfrgrxrrrrs    7