Update server/state.py

server/state.py

@@ -19,6 +19,29 @@ class PinkSkyState:
             cls._instance = super().__new__(cls)
         return cls._instance
 
+    def load_all(self):
+        """Загружает модели, роли и кондукторы из файлов."""
+        self._load_models()
+        self._load_roles()
+        self._load_conductors()
+        self._load_history()
+
+    def _load_models(self):
+        defaults = {
+            name: self._build_model_config(name, data)
+            for name, data in MODEL_RANKING.items()
+        }
+        if os.path.exists(MODELS_FILE):
+            try:
+                with open(MODELS_FILE, "r", encoding="utf-8") as f:
+                    custom = json.load(f)
+                for k, v in custom.items():
+                    if k not in defaults:
+                        defaults[k] = ModelConfig(**v)
+            except Exception as e:
+                self.logger.error(f"Ошибка загрузки models.json: {e}")
+        self.models = defaults
+        
     def __init__(self):
         if PinkSkyState._initialized:
             return
@@ -343,7 +366,37 @@ FORMAT (STRICT JSON):
                     self.build_history = data.get("build", [])
             except Exception as e:
                 print(f"⚠️ Ошибка загрузки истории: {e}")
-
+    def __init__(self):
+        if PinkSkyState._initialized:
+            return
+        PinkSkyState._initialized = True
+        self.logger = logging.getLogger(__name__)
+        self.models: Dict[str, ModelConfig] = {}
+        self.roles: Dict[str, Role] = {}
+        self.conductors: Dict[str, Conductor] = {}
+        self.current_mode: str = "chat"
+        self.current_conductor: str = "default"
+        self.current_role: str = "universal"
+        self.current_model: str = "deepseek-v4-pro"
+        self.chat_history: List[Dict[str, str]] = []
+        self.skill_history: List[Dict[str, str]] = []
+        self.build_history: List[Dict[str, str]] = []
+        self.build_context: Dict[str, Any] = {
+            "spec": "",
+            "agents": 3,
+            "models_tier": "tier1",
+            "skills_count": 2,
+            "files_count": 3,
+            "role": "universal",
+            "strategy": "parallel",
+            "use_interpreter": True,
+            "notifications": True,
+            "internet_access": True,
+        }
+        self.cancel_flag: bool = False
+        self.last_history_save: Optional[datetime] = None
+        self.load_all()  # Теперь метод load_all доступен
+        
     def save_roles(self):
         data = {k: {"name": v.name, "prompt": v.prompt, "description": v.description,
                     "preferred_models": v.preferred_models, "complexity": v.complexity, "tags": v.tags}