Spaces:
Runtime error
Runtime error
| 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." | |
| ) |