fixes

Author: codelion

README.md

@@ -1,7 +1,7 @@
 # OptiLLM
 
 <p align="center">
-  <img src="optillm-logo.png" alt="OptiLLM Logo" width="200" />
+  <img src="optillm-logo.png" alt="OptiLLM Logo" width="400" />
 </p>
 
 <p align="center">

optillm/plugins/json_plugin.py

@@ -22,7 +22,7 @@ def get_device(self):
         else:
             return torch.device("cpu")
 
-    def __init__(self, model_name: str = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"):
+    def __init__(self, model_name: str = "google/gemma-3-270m-it"):
         """Initialize the JSON generator with a specific model."""
         self.device = self.get_device()
         logger.info(f"Using device: {self.device}")

tests/test_json_plugin.py

@@ -82,11 +82,19 @@ def test_json_generator_init(self, mock_tokenizer, mock_from_transformers):
         self.assertIsNotNone(generator.model)
         self.assertIsNotNone(generator.tokenizer)
 
-    def test_parse_json_schema_to_pydantic(self):
+    @patch('optillm.plugins.json_plugin.outlines.from_transformers')
+    @patch('optillm.plugins.json_plugin.AutoModelForCausalLM.from_pretrained')
+    @patch('optillm.plugins.json_plugin.AutoTokenizer.from_pretrained')
+    def test_parse_json_schema_to_pydantic(self, mock_tokenizer, mock_model, mock_from_transformers):
         """Test JSON schema to Pydantic model conversion."""
         if not PLUGIN_AVAILABLE:
             self.skipTest("JSON plugin not available")
 
+        # Mock the dependencies
+        mock_model.return_value = Mock()
+        mock_tokenizer.return_value = Mock()
+        mock_from_transformers.return_value = Mock()
+            
         # Create JSONGenerator instance  
         generator = JSONGenerator()