agent.py 985 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from enum import Enum
  2. class AgentState(str, Enum):
  3. LOADING = 'loading'
  4. """The agent is loading.
  5. """
  6. INIT = 'init'
  7. """The agent is initialized.
  8. """
  9. RUNNING = 'running'
  10. """The agent is running.
  11. """
  12. AWAITING_USER_INPUT = 'awaiting_user_input'
  13. """The agent is awaiting user input.
  14. """
  15. PAUSED = 'paused'
  16. """The agent is paused.
  17. """
  18. STOPPED = 'stopped'
  19. """The agent is stopped.
  20. """
  21. FINISHED = 'finished'
  22. """The agent is finished with the current task.
  23. """
  24. REJECTED = 'rejected'
  25. """The agent rejects the task.
  26. """
  27. ERROR = 'error'
  28. """An error occurred during the task.
  29. """
  30. AWAITING_USER_CONFIRMATION = 'awaiting_user_confirmation'
  31. """The agent is awaiting user confirmation.
  32. """
  33. USER_CONFIRMED = 'user_confirmed'
  34. """The user confirmed the agent's action.
  35. """
  36. USER_REJECTED = 'user_rejected'
  37. """The user rejected the agent's action.
  38. """