Spaces:
Running
Running
File size: 1,494 Bytes
70f2179 | 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 | # Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
"""Task payload accepted by :class:`OpenCodeSessionFactory`."""
from __future__ import annotations
from typing import Any
from pydantic import BaseModel, Field
class OpenCodeTask(BaseModel):
"""One task for an OpenCode rollout.
The primitive only needs ``instruction`` (the prompt handed to ``opencode
run``). Callers may attach ``setup_shell`` (run once inside the sandbox
before the agent starts) and ``upload_files`` (written into the sandbox at
absolute paths). Any additional metadata belongs in ``metadata`` and is
passed through to the verifier untouched.
"""
instruction: str
setup_shell: str | None = None
upload_files: dict[str, str] = Field(default_factory=dict)
metadata: dict[str, Any] = Field(default_factory=dict)
@classmethod
def coerce(cls, value: Any) -> "OpenCodeTask":
"""Accept a bare string, a dict, or an existing ``OpenCodeTask``."""
if isinstance(value, cls):
return value
if isinstance(value, str):
return cls(instruction=value)
if isinstance(value, dict):
return cls(**value)
raise TypeError(
f"Cannot coerce {type(value).__name__} to OpenCodeTask; "
"pass a str, dict, or OpenCodeTask."
)
|