Skip to content

Commit 5bfe5a2

Browse files
committed
fixup! test(llm_e2e): run smart-alignment e2e when OpenAI or Gemini key present
1 parent 5dcff51 commit 5bfe5a2

1 file changed

Lines changed: 8 additions & 12 deletions

File tree

application/tests/test_smart_embeddings_e2e_llm.py

Lines changed: 8 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,7 @@
2424
def _skip_no_llm() -> None:
2525
if os.environ.get("CRE_EMBED_ALIGN_E2E", "1").lower() in ("0", "false", "no"):
2626
pytest.skip("CRE_EMBED_ALIGN_E2E disabled")
27-
if not (
28-
os.environ.get("OPENAI_API_KEY") or os.environ.get("GEMINI_API_KEY")
29-
):
27+
if not (os.environ.get("OPENAI_API_KEY") or os.environ.get("GEMINI_API_KEY")):
3028
pytest.skip("No LLM credentials (set OPENAI_API_KEY or GEMINI_API_KEY)")
3129

3230

@@ -104,12 +102,10 @@ def test_owasp_ai_exchange_live_alignment(
104102
os.environ.get("CRE_EMBED_SMART_CONFIDENCE", "0.55")
105103
),
106104
)
107-
assert out.used_excerpt, (
108-
f"expected excerpt mode ({provider}): {out.rationale!r}"
109-
)
110-
assert expected_fragment in out.resolved_embeddings_url, (
111-
f"expected #{expected_fragment} in resolved URL ({provider}), got {out.resolved_embeddings_url!r}"
112-
)
113-
assert len(out.embed_plain_text) < len(full_clean) * 0.98, (
114-
f"excerpt should be materially shorter than full cleaned body ({provider})"
115-
)
105+
assert out.used_excerpt, f"expected excerpt mode ({provider}): {out.rationale!r}"
106+
assert (
107+
expected_fragment in out.resolved_embeddings_url
108+
), f"expected #{expected_fragment} in resolved URL ({provider}), got {out.resolved_embeddings_url!r}"
109+
assert (
110+
len(out.embed_plain_text) < len(full_clean) * 0.98
111+
), f"excerpt should be materially shorter than full cleaned body ({provider})"

0 commit comments

Comments
 (0)