INNER CODE UNIT · Python
_build_backends
F-R-L/forge-film · forge/cli.py:149
def _build_backends(backend_override: str, cfg) -> dict:
"""Build the backends dict based on override or forge.yaml routing needs."""
from forge.generation.mock_pipeline import MockPipeline
mock = MockPipeline()
backends = {"mock": mock}
# Determine which backends are actually needed
needed = set(cfg.routing.values())
if backend_override:
needed = {backend_override}
if "kling_light" in needed or "kling_heavy" in needed or backend_override == "kling":
try:
from forge.generation.light_pipeline import LightPipeline
from forge.generation.heavy_pipeline import HeavyPipeline
backends["kling_light"] = LightPipeline()
backends["kling_heavy"] = HeavyPipeline()
except Exception as e: