Tribh/devops-copilot
0
1�
2 њi� � � � d dl mZmZmZmZmZmZ d dlmZm Z m3Z4m Z m Z d dl
Z
d dlmZ G d� de� � Z G d� d� � Z e� � ZdS ) � )�Any�Callable�Dict�List�Optional�Type)� BaseModel�5ConfigDict�Field� create_model�
validate_callN)�loggerc �l � e Zd ZU dZeed< eed< ee ed< eed< e d�� � Z6de fd �Z d7S ) �Toolz&Represents a tool available to agents.�name� description�parameters_schema�funcT)�arbitrary_types_allowed�returnc � � t j d| j � d|� �� � | j di |��} | j di |� � � ��S )z�8 Executes the tool with validation.9 IMPORTANT: All arguments are validated against parameters_schema before execution.10 zExecuting tool: z with params: � )r �infor r r �11model_dump)�self�kwargs�validated_paramss �S/Users/tribhangind/Documents/GitHub/devops-copilot/devops_copilot/tools/registry.py�executez Tool.execute se � �12 � �H�t�y�H�H��H�H�I�I�I�1�4�1�;�;�F�;�;���t�y�9�9�+�6�6�8�8�9�9�9� N)
�__name__�13__module__� __qualname__�__doc__�str�__annotations__r r r r14 � model_configr r r r r r r sw � � � � � � �0�0�15
�I�I�I������I��&�&�&�16�N�N�N� �:�d�;�;�;�L�:�3� :� :� :� :� :� :r r c �l � e Zd ZdZd� Zdedefd�Zdedee fd�Z de17e ee f fd�Z
d S )18� ToolRegistryz+Registry for managing and validating tools.c � � i | _ d S �N)�_tools�r s r �__init__zToolRegistry.__init__ s
� �')�� � � r r r c �* � ��� dt f��� fd� }|S )z+Decorator to register a function as a tool.r c � �� t j | � � }i }|j � � � D ]Y\ }}|j t j j k r
t df||< �,|j |j t j j k rdn|j f||< �Zt | j19 � d�fi |��}t ��|| �� � }|� j �< t j d�� �� � | S )N.�_Schema)r r r r zRegistered tool: )�inspect� signature�20parameters�items�21annotation� Parameter�emptyr �defaultr r! r r, r r )22r �sig�fields�23param_name�param�pydantic_model�toolr r r s24 ���r � decoratorz(ToolRegistry.register.<locals>.decorator s� �� ��#�D�)�)�C��F�%(�^�%9�%9�%;�%;�
@�
@�!�25�E��#�w�'8�'>�>�>�*-�s��F�:�&�&�*/�*:�5�=�T[�Te�Tk�Ck�Ck�C�C�qv�q~�)�F�:�&�&� )�T�]�*C�*C�*C� N� N�v� N� N�N��� '�"0�� � � �D� !%�D�K�� � �K�2�D�2�2� 3� 3� 3��Kr )r )r r r r@ s ``` r �registerzToolRegistry.register s< ���� � �H� � � � � � � � �* �r r c �6 � | j � |� � S r+ )r, �get)r r s r �get_toolzToolRegistry.get_tool7 s � ��{���t�$�$�$r c �H � d� | j � � � D � � S )Nc �\ � g | ])}|j |j |j � � � d ���*S ))r r r4 )r r r �model_json_schema)�.0r? s r �26<listcomp>z+ToolRegistry.list_tools.<locals>.<listcomp>; sN � � 27� 28� 29� � � � #� /� "� 4� F� F� H� H�
�
�30� 31� 32r )r, �valuesr- s r �33list_toolszToolRegistry.list_tools: s3 � �34� 35� � �*�*�,�,�
36� 37� 38� 39r N)r! r" r# r$ r. r% rA r r rD r r r rK r r r r) r) s� � � � � � �5�5�*� *� *��S� �s� � � � �2%�S� %�X�d�^� %� %� %� %�40�D��c�3�h� � 0� 41� 42� 43� 44� 45� 46r r) )�typingr r r r r r �pydanticr r47 r r r
r2 �devops_copilot.utils.loggerr r r) �registryr r r �<module>rP s� �� <� <� <� <� <� <� <� <� <� <� <� <� <� <� <� <� N� N� N� N� N� N� N� N� N� N� N� N� N� N� ���� .� .� .� .� .� .�:� :� :� :� :�9� :� :� :�$*48� *49� *50� *51� *52� *53� *54� *55�Z �<�>�>���r