|
|
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) |
|
|
|
|
|
|
