Goodguygregory93/ai-agents-certification-code
0
1�
2 ���h� � �| � S SK Jr S SKJr S SKJr S SKJr S SKJ r S SK3J r S SK r S SK
Jr S SKr " S S45 rg) � )�
AgentWorkflow)�HuggingFaceInferenceAPI)� FunctionTool)�Ollama)�Tools)� RetrieverN)� load_dotenvc �j � \ rS rSr\" 5 S rS
S\S\4S jjrS\S\4S jrS\ \5 4S jr S6 r S r
g )�7BasicAgent� c �� � [ [ R R S- 5 [ R8 R
S5 n[ R9 R
S5 n[ R10 R
S5 nUc [ S5 eUc [ S5 eU R XS9U l U R 5 U l U R 5 U l
X l g )NzBasicAgent initialized.� LLM_MODEL�11SYS_PROMPT�HF_TOKENz3Missing SYS_PROMPT: add SYS_PROMPT inside your .envz/Missing HF_TOKEN: add HF_TOKEN inside your .env)r r )�print�colorama�Fore�GREEN�os�environ�get� Exception�_initialize_llm� agent_llm�_initialize_agent_tools� agent_tools�_initialize_workflow�agent_work_flow�
system_prompt)�selfr r r s �I/Users/gregwitt/Github/clones/ai-agents-certification-code/basic_agent.py�__init__�BasicAgent.__init__ s� � �
�h�m�m�!�!�$=�=�>��J�J�N�N�;�/� ��Z�Z�^�^�L�1�12��:�:�>�>�*�-�� � �� Q�R� R� � �� M�N� N� �-�-� �-�U��� �7�7�9��� #� 8� 8� :�� � '�� �question�returnc � �d # � U( aZ U R SU 3- nU R R U5 I Sh v�N nUR R S R13 nU$ U R R U5 I Sh v�N nUR R S R14 nU$ Nq N+7f)a5 15the __call__ method is the method that is triggered when BasicAgent()16is called the `question` field is expected to be applied to the Agent's 17LlamaIndex Workflow and produce an LLM response 18 19the sys_prompt is defaulted to True and is found here: 20 21[GAIA HuggingFace Paper](https://huggingface.co/spaces/gaia-benchmark/leaderboard)22 23Args:24 question (str): the desired prompt that the user desires information25 sys_prompt (bool): defaults to True as the GAIA specifications prompt is supplied in the study26Returns:27 response (str): response from the LLM 28z QUESTION: Nr )r r �run�response�blocks�text)r r% �29sys_prompt�prompt� agent_outputr) s r! �__call__�BasicAgent.__call__! s� � � � ��'�'�K��z�*B�B�F�!%�!5�!5�!9�!9�&�!A�A�L�#�,�,�3�3�A�6�;�;�H� �� "&�!5�!5�!9�!9�(�!C�C�L�#�,�,�3�3�A�6�;�;�H� �� B� D�s" �8B0�B,�AB0�B.�*B0�.B0r r c �~ � [ [ R R S- 5 [ S5 [ UUSS9nU$ )z�30initialize the llm brain of the Agent using an Ollama installed31model. Returns the Ollama object32 33Args:34 LLM_MODEL (str): provider model name ex: "llama3.2:latest"35 36Returns:37 llm: returns the HuggingFaceInferenceAPI Model38u& 🧠 initializing an LLM for the Agentz---------------�auto)�39model_name�token�provider)r r r �YELLOWr )r r r �llms r! r �BasicAgent._initialize_llm< s? � � �h�m�m�"�"�%M�M�N�
�f�
� &� ����� �40r$ c �T � / n[ 5 R nUR U5 U$ )a 41returns the FunctionalTools from the Tool class calling the42tool_belt and collecting the tools into a list. also loads43the Retriever tool `similar_question_tool` for question querying44 45Args:46 None47Returns:48 found_tools (list[FunctionTool]): the tools initiated from the Tool agent49)r �similar_question_tool�append)r � found_toolsr: s r! r �"BasicAgent._initialize_agent_toolsT s0 � � � � !*� � A� A� ����0�1�50 �r$ c �X � [ R " U R U R S9nU$ )af 51creates the BasicAgent Workflow for the agent that will leverage the52Tools `agent_tools` and the `agent_llm` from the HuggingFaceInterfaceAPI53 54Args: 55 None56Returns:57 basic_agent_workflow (AgentWorkflow): leverages the agent_tools, and agent_llm to create a workflow58 that will create the workflow59)r7 )r �from_tools_or_functionsr r )r �basic_agent_workflows r! r � BasicAgent._initialize_workflowk s- � � -� D� D�� � ��N�N� � �60 $�#r$ )r r r r N)T)�__name__�61__module__� __qualname__�__firstlineno__r r" �strr/ r �listr r r �__static_attributes__� r$ r! r r sH � ��M�(�&�s� �� �6�� �� �0��l�);� �.$r$ r )� llama_index.core.agent.workflowr � llama_index.llms.huggingface_apir �llama_index.core.toolsr �llama_index.llms.ollamar �toolsr � retrieverr r �dotenvr r r rI r$ r! �<module>rQ s, �� 9� D� /� *� � � � � �p$� p$r$