Merge restoration policy into system message

2026-09-08 08:37:53 +02:00
parent 118e32005e
commit 636e48ce93
2 changed files with 27 additions and 1 deletions
@@ -294,6 +294,21 @@ class RestorationSystemPolicyTests(unittest.TestCase):
        self.assertIn("Photo-restoration mode", normalized["instructions"])
        self.assertTrue(normalized["instructions"].endswith("Client policy."))

+    def test_chat_policy_merges_with_existing_leading_system_message(self) -> None:
+        request = {"messages": [
+            {"role": "system", "content": "Global policy."},
+            {"role": "user", "content": "Mach schöner"},
+        ]}
+        normalized = _inject_restoration_system_policy(
+            request, "/v1/chat/completions")
+        system_messages = [
+            message for message in normalized["messages"]
+            if message.get("role") == "system"
+        ]
+        self.assertEqual(len(system_messages), 1)
+        self.assertIn("Photo-restoration mode", system_messages[0]["content"])
+        self.assertTrue(system_messages[0]["content"].endswith("Global policy."))
+

 class RetentionTests(unittest.TestCase):
    def test_oldest_pairs_are_removed(self) -> None: