observation.py 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. from pydantic import BaseModel, Field
  2. __all__ = ['ObservationType']
  3. class ObservationTypeSchema(BaseModel):
  4. READ: str = Field(default='read')
  5. """The content of a file
  6. """
  7. WRITE: str = Field(default='write')
  8. EDIT: str = Field(default='edit')
  9. BROWSE: str = Field(default='browse')
  10. """The HTML content of a URL
  11. """
  12. RUN: str = Field(default='run')
  13. """The output of a command
  14. """
  15. RUN_IPYTHON: str = Field(default='run_ipython')
  16. """Runs a IPython cell.
  17. """
  18. CHAT: str = Field(default='chat')
  19. """A message from the user
  20. """
  21. DELEGATE: str = Field(default='delegate')
  22. """The result of a task delegated to another agent
  23. """
  24. MESSAGE: str = Field(default='message')
  25. ERROR: str = Field(default='error')
  26. SUCCESS: str = Field(default='success')
  27. NULL: str = Field(default='null')
  28. AGENT_STATE_CHANGED: str = Field(default='agent_state_changed')
  29. USER_REJECTED: str = Field(default='user_rejected')
  30. ObservationType = ObservationTypeSchema()