18 lines
350 B
Python
18 lines
350 B
Python
|
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)
|