fix: model name in Guardrails tests

adolfo-ab · adolfo-ab · commit 6c2c009c399d · 2025-09-09T17:33:23.000+02:00
diff --git a/tests/model_explainability/guardrails/test_guardrails.py b/tests/model_explainability/guardrails/test_guardrails.py
@@ -26,7 +26,6 @@
     BUILTIN_DETECTOR_CONFIG,
     MinIo,
     QWEN_MODEL_NAME,
-    QWEN_ISVC_NAME,
 )
 from utilities.plugins.constant import OpenAIEnpoints
 
@@ -246,7 +245,7 @@ def test_guardrails_builtin_detectors_negative_detection(
                     "config.yaml": yaml.dump({
                         "openai": {
                             "service": {
-                                "hostname": f"{QWEN_ISVC_NAME}-predictor",
+                                "hostname": f"{QWEN_MODEL_NAME}-predictor",
                                 "port": 8032,
                             }
                         },
@@ -383,7 +382,7 @@ def test_guardrails_standalone_detector_endpoint(
                     "config.yaml": yaml.dump({
                         "openai": {
                             "service": {
-                                "hostname": f"{QWEN_ISVC_NAME}-predictor",
+                                "hostname": f"{QWEN_MODEL_NAME}-predictor",
                                 "port": 8032,
                             }
                         },
diff --git a/utilities/constants.py b/utilities/constants.py
@@ -368,8 +368,9 @@ class RunTimeConfig:
 }
 
 QWEN_ISVC_NAME = "qwen-isvc"
+QWEN_MODEL_NAME: str = "qwen25-05b-instruct"
+
 CHAT_GENERATION_CONFIG: Dict[str, Any] = {
-    "service": {"hostname": f"{QWEN_ISVC_NAME}-predictor", "port": 8032, "request_timeout": 600}
+    "service": {"hostname": f"{QWEN_MODEL_NAME}-predictor", "port": 8032, "request_timeout": 600}
 }
 TRUSTYAI_SERVICE_NAME: str = "trustyai-service"
-QWEN_MODEL_NAME: str = "qwen25-05b-instruct"

Original file line number	Diff line number	Diff line change
`@@ -368,8 +368,9 @@ class RunTimeConfig:`
`368`	`368`	`}`
`369`	`369`
`370`	`370`	`QWEN_ISVC_NAME = "qwen-isvc"`
	`371`	`+QWEN_MODEL_NAME: str = "qwen25-05b-instruct"`
	`372`	`+`
`371`	`373`	`CHAT_GENERATION_CONFIG: Dict[str, Any] = {`
`372`		`- "service": {"hostname": f"{QWEN_ISVC_NAME}-predictor", "port": 8032, "request_timeout": 600}`
	`374`	`+ "service": {"hostname": f"{QWEN_MODEL_NAME}-predictor", "port": 8032, "request_timeout": 600}`
`373`	`375`	`}`
`374`	`376`	`TRUSTYAI_SERVICE_NAME: str = "trustyai-service"`
`375`		`-QWEN_MODEL_NAME: str = "qwen25-05b-instruct"`