Skip to content

Commit 4ce1bbc

Browse files
yossiovadiaclaude
andcommitted
fix(aibrix): revert PII model to ModernBERT configuration from PR #688
The aibrix E2E profile was designed to use ModernBERT for PII detection (as configured in PR #688), but during the merge it was incorrectly changed to use the LoRA PII model which isn't available in the aibrix deployment. This commit reverts the aibrix PII model configuration to use: - model_id: "models/pii_classifier_modernbert-base_presidio_token_model" - use_modernbert: true This restores the original working configuration from PR #688 and allows the aibrix E2E tests to pass while keeping the LoRA PII auto-detection feature available for other profiles that explicitly configure it. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]> Signed-off-by: Yossi Ovadia <[email protected]>
1 parent e7754b5 commit 4ce1bbc

File tree

1 file changed

+3
-5
lines changed
  • deploy/kubernetes/aibrix/semantic-router-values

1 file changed

+3
-5
lines changed

deploy/kubernetes/aibrix/semantic-router-values/values.yaml

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -438,13 +438,11 @@ config:
438438
use_cpu: true
439439
category_mapping_path: "models/category_classifier_modernbert-base_model/category_mapping.json"
440440
pii_model:
441-
# Support both traditional (modernbert) and LoRA-based PII detection
442-
# When model_type is "auto", the system will auto-detect LoRA configuration
443-
model_id: "models/lora_pii_detector_bert-base-uncased_model"
444-
model_type: "auto" # Enables LoRA auto-detection
441+
model_id: "models/pii_classifier_modernbert-base_presidio_token_model"
442+
use_modernbert: true
445443
threshold: 0.7
446444
use_cpu: true
447-
pii_mapping_path: "models/lora_pii_detector_bert-base-uncased_model/pii_type_mapping.json"
445+
pii_mapping_path: "models/pii_classifier_modernbert-base_presidio_token_model/pii_type_mapping.json"
448446

449447
keyword_rules:
450448
- category: "thinking"

0 commit comments

Comments
 (0)