repositories.py

from dataclasses import dataclass
from typing import Self

from ai.backend.manager.repositories.agent.repositories import AgentRepositories
from ai.backend.manager.repositories.app_config.repositories import AppConfigRepositories
from ai.backend.manager.repositories.artifact.repositories import ArtifactRepositories
from ai.backend.manager.repositories.artifact_registry.repositories import (
    ArtifactRegistryRepositories,
)
from ai.backend.manager.repositories.audit_log.repositories import AuditLogRepositories
from ai.backend.manager.repositories.auth.repositories import AuthRepositories
from ai.backend.manager.repositories.container_registry.repositories import (
    ContainerRegistryRepositories,
)
from ai.backend.manager.repositories.deployment.repositories import DeploymentRepositories
from ai.backend.manager.repositories.deployment_revision_preset.repositories import (
    DeploymentRevisionPresetRepositories,
)
from ai.backend.manager.repositories.domain.repositories import DomainRepositories
from ai.backend.manager.repositories.dotfile.repositories import DotfileRepositories
from ai.backend.manager.repositories.error_log.repositories import ErrorLogRepositories
from ai.backend.manager.repositories.etcd_config.repositories import EtcdConfigRepositories
from ai.backend.manager.repositories.events.repositories import EventsRepositories
from ai.backend.manager.repositories.export.repositories import ExportRepositories
from ai.backend.manager.repositories.fair_share.repositories import FairShareRepositories
from ai.backend.manager.repositories.group.repositories import GroupRepositories
from ai.backend.manager.repositories.huggingface_registry.repositories import (
    HuggingFaceRegistryRepositories,
)
from ai.backend.manager.repositories.image.repositories import ImageRepositories
from ai.backend.manager.repositories.keypair_resource_policy.repositories import (
    KeypairResourcePolicyRepositories,
)
from ai.backend.manager.repositories.manager_admin.repositories import ManagerAdminRepositories
from ai.backend.manager.repositories.metric.repositories import MetricRepositories
from ai.backend.manager.repositories.model_card.repositories import ModelCardRepositories
from ai.backend.manager.repositories.model_serving.repositories import ModelServingRepositories
from ai.backend.manager.repositories.notification.repositories import NotificationRepositories
from ai.backend.manager.repositories.object_storage.repositories import ObjectStorageRepositories
from ai.backend.manager.repositories.permission_controller.repositories import (
    PermissionControllerRepositories,
)
from ai.backend.manager.repositories.project_resource_policy.repositories import (
    ProjectResourcePolicyRepositories,
)
from ai.backend.manager.repositories.prometheus_query_preset import (
    PrometheusQueryPresetRepositories,
)
from ai.backend.manager.repositories.prometheus_query_preset_category.repositories import (
    PrometheusQueryPresetCategoryRepositories,
)
from ai.backend.manager.repositories.reservoir_registry.repositories import (
    ReservoirRegistryRepositories,
)
from ai.backend.manager.repositories.resource_preset.repositories import ResourcePresetRepositories
from ai.backend.manager.repositories.resource_slot.repositories import ResourceSlotRepositories
from ai.backend.manager.repositories.resource_usage_history.repositories import (
    ResourceUsageHistoryRepositories,
)
from ai.backend.manager.repositories.role_preset.repositories import RolePresetRepositories
from ai.backend.manager.repositories.runtime_variant.repositories import RuntimeVariantRepositories
from ai.backend.manager.repositories.runtime_variant_preset.repositories import (
    RuntimeVariantPresetRepositories,
)
from ai.backend.manager.repositories.scaling_group.repositories import ScalingGroupRepositories
from ai.backend.manager.repositories.scheduler.repositories import SchedulerRepositories
from ai.backend.manager.repositories.scheduling_history.repositories import (
    SchedulingHistoryRepositories,
)
from ai.backend.manager.repositories.session.repositories import SessionRepositories
from ai.backend.manager.repositories.storage_namespace.repositories import (
    StorageNamespaceRepositories,
)
from ai.backend.manager.repositories.stream.repositories import StreamRepositories
from ai.backend.manager.repositories.template.repositories import TemplateRepositories
from ai.backend.manager.repositories.types import RepositoryArgs
from ai.backend.manager.repositories.user.repositories import UserRepositories
from ai.backend.manager.repositories.user_resource_policy.repositories import (
    UserResourcePolicyRepositories,
)
from ai.backend.manager.repositories.vfolder.repositories import VfolderRepositories
from ai.backend.manager.repositories.vfs_storage.repositories import VFSStorageRepositories


