File size: 2,201 Bytes
246d201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
from pydantic import BaseModel, Field
__all__ = ['ActionType']
class ActionTypeSchema(BaseModel):
INIT: str = Field(default='initialize')
"""Initializes the agent. Only sent by client.
"""
MESSAGE: str = Field(default='message')
"""Represents a message.
"""
START: str = Field(default='start')
"""Starts a new development task OR send chat from the user. Only sent by the client.
"""
READ: str = Field(default='read')
"""Reads the content of a file.
"""
WRITE: str = Field(default='write')
"""Writes the content to a file.
"""
EDIT: str = Field(default='edit')
"""Edits a file by providing a draft.
"""
RUN: str = Field(default='run')
"""Runs a command.
"""
RUN_IPYTHON: str = Field(default='run_ipython')
"""Runs a IPython cell.
"""
BROWSE: str = Field(default='browse')
"""Opens a web page.
"""
BROWSE_INTERACTIVE: str = Field(default='browse_interactive')
"""Interact with the browser instance.
"""
DELEGATE: str = Field(default='delegate')
"""Delegates a task to another agent.
"""
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.
"""
REJECT: str = Field(default='reject')
"""If you're absolutely certain that you cannot complete the task with given requirements,
use the reject action to stop working.
"""
NULL: str = Field(default='null')
SUMMARIZE: str = Field(default='summarize')
PAUSE: str = Field(default='pause')
"""Pauses the task.
"""
RESUME: str = Field(default='resume')
"""Resumes the task.
"""
STOP: str = Field(default='stop')
"""Stops the task. Must send a start action to restart a new task.
"""
CHANGE_AGENT_STATE: str = Field(default='change_agent_state')
PUSH: str = Field(default='push')
"""Push a branch to github."""
SEND_PR: str = Field(default='send_pr')
"""Send a PR to github."""
ActionType = ActionTypeSchema()
|