from .base import BaseLanguageAgent
from .de import GermanAgent
from .en import EnglishAgent
from .es import SpanishAgent
from .fr import FrenchAgent
from .it import ItalianAgent
from .nl import DutchAgent
from .pt import PortugueseAgent
from .ru import RussianAgent

AGENT_REGISTRY = {
    "nl": DutchAgent,
    "en": EnglishAgent,
    "de": GermanAgent,
    "fr": FrenchAgent,
    "es": SpanishAgent,
    "it": ItalianAgent,
    "pt": PortugueseAgent,
    "ru": RussianAgent,
}


def get_language_agent(locale_code: str) -> BaseLanguageAgent:
    agent_class = AGENT_REGISTRY.get(locale_code, BaseLanguageAgent)
    return agent_class()