Ë
    µŒj9  ã                   ó”   — d dl mZ d dlmZmZmZmZ d dlmZm	Z	 d dl
mZ d dlmZ d dlmZ defd„Z G d	„ d
«      Z G d„ dee«      Zy)é    )Údeepcopy)ÚAnyÚDictÚListÚOptional)ÚAgentActionÚAgentFinish)ÚBaseCallbackHandler)Ú	LLMResult©Úguard_importÚreturnc                  ó   — t        d«      S )zHImport the aim python package and raise an error if it is not installed.Úaimr   © ó    út/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/callbacks/aim_callback.pyÚ
import_aimr   
   s   € ä˜ÓÐr   c                   ó¤   — e Zd ZdZdd„Zedefd„«       Zedefd„«       Zedefd„«       Z	edefd„«       Z
edefd	„«       Zdeeef   fd
„Zdd„Zy)ÚBaseMetadataCallbackHandleraš  Callback handler for the metadata and associated function states for callbacks.

    Attributes:
        step (int): The current step.
        starts (int): The number of times the start method has been called.
        ends (int): The number of times the end method has been called.
        errors (int): The number of times the error method has been called.
        text_ctr (int): The number of times the text method has been called.
        ignore_llm_ (bool): Whether to ignore llm callbacks.
        ignore_chain_ (bool): Whether to ignore chain callbacks.
        ignore_agent_ (bool): Whether to ignore agent callbacks.
        ignore_retriever_ (bool): Whether to ignore retriever callbacks.
        always_verbose_ (bool): Whether to always be verbose.
        chain_starts (int): The number of times the chain start method has been called.
        chain_ends (int): The number of times the chain end method has been called.
        llm_starts (int): The number of times the llm start method has been called.
        llm_ends (int): The number of times the llm end method has been called.
        llm_streams (int): The number of times the text method has been called.
        tool_starts (int): The number of times the tool start method has been called.
        tool_ends (int): The number of times the tool end method has been called.
        agent_ends (int): The number of times the agent end method has been called.
    r   Nc                 ó   — d| _         d| _        d| _        d| _        d| _        d| _        d| _        d| _        d| _        d| _	        d| _
        d| _        d| _        d| _        d| _        d| _        d| _        d| _        y )Nr   F)ÚstepÚstartsÚendsÚerrorsÚtext_ctrÚignore_llm_Úignore_chain_Úignore_agent_Úignore_retriever_Úalways_verbose_Úchain_startsÚ
chain_endsÚ
llm_startsÚllm_endsÚllm_streamsÚtool_startsÚ	tool_endsÚ
agent_ends©Úselfs    r   Ú__init__z$BaseMetadataCallbackHandler.__init__'   sˆ   € ØˆŒ	àˆŒØˆŒ	ØˆŒØˆŒà ˆÔØ"ˆÔØ"ˆÔØ!&ˆÔØ$ˆÔàˆÔØˆŒàˆŒØˆŒØˆÔàˆÔØˆŒàˆ�r   c                 ó   — | j                   S )z;Whether to call verbose callbacks even if verbose is False.)r!   r*   s    r   Úalways_verbosez*BaseMetadataCallbackHandler.always_verboseA   s   € ð ×#Ñ#Ð#r   c                 ó   — | j                   S )z Whether to ignore LLM callbacks.)r   r*   s    r   Ú
ignore_llmz&BaseMetadataCallbackHandler.ignore_llmF   s   € ð ×ÑÐr   c                 ó   — | j                   S )z"Whether to ignore chain callbacks.)r   r*   s    r   Úignore_chainz(BaseMetadataCallbackHandler.ignore_chainK   ó   € ð ×!Ñ!Ð!r   c                 ó   — | j                   S )z"Whether to ignore agent callbacks.)r   r*   s    r   Úignore_agentz(BaseMetadataCallbackHandler.ignore_agentP   r3   r   c                 ó   — | j                   S )z&Whether to ignore retriever callbacks.)r    r*   s    r   Úignore_retrieverz,BaseMetadataCallbackHandler.ignore_retrieverU   s   € ð ×%Ñ%Ð%r   c                 ó&  — | j                   | j                  | j                  | j                  | j                  | j
                  | j                  | j                  | j                  | j                  | j                  | j                  | j                  dœS )N©r   r   r   r   r   r"   r#   r$   r%   r&   r'   r(   r)   r9   r*   s    r   Úget_custom_callback_metaz4BaseMetadataCallbackHandler.get_custom_callback_metaZ   sm   € à—I‘IØ—k‘kØ—I‘IØ—k‘kØŸ™Ø ×-Ñ-ØŸ/™/ØŸ/™/ØŸ™Ø×+Ñ+Ø×+Ñ+ØŸ™ØŸ/™/ñ
