github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2025-12-26 05:48:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add pydantic (#868)

Browse Source 
* Update action.py

* Update observation.py

* Update observation.py

* Update action.py

* Update observation.py

* Update opendevin/schema/action.py

* Update action.py

* Update observation.py

* Update action.py

* Update observation.py

* Update observation.py

* Update observation.py

* Update observation.py

* Update action.py

* Update action.py

* Update action.py

* Update observation.py

* Update observation.py

---------

Co-authored-by: Robert Brennan <accounts@rbren.io>
Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
This commit is contained in:
yaroslavMain 2024-04-21 22:41:04 +03:00 committed by GitHub
parent 900de8728c
commit 454e9613b0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 45 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
opendevin/schema/action.py
View File

@ -1,68 +1,73 @@
from enum import Enum
from pydantic import BaseModel, Field
__all__ = [
'ActionType'
]
class ActionType(str, Enum):
INIT = 'initialize'
class ActionTypeSchema(BaseModel):
INIT: str = Field(default='initialize')
"""Initializes the agent. Only sent by client.
"""
START = 'start'
START: str = Field(default='start')
"""Starts a new development task. Only sent by the client.
"""
READ = 'read'
READ: str = Field(default='read')
"""Reads the content of a file.
"""
WRITE = 'write'
WRITE: str = Field(default='write')
"""Writes the content to a file.
"""
RUN = 'run'
RUN: str = Field(default='run')
"""Runs a command.
"""
KILL = 'kill'
KILL: str = Field(default='kill')
"""Kills a background command.
"""
BROWSE = 'browse'
BROWSE: str = Field(default='browse')
"""Opens a web page.
"""
RECALL = 'recall'
RECALL: str = Field(default='recall')
"""Searches long-term memory
"""
THINK = 'think'
THINK: str = Field(default='think')
"""Allows the agent to make a plan, set a goal, or record thoughts
"""
FINISH = 'finish'
FINISH: str = Field(default='finish')
"""If you're absolutely certain that you've completed your task and have tested your work,
use the finish action to stop working.
"""
CHAT = 'chat'
NULL: str = Field(default='null')
SUMMARIZE = 'summarize'
SUMMARIZE: str = Field(default='summarize')
ADD_TASK = 'add_task'
ADD_TASK: str = Field(default='add_task')
MODIFY_TASK = 'modify_task'
MODIFY_TASK: str = Field(default='modify_task')
NULL = 'null'
PAUSE = 'pause'
PAUSE: str = Field(default='pause')
"""Pauses the task.
"""
RESUME = 'resume'
RESUME: str = Field(default='resume')
"""Resumes the task.
"""
STOP = 'stop'
STOP: str = Field(default='stop')
"""Stops the task. Must send a start action to restart a new task.
"""
CHANGE_TASK_STATE = 'change_task_state'
CHANGE_TASK_STATE: str = Field(default='change_task_state')
ActionType = ActionTypeSchema()

29
opendevin/schema/observation.py
View File

@ -1,31 +1,38 @@
from enum import Enum
from pydantic import BaseModel, Field
__all__ = [
'ObservationType'
]
class ObservationType(str, Enum):
READ = 'read'
class ObservationTypeSchema(BaseModel):
READ: str = Field(default='read')
"""The content of a file
"""
WRITE = 'write'
WRITE: str = Field(default='write')
BROWSE = 'browse'
BROWSE: str = Field(default='browse')
"""The HTML content of a URL
"""
RUN = 'run'
RUN: str = Field(default='run')
"""The output of a command
"""
RECALL = 'recall'
RECALL: str = Field(default='recall')
"""The result of a search
"""
CHAT = 'chat'
CHAT: str = Field(default='chat')
"""A message from the user
"""
MESSAGE = 'message'
MESSAGE: str = Field(default='message')
ERROR = 'error'
ERROR: str = Field(default='error')
NULL = 'null'
NULL: str = Field(default='null')
ObservationType = ObservationTypeSchema()