File size: 239 Bytes
a47e415 |
1 2 3 4 5 6 7 8 |
import os
from langchain_openai import ChatOpenAI
def init_main_model(llm_model_name: str):
openai_api_key = os.getenv("openai_api_key")
llm = ChatOpenAI(api_key=openai_api_key, model=llm_model_name, temperature=0)
return llm |