@dataclass
class Repositories:
    agent: AgentRepositories
    app_config: AppConfigRepositories
    auth: AuthRepositories
    container_registry: ContainerRegistryRepositories
    deployment: DeploymentRepositories
    domain: DomainRepositories
    dotfile: DotfileRepositories
    error_log: ErrorLogRepositories
    etcd_config: EtcdConfigRepositories
    export: ExportRepositories
    fair_share: FairShareRepositories
    group: GroupRepositories
    image: ImageRepositories
    keypair_resource_policy: KeypairResourcePolicyRepositories
    manager_admin: ManagerAdminRepositories
    metric: MetricRepositories
    model_serving: ModelServingRepositories
    notification: NotificationRepositories
    permission_controller: PermissionControllerRepositories
    project_resource_policy: ProjectResourcePolicyRepositories
    prometheus_query_preset: PrometheusQueryPresetRepositories
    prometheus_query_preset_category: PrometheusQueryPresetCategoryRepositories
    reservoir_registry: ReservoirRegistryRepositories
    resource_preset: ResourcePresetRepositories
    resource_slot: ResourceSlotRepositories
    role_preset: RolePresetRepositories
    runtime_variant: RuntimeVariantRepositories
    runtime_variant_preset: RuntimeVariantPresetRepositories
    deployment_revision_preset: DeploymentRevisionPresetRepositories
    model_card: ModelCardRepositories
    resource_usage_history: ResourceUsageHistoryRepositories
    scaling_group: ScalingGroupRepositories
    scheduler: SchedulerRepositories
    scheduling_history: SchedulingHistoryRepositories
    session: SessionRepositories
    template: TemplateRepositories
    user: UserRepositories
    user_resource_policy: UserResourcePolicyRepositories
    vfolder: VfolderRepositories
    object_storage: ObjectStorageRepositories
    vfs_storage: VFSStorageRepositories
    huggingface_registry: HuggingFaceRegistryRepositories
    artifact: ArtifactRepositories
    artifact_registry: ArtifactRegistryRepositories
    stream: StreamRepositories
    events: EventsRepositories
    storage_namespace: StorageNamespaceRepositories
    audit_log: AuditLogRepositories

    @classmethod
    def create(cls, args: RepositoryArgs) -> Self:
        agent_repositories = AgentRepositories.create(args)
        app_config_repositories = AppConfigRepositories.create(args)
        auth_repositories = AuthRepositories.create(args)
        container_registry_repositories = ContainerRegistryRepositories.create(args)
        deployment_repositories = DeploymentRepositories.create(args)
        domain_repositories = DomainRepositories.create(args)
        dotfile_repositories = DotfileRepositories.create(args)
        error_log_repositories = ErrorLogRepositories.create(args)
        etcd_config_repositories = EtcdConfigRepositories.create(args)
        export_repositories = ExportRepositories.create(args)
        fair_share_repositories = FairShareRepositories.create(args)
        group_repositories = GroupRepositories.create(args)
        image_repositories = ImageRepositories.create(args)
        keypair_resource_policy_repositories = KeypairResourcePolicyRepositories.create(args)
        manager_admin_repositories = ManagerAdminRepositories.create(args)
        metric_repositories = MetricRepositories.create(args)
        model_serving_repositories = ModelServingRepositories.create(args)
        notification_repositories = NotificationRepositories.create(args)
        permission_controller_repositories = PermissionControllerRepositories.create(args)
        project_resource_policy_repositories = ProjectResourcePolicyRepositories.create(args)
        prometheus_query_preset_repositories = PrometheusQueryPresetRepositories.create(args)
        prometheus_query_preset_category_repositories = (
            PrometheusQueryPresetCategoryRepositories.create(args)
        )
        reservoir_registry_repositories = ReservoirRegistryRepositories.create(args)
        resource_preset_repositories = ResourcePresetRepositories.create(args)
        resource_slot_repositories = ResourceSlotRepositories.create(args)
        role_preset_repositories = RolePresetRepositories.create(args)
        runtime_variant_repositories = RuntimeVariantRepositories.create(args)
        runtime_variant_preset_repositories = RuntimeVariantPresetRepositories.create(args)
        deployment_revision_preset_repositories = DeploymentRevisionPresetRepositories.create(args)
        model_card_repositories = ModelCardRepositories.create(args)
        resource_usage_history_repositories = ResourceUsageHistoryRepositories.create(args)
        scaling_group_repositories = ScalingGroupRepositories.create(args)
        scheduler_repositories = SchedulerRepositories.create(args)
        scheduling_history_repositories = SchedulingHistoryRepositories.create(args)
        session_repositories = SessionRepositories.create(args)
        template_repositories = TemplateRepositories.create(args)
        user_repositories = UserRepositories.create(args)
        user_resource_policy_repositories = UserResourcePolicyRepositories.create(args)
        vfolder_repositories = VfolderRepositories.create(args)
        object_storage_repositories = ObjectStorageRepositories.create(args)
        vfs_storage_repositories = VFSStorageRepositories.create(args)
        artifact_repositories = ArtifactRepositories.create(args)
        huggingface_registry_repositories = HuggingFaceRegistryRepositories.create(args)
        artifact_registries = ArtifactRegistryRepositories.create(args)
        stream_repositories = StreamRepositories.create(args)
        events_repositories = EventsRepositories.create(args)
        storage_namespace_repositories = StorageNamespaceRepositories.create(args)
        audit_log_repositories = AuditLogRepositories.create(args)

        return cls(
            agent=agent_repositories,
            app_config=app_config_repositories,
            auth=auth_repositories,
            container_registry=container_registry_repositories,
            deployment=deployment_repositories,
            domain=domain_repositories,
            dotfile=dotfile_repositories,
            error_log=error_log_repositories,
            etcd_config=etcd_config_repositories,
            export=export_repositories,
            fair_share=fair_share_repositories,
            group=group_repositories,
            image=image_repositories,
            keypair_resource_policy=keypair_resource_policy_repositories,
            manager_admin=manager_admin_repositories,
            metric=metric_repositories,
            model_serving=model_serving_repositories,
            notification=notification_repositories,
            permission_controller=permission_controller_repositories,
            project_resource_policy=project_resource_policy_repositories,
            prometheus_query_preset=prometheus_query_preset_repositories,
            prometheus_query_preset_category=prometheus_query_preset_category_repositories,
            reservoir_registry=reservoir_registry_repositories,
            resource_preset=resource_preset_repositories,
            resource_slot=resource_slot_repositories,
            role_preset=role_preset_repositories,
            runtime_variant=runtime_variant_repositories,
            runtime_variant_preset=runtime_variant_preset_repositories,
            deployment_revision_preset=deployment_revision_preset_repositories,
            model_card=model_card_repositories,
            resource_usage_history=resource_usage_history_repositories,
            scaling_group=scaling_group_repositories,
            scheduler=scheduler_repositories,
            scheduling_history=scheduling_history_repositories,
            session=session_repositories,
            template=template_repositories,
            user=user_repositories,
            user_resource_policy=user_resource_policy_repositories,
            vfolder=vfolder_repositories,
            object_storage=object_storage_repositories,
            vfs_storage=vfs_storage_repositories,
            huggingface_registry=huggingface_registry_repositories,
            artifact=artifact_repositories,
            artifact_registry=artifact_registries,
            stream=stream_repositories,
            events=events_repositories,
            storage_namespace=storage_namespace_repositories,
            audit_log=audit_log_repositories,
        )