mirror of
https://github.com/vegu-ai/talemate.git
synced 2025-12-16 11:47:48 +01:00
33 lines
668 B
Python
33 lines
668 B
Python
import pytest
|
|
from talemate.client.base import ClientBase
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"kind",
|
|
[
|
|
"narrate",
|
|
"director",
|
|
"create",
|
|
"roleplay",
|
|
"conversation",
|
|
"editor",
|
|
"world_state",
|
|
"analyze_freeform",
|
|
"analyst",
|
|
"analyze",
|
|
"summarize",
|
|
],
|
|
)
|
|
def test_system_message(kind):
|
|
client = ClientBase()
|
|
|
|
assert client.get_system_message(kind) is not None
|
|
|
|
assert "explicit" in client.get_system_message(kind)
|
|
|
|
client.decensor_enabled = False
|
|
|
|
assert client.get_system_message(kind) is not None
|
|
|
|
assert "explicit" not in client.get_system_message(kind)
|