ð 	
r   c                 óò   — d| _         d| _        d| _        d| _        d| _        d| _        d| _        d| _        d| _        d| _	        d| _
        d| _        d| _        d| _        d| _        d| _        d| _        y)zReset the callback metadata.r   FN)r   r   r   r   r   r   r   r   r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   s    r   Úreset_callback_metaz/BaseMetadataCallbackHandler.reset_callback_metak   sƒ   € àˆŒ	àˆŒØˆŒ	ØˆŒØˆŒà ˆÔØ"ˆÔØ"ˆÔØ$ˆÔàˆÔØˆŒàˆŒØˆŒØˆÔàˆÔØˆŒàˆŒàr   )r   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r,   ÚpropertyÚboolr.   r0   r2   r5   r7   r   Ústrr   r:   r<   r   r   r   r   r      s¬   „ ñó.ð4 ð$ ò $ó ð$ð ð ˜Dò  ó ð ð ð"˜dò "ó ð"ð ð"˜dò "ó ð"ð ð& $ò &ó ð&ð
¨$¨s°C¨x©.ó 
ô"r   r   c                   óö  ‡ — e Zd ZdZ	 	 	 	 d'dee   dee   dee   deddf
ˆ fd„Zd	e	ddfd
„Z
deee	f   dee   d	e	ddfd„Zded	e	ddfd„Zded	e	ddfd„Zded	e	ddfd„Zdeee	f   deee	f   d	e	ddfd„Zdeee	f   d	e	ddfd„Zded	e	ddfd„Zdeee	f   ded	e	ddfd„Zde	d	e	ddfd„Zded	e	ddfd„Zded	e	ddfd„Zd ed	e	ddfd!„Zd"ed	e	de	fd#„Z	 	 	 	 	 	 	 d(dee   dee   dee   ded$e	d%ed eddfd&„Zˆ xZS ))ÚAimCallbackHandlera¿  Callback Handler that logs to Aim.

    Parameters:
        repo (:obj:`str`, optional): Aim repository path or Repo object to which
            Run object is bound. If skipped, default Repo is used.
        experiment_name (:obj:`str`, optional): Sets Run's `experiment` property.
            'default' if not specified. Can be used later to query runs/sequences.
        system_tracking_interval (:obj:`int`, optional): Sets the tracking interval
            in seconds for system usage metrics (CPU, Memory, etc.). Set to `None`
             to disable system metrics tracking.
        log_system_params (:obj:`bool`, optional): Enable/Disable logging of system
            params such as installed packages, git info, environment variables, etc.

    This handler will utilize the associated callback method called and formats
    the input of each callback function with metadata regarding the state of LLM run
    and then logs the response to Aim.
    NÚrepoÚexperiment_nameÚsystem_tracking_intervalÚlog_system_paramsr   c                 ó6  •— t         ‰| �  «        t        «       }|| _        || _        || _        || _        |j                  | j                  | j                  | j
                  | j                  ¬«      | _        | j                  j                  | _
        g | _        y)zInitialize callback handler.©rF   Ú
experimentrH   rI   N)Úsuperr,   r   rF   rG   rH   rI   ÚRunÚ_runÚhashÚ	_run_hashÚaction_records)r+   rF   rG   rH   rI   r   Ú	__class__s         €r   r,   zAimCallbackHandler.__init__›   s†   ø€ ô 	‰ÑÔä‹lˆØˆŒ	Ø.ˆÔØ(@ˆÔ%Ø!2ˆÔØ—G‘GØ—‘Ø×+Ñ+Ø%)×%BÑ%BØ"×4Ñ4ð	 ó 
ˆŒ	ð Ÿ™Ÿ™ˆŒØ$&ˆÕr   Úkwargsc                 óä  — t        «       }| j                  s¡| j                  r8|j                  | j                  | j                  | j
                  ¬«      | _        n]|j                  | j                  | j                  | j
                  | j                  ¬«      | _        | j                  j                  | _        |r7|j                  «       D ]#  \  }}| j                  j                  ||d¬«       Œ% y y )N)rF   rH   rK   F©Ústrict)r   rO   rQ   rN   rF   rH   rG   rI   rP   ÚitemsÚset)r+   rT   r   ÚkeyÚvalues        r   ÚsetupzAimCallbackHandler.setup´   s¿   € Ü‹lˆà�yŠyØ�~Š~ØŸG™GØ—N‘NØŸ™Ø-1×-JÑ-Jð $ó �•	ð  ŸG™GØŸ™Ø#×3Ñ3Ø-1×-JÑ-JØ&*×&<Ñ&<ð	 $ó �”	ð "&§¡§¡�”áØ$Ÿl™lžn‘
