import sys
from pathlib import Path

import pytest

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))

from llamacppctl.prompt_io import InputPolicy  # noqa: E402


@pytest.fixture
def policy() -> InputPolicy:
    return InputPolicy()


@pytest.fixture
def small_policy() -> InputPolicy:
    return InputPolicy(max_input_bytes=16)