Reformatted

Author: tisnik

tests/e2e/features/steps/okp_rag.py

@@ -375,9 +375,9 @@ def check_tool_call_name(context: Context, name: str) -> None:
     tool_calls = _get_tool_calls(context)
     assert tool_calls, "No tool_calls to check"
     names = [tc.get("name") for tc in tool_calls]
-    assert name in names, (
-        f"Expected tool_call with name {name!r}, but found names {names!r}"
-    )
+    assert (
+        name in names
+    ), f"Expected tool_call with name {name!r}, but found names {names!r}"
 
 
 # ── Then Steps: Content and Results Assertions ──

tests/e2e/features/steps/proxy.py

@@ -618,9 +618,7 @@ def configure_ogx_ciphers(context: Context, ciphers: str) -> None:
 # --- Proxy Verification Steps ---
 
 
-@then(
-    "The tunnel proxy handled at least {count:d} CONNECT request to the LLM provider"
-)
+@then("The tunnel proxy handled at least {count:d} CONNECT request to the LLM provider")
 def verify_tunnel_proxy_used(context: Context, count: int) -> None:
     """Verify the tunnel proxy received CONNECT requests."""
     if is_prow_environment():
@@ -634,9 +632,9 @@ def verify_tunnel_proxy_used(context: Context, count: int) -> None:
         return
 
     proxy = context.tunnel_proxy
-    assert proxy.connect_count >= count, (
-        f"Expected at least {count} CONNECT requests, got {proxy.connect_count}"
-    )
+    assert (
+        proxy.connect_count >= count
+    ), f"Expected at least {count} CONNECT requests, got {proxy.connect_count}"
     assert proxy.last_connect_target is not None, "No CONNECT target recorded"
 
 

tests/unit/utils/test_saved_prompts.py

@@ -82,29 +82,23 @@ def test_rejects_count_equal_to_max(self) -> None:
         """Test create is rejected when current count equals the inclusive max."""
         with pytest.raises(
             SavedPromptLimitExceededError,
-            match=(
-                r"Saved prompt limit exceeded: 50 existing prompts, maximum is 50"
-            ),
+            match=(r"Saved prompt limit exceeded: 50 existing prompts, maximum is 50"),
         ):
             validate_saved_prompt_quota(50, 50)
 
     def test_rejects_count_above_max(self) -> None:
         """Test create is rejected when current count is above the max."""
         with pytest.raises(
             SavedPromptLimitExceededError,
-            match=(
-                r"Saved prompt limit exceeded: 51 existing prompts, maximum is 50"
-            ),
+            match=(r"Saved prompt limit exceeded: 51 existing prompts, maximum is 50"),
         ):
             validate_saved_prompt_quota(51, 50)
 
     def test_rejects_when_max_is_zero(self) -> None:
         """Test max_prompts_per_user of 0 rejects even a zero current count."""
         with pytest.raises(
             SavedPromptLimitExceededError,
-            match=(
-                r"Saved prompt limit exceeded: 0 existing prompts, maximum is 0"
-            ),
+            match=(r"Saved prompt limit exceeded: 0 existing prompts, maximum is 0"),
         ):
             validate_saved_prompt_quota(0, 0)