Integrate Qwen image prompt enhancers

2026-09-21 14:51:29 +02:00
parent b832157226
commit 5106d0d2ed
5 changed files with 403 additions and 10 deletions
@@ -13,6 +13,18 @@ import ai_profile_router as router


 class CoordinationTests(unittest.TestCase):
+    def test_prompt_enhancer_accepts_plain_and_fenced_json(self):
+        plain = router._parse_prompt_enhancer_result(
+            '{"rewritten_prompt":"new scene","wh_ratio":"3:2"}')
+        fenced = router._parse_prompt_enhancer_result(
+            '```json\n{"rewritten_prompt":"portrait","wh_ratio":"3:4"}\n```')
+        self.assertEqual(plain['rewritten_prompt'], 'new scene')
+        self.assertEqual(fenced['wh_ratio'], '3:4')
+
+    def test_prompt_enhancer_rejects_missing_rewrite(self):
+        with self.assertRaisesRegex(RuntimeError, 'rewritten_prompt'):
+            router._parse_prompt_enhancer_result('{"wh_ratio":"1:1"}')
+
    def test_mode_uses_one_consistent_controller_snapshot(self):
        with patch.object(router, 'PROFILE_CONTROL_URL', 'http://controller'), \
             patch.object(router, '_profile_controller_request', return_value={