fix: update model manager initialization

- Added optional model_dir parameter to ModelManager
- Updated LocalLLMStrategy to pass model_dir correctly
- Fixed TypeError in ModelManager.__init__()

reasoning/local_llm.py

@@ -17,8 +17,9 @@ class LocalLLMStrategy(ReasoningStrategy):
         super().__init__()
         self.config = config or {}
         
-        # Initialize model manager
-        self.model_manager = ModelManager(self.config.get('model_dir', "models"))
+        # Initialize model manager with model_dir from config
+        model_dir = self.config.get('model_dir')
+        self.model_manager = ModelManager(model_dir)
         
         # Standard reasoning parameters
         self.min_confidence = self.config.get('min_confidence', 0.7)

reasoning/model_manager.py

@@ -30,9 +30,9 @@ class ModelConfig:
 class ModelManager:
     """Manages multiple LLM models for different tasks in Spaces."""
     
-    def __init__(self):
+    def __init__(self, model_dir: Optional[str] = None):
         # In Spaces, models are stored in the cache directory
-        self.model_dir = os.getenv('SPACE_CACHE_DIR', '/tmp/models')
+        self.model_dir = model_dir or os.getenv('SPACE_CACHE_DIR', '/tmp/models')
         self.models: Dict[str, Llama] = {}
         self.logger = logging.getLogger(__name__)