llm_loader.py

from langchain_openai import ChatOpenAI

def load_model(openai_api_key):
    return ChatOpenAI(
        model_name="gpt-4o-mini",
        openai_api_key=openai_api_key,
        temperature=0.01,
        max_tokens=4096,
        top_p=0.9
    )