api/llms/base.py

from api.llms.llm_config import LLMConfig
import api.llms.gpt4o_mini as gpt4o

def get_LLM(model_name) -> LLMConfig:
    """
        Hàm điều hướng trung gian xử lý tác vụ tạo LLM, config và prompt template liên quan
    """
    if model_name == "gpt4o_mini":
        return gpt4o.GPT4oMini()
    else:
        raise NotImplementedError("This type of model is not yet supported")