
    ~zg'                     x    d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZ  G d	 d
      Zy)   )SummarizeTool)WriteArticleTool)SanitizeDataTool)SummarizeValidatorAgent)WriteArticleValidatorAgent)SanitizeDataValidatorAgent)RefinerAgent)ValidatorAgentc                       e Zd ZddZd Zy)AgentManagerc                     t        ||      t        ||      t        ||      t        ||      t	        ||      t        ||      t        ||      t        ||      d| _        y )N)max_retriesverbose)	summarizewrite_articlesanitize_datasummarize_validatorwrite_article_validatorsanitize_data_validatorrefiner	validator)	r   r   r   r   r   r   r	   r
   agents)selfr   r   s      rC:\Users\bruma\Desktop\finale\rulare ollama remote cu colab\AI-Agents-from-Scratch-using-Ollama\agents\__init__.py__init__zAgentManager.__init__   s^    &;P-+wW-+wW#:{\c#d'Akcj'k'Akcj'k#WM'KQ	
    c                 ^    | j                   j                  |      }|st        d| d      |S )NzAgent 'z' not found.)r   get
ValueError)r   
agent_nameagents      r   	get_agentzAgentManager.get_agent   s0    
+wzl,?@@r   N)   T)__name__
__module____qualname__r   r"    r   r   r   r      s    

r   r   N)summarize_toolr   write_article_toolr   sanitize_data_toolr   summarize_validator_agentr   write_article_validator_agentr   sanitize_data_validator_agentr   refiner_agentr	   validator_agentr
   r   r'   r   r   <module>r0      s)    * 0 0 > E E ' + r   