o
    D±h{  ã                
   @   sÔ   U d dl mZmZmZmZmZmZmZmZm	Z	 ddl
mZmZmZmZmZmZmZ ddlmZ ddlmZ dededefd	d
„Zedeƒedeƒedeƒedeƒedeƒedeƒedeƒede	ƒgZee ed< dS )é   )	ÚToolDefinitionÚFetchNewsParamsÚFetchArxivParamsÚFetchDDGParamsÚFetchTorrentsParamsÚFetchWikiParamsÚFetchIMDBParamsÚFetchYouTubeParamsÚFetchFinanceParamsé    )Úfetch_torrentsÚ
fetch_newsÚfetch_wikipediaÚ
fetch_imdbÚfetch_youtubeÚfetch_financeÚsearch_google)r   )Úfetch_arxivÚnameÚmodelÚreturnc                 C   s   t | |j ¡ | ¡ d�S )N)r   ÚdescriptionÚ
parameters)r   Ú__doc__ÚstripZmodel_json_schema)r   r   © r   úCC:\Users\shubh\Videos\agent\agent\modules\tools\tool_definitions.pyÚschema_to_tool   s
   ýr   r   r   r   r   r   r   r   r   Útool_definitionsN)Zschemasr   r   r   r   r   r   r   r	   r
   Zmodules.search.webr   r   r   r   r   r   r   Zmodules.search.newsZmodules.search.arxivr   ÚstrÚtyper   r   ÚlistÚ__annotations__r   r   r   r   Ú<module>   s   . $	ø