Update config.py

config.py

@@ -20,24 +20,21 @@ class Settings(BaseSettings):
     shipper_id: str = Field(default="default_shipper")
     
     class Config:
-        # This enables the automatic reloading of values when they change
         validate_assignment = True
     
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self.set_model()
     
-    # Function to dynamically select model and processor based on shipper_id
+    # Dynamically select model and processor based on shipper_id
     def set_model(self):
-        # IMPORTANT: Make sure model names are consistent!
-        # Map can handle both string and integer keys by converting to string
         shipper_model_map = {
-            # Group 1 models - using donut-v16
+            # Shippers using donut-v16
             "61": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
             "81": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
             "139": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
             
-            # Group 2 models - using donut-v17
+            # Shippers using donut-v17
             "165": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
             "127": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
             "145": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
@@ -46,7 +43,6 @@ class Settings(BaseSettings):
         previous_model = self.model
         previous_processor = self.processor
         
-        # Convert shipper_id to string to handle both numeric and string values
         shipper_id_str = str(self.shipper_id)
         
         config = shipper_model_map.get(
@@ -57,18 +53,12 @@ class Settings(BaseSettings):
         self.model = config["model"]
         self.processor = config["processor"]
         
-        # Log changes for debugging
-        logger.info(f"Shipper ID set to: {self.shipper_id}")
-        logger.info(f"Changed model from {previous_model} to {self.model}")
-        logger.info(f"Changed processor from {previous_processor} to {self.processor}")
-        
         return self.model, self.processor
 
-# Create a single instance
 settings = Settings()
 logger.info(f"Initial model setup: {settings.model}")
 
-# Function to update shipper and trigger model change
+# trigger model change
 def update_shipper(new_shipper_id):
     """
     Update the shipper ID and change the model accordingly