gddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z m Z mZmZmZddlmZmZmZmZmZddlmZmZmZddlmZddlmZmZm Z dd l!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+d d l,m-Z-m.Z.e+j/e0Z1e)rddl2Z2e(r dd l3m4Z4ddl5m6Z6dZ7d1dZ8dZ9dZ:e:GddZ;dZedZ?Gdde;Z@de;fdZAdd d!d"d#d$d%d&ZBd2d'ZCd(ZDGd)d*ZEd+e e=e=fde e=e=ffd,ZFGd-d.ZGd/e de;fd0ZHdS)3N) lru_cachewraps)AnyCallableDictListOptionalUnion) create_repoget_collectionhf_hub_downloadmetadata_update upload_folder)RepositoryNotFoundErrorbuild_hf_headers get_session)version)custom_object_saveget_class_from_dynamic_module get_imports) AutoProcessor) CONFIG_NAMETypeHintParsingException cached_fileget_json_schemais_accelerate_availableis_torch_availableis_vision_availablelogging)handle_agent_inputshandle_agent_outputs) PartialState)send_to_devicetool_config.jsonc ||S t|tfddi|dS#t$rH t|tfddi|YdS#t$rtd|dt$rYYdSwxYwt$rYdSwxYw)N repo_typespacemodel`z9` does not seem to be a valid repo identifier on the Hub.)r TOOL_CONFIG_FILErEnvironmentError Exception)repo_idr( hub_kwargss U/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/agents/tools.py get_repo_typer2@s !1SSWS SSSw "  G%5 W W WJ W W W77& k k k"#iw#i#i#ijj j   777  wws, A;A&A+&A;*A++ A;:A;zufrom transformers import launch_gradio_demo from {module_name} import {class_name} launch_gradio_demo({class_name}) cX|jtfd}||_|S)Ncv|g|Ri|t|ts|dSdSN) isinstance PipelineToolvalidate_arguments)selfargskwargs original_inits r1new_initz%validate_after_init..new_init]sW d,T,,,V,,,$ -- &  # # % % % % % & &)__init__r)clsr=r<s @r1validate_after_initrAZsDLM =&&&&& CL Jr>c PeZdZUdZeed<eed<eeeeeeefffed<eed<dZ dZ dZ d Z d Z d Ze dd edeedeefdZ dd ededeedeeeefdedef dZedZedZd S)Toolak A base class for the functions used by the agent. Subclass this and implement the `__call__` method as well as the following class attributes: - **description** (`str`) -- A short description of what your tool does, the inputs it expects and the output(s) it will return. For instance 'This is a tool that downloads a file from a `url`. It takes the `url` as input, and returns the text contained in the file'. - **name** (`str`) -- A performative name that will be used for your tool in the prompt to the agent. For instance `"text-classifier"` or `"image_generator"`. - **inputs** (`Dict[str, Dict[str, Union[str, type]]]`) -- The dict of modalities expected for the inputs. It has one `type`key and a `description`key. This is used by `launch_gradio_demo` or to make a nice space from your tool, and also can be used in the generated description for your tool. - **output_type** (`type`) -- The type of the tool output. This is used by `launch_gradio_demo` or to make a nice space from your tool, and also can be used in the generated description for your tool. You can also override the method [`~Tool.setup`] if your tool as an expensive operation to perform before being usable (such as loading a model). [`~Tool.setup`] will be called the first time you use your tool, but not at instantiation. name descriptioninputs output_typecd|_dS)NFis_initializedr9r:r;s r1r?z Tool.__init__s#r>c  ttttd}gd}|D]A\}}t||d}t ||st d|d|jdB|jD]R\}}d|vsJd|d|d|vrtd|d |dd |dd |vsJd|d St|d d|vsJt |tsvtj |j }t|jt|jkstddSdS)N)rErDrFrG)stringintegernumberimageaudioanybooleanzYou must set an attribute z of type .typezInput 'z' should specify a type.z ': type 'z/' is not an authorized value, should be one of rEz' should have a description.rGzTool's 'forward' method should take 'self' as its first argument, then its next arguments should match the keys of tool attribute 'inputs'.)strritemsgetattrr6 TypeError__name__rFr.r7inspect signatureforwardset parameterskeys) r9required_attributesauthorized_typesattr expected_type attr_value input_name input_contentr\s r1r8zTool.validate_argumentss    _^^#6#<#<#>#> g g D- tT22Jj-88 g eT e eMLb e e efff g)-):):)<)< f f %J ]***,Zj,Z,Z,Z***V$,<<<MjMM=3HMMzJMMM!M1113eZ3e3e3e1111t]D115EEEEE$ -- )$,77Iy+002233s4;;K;K;M;M7N7NNNb  NNr>c tdS)Nz-Write this method in your subclass of `Tool`.)NotImplementedrKs r1r]z Tool.forwardsMNNNr>cdt|i|\}}|j|i|}t||jSr5)r"r]r#rG)r9r:r;outputss r1__call__z Tool.__call__sA*D;F;; f$,///#GT-=>>>r>cd|_dS)z Overwrite this method here for any operation that is expensive and needs to be executed before you start using your tool. Such as loading a big model. TNrI)r9s r1setupz Tool.setups #r>ctj|d|jdkrtd|d|dt ||}|jj}|dd}|d|jj}tj |d }tj |r?t|d d 5}tj |}d d d n #1swxYwYni}||j|j|jt#|jd}t|dd 5}|tj|dddzd d d n #1swxYwYtj |d} t| dd 5}|t*||jjd d d n #1swxYwYtj |d} g} |D]$} | t1| %t3t5| } t| dd 5}|d | dzd d d d S#1swxYwYd S)a Saves the relevant code files for your tool so it can be pushed to the Hub. This will copy the code of your tool in `output_dir` as well as autogenerate: - a config file named `tool_config.json` - an `app.py` file so that your tool can be converted to a space - a `requirements.txt` containing the names of the module used by your tool (as detected when inspecting its code) You should only use this method to save tools that are defined in a separate module (not `__main__`). Args: output_dir (`str`): The folder in which you want to save your tool. T)exist_ok__main__z We can't save the code defining z in z{ as it's been defined in __main__. You have to put this code in a separate module so we can include it in the saved folder.rTr&rutf-8encodingN) tool_classrErDrFrGwr)indent sort_keys zapp.py) module_name class_namezrequirements.txt)osmakedirs __module__ ValueErrorr __class__splitrZpathjoinisfileopenjsonloadrErDrFrVrGwritedumpsAPP_FILE_TEMPLATEformatextendrlistr^) r9 output_dir module_filesr| last_module full_name config_filef tool_configapp_filerequirements_fileimportsmodules r1savez Tool.saves J.... ?j ( (g4ggZggg *$ ;; n/ !'',,R0 ">>T^%<>> gll:/ABB 7>>+ & & k3999 +Q"ill  + + + + + + + + + + + + + + +K$+Ikt/00   +sW 5 5 5 N GGDJ{1EEEL M M M N N N N N N N N N N N N N N N7<< H55 (C' 2 2 2 ka GG%,,QUQ_Qh,ii j j j k k k k k k k k k k k k k k kGLL5GHH" 0 0F NN;v.. / / / /s7||$$ #S7 ; ; ; /q GGDIIg&&- . . . / / / / / / / / / / / / / / / / / /sHC22C69C6;.E55E9<E92:G88G<?G<:,J33J7:J7Nr/ model_repo_idtokenc  gd fd|D}t|fi||d<t|tfd|i|dddd}|du}|t|tfd|i|dddd}|t |dt |d 5}tj|} dddn #1swxYwY|sd | vrt |d | d } n| } | d } t| |fd|i|} t| j dkr | d| _ | j | dkr/t | jd| d| _ t| jdkr | d| _| j| dkr/t | jd| d| _| j| dkr | d| _| j| dkr | d| _| di|S)a Loads a tool defined on the Hub. Loading a tool from the Hub means that you'll download the tool and execute it locally. ALWAYS inspect the tool you're downloading before loading it within your runtime, as you would do when installing a package using pip/npm/apt. Args: repo_id (`str`): The name of the repo on the Hub where your tool is defined. model_repo_id (`str`, *optional*): If your tool uses a model and you want to use a different model than the default, you can pass a second repo ID or an endpoint url to this argument. token (`str`, *optional*): The token to identify you on hf.co. If unset, will use the token generated when running `huggingface-cli login` (stored in `~/.huggingface`). kwargs (additional keyword arguments, *optional*): Additional keyword arguments that will be split in two: all arguments relevant to the Hub (such as `cache_dir`, `revision`, `subfolder`) will be used when downloading the files for your tool, and the others will be passed along to its init. ) cache_dirforce_downloadresume_downloadproxiesrevisionr( subfolderlocal_files_onlyc$i|] \}}|v || Sr).0kvhub_kwargs_namess r1 z!Tool.from_hub..s*OOOtq!>N9N9Na9N9N9Nr>r(rF) _raise_exceptions_for_gated_repo%_raise_exceptions_for_missing_entries'_raise_exceptions_for_connection_errorsNzY does not appear to provide a valid configuration in `tool_config.json` or `config.json`.rtru custom_toolzO does not provide a mapping to custom tools in its configuration `config.json`.rwrrDz_ implements a different name in its configuration and class. Using the tool configuration name.rEzm implements a different description in its configuration and class. Using the tool configuration description.rFrGr)rWr2rr,rr-rrrrlenrDloggerwarningrZrErFrG) r@r/rrr;r0resolved_config_fileis_tool_configreaderconfigrrwrs @r1from_hubz Tool.from_hubs6B    POOOv||~~OOO #0"F"F:"F"F ;*       .32749    .T9  '#.$$$ $ 276;8=$$$$  '"uuu & 9 9 9 'VYv&&F ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' !F**&ooo!/KK K . 2:wbbebWabb z  1 $ $)&1JO ?k&1 1 1 NN&&&&   *&1JO z% & &! + +%0%?J "  ![%? ? ? NN&222   &1%?J "   H 5 5 5 +H 5J   ![%? ? ?%0%?J "z##F###sB88B<?B< Upload toolFcommit_messageprivate create_prreturnc t|||ddd}|j}t|ddgidtj5}||t d|d d tj |t|||||d cd d d S#1swxYwYd S) a Upload the tool to the Hub. For this method to work properly, your tool must have been defined in a separate module (not `__main__`). For instance: ``` from my_tool_module import MyTool my_tool = MyTool() my_tool.push_to_hub("my-username/my-space") ``` Parameters: repo_id (`str`): The name of the repository you want to push your tool to. It should contain your organization name when pushing to a given organization. commit_message (`str`, *optional*, defaults to `"Upload tool"`): Message to commit while pushing. private (`bool`, *optional*): Whether or not the repository created should be private. token (`bool` or `str`, *optional*): The token to use as HTTP bearer authorization for remote files. If unset, will use the token generated when running `huggingface-cli login` (stored in `~/.huggingface`). create_pr (`bool`, *optional*, defaults to `False`): Whether or not to create a PR with the uploaded files or directly commit. Tr)gradio)r/rrrpr( space_sdktagstool)r(z!Uploading the following files to z: ,)r/r folder_pathrrr(N) r r/rtempfileTemporaryDirectoryrrinforr~listdirr)r9r/rrrrrepo_urlwork_dirs r1 push_to_hubzTool.push_to_hub_s3B    "&6(!3wGGGG  ( * * h IIh    KKgGggsxxPRPZ[cPdPdGeGegg h h h -$#!                   sA/CCCcLddlGfddt}||S)z8 Creates a [`Tool`] from a gradio tool. rNc(eZdZfdZdZxZS)+Tool.from_gradio..GradioToolWrapperc(t|j|_|j|_d|_||_t j|jj }d|D|_ dS)NrMci|]}|dS)r)rkeys r1rzHTool.from_gradio..GradioToolWrapper.__init__..s<<<3sB<<) superr?rDrErG _gradio_toolrr\runr_r`rF)r9r func_argsrr[s r1r?z4Tool.from_gradio..GradioToolWrapper.__init__s  """(- #/#; #+ $0! !2!2<3C!D!D!O!T!T!V!VWW <<)<<< r>c&|jj|i|Sr5)rrrKs r1r]z3Tool.from_gradio..GradioToolWrapper.forwards,t(,d=f===r>rZr __qualname__r?r] __classcell__)rr[s@r1GradioToolWrapperrsM = = = = = = > > > > > > >r>r)r[rC) gradio_toolrr[s @r1 from_gradiozTool.from_gradiosU  > > > > > > > > > >! ---r>c>Gddt}||S)z; Creates a [`Tool`] from a langchain tool. c$eZdZfdZdZxZS)1Tool.from_langchain..LangChainToolWrapperct|j|_|j|_t |j|_d|_||_ dS)NrM) rr?rDlowerrEparse_langchain_argsr:rFrGlangchain_tool)r9_langchain_toolrs r1r?z:Tool.from_langchain..LangChainToolWrapper.__init__sb  """+06688 #2#> 2?3GHH #+ &5###r>c|}t|D]C\}}|t|jkr&t t |j}|||<D|j|Sr5)copy enumeraterrFnextiterrr)r9r:r; tool_inputindexargument input_keys r1r]z9Tool.from_langchain..LangChainToolWrapper.forwardsv#[[]] '099OE8s4;////$(dk):):$;$; 08 9-*..z:::r>rrs@r1LangChainToolWrapperrsG 6 6 6 6 6 ; ; ; ; ; ; ;r>r)rC)rrs r1from_langchainzTool.from_langchains?  ; ; ; ; ;4 ; ; ;"$#N333r>NN)rNNF)rZrr__doc__rV__annotations__rr rUr?r8r]rlrnr classmethodr rboolr staticmethodrrrr>r1rCrCgs* III d3c4i 0011 2222$$$<OOO??? ###:/:/:/x(,# m$m$m$ }m$} m$m$m$[m$d,"&,0 7777$ 7 dCi() 7  7 7777r..\.*44\444r>rCz - {{ tool.name }}: {{ tool.description }} Takes inputs: {{tool.inputs}} Returns an output of type: {{tool.output_type}} rdescription_templatercPt|}||}|S)N)r)compile_jinja_templaterender)rrcompiled_templaterendereds r1get_tool_description_with_argsrs5./CDD '' (H Or>cR ddl}ddlmddlm}n#t $rt dwxYwt j|jt jdkrt d|jdfd}|d d }||j d <| |S) Nr) TemplateError)ImmutableSandboxedEnvironmentz)template requires jinja2 to be installed.z3.1.0zAtemplate requires jinja2>=3.1.0 to be installed. Your version is rTc|r5r)messagers r1raise_exceptionz/compile_jinja_template..raise_exceptionsmG$$$r>T) trim_blocks lstrip_blocksr) jinja2jinja2.exceptionsrjinja2.sandboxr ImportErrorrparse __version__globals from_string)templaterrr jinja_envrs @r1rrsG 333333@@@@@@@ GGGEFFFG}V'((7=+A+AAAvagasvvvwww%%%%%.-$dSSSI+:I'(   * **s.cxeZdZdZeZdZeZdZdZ dZ de iZ e Z d fd ZfdZdZd Zd Zd ZxZS) r7a0 A [`Tool`] tailored towards Transformer models. On top of the class attributes of the base class [`Tool`], you will need to specify: - **model_class** (`type`) -- The class to use to load the model in this tool. - **default_checkpoint** (`str`) -- The default checkpoint that should be used when the user doesn't specify one. - **pre_processor_class** (`type`, *optional*, defaults to [`AutoProcessor`]) -- The class to use to load the pre-processor - **post_processor_class** (`type`, *optional*, defaults to [`AutoProcessor`]) -- The class to use to load the post-processor (when different from the pre-processor). Args: model (`str` or [`PreTrainedModel`], *optional*): The name of the checkpoint to use for the model, or the instantiated model. If unset, will default to the value of the class attribute `default_checkpoint`. pre_processor (`str` or `Any`, *optional*): The name of the checkpoint to use for the pre-processor, or the instantiated pre-processor (can be a tokenizer, an image processor, a feature extractor or a processor). Will default to the value of `model` if unset. post_processor (`str` or `Any`, *optional*): The name of the checkpoint to use for the post-processor, or the instantiated pre-processor (can be a tokenizer, an image processor, a feature extractor or a processor). Will default to the `pre_processor` if unset. device (`int`, `str` or `torch.device`, *optional*): The device on which to execute the model. Will default to any accelerator available (GPU, MPS etc...), the CPU otherwise. device_map (`str` or `dict`, *optional*): If passed along, will be used to instantiate the model. model_kwargs (`dict`, *optional*): Any keyword argument to send to the model instantiation. token (`str`, *optional*): The token to use as HTTP bearer authorization for remote files. If unset, will use the token generated when running `huggingface-cli login` (stored in `~/.huggingface`). hub_kwargs (additional keyword arguments, *optional*): Any additional keyword argument to send to the methods that will load the data from the Hub. NzThis is a pipeline toolpipelinepromptc tstdtstd||jt d|j}||}||_||_||_||_||_ |in||_ | ||j d<||_ ||j d<t dS)Nz/Please install torch in order to use this tool.z4Please install accelerate in order to use this tool.zHThis tool does not implement a default checkpoint, you need to pass one. device_mapr)rr rdefault_checkpointrr* pre_processorpost_processordevicer model_kwargsr0rr?) r9r*rrrrrrr0rs r1r?zPipelineTool.__init__s"## QOPP P&(( VTUU U =&. !klll+E  !M *, $"."6BBL  !.8D l +$#(  r>ct|jtr"|jj|jfi|j|_t|jtr)|jj|jfi|j|j|_|j |j|_ nc,||S)zQ Uses the `pre_processor` to prepare the inputs for the `model`. )r)r9 raw_inputss r1encodezPipelineTool.encodeYs!!*---r>cvtj5|jdi|cdddS#1swxYwYdS)z7 Sends the inputs through the `model`. Nr)torchno_gradr*)r9rFs r1r]zPipelineTool.forward_s]__ ( (4:'''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s .22c,||S)zG Uses the `post_processor` to decode the model output. )r)r9rks r1decodezPipelineTool.decodefs""7+++r>ct|i|\}}|js||j|i|}d|D}d|D}t ||j}|i||}t |d}||}t||j S)NcNi|]"\}}t|tj||#Srr6r(Tensorrrrs r1rz)PipelineTool.__call__..ts1```$!QJqRWR^D_D_`A```r>cNi|]"\}}t|tj||#Srr.r0s r1rz)PipelineTool.__call__..us2hhhdaJWXZ_ZfLgLghQhhhr>cpu) r"rJrnr&rWr%rr]r+r#rG)r9r:r;encoded_inputs tensor_inputsnon_tensor_inputsrkdecoded_outputss r1rlzPipelineTool.__call__ls*D;F;; f"  JJLLL$d5f55``.*>*>*@*@``` hhn.B.B.D.Dhhh' t{CC,,F.F4EFGG %00++g..#OT5EFFFr>)NNNNNNN)rZrrrrrrrrrErDrVrFrGr?rnr&r]r+rlrrs@r1r7r7s##J(K(+K D_FK######J4... (((,,, GGGGGGGr>r7rwc ddl}n#t$rtdwxYw| fd}g}|jD]\}}|d}|dkr*|||=|dkr*|||m|d vr*|||d |d }t||j }|d vsJd |d | ||||j j  dS)z Launches a gradio demo for a tool. The corresponding tool class needs to properly implement the class attributes `inputs` and `output_type`. Args: tool_class (`type`): The class of the tool for which to launch the demo. rNz.fnstT$V$$$r>rUrP)labelrQ)rMrNrOz Input type 'z' not supported.)rMrPrQz Output type ')r9rFrktitlearticle)rr rFrWappendImageAudioTextboxrrG InterfacerZrElaunch) rwgrr9 gradio_inputsrf input_details input_type error_message gradio_outputrs @r1launch_gradio_demorIsZ ZZZXYYYZ :<%> , ,! M"6*    !;!; < < < < 7 " "   !;!; < < < < : : :  *!=!= > > > >G:GGGM]++ +*M 8 8 8 8:i-:i:i:i 8 8 8LL !   fhhhhhs"DocumentQuestionAnsweringToolImageQuestionAnsweringToolSpeechToTextToolTextToSpeechToolTranslationToolPythonInterpreterToolDuckDuckGoSearchTool)document_question_answeringimage_question_answeringspeech_to_texttext_to_speech translationpython_interpreter web_searchc |tvrCt|}tjd}|j}t ||}||fd|i|St d|dtj|f||d|S)a Main function to quickly load a tool, be it on the Hub or in the Transformers library. Loading a tool means that you'll download the tool and execute it locally. ALWAYS inspect the tool you're downloading before loading it within your runtime, as you would do when installing a package using pip/npm/apt. Args: task_or_repo_id (`str`): The task for which to load the tool or a repo ID of a tool on the Hub. Tasks implemented in Transformers are: - `"document_question_answering"` - `"image_question_answering"` - `"speech_to_text"` - `"text_to_speech"` - `"translation"` model_repo_id (`str`, *optional*): Use this argument to use a different model than the default one for the tool you selected. token (`str`, *optional*): The token to identify you on hf.co. If unset, will use the token generated when running `huggingface-cli login` (stored in `~/.huggingface`). kwargs (additional keyword arguments, *optional*): Additional keyword arguments that will be split in two: all arguments relevant to the Hub (such as `cache_dir`, `revision`, `subfolder`) will be used when downloading the files for your tool, and the others will be passed along to its init. transformersrz(You're loading a tool from the Hub from a. Please make sure this is a source that you trust as the code within that tool will be executed on your machine. Always verify the code of the tools that you load. We recommend specifying a `revision` to ensure you're loading the code that you have checked.)rr) TOOL_MAPPING importlib import_moduleagentsrXr warning_oncerCr)task_or_repo_idrrr;tool_class_name main_module tools_modulerws r1 load_toolrcsB,&&&7-n== ") \?;; z-??u???? +} + + +   }_aMQVaaZ`aaar>cfd}|S)z< A decorator that adds a description to a function. c.|_|j|_|Sr5)rErZrD)funcrEs r1innerzadd_description..inners&M  r>r)rErgs` r1add_descriptionrhs#  Lr>c eZdZddedeefdZedZedZ ddee ee e ee e efd ee d ee d e d ef d ZdS)EndpointClientN endpoint_urlrcJit|ddi|_||_dS)Nrz Content-Typezapplication/json)rheadersrk)r9rkrs r1r?zEndpointClient.__init__s> U+++ .   )r>ctj}||dtj|}|dS)NPNG)rrt)ioBytesIOrbase64 b64encodegetvaluer+)rP_bytesb64s r1 encode_imagezEndpointClient.encode_imagesO 6% (((v0011zz'"""r>ctstdddlm}t j|}t j|}||S)NzbThis tool returned an image but Pillow is not installed. Please install it (`pip install Pillow`).r)r>) rr PILr>rs b64decoderqrrr) raw_imager>rwrvs r1 decode_imagezEndpointClient.decode_imagesh"$$ t  y))Czz&!!!r>FrFparamsdata output_imagerci}|r||d<|r||d<t|j|j||}|r||jS|S)NrFr_)rnrr)rpostrkrnr}contentr)r9rFr~rrpayloadresponses r1rlzEndpointClient.__call__s  ' &GH   +$*GL !==%%d&7T[bf%gg  #$$X%566 6==?? "r>r5)NNNF)rZrrrVr r?rrxr}r rrbytesrrrlrr>r1rjrjs))S)#))))##\#  " "\ "JN!% $" ##sD$s)T$s)_DEF##uo #  # ######r>rjr:c|}|D]}d|vr|d|S)zTParse the args attribute of a LangChain tool to create a matching inputs dictionary.r;)rr!pop)r:rF arg_detailss r1rr(sI YY[[F}}%% k ! ! OOG $ $ $ Mr>c0eZdZdZddedeefdZdS)ToolCollectionai Tool collections enable loading all Spaces from a collection in order to be added to the agent's toolbox. > [!NOTE] > Only Spaces will be fetched, so you can feel free to add models and datasets to your collection if you'd > like for this collection to showcase them. Args: collection_slug (str): The collection slug referencing the collection. token (str, *optional*): The authentication token if the collection is private. Example: ```py >>> from transformers import ToolCollection, ReactCodeAgent >>> image_tool_collection = ToolCollection(collection_slug="huggingface-tools/diffusion-tools-6630bb19a942c2306a2cdb6f") >>> agent = ReactCodeAgent(tools=[*image_tool_collection.tools], add_base_tools=True) >>> agent.run("Please draw me a picture of rivers and lakes.") ``` Ncollection_slugrct|||_d|jjD|_d|jD|_dS)Nrmc2h|]}|jdk |jS)r)) item_typeitem_id)ritems r1 z*ToolCollection.__init__..Ms)kkktQUQ_cjQjQjdlQjQjQjr>cBh|]}t|Sr)rCr)rr/s r1rz*ToolCollection.__init__..Ns$OOOdmmG,,OOOr>)r _collectionrW _hub_repo_idstools)r9rrs r1r?zToolCollection.__init__KsO)/GGGkkt7G7MkkkOODr5)rZrrrrVr r?rr>r1rr1sR2PPPHSMPPPPPPr>r tool_functionctddvrtddd}Gfddt}t j}t jdtjjgt|j z}| | }||j _ ||_|S) a& Converts a function into an instance of a Tool subclass. Args: tool_function: Your function. Should have type hints for each input and a type hint for the output. Should also have a docstring description including an 'Args:' part where each argument is described. functionrzKTool return type not found: make sure your function has a return type hint!rDrCceZdZdZdZddZddZefdZdS) tool..SpecificToolrDrEr_ propertiesrrUc|i|Sr5r)r9r:r;rs r1r]z"tool..SpecificTool.forwardds =$1&11 1r>N) rZrrrDrErFrGrr])r_rsr1 SpecificToolr^su&! / L),7 *62 }   2 2 2 2   2 2 2r>rr9)r_)rr capitalizerCr[r\ ParameterPOSITIONAL_OR_KEYWORDrr_r!replacer] __signature__rZ)rr}roriginal_signaturenew_parameters new_signaturer_s` @r1rrQs!// ;Jz!!&'tuuuv&1133999J22222222t222!*=99'0A0WXXY\`%,,..]]N'....IIM)6L&&L <>>r>r5r)Irsr[r[rqrr~r functoolsrrtypingrrrrr r huggingface_hubr r r rrhuggingface_hub.utilsrrr packagingrdynamic_module_utilsrrr models.autorutilsrrrrrrrr agent_typesr"r# get_loggerrZrr( accelerater$accelerate.utilsr%r,r2rrArC!DEFAULT_TOOL_DESCRIPTION_TEMPLATErVrrr7rIrZrcrhrjrrrrr>r1rs:"  &&&&&&&&================hhhhhhhhhhhhhhXXXXXXXXXX ('''''                    CBBBBBBB  H % %LLL0''''''//////&&   ]4]4]4]4]4]4]4]4@ %!LmSqt ++ +&QGQGQGQGQG4QGQGQGh(4((((X$C <(($1( .b.b.b.bb   1#1#1#1#1#1#1#1#htCH~$sCx.PPPPPPPP@Tr>