Update config.py

config.py

@@ -1,8 +1,7 @@
 from pydantic import BaseSettings, Field
 import logging
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
+# Configure logger
 logger = logging.getLogger(__name__)
 
 class Settings(BaseSettings):
@@ -31,7 +30,7 @@ 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
+        # Map can handle both string and integer keys by converting to string
         shipper_model_map = {
             # Group 1 models - using donut-v16
             "61": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
@@ -47,8 +46,11 @@ 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(
-            self.shipper_id, 
+            shipper_id_str, 
             {"model": self.base_model, "processor": self.base_processor}
         )
         
@@ -62,7 +64,7 @@ class Settings(BaseSettings):
         
         return self.model, self.processor
 
-# Create a singleton instance
+# Create a single instance
 settings = Settings()
 logger.info(f"Initial model setup: {settings.model}")