"""Cloud provider for worker management."""

from dataclasses import dataclass


@dataclass
class Instance:
    id: str
    name: str
    ip: str
    status: str
    provider: str
    type: str


def get_provider(provider_name: str):
    if provider_name == "hetzner":
        from beanflows_pipeline.providers import hetzner

        return hetzner
    else:
        raise ValueError(f"Unknown provider: {provider_name}")
