from langchain.chat_models import ChatOpenAI
from langchain.base_language import BaseLanguageModel

def llm(temperature=0) -> BaseLanguageModel:
    # gpt-3.5
    return ChatOpenAI(temperature=temperature)
    # gpt-4
    # return ChatOpenAI(temperature=temperature, model_name="gpt-4")