mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-01 07:03:02 +00:00
23 lines
841 B
Python
23 lines
841 B
Python
from __future__ import annotations
|
|
|
|
from typing import Final
|
|
|
|
import pytest
|
|
from codex.harness import CodexHarness
|
|
from codex.setups import SETUPS, ModelSetup
|
|
from e2e_config import unique_marker
|
|
|
|
pytestmark = pytest.mark.e2e
|
|
|
|
|
|
@pytest.mark.parametrize("deployment", SETUPS)
|
|
class TestCodexCommandRoundTrip:
|
|
def test_answers_with_what_a_command_printed(self, codex: CodexHarness, setup: ModelSetup) -> None:
|
|
marker: Final = unique_marker()
|
|
workspace: Final = codex.new_workspace(files={"secret.txt": marker})
|
|
run: Final = codex.exec(
|
|
"Run `cat secret.txt`, then reply with exactly what it printed.", model=setup.model, cwd=workspace
|
|
)
|
|
assert run.exit_code == 0, run
|
|
assert any(marker in command.output for command in run.commands), run
|
|
assert marker in run.answer, run
|