Goodguygregory93/ai-agents-certification-code
0
1�
2 c��hQ � �l � S SK Jr S SKJr S SKJr S SKJr S SKrS SK r S SK3J r S SK r " S S5 r
g) � )�DuckDuckGoSearchToolSpec)� FunctionTool)�WikipediaReader)�YouTubeTranscriptApiN)�
BeautifulSoupc �J � \ rS rSrSrS S jrS rS\4S jrS\4S jr S r4S5r g )
�Tools�! ar 6Tools class which holds all of the the tools the Basic Agent will7leverage to answer the GAIA Challenge8 9Included Tools: 10 11* duck_duck_go_search_tool: allows for searching Duck Duck Go for answers *see create_duck_duck_go_tool for usage example*12* wikipedia_search_tool: allows for searching wikipedia for specific details about a subject matter, leveraging wikipedia13 14c � � Xl U R 5 U l [ R " U R15 5 U l [ R " U R 5 U l [ R " U R 5 U l U( a} [ [ R R S- 5 [ [ R R S- 5 [ [ R R S- 5 [ S5 U R U R16 U R U R /U l g )Nu/ ✅ 📖 Wikipedia Search tool has been createdu; ✅ 📹 Youtube Transcription Search tool has been createdu6 ✅ 💻 📰 Wired Top Articles tool has been createdz---------------)
�status_updates�create_duck_duck_go_tool�duck_duck_go_search_toolr �
from_defaults�wikipedia_search_tool�youtube_transcription_tool�top_wired_articles_tool�print�colorama�Fore�YELLOW� tool_belt)�selfr s �C/Users/gregwitt/Github/clones/ai-agents-certification-code/tools.py�__init__�Tools.__init__- s� � � ,��(,�(E�(E�(G��%�%1�%?�%?��@Z�@Z�%[��"�*6�*D�*D�T�Ed�Ed�*e��'�'3�'A�'A�$�B^�B^�'_��$� � �(�-�-�&�&�)Z�Z� [� �(�-�-�&�&�+f�g� h� �(�-�-�&�&�+a�b� c� �&�M� �7�7��9S�9S�UY�Ut�Ut� �4�4�6��� c �� � [ 5 n[ R " UR 5 nU R ( a& [ [ R R S- 5 U$ )a� 17creates the DuckDuckGo Search tool from LlamaIndex18and returns the tool for later use19 20Example usage:21 22```python23response = self.duck_duck_go_search_tool("Who is the silver robot from daft punk?")24 25print(response.raw_output[-1]['body'])26```27 28[Tool Documentation](https://llamahub.ai/l/tools/llama-index-tools-duckduckgo?from=)29 30Args: 31 self : class instance32Returns:33 search_tool (FunctionTool): represents the FunctionTool Spec for DuckDuckGoSearch34u+ ✅ 🦆 Duck Duck Go tool has been created) r r r �duckduckgo_full_searchr r r r r )r � tool_spec� search_tools r r
� Tools.create_duck_duck_go_tool= sI � �* -�.� �"�0�0��1Q�1Q�R� � � � � �(�-�-�&�&�+V�W� X��r �queryc �: � [ 5 nUR U/S9nU$ )a� 35creates a Wikipedia Search FunctionTool for searching wikipedia articles for specific pages36the details from the specified query37 38[Tool Documentation](https://llamahub.ai/l/readers/llama-index-readers-wikipedia?from=)39 40Args: 41 self: class instance 42 query (str): the specific article in question to be searched in wikipedia.43Returns: 44 found_document : the response from the Wikipedia Reader which is the found article 45)�pages)r � load_data)r r" � wiki_reader�found_documents r r �Tools.wikipedia_search_tool\ s' � � &�'� �$�.�.�e�W�.�=�� � r �46video_linkc � � [ R " S5 nUR U5 nUc gSnUR 5 R S5 S n[ 5 nUR
US/S9nU H nXHR S- -
nM U$ ) aY 47checks if the supplied video(s) is in fact a valid youtube video and then proceeds to load the entire transcription 48of the provided youtube video for quick searching and relevance querying for your llm.49leverages the `YoutubeTranscriptionReader` from LlamaIndex to provide a string response of the 50youtube video in question. 51 52 53[Tool Documentation](https://pypi.org/project/youtube-transcript-api/)54 55Args: 56 video_link (list[str]): videos or video in question that you desire information about.57Returns:58 transcript_text: the converted youtube video's audio transcription content in a string59zv=(\w+)z0You haven't provided a legitimate youtube video.� �=� zen-US)�video_id� languages� )�re�compile�search�group�splitr �fetch�text) r r) � videoIdRegex�mo�transcript_text�videoId�ytt_api�fetched_transcript�snippets r r � Tools.youtube_transcription_toolp s� � � �z�z�*�-� �
�
�
��
,��
�:�E� �O��h�h�j�&�&�s�+�A�.�G�*�,�G�!(� � ��G�9� �!U� �-�� �<�<�#�#5�5�� .� #� "r c �� � [ R " 5 nSnUR U5 n[ UR S5 nUR SS[ R " S5 05 n0 nU H� nUR S5 R nUR S5 R n UR S5 R S n60UR SS[ R " S 5 05 R n M� [ $ )61z'62 63 64Returns:65 top_articles (dict): 66 67z&https://www.wired.com/about/rss-feeds/z html.parser�div�class�SummaryItemContent�span�a�href�68BylineName) �requests�Session�getr r7 �find_allr1 r2 �find�attrs� popular_wired_unordered_list) r �session� wired_url�rss_feed_page_resp�soup� popular_wired_popular_li_list�found_articles�69article_li�category�
article_title� article_link�article_authors r r � Tools.top_wired_articles_tool� s� � � �"�"�$��<� � $�[�[�� 3�� � /� 4� 4�m�D�� )-�
�
�e�g�r�z�z�Rf�Gg�=h�(i�%���7�J�!���v�.�3�3�H� &�O�O�C� 0� 5� 5�M�%�?�?�3�/�5�5�f�=�L� '�_�_�V�g�r�z�z�,�?W�5X� Y� ^� ^�N� 8� ,�+r )r r r r r r N)F) �__name__�70__module__� __qualname__�__firstlineno__�__doc__r r
�strr r r �__static_attributes__� r r r r ! s/ � � �6� �> �3� �(!#�S� !#�F ,r r )� llama_index.tools.duckduckgor �llama_index.core.toolsr � llama_index.readers.wikipediar �youtube_transcript_apir r1 rH �bs4r r r rb r r �<module>rh s) �� A� /� 9� 7� � � � �0O,� O,r