pydantic models refactor + add to autodoc + fixes

tests/patched/test_validation.py

@@ -11,10 +11,10 @@ from pydantic import ValidationError
 
 from axolotl.utils import is_comet_available
 from axolotl.utils.config import validate_config
-from axolotl.utils.config.models.input.v0_4_1 import AxolotlConfigWCapabilities
 from axolotl.utils.dict import DictDefault
 from axolotl.utils.mlflow_ import setup_mlflow_env_vars
 from axolotl.utils.models import check_model_config
+from axolotl.utils.schemas.config import AxolotlConfigWCapabilities
 from axolotl.utils.wandb_ import setup_wandb_env_vars
 
 warnings.filterwarnings("error")

tests/test_validation_dataset.py

@@ -6,8 +6,8 @@ from typing import Optional
 import pytest
 
 from axolotl.utils.config import validate_config
-from axolotl.utils.config.models.input.v0_4_1 import ChatTemplate
 from axolotl.utils.dict import DictDefault
+from axolotl.utils.schemas.datasets import ChatTemplate
 
 warnings.filterwarnings("error")
 

tests/utils/test_models.py

@@ -119,7 +119,7 @@ class TestModelsUtils:
 
     def test_message_property_mapping(self):
         """Test message property mapping configuration validation"""
-        from axolotl.utils.config.models.input.v0_4_1 import SFTDataset
+        from axolotl.utils.schemas.datasets import SFTDataset
 
         # Test legacy fields are mapped orrectly
         dataset = SFTDataset(