��UØ—	‘	—‘˜c 5°�Õ7ñ -ð r   Ú
serializedÚpromptsc                 ór  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j	                  | j                  «       «       t        |«      }| j                  j                  |D �cg c]  }|j                  |«      ‘Œ c}d|¬«       yc c}w )zRun when LLM starts.é   ÚactionÚon_llm_start©ÚnameÚcontextN)
r   r   r$   r   Úupdater:   r   rO   ÚtrackÚText)r+   r]   r^   rT   r   ÚrespÚprompts_resÚprompts           r   rb   zAimCallbackHandler.on_llm_startË   s–   € ô ‹lˆà�	Š	�Q‰�	Ø�Š˜1Ñ�Ø�Š�qÑ�à˜.Ð)ˆØ�‰�D×1Ñ1Ó3Ô4ä˜wÓ'ˆà�	‰	�‰Ù,7Ó8©K &ˆS�X‰X�fÕ¨KÑ8ØØð 	õ 	
ùÚ8s   ÂB4Úresponsec                 ó°  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j	                  | j                  «       «       t        |«      }|j                  D ��cg c]$  }|D ]  }|j                  |j                  «      ‘Œ Œ& }}}| j                  j                  |d|¬«       yc c}}w )zRun when LLM ends running.r`   ra   Ú
on_llm_endrc   N)r   r   r%   r   rf   r:   r   Úgenerationsrh   ÚtextrO   rg   )	r+   rl   rT   r   ri   Úresponse_resro   Ú
generationÚ	generateds	            r   rn   zAimCallbackHandler.on_llm_endà   sÁ   € ä‹lˆØ�	Š	�Q‰�	Ø�Š˜Ñ�Ø�	Š	�Q‰�	à˜,Ð'ˆØ�‰�D×1Ñ1Ó3Ô4ä Ó)ˆð  ,×7Ò7ô
á7�Û)�
ð �H‰H�Z—_‘_Õ%à)ð &Ø7ð 	ñ 
ð
 	�	‰	�‰ØØØð 	õ 	
ùó
s   Â)CÚtokenc                 óX   — | xj                   dz  c_         | xj                  dz  c_        y)z#Run when LLM generates a new token.r`   N)r   r&   )r+   rt   rT   s      r   Úon_llm_new_tokenz#AimCallbackHandler.on_llm_new_token÷   s    € à�	Š	�Q‰�	Ø×Ò˜AÑÖr   Úerrorc                 óX   — | xj                   dz  c_         | xj                  dz  c_        y)zRun when LLM errors.r`   N©r   r   ©r+   rw   rT   s      r   Úon_llm_errorzAimCallbackHandler.on_llm_errorü   ó   € à�	Š	�Q‰�	Ø�Š�qÑŽr   Úinputsc                 óR  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j	                  | j                  «       «       t        |«      }| j                  j                  |j                  |d   «      d|¬«       y)zRun when chain starts running.r`   ra   Úon_chain_startÚinputrc   N)
r   r   r"   r   rf   r:   r   rO   rg   rh   )r+   r]   r}   rT   r   ri   Ú
inputs_ress          r   r   z!AimCallbackHandler.on_chain_start  sŒ   € ô ‹lˆØ�	Š	�Q‰�	Ø×Ò˜QÑÕØ�Š�qÑ�àÐ*Ð+ˆØ�‰�D×1Ñ1Ó3Ô4ä˜fÓ%ˆ
à�	‰	�‰Ø�H‰H�Z Ñ(Ó)Ð0@È$ð 	õ 	
r   Úoutputsc                 óR  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j	                  | j                  «       «       t        |«      }| j                  j                  |j                  |d   «      d|¬«       y)zRun when chain ends running.r`   ra   Úon_chain_endÚoutputrc   N)
r   r   r#   r   rf   r:   r   rO   rg   rh   )r+   r‚   rT   r   ri   Úoutputs_ress         r   r„   zAimCallbackHandler.on_chain_end  s„   € ä‹lˆØ�	Š	�Q‰�	Ø�Š˜1Ñ�Ø�	Š	�Q‰�	à˜.Ð)ˆØ�‰�D×1Ñ1Ó3Ô4ä˜wÓ'ˆà�	‰	�‰Ø�H‰H�[ Ñ*Ó+°.È$ð 	õ 	
r   c                 óX   — | xj                   dz  c_         | xj                  dz  c_        y)zRun when chain errors.r`   Nry   rz   s      r   Úon_chain_errorz!AimCallbackHandler.on_chain_error#  r|   r   Ú	input_strc                 ó6  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j	                  | j                  «       «       | j                  j                  |j                  |«      d|¬«       y)zRun when tool starts running.r`   ra   Úon_tool_startrc   N)	r   r   r'   r   rf   r:   rO   rg   rh   )r+   r]   r‰   rT   r   ri   s         r   r‹   z AimCallbackHandler.on_tool_start(  su   € ô ‹lˆØ�	Š	�Q‰�	Ø×Ò˜AÑÕØ�Š�qÑ�à˜/Ð*ˆØ�‰�D×1Ñ1Ó3Ô4à�	‰	�‰˜Ÿ™ Ó+°/È4ˆÕPr   r…   c                 óL  — t        |«      }t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j                  | j                  «       «       | j                  j                  |j                  |«      d|¬«       y)zRun when tool ends running.r`   ra   Úon_tool_endrc   N)
rC   r   r   r(   r   rf   r:   rO   rg   rh   )r+   r…   rT   r   ri   s        r   r�   zAimCallbackHandler.on_tool_end6  sx   € ä�V“ˆÜ‹lˆØ�	Š	�Q‰�	Ø�Š˜!Ñ�Ø�	Š	�Q‰�	à˜-Ð(ˆØ�‰�D×1Ñ1Ó3Ô4à�	‰	�‰˜Ÿ™ Ó(¨}ÀdˆÕKr   c                 óX   — | xj                   dz  c_         | xj                  dz  c_        y)zRun when tool errors.r`   Nry   rz   s      r   Úon_tool_errorz AimCallbackHandler.on_tool_errorC  r|   r   rp   c                 óX   — | xj                   dz  c_         | xj                  dz  c_        y)z+
        Run when agent is ending.
        r`   N)r   r   )r+   rp   rT   s      r   Úon_textzAimCallbackHandler.on_textH  s   € ð 	�	Š	�Q‰�	Ø�Š˜ÑŽr   Úfinishc                 óž  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        ddi}|j	                  | j                  «       «       t        |«      }dj                  |j                  d   |j                  «      }| j                  j                  |j                  |«      d|¬«       y)zRun when agent ends running.r`   ra   Úon_agent_finishzOUTPUT:
{}

LOG:
{}r…   rc   N)r   r   r)   r   rf   r:   r   ÚformatÚreturn_valuesÚlogrO   rg   rh   )r+   r’   rT   r   ri   Ú
finish_resrp   s          r   r”   z"AimCallbackHandler.on_agent_finishO  s    € ä‹lˆØ�	Š	�Q‰�	Ø�Š˜1Ñ�Ø�	Š	�Q‰�	àÐ+Ð,ˆØ�‰�D×1Ñ1Ó3Ô4ä˜fÓ%ˆ
à(×/Ñ/Ø×$Ñ$ XÑ.°
·±ó
ˆð 	�	‰	�‰˜Ÿ™ ›Ð->ÈˆÕMr   ra   c                 ó®  — t        «       }| xj                  dz  c_        | xj                  dz  c_        | xj                  dz  c_        d|j                  dœ}|j                  | j                  «       «       t        |«      }dj                  |j                  |j                  «      }| j                  j                  |j                  |«      d|¬«       y)zRun on agent action.r`   Úon_agent_action)ra   ÚtoolzTOOL INPUT:
{}

