model/init_model.py

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