48 lines
2.0 KiB
Python
48 lines
2.0 KiB
Python
import json
|
|
from pathlib import Path
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
|
|
class ModelConfig:
|
|
"""Класс для работы с конфигурацией моделей AI провайдеров"""
|
|
|
|
def __init__(self, config_path: str = "config/models.json"):
|
|
self.config_path = Path(config_path)
|
|
self.config = self._load_config()
|
|
|
|
def _load_config(self) -> Dict[str, Any]:
|
|
"""Загрузить конфигурацию из JSON файла"""
|
|
if not self.config_path.exists():
|
|
raise FileNotFoundError(f"Config file not found: {self.config_path}")
|
|
|
|
with open(self.config_path, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
|
|
def get_available_models(self, provider: str) -> List[Dict[str, Any]]:
|
|
"""Получить список доступных моделей для провайдера"""
|
|
if provider not in self.config:
|
|
raise ValueError(f"Unknown provider: {provider}")
|
|
return self.config[provider]["available_models"]
|
|
|
|
def get_default_model(self, provider: str) -> str:
|
|
"""Получить модель по умолчанию для провайдера"""
|
|
if provider not in self.config:
|
|
raise ValueError(f"Unknown provider: {provider}")
|
|
return self.config[provider]["default_model"]
|
|
|
|
def get_model_info(self, provider: str, model_id: str) -> Optional[Dict[str, Any]]:
|
|
"""Получить информацию о конкретной модели"""
|
|
models = self.get_available_models(provider)
|
|
for model in models:
|
|
if model["id"] == model_id:
|
|
return model
|
|
return None
|
|
|
|
def validate_model(self, provider: str, model_id: str) -> bool:
|
|
"""Проверить, существует ли модель для провайдера"""
|
|
return self.get_model_info(provider, model_id) is not None
|
|
|
|
|
|
# Singleton instance
|
|
model_config = ModelConfig()
|