mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-09-30 21:39:52 +02:00
support prompt template loading
This commit is contained in:
1 parent
911b629217
commit
af75e9aabf
22 files changed
+291
-97
No files matched your search
@@ -1,21 +1,7 @@
|
||||
import pytest
|
||||
from src.text_generation.domain.average import Average
|
||||
from src.text_generation.domain.guardrail_analyzed_response import GuardrailAnalyzedResponse
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_get_average():
|
||||
scores = [
|
||||
0.12765,
|
||||
0.00282,
|
||||
0.63945,
|
||||
0.97123,
|
||||
0.38921
|
||||
]
|
||||
avg_1 = Average().from_list_of_floats(scores)
|
||||
assert avg_1 == 0.426072
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_guardrail_analyzed_response():
|
||||
response = GuardrailAnalyzedResponse(
|
||||
|
||||
Reference in new issue
Block a user