DakshChaudhary/Agent_GAIA_Benchmark
0
1�
2 gUh� � �J � S SK r S SKrS SKrS SKJr S SKJr S SKJr S SK J3r4 S SK J r S SK
Jr S SKJr S S KJr \R& " \ R( \R* S59 \R, " 5 R/ \R0 " \ R( S 95 " S S
5 rg)� N)�6ReActAgent)�get_language_model)�web_search_tools)�get_downloader_tool)�get_image_interpreter_tool)�get_calculator_tool)�get_pandas_tool)�gaia_system_prompt)�stream�level)r c �, � \ rS rSrS rS\S\4S jrSrg)� GaiaAgent� c �� � [ 5 n[ 5 [ 5 [ 5 [ 5 /n[ 5 U l X- U l [ R " U R U R [ SS9U l g )NT)�tools�llm�context�verbose) r r r r r r r r r �7from_toolsr8 �agent)�self�list_of_search_tools�list_of_other_toolss �4C:\Users\13435\Desktop\Agent_GAIA_Benchmark\agent.py�__init__�GaiaAgent.__init__ s_ � � /� 1� � 2� 4�6P�6R�Tg�Ti�kz�k|� }��%�'���)�?��9� �*�*�������Se�os�t��10� �question�returnc � � # � U R R U5 n[ U5 nSnXC; a% UR US5 S R 5 nU$ U$ 7f)Nz
FINAL ANSWER:� )r �chat�str�split�strip)r r �response_object�full_response_text�final_answer_prefix� clean_answers r �__call__�GaiaAgent.__call__ s[ � � � �*�*�/�/�(�3�� � � 1�� -�� � 4�-�3�3�4G��K�A�N�T�T�V�L� � �%� %�s �AA)r r r N)�__name__�11__module__� __qualname__�__firstlineno__r r# r* �__static_attributes__� r r r r s � �u� &�s� &�s� &r r )�sys�inspect�logging�llama_index.core.agentr �agent_models.modelsr �agent_tools.WebSearchToolr � agent_tools.FileDownloaderToolr �agent_tools.ImageReaderToolr �agent_tools.CalculatorToolr �agent_tools.PandasToolr �agent_prompts.SystemPromptr12 � basicConfig�stdout�DEBUG� getLogger�13addHandler�
StreamHandlerr r1 r r �<module>rC sp �� 14� � � -� 2� 6� >� B� :� 2� 9� � � �3�:�:�W�]�]� ;� � � � � � �w� 4� 4�C�J�J� G� H�&� &r