Работающий импорт с изменением БД

superset_tool/models.py

@@ -1,10 +1,21 @@
-from pydantic import BaseModel
+# models.py
+from pydantic import BaseModel, validator
+from typing import Optional
+from .utils.logger import SupersetLogger
 
 class SupersetConfig(BaseModel):
     base_url: str
-    auth: dict  # Словарь с параметрами аутентификации
+    auth: dict
     verify_ssl: bool = True
     timeout: int = 30
+    logger: Optional[SupersetLogger] = None
+
+    class Config:
+        arbitrary_types_allowed = True  # Разрешаем произвольные типы
 
 class DatabaseConfig(BaseModel):
-    database_config: dict
+    database_config: dict
+    logger: Optional[SupersetLogger] = None
+
+    class Config:
+        arbitrary_types_allowed = True