mirror of
https://github.com/vegu-ai/talemate.git
synced 2025-12-28 16:06:38 +01:00
fix gpt-4 censorship triggered by system message (#74)
This commit is contained in:
34
tests/test_system_messages.py
Normal file
34
tests/test_system_messages.py
Normal file
@@ -0,0 +1,34 @@
|
||||
import pytest
|
||||
from talemate.client.base import ClientBase
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"kind",
|
||||
[
|
||||
"narrate",
|
||||
"story",
|
||||
"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 "crude" in client.get_system_message(kind)
|
||||
|
||||
client.decensor_enabled = False
|
||||
|
||||
assert client.get_system_message(kind) is not None
|
||||
|
||||
assert "crude" not in client.get_system_message(kind)
|
||||
|
||||
Reference in New Issue
Block a user