observation.py 1004 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  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. BROWSE: str = Field(default='browse')
  9. """The HTML content of a URL
  10. """
  11. RUN: str = Field(default='run')
  12. """The output of a command
  13. """
  14. RUN_IPYTHON: str = Field(default='run_ipython')
  15. """Runs a IPython cell.
  16. """
  17. CHAT: str = Field(default='chat')
  18. """A message from the user
  19. """
  20. DELEGATE: str = Field(default='delegate')
  21. """The result of a task delegated to another agent
  22. """
  23. MESSAGE: str = Field(default='message')
  24. ERROR: str = Field(default='error')
  25. SUCCESS: str = Field(default='success')
  26. NULL: str = Field(default='null')
  27. AGENT_STATE_CHANGED: str = Field(default='agent_state_changed')
  28. USER_REJECTED: str = Field(default='user_rejected')
  29. ObservationType = ObservationTypeSchema()