chore: lint

Author: northdpole

application/tests/chat_completion_test.py

@@ -37,9 +37,7 @@ def test_completion_returns_503_json_on_gemini_429(self) -> None:
             },
             None,
         )
-        with patch(
-            "application.prompt_client.prompt_client.PromptHandler"
-        ) as mock_ph:
+        with patch("application.prompt_client.prompt_client.PromptHandler") as mock_ph:
             mock_ph.return_value.generate_text.side_effect = err
             with self.app.test_client() as client:
                 response = client.post(
@@ -56,12 +54,16 @@ def test_completion_propagates_non_429_genai_error(self) -> None:
         os.environ["NO_LOGIN"] = "1"
         err = genai_errors.ClientError(
             400,
-            {"error": {"code": 400, "message": "Bad request", "status": "INVALID_ARGUMENT"}},
+            {
+                "error": {
+                    "code": 400,
+                    "message": "Bad request",
+                    "status": "INVALID_ARGUMENT",
+                }
+            },
             None,
         )
-        with patch(
-            "application.prompt_client.prompt_client.PromptHandler"
-        ) as mock_ph:
+        with patch("application.prompt_client.prompt_client.PromptHandler") as mock_ph:
             mock_ph.return_value.generate_text.side_effect = err
             with self.app.test_client() as client:
                 with self.assertRaises(genai_errors.ClientError) as ctx:

application/tests/vertex_prompt_client_test.py

@@ -72,7 +72,13 @@ def test_is_genai_rate_limit_error_recognizes_clienterror_code_429(self) -> None
     def test_is_genai_rate_limit_error_false_for_other_clienterror(self) -> None:
         err = genai_errors.ClientError(
             400,
-            {"error": {"code": 400, "message": "Bad request", "status": "INVALID_ARGUMENT"}},
+            {
+                "error": {
+                    "code": 400,
+                    "message": "Bad request",
+                    "status": "INVALID_ARGUMENT",
+                }
+            },
             None,
         )
         self.assertFalse(_is_genai_rate_limit_error(err))

application/web/web_main.py

@@ -927,14 +927,17 @@ def chat_cre() -> Any:
     except genai_errors.ClientError as e:
         # google.genai APIError uses ``code`` (HTTP status), not ``status_code``.
         if getattr(e, "code", None) == 429:
-            return jsonify(
-                {
-                    "error": (
-                        "The AI service is temporarily rate-limited. "
-                        "Please try again in a minute."
-                    )
-                }
-            ), 503
+            return (
+                jsonify(
+                    {
+                        "error": (
+                            "The AI service is temporarily rate-limited. "
+                            "Please try again in a minute."
+                        )
+                    }
+                ),
+                503,
+            )
         raise
     return jsonify(response)