dify/api/services/feature_query_service.py

"""Application service for feature queries exposed by API adapters."""

from typing import Protocol

from machinery.context import RequestContext
from services.entities.feature_entities import (
    FeatureModel,
    LicenseModel,
    SystemFeatureModel,
    VectorSpaceLimitationModel,
)


class FeatureQueryGateway(Protocol):
    """Read dynamic feature resources without exposing their current implementation."""

    def get_workspace_features(self, workspace_id: str) -> FeatureModel: ...

    def get_trial_models(self, workspace_id: str) -> list[str]: ...

    def get_vector_space(self, workspace_id: str) -> VectorSpaceLimitationModel: ...

    def get_public_system_features(self) -> SystemFeatureModel: ...

    def get_license(self) -> LicenseModel: ...


class FeatureQueryService:
    def __init__(
        self,
        *,
        features: FeatureQueryGateway,
        app_dsl_version: str,
    ) -> None:
        self._features = features
        self._app_dsl_version = app_dsl_version

    def get_features(self, context: RequestContext) -> FeatureModel:
        return self._features.get_workspace_features(self._require_active_workspace(context))

    def get_vector_space(self, context: RequestContext) -> VectorSpaceLimitationModel:
        return self._features.get_vector_space(self._require_active_workspace(context))

    def get_trial_models(self, context: RequestContext) -> list[str]:
        return self._features.get_trial_models(self._require_active_workspace(context))

    def get_app_dsl_version(self) -> str:
        return self._app_dsl_version

    def get_system_features(self) -> SystemFeatureModel:
        return self._features.get_public_system_features()

    def get_license(self) -> LicenseModel:
        return self._features.get_license()

    @staticmethod
    def _require_active_workspace(context: RequestContext) -> str:
        workspace_id = context.active_workspace_id
        if workspace_id is None:
            raise RuntimeError("Console account admission did not resolve an active workspace")
        return workspace_id