from .openai_llm import OpenAILLM

def get_llm(engine):
    if engine == "openai":
        return OpenAILLM()
    raise ValueError("Unsupported LLM engine")