from typing import Type

from ..config import get_settings
from .base import BaseAttributionService
from .service_anthropic import AnthropicService
from .service_openai import OpenAIService

settings = get_settings()


class AIServiceFactory:
    _services = {"openai": OpenAIService, "anthropic": AnthropicService}

    @classmethod
    def get_service(cls, ai_vendor: str = None) -> BaseAttributionService:
        ai_vendor = ai_vendor or settings.DEFAULT_VENDOR
        service_class = cls._services.get(ai_vendor.lower())
        if not service_class:
            raise ValueError(f"Unsupported ai_vendor: {ai_vendor}")
        return service_class()