hashiruAI / src /tools /default_tools /get_agents_tool.py
helloparthshah's picture
Some more refactor
0e58feb
raw
history blame contribute delete
736 Bytes
from src.manager.agent_manager import AgentManager
__all__ = ['GetAgents']
class GetAgents():
dependencies = []
inputSchema = {
"name": "GetAgents",
"description": "Retrieves a list of available AI agents. This tool is used to get the list of available models that can be invoked using the AskAgent tool.",
"parameters": {
"type": "object",
"properties": {},
"required": [],
},
}
def run(self, **kwargs):
agent_manger = AgentManager()
agents = agent_manger.list_agents()
return {
"status": "success",
"message": "Agents list retrieved successfully",
"agents": agents,
}