You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
3.5 KiB
123 lines
3.5 KiB
from __future__ import annotations
|
|
|
|
from enum import Enum
|
|
from typing import Any, List, Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class TaskInput(BaseModel):
|
|
__root__: Any = Field(
|
|
...,
|
|
description="The input parameters for the task. Any value is allowed.",
|
|
example='{\n"debug": false,\n"mode": "benchmarks"\n}',
|
|
)
|
|
|
|
|
|
class Artifact(BaseModel):
|
|
artifact_id: str = Field(
|
|
...,
|
|
description="Id of the artifact",
|
|
example="b225e278-8b4c-4f99-a696-8facf19f0e56",
|
|
)
|
|
file_name: str = Field(
|
|
..., description="Filename of the artifact", example="main.py"
|
|
)
|
|
relative_path: Optional[str] = Field(
|
|
None,
|
|
description="Relative path of the artifact in the agent's workspace",
|
|
example="python/code/",
|
|
)
|
|
|
|
|
|
class ArtifactUpload(BaseModel):
|
|
file: bytes = Field(..., description="File to upload")
|
|
relative_path: Optional[str] = Field(
|
|
None,
|
|
description="Relative path of the artifact in the agent's workspace",
|
|
example="python/code/",
|
|
)
|
|
|
|
|
|
class StepInput(BaseModel):
|
|
__root__: Any = Field(
|
|
...,
|
|
description="Input parameters for the task step. Any value is allowed.",
|
|
example='{\n"file_to_refactor": "models.py"\n}',
|
|
)
|
|
|
|
|
|
class StepOutput(BaseModel):
|
|
__root__: Any = Field(
|
|
...,
|
|
description="Output that the task step has produced. Any value is allowed.",
|
|
example='{\n"tokens": 7894,\n"estimated_cost": "0,24$"\n}',
|
|
)
|
|
|
|
|
|
class TaskRequestBody(BaseModel):
|
|
input: Optional[str] = Field(
|
|
None,
|
|
description="Input prompt for the task.",
|
|
example="Write the words you receive to the file 'output.txt'.",
|
|
)
|
|
additional_input: Optional[TaskInput] = None
|
|
|
|
|
|
class Task(TaskRequestBody):
|
|
task_id: str = Field(
|
|
...,
|
|
description="The ID of the task.",
|
|
example="50da533e-3904-4401-8a07-c49adf88b5eb",
|
|
)
|
|
artifacts: List[Artifact] = Field(
|
|
[],
|
|
description="A list of artifacts that the task has produced.",
|
|
example=[
|
|
"7a49f31c-f9c6-4346-a22c-e32bc5af4d8e",
|
|
"ab7b4091-2560-4692-a4fe-d831ea3ca7d6",
|
|
],
|
|
)
|
|
|
|
|
|
class StepRequestBody(BaseModel):
|
|
input: Optional[str] = Field(
|
|
None, description="Input prompt for the step.", example="Washington"
|
|
)
|
|
additional_input: Optional[StepInput] = None
|
|
|
|
|
|
class Status(Enum):
|
|
created = "created"
|
|
running = "running"
|
|
completed = "completed"
|
|
|
|
|
|
class Step(StepRequestBody):
|
|
task_id: str = Field(
|
|
...,
|
|
description="The ID of the task this step belongs to.",
|
|
example="50da533e-3904-4401-8a07-c49adf88b5eb",
|
|
)
|
|
step_id: str = Field(
|
|
...,
|
|
description="The ID of the task step.",
|
|
example="6bb1801a-fd80-45e8-899a-4dd723cc602e",
|
|
)
|
|
name: Optional[str] = Field(
|
|
None, description="The name of the task step.", example="Write to file"
|
|
)
|
|
status: Status = Field(..., description="The status of the task step.")
|
|
output: Optional[str] = Field(
|
|
None,
|
|
description="Output of the task step.",
|
|
example="I am going to use the write_to_file command and write Washington to a file called output.txt <write_to_file('output.txt', 'Washington')",
|
|
)
|
|
additional_output: Optional[StepOutput] = None
|
|
artifacts: List[Artifact] = Field(
|
|
[], description="A list of artifacts that the step has produced."
|
|
)
|
|
is_last: Optional[bool] = Field(
|
|
False, description="Whether this is the last step in the task."
|
|
)
|