LOG:
{}rc   N)r   r   r'   r   r›   rf   r:   r   r•   Ú
tool_inputr—   rO   rg   rh   )r+   ra   rT   r   ri   Ú
action_resrp   s          r   rš   z"AimCallbackHandler.on_agent_action`  s©   € ä‹lˆØ�	Š	�Q‰�	Ø×Ò˜AÑÕØ�Š�qÑ�ð (Ø—K‘Kñ
ˆð 	�‰�D×1Ñ1Ó3Ô4ä˜fÓ%ˆ
à,×3Ñ3Ø×!Ñ! :§>¡>ó
ˆð 	�	‰	�‰˜Ÿ™ ›Ð->ÈˆÕMr   Úlangchain_assetÚresetc                 ó˜  — |rF	 |j                  «       j                  «       D ]#  \  }}	| j                  j                  ||	d¬«       Œ% 	 |s|r*| j                  j                  «        | j                  «        |rÃt        «       }
|r|n| j                  | _        |r|n| j                  | _	        |r|n| j                  | _
        |r|n| j                  | _        |
j                  | j                  | j                  | j                  | j                  ¬«      | _        | j                  j                  | _        g | _        yy# t        $ r Y Œÿw xY w)aã  Flush the tracker and reset the session.

        Args:
            repo (:obj:`str`, optional): Aim repository path or Repo object to which
                Run object is bound. If skipped, default Repo is used.
            experiment_name (:obj:`str`, optional): Sets Run's `experiment` property.
                'default' if not specified. Can be used later to query runs/sequences.
            system_tracking_interval (:obj:`int`, optional): Sets the tracking interval
                in seconds for system usage metrics (CPU, Memory, etc.). Set to `None`
                 to disable system metrics tracking.
            log_system_params (:obj:`bool`, optional): Enable/Disable logging of system
                params such as installed packages, git info, environment variables, etc.
            langchain_asset: The langchain asset to save.
            reset: Whether to reset the session.
            finish: Whether to finish the run.

            Returns:
                None
        FrV   rK   N)ÚdictrX   rO   rY   Ú	ExceptionÚcloser<   r   rF   rG   rH   rI   rN   rP   rQ   rR   )r+   rF   rG   rH   rI   rž   rŸ   r’   rZ   r[   r   s              r   Úflush_trackerz AimCallbackHandler.flush_trackert  s0  € ñ< ðØ"1×"6Ñ"6Ó"8×">Ñ">Ö"@‘J�C˜Ø—I‘I—M‘M # u°U�MÕ;ñ #Añ
 ‘UØ�I‰I�O‰OÔØ×$Ñ$Ô&ÙÜ“,ˆCÙ $™¨$¯)©)ˆDŒIá#2‘¸×8LÑ8Lð Ô ñ
 ,ñ )à×2Ñ2ð Ô)ñ &7Ñ!¸D×<RÑ<Rð Ô"ð Ÿ™Ø—Y‘YØ×/Ñ/Ø)-×)FÑ)FØ"&×"8Ñ"8ð	  ó ˆDŒIð "ŸY™YŸ^™^ˆDŒNØ"$ˆDÕð- øô ò Ùðús   „AD= Ä=	E	ÅE	)NNé
   T)NNr¥   TNTF) r=   r>   r?   r@   r   rC   ÚintrB   r,   r   r\   r   r   rb   r   rn   rv   ÚBaseExceptionr{   r   r„   rˆ   r‹   r�   r�   r‘   r	   r”   r   rš   r¤   Ú__classcell__)rS   s   @r   rE   rE   ˆ   s�  ø„ ñð( #Ø)-Ø24Ø"&ñ'à�s‰mð'ð " #™ð'ð #+¨3¡-ð	'ð
  ð'ð 
õ'ð28˜cð 8 dó 8ð.
Ø˜s C˜x™.ð
Ø37¸±9ð
ØHKð
à	ó
ð*
 9ð 
¸ð 
Àó 
ð. cð °Sð ¸Tó ð
 -ð ¸3ð À4ó ð

Ø˜s C˜x™.ð
Ø26°s¸C°x±.ð
ØLOð
à	ó
ð$
 D¨¨c¨¡Nð 
¸cð 
Àdó 
ð  Mð ¸Sð ÀTó ð
QØ˜s C˜x™.ðQØ58ðQØDGðQà	óQðL #ð L°ð L¸ó Lð =ð ¸Cð ÀDó ð
˜Cð ¨3ð °4ó ðN kð N¸Sð NÀTó Nð"N kð N¸Sð NÀSó Nð, #Ø)-Ø24Ø"&Ø#ØØñ>%à�s‰mð>%ð " #™ð>%ð #+¨3¡-ð	>%ð
  ð>%ð ð>%ð ð>%ð ð>%ð 
÷>%r   rE   N)Úcopyr   Útypingr   r   r   r   Úlangchain_core.agentsr   r	   Úlangchain_core.callbacksr
   Úlangchain_core.outputsr   Úlangchain_core.utilsr   r   r   rE   r   r   r   Ú<module>r¯      sE   ðÝ ß ,Ó ,ç :Ý 8Ý ,Ý -ð�Có ÷
vñ vôrj%Ð4Ð6Iõ j%r   