
    ~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 )SanitizeDataToolc                 *    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_tool.pyr   zSanitizeDataTool.__init__   s    0kSZ[    c                 H    ddddd| ddg}| j                  |d      }|S )	NsystemzcYou are an AI assistant that sanitizes medical data by removing Protected Health Information (PHI).)rolecontentuserz)Remove all PHI from the following data:

z

Sanitized Data:i  )
max_tokens)
call_llama)r   medical_datamessagessanitized_datas       r   executezSanitizeDataTool.execute	   sJ      +P  QA#n$79	
 cBr   )   T)__name__
__module____qualname__r   r   __classcell__)r   s   @r   r   r      s    \r   r   N)
agent_baser   r    r   r   <module>r"      s    "y r   