feat: preserve image profile configuration

2026-07-10 05:26:09 +08:00
parent 58638d9fe3
commit c52afce684
4 changed files with 94 additions and 11 deletions
@@ -409,6 +409,11 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
            "reasoning_equals": [516, 1034],
            "raw": {
                "form": {
+                    "image_base_url": "https://images.example/v1",
+                    "image_auth_mode": "manual_bearer",
+                    "image_manual_secret_file": "/run/secrets/images-token",
+                    "image_manual_secret_configured": True,
+                    "image_auth_json_key": "IMAGE_API_KEY",
                    "retryable_error_messages": [
                        "Selected model is at capacity. Please try a different model.",
                        "stream disconnected before completion: Concurrency limit exceeded for account, please retry later",
@@ -419,6 +424,17 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
        text = mod.profile_editor_document(row)
        payload = mod.profile_payload_from_editor_text(text)
        self.assertEqual(payload["name"], "pc")
+        self.assertEqual(payload["image_base_url"], "https://images.example/v1")
+        self.assertEqual(payload["image_auth_mode"], "manual_bearer")
+        self.assertEqual(payload["image_manual_secret_file"], "/run/secrets/images-token")
+        self.assertTrue(payload["image_manual_secret_configured"])
+        self.assertEqual(payload["image_manual_secret"], "")
+        self.assertNotIn("test-image-profile-secret", text)
+        saved_payload = mod.profile_payload_from_row(row)
+        self.assertEqual(saved_payload["image_base_url"], "https://images.example/v1")
+        self.assertEqual(saved_payload["image_auth_mode"], "manual_bearer")
+        self.assertEqual(saved_payload["image_manual_secret_file"], "/run/secrets/images-token")
+        self.assertEqual(saved_payload["image_manual_secret"], "")
        self.assertIn(
            "stream disconnected before completion: Concurrency limit exceeded for account, please retry later",
            payload["retryable_error_messages"],