"""
Module to initialize the Azure GPT model.
"""
import os
from langchain_openai import AzureChatOpenAI
from dotenv import load_dotenv

load_dotenv()

AZURE_OPENAI_API_KEY = os.getenv("AZURE_OPENAI_API_KEY")
AZURE_OPENAI_ENDPOINT = os.getenv("AZURE_OPENAI_ENDPOINT")

class GPTModel(AzureChatOpenAI):
    """
    GPTModel class that extends AzureChatOpenAI.

    This class initializes a GPT model with specific deployment settings and a callback function.

    Attributes:
        callback (function): The callback function to be used with the model.

    Methods:
        __init__(callback):
            Initializes the GPTModel with the specified callback function.
    """
    def __init__(self):
        super().__init__(
            deployment_name="gpt-4.1-mini",
            api_version="2024-12-01-preview",
            azure_endpoint='https://openai-oe.openai.azure.com/',
            api_key='b9135a15c242432cb20ddc43fea3a413',
            streaming=True,
            temperature=0
        )