address review comment

Signed-off-by: Huamin Chen <[email protected]>

Author: rootfs

src/semantic-router/pkg/config/config.go

@@ -418,6 +418,8 @@ func ReplaceGlobalConfig(newCfg *RouterConfig) {
 
 // GetConfig returns the current configuration
 func GetConfig() *RouterConfig {
+	configMu.RLock()
+	defer configMu.RUnlock()
 	return config
 }
 

src/semantic-router/pkg/extproc/request_handler.go

@@ -623,8 +623,8 @@ func (r *OpenAIRouter) handleModelRouting(openAIRequest *openai.ChatCompletionNe
 							observability.Infof("Added category-specific system prompt for category: %s (mode: %s)", categoryName, mode)
 						}
 
-						// Log the complete message structure after system prompt injection
-						observability.Infof("Complete request body after system prompt injection: %s", string(modifiedBody))
+						// Log metadata about system prompt injection (avoid logging sensitive user data)
+						observability.Infof("System prompt injection completed for category: %s, body size: %d bytes", categoryName, len(modifiedBody))
 					} else if category != nil && category.SystemPrompt != "" && !category.IsSystemPromptEnabled() {
 						observability.Infof("System prompt disabled for category: %s", categoryName)
 					}

src/semantic-router/pkg/services/classification.go

@@ -3,6 +3,7 @@ package services
 import (
 	"fmt"
 	"os"
+	"sync"
 	"time"
 
 	"github.com/vllm-project/semantic-router/src/semantic-router/pkg/config"
@@ -18,6 +19,7 @@ type ClassificationService struct {
 	classifier        *classification.Classifier
 	unifiedClassifier *classification.UnifiedClassifier // New unified classifier
 	config            *config.RouterConfig
+	configMutex       sync.RWMutex // Protects config access
 }
 
 // NewClassificationService creates a new classification service
@@ -488,11 +490,15 @@ func (s *ClassificationService) GetUnifiedClassifierStats() map[string]interface
 
 // GetConfig returns the current configuration
 func (s *ClassificationService) GetConfig() *config.RouterConfig {
+	s.configMutex.RLock()
+	defer s.configMutex.RUnlock()
 	return s.config
 }
 
 // UpdateConfig updates the configuration
 func (s *ClassificationService) UpdateConfig(newConfig *config.RouterConfig) {
+	s.configMutex.Lock()
+	defer s.configMutex.Unlock()
 	s.config = newConfig
 	// Update the global config as well
 	config.ReplaceGlobalConfig(newConfig)