Refactor: move shared constants to utilities/constants.py and update guardrails tests

Signed-off-by: Sandeep20013 <sandeepm20013@gmail.com>

Author: Sandeep20013

tests/fixtures/guardrails.py

@@ -13,7 +13,7 @@
 from ocp_resources.resource import ResourceEditor
 from ocp_resources.route import Route
 
-from tests.model_explainability.guardrails.constants import HAP_DETECTOR, PROMPT_INJECTION_DETECTOR
+from utilities.constants import HAP_DETECTOR, PROMPT_INJECTION_DETECTOR
 from utilities.constants import BUILTIN_DETECTOR_CONFIG, Annotations, Labels, VLLMGPUConfig
 from utilities.guardrails import check_guardrails_health_endpoint
 

tests/model_explainability/guardrails/constants.py

@@ -12,8 +12,7 @@
 PII_ENDPOINT: str = "/pii"
 AUTOCONFIG_GATEWAY_ENDPOINT: str = "/all"
 STANDALONE_DETECTION_ENDPOINT: str = "api/v2/text/detection/content"
-PROMPT_INJECTION_DETECTOR: str = "prompt-injection-detector"
-HAP_DETECTOR: str = "hap-detector"
+
 
 
 @dataclass

tests/model_explainability/guardrails/test_guardrails.py

@@ -8,13 +8,11 @@
     AUTOCONFIG_DETECTOR_LABEL,
     AUTOCONFIG_GATEWAY_ENDPOINT,
     CHAT_COMPLETIONS_DETECTION_ENDPOINT,
-    HAP_DETECTOR,
     HAP_INPUT_DETECTION_PROMPT,
     HARMLESS_PROMPT,
     PII_ENDPOINT,
     PII_INPUT_DETECTION_PROMPT,
     PII_OUTPUT_DETECTION_PROMPT,
-    PROMPT_INJECTION_DETECTOR,
     PROMPT_INJECTION_INPUT_DETECTION_PROMPT,
     STANDALONE_DETECTION_ENDPOINT,
 )
@@ -31,7 +29,7 @@
     BUILTIN_DETECTOR_CONFIG,
     LLM_D_CHAT_GENERATION_CONFIG,
     LLMdInferenceSimConfig,
-    Timeout,
+    Timeout, PROMPT_INJECTION_DETECTOR, HAP_DETECTOR,
 )
 from utilities.plugins.constant import OpenAIEnpoints
 

tests/model_explainability/guardrails/test_guardrails_gpu.py

@@ -3,13 +3,11 @@
 
 from tests.model_explainability.guardrails.constants import (
     CHAT_COMPLETIONS_DETECTION_ENDPOINT,
-    HAP_DETECTOR,
     HAP_INPUT_DETECTION_PROMPT,
     HARMLESS_PROMPT,
     PII_ENDPOINT,
     PII_INPUT_DETECTION_PROMPT,
     PII_OUTPUT_DETECTION_PROMPT_QWEN,
-    PROMPT_INJECTION_DETECTOR,
     PROMPT_INJECTION_INPUT_DETECTION_PROMPT,
     STANDALONE_DETECTION_ENDPOINT,
 )
@@ -22,7 +20,7 @@
     verify_health_info_response,
 )
 from utilities.constants import (
-    VLLMGPUConfig,
+    VLLMGPUConfig, PROMPT_INJECTION_DETECTOR, HAP_DETECTOR,
 )
 from utilities.plugins.constant import OpenAIEnpoints
 

utilities/constants.py

@@ -518,3 +518,6 @@ def get_hostname(cls, namespace: str) -> str:
 
 class PodNotFound(Exception):
     """Pod not found"""
+
+PROMPT_INJECTION_DETECTOR: str = "prompt-injection-detector"
+HAP_DETECTOR: str = "hap-detector"