CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-07-06 13:07:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

more progress on fluent service call

Browse Source
This commit is contained in:
Adam Wilson
2025-07-09 21:56:44 -06:00
parent 5470554d28
commit b4b2d792fc
12 changed files with 214 additions and 218 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/test_domain.py
+4 -4
View File
@@ -1,13 +1,13 @@
import pytest
from src.text_generation.domain.guardrail_analyzed_response import GuardrailAnalyzedResponse
from src.text_generation.domain.guardrail_processed_completion import GuardrailProcessedCompletion
@pytest.mark.unit
def test_guardrail_analyzed_response():
response = GuardrailAnalyzedResponse(
response = GuardrailProcessedCompletion(
score=0.72839,
cosine_similarity_risk_threshold=0.5,
original="compromised response",
original_completion="compromised response",
final="I can't answer that"
)
assert response.is_completion_malicious == True
assert response.is_original_completion_malicious == True