File size: 962 Bytes
5374a2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from .agent import Agent
from ..actions.agent_generation import AgentGeneration
from ..prompts.agent_generator import AGENT_GENERATOR
class AgentGenerator(Agent):
"""
An agent responsible for generating agents for a task.
"""
def __init__(self, **kwargs):
name = kwargs.pop("name") if "name" in kwargs else AGENT_GENERATOR["name"]
description = kwargs.pop("description") if "description" in kwargs else AGENT_GENERATOR["description"]
system_prompt = kwargs.pop("system_prompt") if "system_prompt" in kwargs else AGENT_GENERATOR["system_prompt"]
actions = kwargs.pop("actions") if "actions" in kwargs else [AgentGeneration(tools=kwargs.pop("tools", []))]
super().__init__(name=name, description=description, system_prompt=system_prompt, actions=actions, **kwargs)
@property
def agent_generation_action_name(self):
return self.get_action_name(action_cls=AgentGeneration)
|