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