File size: 736 Bytes
2f85c93
8157183
 
 
 
 
 
 
 
 
 
 
2ea8556
 
 
8157183
 
 
fcb1a95
 
 
 
8157183
 
 
fcb1a95
8157183
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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,
        }