llm_loader.py

# llm_loader.py
from langchain.chat_models import ChatOpenAI

def load_model(openai_api_key):
    return ChatOpenAI(
        model_name="gpt-4o",
        openai_api_key=openai_api_key,
        temperature=0.01,
        max_tokens=2096,
        top_p=0.95,
        top_k=25
    )