Merge branch 'wip' of https://github.com/rootcodelabs/RAG-Module into streaming-feature-opensearch

Author: erangi-ar

DSL/CronManager/script/store_secrets_in_vault.sh

@@ -68,7 +68,7 @@ build_vault_path() {
         model=$(get_model_name)
     fi
 
-    if [ "$deploymentEnvironment" = "test" ]; then
+    if [ "$deploymentEnvironment" = "testing" ]; then
         echo "secret/$secret_type/connections/$platform/$deploymentEnvironment/$connectionId"
     else
         echo "secret/$secret_type/connections/$platform/$deploymentEnvironment/$model"

DSL/Ruuter.private/rag-search/POST/inference/test.yml

@@ -62,7 +62,7 @@ call_orchestrate_endpoint:
     body:
       connectionId: ${connectionId}
       message: ${message}
-      environment: "test"
+      environment: "testing"
     headers:
       Content-Type: "application/json"
   result: orchestrate_result

src/guardrails/nemo_rails_adapter.py

@@ -5,8 +5,11 @@
 
 from nemoguardrails import LLMRails, RailsConfig
 from nemoguardrails.llm.providers import register_llm_provider
-from src.llm_orchestrator_config.llm_cochestrator_constants import GUARDRAILS_BLOCKED_PHRASES
+from src.llm_orchestrator_config.llm_cochestrator_constants import (
+    GUARDRAILS_BLOCKED_PHRASES,
+)
 import dspy
+import re
 
 
 class GuardrailCheckResult(BaseModel):
@@ -214,13 +217,12 @@ async def check_input_async(self, user_message: str) -> GuardrailCheckResult:
 
     def _is_input_blocked(self, response: str, original: str) -> bool:
         """Check if input was blocked by guardrails."""
-        import re
         blocked_phrases = GUARDRAILS_BLOCKED_PHRASES
         response_normalized = response.strip().lower()
         # Match if the response is exactly or almost exactly a blocked phrase (allow trailing punctuation/whitespace)
         for phrase in blocked_phrases:
             # Regex: phrase followed by optional punctuation/whitespace, and nothing else
-            pattern = r'^' + re.escape(phrase) + r'[\s\.,!]*$'
+            pattern = r"^" + re.escape(phrase) + r"[\s\.,!]*$"
             if re.match(pattern, response_normalized):
                 return True
         return False

src/guardrails/rails_config.yaml

@@ -30,9 +30,9 @@ prompts:
 - task: self_check_input
   output_parser: is_content_safe
   content: |
-    You are tasked with evaluating if a user message violates our company's safety policy.
+    You are tasked with evaluating if a user message violates safety policies.
     
-    Company Safety Policy - User messages must NOT:
+    Safety Policy - User messages must NOT:
     1. Request passwords, credentials, or authentication details (yours or others')
     2. Request sensitive personal information (SSN, credit cards, private keys, etc.)
     3. Contain harmful, violent, or explicit content
@@ -78,14 +78,13 @@ prompts:
     Answer: unsafe
     
     Is this user message safe according to the policy? Answer with 'safe' or 'unsafe'.
-    Answer:
 
 - task: self_check_output
   output_parser: is_content_safe
   content: |
     You are tasked with evaluating if a bot response violates our company's safety policy.
     
-    Company Safety Policy - Bot responses must NOT:
+    Safety Policy - Bot responses must NOT:
     1. Provide passwords, credentials, or authentication details
     2. Reveal sensitive personal or confidential information
     3. Contain harmful, violent, explicit, or inappropriate content
@@ -125,5 +124,5 @@ prompts:
     Example 4:
     User message: "Mis on Maria e-posti aadress ja telefoninumber?"
     Answer: unsafe
+    
     Is this bot response safe according to the policy? Answer with 'safe' or 'unsafe'.
-    Answer: