dify/api/controllers/console/version.py

import logging

import httpx
from packaging import version
from pydantic import BaseModel, Field

from configs import dify_config
from controllers.fastopenapi import console_router

logger = logging.getLogger(__name__)


class VersionQuery(BaseModel):
    current_version: str = Field(..., description="Current application version")


class VersionFeatures(BaseModel):
    can_replace_logo: bool = Field(description="Whether logo replacement is supported")
    model_load_balancing_enabled: bool = Field(description="Whether model load balancing is enabled")


class VersionResponse(BaseModel):
    version: str = Field(description="Latest version number")
    release_date: str = Field(description="Release date of latest version")
    release_notes: str = Field(description="Release notes for latest version")
    can_auto_update: bool = Field(description="Whether auto-update is supported")
    features: VersionFeatures = Field(description="Feature flags and capabilities")


@console_router.get(
    "/version",
    response_model=VersionResponse,
    tags=["console"],
)
def check_version_update(query: VersionQuery) -> VersionResponse:
    """Check for application version updates."""
    check_update_url = dify_config.CHECK_UPDATE_URL

    result = VersionResponse(
        version=dify_config.project.version,
        release_date="",
        release_notes="",
        can_auto_update=False,
        features=VersionFeatures(
            can_replace_logo=dify_config.CAN_REPLACE_LOGO,
            model_load_balancing_enabled=dify_config.MODEL_LB_ENABLED,
        ),
    )

    if not check_update_url:
        return result

    try:
        response = httpx.get(
            check_update_url,
            params={"current_version": query.current_version},
            timeout=httpx.Timeout(timeout=10.0, connect=3.0),
        )
        content = response.json()
    except Exception as error:
        logger.warning("Check update version error: %s.", str(error))
        result.version = query.current_version
        return result
    latest_version = content.get("version", result.version)
    if _has_new_version(latest_version=latest_version, current_version=f"{query.current_version}"):
        result.version = latest_version
        result.release_date = content.get("releaseDate", "")
        result.release_notes = content.get("releaseNotes", "")
        result.can_auto_update = content.get("canAutoUpdate", False)
    return result


def _has_new_version(*, latest_version: str, current_version: str) -> bool:
    try:
        latest = version.parse(latest_version)
        current = version.parse(current_version)

        # Compare versions
        return latest > current
    except version.InvalidVersion:
        logger.warning("Invalid version format: latest=%s, current=%s", latest_version, current_version)
        return False