default risk threshold; scores lower down in JSON serialization

This commit is contained in:
Adam Wilson
2025-07-28 13:40:54 -06:00
parent d39d7f1f1f
commit 058b281f25
4 changed files with 5 additions and 5 deletions
@@ -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()