from __future__ import annotations

from pathlib import Path

from src.schemas import ExternalAgentResult, SanitizedTask
from src.tools.codex_runner import run_codex_or_mock


class CodexBigAgent:
    def __init__(self, base_dir: str | Path = "data") -> None:
        self.base_dir = Path(base_dir)

    def run(self, task: SanitizedTask) -> ExternalAgentResult:
        return run_codex_or_mock(task, self.base_dir)