llm-proxy / app /factory.py
knighter75's picture
Fix actual models
ea89131
Raw
History Blame Contribute Delete
422 Bytes
from .providers.hf_openai import HFOpenAIProvider, configured_models
class ProviderFactory:
_instance = None
@classmethod
def get_provider(cls, model_name: str) -> HFOpenAIProvider:
if model_name not in configured_models():
raise ValueError(f"Unsupported model: {model_name}")
if cls._instance is None:
cls._instance = HFOpenAIProvider()
return cls._instance