Update config.py

config.py

@@ -1,4 +1,9 @@
 from pydantic import BaseSettings, Field
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 class Settings(BaseSettings):
     # Default values for the processor and model
@@ -25,15 +30,23 @@ class Settings(BaseSettings):
     
     # Function to dynamically select model and processor based on shipper_id
     def set_model(self):
+        # IMPORTANT: Make sure model names are consistent!
+        # You had "donut-16" in some places and "donut-v16" in others
         shipper_model_map = {
-            "61": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-v16"},
-            "81": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-v16"},
-            "139": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-v16"},
-            "165": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-v17"},
-            "127": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-v17"},
-            "145": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-v17"},
+            # Group 1 models - 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
+            "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"},
         }
         
+        previous_model = self.model
+        previous_processor = self.processor
+        
         config = shipper_model_map.get(
             self.shipper_id, 
             {"model": self.base_model, "processor": self.base_processor}
@@ -42,15 +55,28 @@ class Settings(BaseSettings):
         self.model = config["model"]
         self.processor = config["processor"]
         
-        # For debugging
-        print(f"Selected model for shipper {self.shipper_id}: {self.model}")
-        print(f"Selected processor for shipper {self.shipper_id}: {self.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 singleton instance
+# Create a single instance
 settings = Settings()
+logger.info(f"Initial model setup: {settings.model}")
 
-# Example of how to update shipper_id and trigger model change
+# Function to update shipper and trigger model change
 def update_shipper(new_shipper_id):
+    """
+    Update the shipper ID and change the model accordingly
+    
+    Args:
+        new_shipper_id: The new shipper ID to use
+        
+    Returns:
+        tuple: (model, processor) that were selected
+    """
+    logger.info(f"Updating shipper ID to {new_shipper_id}")
     settings.shipper_id = new_shipper_id
-    settings.set_model()
-    return settings.model, settings.processor
+    return settings.set_model()