From 058b281f253651ab5c7f6a807b6abfa17088d677 Mon Sep 17 00:00:00 2001 From: Adam Wilson Date: Mon, 28 Jul 2025 13:40:54 -0600 Subject: [PATCH] default risk threshold; scores lower down in JSON serialization --- src/text_generation/domain/guardrails_result.py | 2 +- src/text_generation/domain/guidelines_result.py | 2 +- src/text_generation/domain/original_completion_result.py | 2 +- src/text_generation/domain/semantic_similarity_result.py | 4 ++-- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/text_generation/domain/guardrails_result.py b/src/text_generation/domain/guardrails_result.py index 1ba920158..b0587ec5b 100644 --- a/src/text_generation/domain/guardrails_result.py +++ b/src/text_generation/domain/guardrails_result.py @@ -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 diff --git a/src/text_generation/domain/guidelines_result.py b/src/text_generation/domain/guidelines_result.py index 84f7e2d65..3fbd3bc36 100644 --- a/src/text_generation/domain/guidelines_result.py +++ b/src/text_generation/domain/guidelines_result.py @@ -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 diff --git a/src/text_generation/domain/original_completion_result.py b/src/text_generation/domain/original_completion_result.py index 14c240ccb..36ae4c967 100644 --- a/src/text_generation/domain/original_completion_result.py +++ b/src/text_generation/domain/original_completion_result.py @@ -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 diff --git a/src/text_generation/domain/semantic_similarity_result.py b/src/text_generation/domain/semantic_similarity_result.py index 533da1470..c9ddc10d2 100644 --- a/src/text_generation/domain/semantic_similarity_result.py +++ b/src/text_generation/domain/semantic_similarity_result.py @@ -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()) \ No newline at end of file + self.scores: List[float] = scores.tolist() \ No newline at end of file