
    ~zg                     &    d dl mZ  G d de      Zy)   )	AgentBasec                   &     e Zd Zd fd	Zd Z xZS )SanitizeDataValidatorAgentc                 *    t         |   d||       y )Nr   )namemax_retriesverbose)super__init__)selfr   r	   	__class__s      C:\Users\bruma\Desktop\finale\rulare ollama remote cu colab\AI-Agents-from-Scratch-using-Ollama\agents\sanitize_data_validator_agent.pyr   z#SanitizeDataValidatorAgent.__init__   s    :]de    c                 V    d}d| d| d}d|dd|dg}| j                  |d	      }|S )
NzYou are an AI assistant that validates the sanitization of medical data by checking for the removal of Protected Health Information (PHI).zGiven the original data and the sanitized data, verify that all PHI has been removed.
List any remaining PHI in the sanitized data and rate the sanitization process on a scale of 1 to 5, where 5 indicates complete sanitization.

Original Data:
z

Sanitized Data:
z

Validation:system)rolecontentuseri   )
max_tokens)
call_llama)r   original_datasanitized_datasystem_messageuser_contentmessages
validations          r   executez"SanitizeDataValidatorAgent.execute	   sa     f,o .  ./ 0 	 .95
 __X#_>
r   )   T)__name__
__module____qualname__r   r   __classcell__)r   s   @r   r   r      s    fr   r   N)
agent_baser   r    r   r   <module>r%      s    " r   