CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-02-12 14:42:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test__generative_ai_guidelines.py

Browse Source
This commit is contained in:
Adam Wilson
2025-09-01 18:17:31 -06:00
committed by GitHub
parent 79851ed1b5
commit 1381450263
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/cicd/test__generative_ai_guidelines.py
View File

@@ -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
)