mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-07-01 19:05:33 +02:00
default risk threshold; scores lower down in JSON serialization
This commit is contained in:
@@ -11,7 +11,7 @@ class GuardrailsResult:
|
||||
completion_text: str,
|
||||
full_prompt: dict[str, Any],
|
||||
llm_config: dict,
|
||||
cosine_similarity_risk_threshold: float = 0.0
|
||||
cosine_similarity_risk_threshold: float = 0.8
|
||||
):
|
||||
self.user_prompt = user_prompt
|
||||
self.guardrails_completion_text = completion_text
|
||||
|
||||
@@ -11,7 +11,7 @@ class GuidelinesResult(AbstractGuidelinesProcessedCompletion):
|
||||
completion_text: str,
|
||||
full_prompt: dict[str, Any],
|
||||
llm_config: dict,
|
||||
cosine_similarity_risk_threshold: float = 0.0
|
||||
cosine_similarity_risk_threshold: float = 0.8
|
||||
):
|
||||
self.user_prompt = user_prompt
|
||||
self.guidelines_completion_text = completion_text
|
||||
|
||||
@@ -19,7 +19,7 @@ class OriginalCompletionResult(AbstractTextGenerationCompletionResult):
|
||||
self.completion_text = completion_text
|
||||
self.full_prompt = full_prompt
|
||||
self.llm_config = llm_config
|
||||
self.cosine_similarity_risk_threshold: float = 0.0
|
||||
self.cosine_similarity_risk_threshold: float = 0.8
|
||||
|
||||
def append_semantic_similarity_result(self, semantic_similarity_result: SemanticSimilarityResult):
|
||||
self.semantic_similarity_result = semantic_similarity_result
|
||||
|
||||
@@ -8,6 +8,6 @@ from src.text_generation.domain.abstract_semantic_similarity_result import Abstr
|
||||
class SemanticSimilarityResult(AbstractSemanticSimilarityResult):
|
||||
def __init__(self, scores: ndarray):
|
||||
super().__init__()
|
||||
self.scores: List[float] = scores.tolist()
|
||||
self.max: float = float(scores.max())
|
||||
self.mean: float = float(scores.mean())
|
||||
self.max: float = float(scores.max())
|
||||
self.scores: List[float] = scores.tolist()
|
||||
Reference in New Issue
Block a user