Format: application/x-python-code
1 o
2
3 ��h�
4 � @ s� d dl mZmZmZ G dd� de�ZG dd� de�Zdededefd d
5 �ZG d d� de�ZG d d� de�ZG d
6 d� de�Z G dd� de�Z
7 G dd� de�Z G dd� de�Z G dd� de�Z
8 G dd� de�ZdS )� )� BaseModel�Field�ValidationErrorc @ �6 e Zd ZU eddd�Zeed< eddd�Zeed< dS ) �FetchNewsParams.zNews search query�� description�query� �Max results to return� max_resultsN� �__name__�
9 __module__� __qualname__r r �str�__annotations__r �int� r r �:C:\Users\shubh\Videos\agent\agent\modules\tools\schemas.pyr �
10 r c @ r ) �FetchArxivParams.zArxiv search queryr r r
11 r r Nr
12 r r r r r r r � tool_name�params�returnc
13 s| zt � fdd�tjD ��}|d }|�|��� W S ty( td� � d��d� ty= } z
14 td|�� � ��|�d}~ww )z-Validate tool parameters against their schemac 3 s � | ] }|d � kr|V qdS )�nameNr )�.0�t�r r r � <genexpr> s � z%validate_tool_args.<locals>.<genexpr>�
15 parameterszTool z
16 not foundNzInvalid parameters: ) �nextZ
17 tool_registry�tool_definitionsZmodel_validateZ
18 model_dump�
19 StopIteration�
20 ValueErrorr �errors)r r Ztool_def�model�er r r �validate_tool_args s ��r( c @ �> e Zd ZU dZeddd�Zeed< edddd d
21 �Ze ed < d S )
22 r z Google News search parameters.� Search termr r �
23 � �2 r ��ge�ler r N�
24 r r r �__doc__r r r r r r r r r r r �
25 c @ r) )
26 r zarXiv search parameters.z arXiv queryr r r+ r, r- zMax papers to returnr. r Nr1 r r r r r r3 c @ r) )
27 �FetchDDGParamsz DuckDuckGo search parameters.r* r r r+ r, r- r r. r Nr1 r r r r r4 $ r3 r4 c @ r) )
28 �FetchTorrentsParamszTorrent search parameters.zBase search termr r r+ r, r- r r. r Nr1 r r r r r5 ) r3 r5 c @ r) )
29 �FetchWikiParamszWikipedia search parameters.r* r r r+ r, r- r r. r Nr1 r r r r r6 / r3 r6 c @ r) )
30 �FetchIMDBParamszIMDB search parameters.r* r r r+ r, r- r r. r Nr1 r r r r r7 4 r3 r7 c @ r) )
31 �FetchYouTubeParamszYouTube search parameters.r* r r r+ r, r- r r. r Nr1 r r r r r8 9 r3 r8 c @ r) )
32 �FetchFinanceParamsz Finance data search parameters.zStock ticker symbol(s)r r r+ r, r- r r. r Nr1 r r r r r9 > r3 r9 N)Zpydanticr r r r r r �dictr( r4 r5 r6 r7 r8 r9 r r r r �<module> s