src/chatbot/core/model.py

from common.utility.openai_model_factory  import OpenAIModelFactory

def get_model_client(provider:str = "openai"):
    if provider.lower() == "google":
        return OpenAIModelFactory.get_model(
            provider="google",
            model_name="gemini-2.5-flash",
            temperature=0
    )
    elif provider.lower() == "openai":
        return OpenAIModelFactory.get_model(
            provider="openai",
            model_name="gpt-4o-mini",
            temperature=0
    )
    elif provider.lower() == "azure":
        return OpenAIModelFactory.get_model(
            provider="azure",
            model_name="gpt-4o-mini",
            temperature=0
    )
    elif provider.lower() == "groq":
        return OpenAIModelFactory.get_model(
            provider="groq",
            model_name="gpt-4o-mini",
            temperature=0
    )
    elif provider.lower() == "ollama":
        return OpenAIModelFactory.get_model(
            provider="ollama",
            model_name="gpt-4o-mini",
            temperature=0
    )
    else: 
        raise  ValueError(f"Unsupported provider: {provider}")