from __future__ import annotations

from app.modules.agent.orchestration.processes.v1.process import V1Process
from app.modules.agent.orchestration.processes.v2.process import V2Process


class ProcessRegistry:
    def __init__(self, v1_process: V1Process, v2_process: V2Process | None = None) -> None:
        self._items = {"v1": v1_process}
        if v2_process is not None:
            self._items["v2"] = v2_process

    def get(self, version: str):
        return self._items.get(version)
