__init__.py 791 B

12345678910111213141516171819202122232425262728
  1. from .micro.registry import all_microagents
  2. from .micro.agent import MicroAgent
  3. from opendevin.agent import Agent
  4. from dotenv import load_dotenv
  5. load_dotenv()
  6. # Import agents after environment variables are loaded
  7. from . import monologue_agent # noqa: E402
  8. from . import codeact_agent # noqa: E402
  9. from . import planner_agent # noqa: E402
  10. from . import SWE_agent # noqa: E402
  11. from . import delegator_agent # noqa: E402
  12. __all__ = ['monologue_agent', 'codeact_agent',
  13. 'planner_agent', 'SWE_agent', 'delegator_agent']
  14. for agent in all_microagents.values():
  15. name = agent['name']
  16. prompt = agent['prompt']
  17. anon_class = type(name, (MicroAgent,), {
  18. 'prompt': prompt,
  19. 'agent_definition': agent,
  20. })
  21. Agent.register(name, anon_class)