Format: application/x-python-code
1 o
2
3 ��hw � @ sD d dl mZmZmZ ddlmZ ddlmZ G dd� d�Ze� ZdS )� )�Callable�Dict�Any� )�available_functions)�tool_definitionsc @ sD e Zd Zdd� Zdd� Zdededdfd d
4 �Zdedefd d �ZdS )
5 � ToolRegistryc C s i | _ | �� d S )N)�tools�_register_core_tools)�self� r �@C:\Users\shubh\Videos\agent\agent\modules\tools\tool_registry.py�__init__ s zToolRegistry.__init__c C s, t D ]}t�|d �}| �|d |� qd S )N�name)r r �get�register)r Ztool�funcr r r
6 r
7
8 s �z!ToolRegistry._register_core_toolsr �method�returnNc C s( || j v r
9 td|� d���|| j |< d S )N�Tool z already registered)r �
10 ValueError)r r r r r r
11 r s
12 zToolRegistry.registerc C s$ || j vr
13 td|� d���| j | S )Nr z not found in registry)r �KeyError)r r r r r
14 �
15 get_method s
16
17 zToolRegistry.get_method) �__name__�
18 __module__� __qualname__r r
19 �strr r r r r r r
20 r s
21 r N) �typingr r r �
22 tool_executorr r r �
23 tool_registryr r r r
24 �<module> s
25
26