diff --git a/src/text_generation/services/guardrails/reflexion_security_guidelines_service.py b/src/text_generation/services/guardrails/reflexion_security_guidelines_service.py index 488558343..9cd58ec69 100644 --- a/src/text_generation/services/guardrails/reflexion_security_guidelines_service.py +++ b/src/text_generation/services/guardrails/reflexion_security_guidelines_service.py @@ -42,7 +42,7 @@ class ReflexionSecurityGuardrailsService( self.llm_configuration_introspection_service = llm_configuration_introspection_service # constants - self.COSINE_SIMILARITY_RISK_THRESHOLD = 0.5 + self.COSINE_SIMILARITY_RISK_THRESHOLD = 0.75 # set up semantic similarity service and supporting texts self.example_prompt_injection_completions = prompt_injection_example_service.get_all_completions() diff --git a/src/text_generation/services/nlp/text_generation_completion_service.py b/src/text_generation/services/nlp/text_generation_completion_service.py index 9648c036e..8e561b356 100644 --- a/src/text_generation/services/nlp/text_generation_completion_service.py +++ b/src/text_generation/services/nlp/text_generation_completion_service.py @@ -57,7 +57,7 @@ class TextGenerationCompletionService( self.reflexion_guardrails: AbstractGeneratedTextGuardrailService = reflexion_guardrails # constants - self.COSINE_SIMILARITY_RISK_THRESHOLD = 0.5 + self.COSINE_SIMILARITY_RISK_THRESHOLD = 0.75 # default guidelines settings self._use_guidelines = False