Replace GuardrailModelFactory with GuardrailClassifierModelFactory and enforce explicit MODEL_PATH env var; rename factory alias accordingly

Author: Paweł Kędzia

llm_router_services/guardrails/inference/factory.py

@@ -48,4 +48,4 @@ def create(
 
 
 # Public alias expected by the Flask app
-GuardrailModelFactory = create
+GuardrailClassifierModelFactory = create

llm_router_services/guardrails/nask/nask_pib_guard_app.py

@@ -4,34 +4,30 @@
 from flask import Flask, request, jsonify
 
 from llm_router_services.guardrails.constants import SERVICES_API_PREFIX
-from llm_router_services.guardrails.inference.factory import GuardrailModelFactory
+from llm_router_services.guardrails.inference.factory import (
+    GuardrailClassifierModelFactory,
+)
 
-# Import the NASK‑specific configuration
 from llm_router_services.guardrails.nask.config import NaskModelConfig
 
-# -----------------------------------------------------------------------
-# Environment prefix – all configuration keys start with this value
-# -----------------------------------------------------------------------
 _ENV_PREFIX = "LLM_ROUTER_NASK_PIB_GUARD_"
 
 app = Flask(__name__)
 
-MODEL_PATH = os.getenv(
-    f"{_ENV_PREFIX}MODEL_PATH",
-    "/mnt/data2/llms/models/community/NASK-PIB/HerBERT-PL-Guard",
-)
+MODEL_PATH = os.getenv(f"{_ENV_PREFIX}MODEL_PATH", None)
+if not MODEL_PATH:
+    raise Exception(
+        f"NASK-PIB guardrail model path is not set! "
+        f"Export {_ENV_PREFIX}MODEL_PATH with proper model path"
+    )
 
-# Keep only a single constant for the device (CPU by default)
 DEFAULT_DEVICE = int(os.getenv(f"{_ENV_PREFIX}DEVICE", "-1"))
 
-# -----------------------------------------------------------------------
-# Build the guardrail object via the factory, passing the NASK‑specific config
-# -----------------------------------------------------------------------
-guardrail = GuardrailModelFactory(
+guardrail = GuardrailClassifierModelFactory(
     model_type="text_classification",
     model_path=MODEL_PATH,
     device=DEFAULT_DEVICE,
-    config=NaskModelConfig(),  # <-- NASK‑specific thresholds & batch size
+    config=NaskModelConfig(),
 )
 
 

run_nask_guardrail.sh

@@ -16,8 +16,6 @@
 : "${LLM_ROUTER_NASK_PIB_GUARD_MODEL_PATH:=NASK-PIB/HerBERT-PL-Guard}"
 : "${LLM_ROUTER_NASK_PIB_GUARD_DEVICE:=0}"
 
-#: "${LLM_ROUTER_NASK_PIB_GUARD_MODEL_PATH:=/mnt/data2/llms/models/community/NASK-PIB/HerBERT-PL-Guard}"
-
 # Export them so the Python process can read them
 export LLM_ROUTER_NASK_PIB_GUARD_FLASK_HOST
 export LLM_ROUTER_NASK_PIB_GUARD_FLASK_PORT