Function init_server

src/memos/api/handlers/component_init.py:108–341

Initialize all server components and configurations. This function orchestrates the creation and initialization of all components required by the MemOS server, including: - Database connections (graph DB, vector DB) - Language models and embedders - Memory systems (text)

()

def init_server() -> dict[str, Any]:
    """
    Initialize all server components and configurations.

    This function orchestrates the creation and initialization of all components
    required by the MemOS server, including:
    - Database connections (graph DB, vector DB)
    - Language models and embedders
    - Memory systems (text)
    - Scheduler and related modules

    Returns:
        A dictionary containing all initialized components with descriptive keys.
        This approach allows easy addition of new components without breaking
        existing code that uses the components.
    """
    logger.info("Initializing MemOS server components...")
    logger.info(
        "[INIT_SERVER] env_MEMSCHEDULER_STREAM_KEY_PREFIX=%s, env_MEMSCHEDULER_REDIS_STREAM_KEY_PREFIX=%s, env_POLAR_DB_DB_NAME=%s",
        os.getenv("MEMSCHEDULER_STREAM_KEY_PREFIX"),
        os.getenv("MEMSCHEDULER_REDIS_STREAM_KEY_PREFIX"),
        os.getenv("POLAR_DB_DB_NAME"),
    )

    # Initialize Redis client first as it is a core dependency for features like scheduler status tracking
    if os.getenv("MEMSCHEDULER_USE_REDIS_QUEUE", "False").lower() == "true":
        try:
            from memos.mem_scheduler.orm_modules.api_redis_model import APIRedisDBManager

            redis_client = APIRedisDBManager.load_redis_engine_from_env()
            if redis_client:
                logger.info("Redis client initialized successfully.")
            else:
                logger.error(
                    "Failed to initialize Redis client. Check REDIS_HOST etc. in environment variables."
                )
        except Exception as e:
            logger.error(f"Failed to initialize Redis client: {e}", exc_info=True)
            redis_client = None  # Ensure redis_client exists even on failure
    else:
        redis_client = None

    # Get default cube configuration
    default_cube_config = APIConfig.get_default_cube_config()

    # Get online bot setting
    dingding_enabled = APIConfig.is_dingding_bot_enabled()

    # Build component configurations
    graph_db_config = build_graph_db_config()
    llm_config = build_llm_config()
    chat_llm_config = build_chat_llm_config()
    playground_chat_llm_config = build_chat_llm_config("PLAYGROUND_CHAT_MODEL_LIST")
    embedder_config = build_embedder_config()
    nli_client_config = build_nli_client_config()
    mem_reader_config = build_mem_reader_config()
    reranker_config = build_reranker_config()
    feedback_reranker_config = build_feedback_reranker_config()