o h @sddlmZmZmZGdddeZGdddeZdededefd d ZGd ddeZGd ddeZGd ddeZ GdddeZ GdddeZ GdddeZ GdddeZ GdddeZdS)) BaseModelFieldValidationErrorc@6eZdZUedddZeed<edddZeed<dS) FetchNewsParams.zNews search query descriptionqueryMax results to return max_resultsN __name__ __module__ __qualname__rr str__annotations__r intrr:C:\Users\shubh\Videos\agent\agent\modules\tools\schemas.pyr rc@r) FetchArxivParams.zArxiv search queryrr r r r Nr rrrrrrr tool_nameparamsreturnc s|ztfddtjD}|d}||WSty(tdddty=}z td||d}~ww)z-Validate tool parameters against their schemac3s |] }|dkr|VqdS)nameNr).0trrr sz%validate_tool_args.. parameterszTool z not foundNzInvalid parameters: ) nextZ tool_registrytool_definitionsZmodel_validateZ model_dump StopIteration ValueErrorrerrors)rrZtool_defmodelerrrvalidate_tool_argss r(c@>eZdZUdZedddZeed<edddd d Ze ed <d S) rzGoogle News search parameters. Search termrr 2r gelerr N rrr__doc__rr rrr rrrrrr c@r)) rzarXiv search parameters.z arXiv queryrr r+r,r-zMax papers to returnr.r Nr1rrrrrr3c@r)) FetchDDGParamszDuckDuckGo search parameters.r*rr r+r,r-r r.r Nr1rrrrr4$r3r4c@r)) FetchTorrentsParamszTorrent search parameters.zBase search termrr r+r,r-r r.r Nr1rrrrr5)r3r5c@r)) FetchWikiParamszWikipedia search parameters.r*rr r+r,r-r r.r Nr1rrrrr6/r3r6c@r)) FetchIMDBParamszIMDB search parameters.r*rr r+r,r-r r.r Nr1rrrrr74r3r7c@r)) FetchYouTubeParamszYouTube search parameters.r*rr r+r,r-r r.r Nr1rrrrr89r3r8c@r)) FetchFinanceParamszFinance data search parameters.zStock ticker symbol(s)rr r+r,r-r r.r Nr1rrrrr9>r3r9N)Zpydanticrrrrrrdictr(r4r5r6r7r8r9rrrrs