Add explicit HYPIR restoration profile

dev/test_router_support.py

@@ -24,10 +24,13 @@ from router_support import (  # noqa: E402
     load_profile_registry,
 )
 from ai_profile_router import (  # noqa: E402
+    RESTORATION_CHAT_MODEL,
+    VIRTUAL_MODELS,
     STATE,
     _cap_chat_generation,
     _context_matches,
     _inject_global_system_policy,
+    _inject_restoration_system_policy,
     _normalize_chat_image,
     _normalize_chat_images,
     _normalize_llamacpp_reasoning,
@@ -80,6 +83,10 @@ class RuntimeStoreTests(unittest.TestCase):
 
 
 class ProfileRegistryTests(unittest.TestCase):
+    def test_restoration_model_maps_to_fast_instruction_profile(self) -> None:
+        self.assertEqual(RESTORATION_CHAT_MODEL, "restauration")
+        self.assertEqual(VIRTUAL_MODELS[RESTORATION_CHAT_MODEL], "fast")
+
     def test_fallback_contains_uncensored_profile(self) -> None:
         registry = load_profile_registry(None)
         self.assertEqual(registry["uncensored"]["context"], 80000)
@@ -270,6 +277,24 @@ class GlobalSystemPolicyTests(unittest.TestCase):
             self._inject(request, "/v1/images/generations"), request)
 
 
+class RestorationSystemPolicyTests(unittest.TestCase):
+    def test_chat_policy_requires_image_tool_and_preservation(self) -> None:
+        request = {"messages": [{"role": "user", "content": "Mach schöner"}]}
+        normalized = _inject_restoration_system_policy(
+            request, "/v1/chat/completions")
+        policy = normalized["messages"][0]["content"]
+        self.assertIn("image generation/editing tool", policy)
+        self.assertIn("Preserve identity", policy)
+        self.assertIn("Do not attempt restoration with Python", policy)
+
+    def test_responses_policy_keeps_client_instructions(self) -> None:
+        request = {"instructions": "Client policy.", "input": "Mach schöner"}
+        normalized = _inject_restoration_system_policy(
+            request, "/v1/responses")
+        self.assertIn("Photo-restoration mode", normalized["instructions"])
+        self.assertTrue(normalized["instructions"].endswith("Client policy."))
+
+
 class RetentionTests(unittest.TestCase):
     def test_oldest_pairs_are_removed(self) -> None:
         with tempfile.TemporaryDirectory() as temp: