From 138145026372bdee868fea796416b44f263daf0a Mon Sep 17 00:00:00 2001 From: Adam Wilson <42450907+lightbroker@users.noreply.github.com> Date: Mon, 1 Sep 2025 18:17:31 -0600 Subject: [PATCH] Update test__generative_ai_guidelines.py --- tests/cicd/test__generative_ai_guidelines.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/cicd/test__generative_ai_guidelines.py b/tests/cicd/test__generative_ai_guidelines.py index 84d15d499..9a7727761 100644 --- a/tests/cicd/test__generative_ai_guidelines.py +++ b/tests/cicd/test__generative_ai_guidelines.py @@ -3,6 +3,7 @@ $ pytest -k test__generative_ai_guidelines -s --disable-warnings """ +import random from typing import Callable, List from src.text_generation.domain.semantic_similarity_result import SemanticSimilarityResult @@ -26,7 +27,7 @@ def test__generative_ai_guidelines( results: list[SemanticSimilarityResult] = test_llm_guidelines( text_generation_completion_service=text_generation_completion_service, semantic_similarity_service=semantic_similarity_service, - prompts=malicious_prompts, + prompts=random.sample(malicious_prompts, 25), comparison_texts=prompt_injection_all_example_responses, service_configurator=configure_service )