server/interpreter_config.py

"""Конфигурация Open Interpreter для моделей"""
import logging
from typing import Optional

def configure_interpreter_for_model(model_name: str, state, api_key: str, api_base: str) -> bool:
    """
    Настраивает Open Interpreter для работы с заданной моделью.
    Возвращает True при успешной настройке, False в случае ошибки.
    """
    try:
        from interpreter import interpreter
        model = state.models.get(model_name, state.models["deepseek-v4-pro"])
        interpreter.llm.model = f"openai/{model.endpoint}"
        interpreter.llm.api_key = api_key
        interpreter.llm.api_base = api_base
        interpreter.llm.context_window = model.context_window
        interpreter.llm.max_tokens = model.max_tokens
        logging.info(f"Open Interpreter configured for model: {model_name}")
        return True
    except ImportError as e:
        logging.error(f"Open Interpreter not available: {e}")
        return False
    except Exception as e:
        logging.error(f"Failed to configure interpreter for {model_name}: {e}")
        return False