| 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") | |