fix gpt-4 censorship triggered by system message (#74)

This commit is contained in:
veguAI
2024-02-01 12:15:30 +02:00
committed by GitHub
parent 4ba635497b
commit eb251d6e37
14 changed files with 125 additions and 32 deletions

View 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)