__init__.py 633 B

123456789101112131415161718192021222324252627282930313233343536
  1. __version__ = "0.2.0"
  2. from pathlib import Path
  3. from sweagent.agent.agents import (
  4. Agent,
  5. AgentArguments,
  6. )
  7. from sweagent.agent.models import (
  8. ModelArguments,
  9. )
  10. from sweagent.environment.swe_env import (
  11. EnvironmentArguments,
  12. SWEEnv,
  13. )
  14. from sweagent.environment.utils import (
  15. get_data_path_name,
  16. )
  17. PACKAGE_DIR = Path(__file__).resolve().parent
  18. assert PACKAGE_DIR.is_dir()
  19. CONFIG_DIR = PACKAGE_DIR.parent / "config"
  20. assert CONFIG_DIR.is_dir()
  21. __all__ = [
  22. "Agent",
  23. "AgentArguments",
  24. "ModelArguments",
  25. "EnvironmentArguments",
  26. "SWEEnv",
  27. "get_data_path_name",
  28. "PACKAGE_DIR",
  29. ]