DeepakJaiz/Extracting_term_defination

1import os2from langchain import OpenAI3from langchain.chat_models import ChatOpenAI4 5 6def get_llm(llm_name, model_temperature, api_key, max_tokens=256):7    os.environ["OPENAI_API_KEY"] = api_key8    if llm_name == "text-davinci-003":9        return OpenAI(10            temperature=model_temperature, model_name=llm_name, max_tokens=max_tokens11        )12    else:13        return ChatOpenAI(14            temperature=model_temperature, model_name=llm_name, max_tokens=max_tokens15        )