def get_gpt4_model()

in evaluate.py [0:0]

• 15 lines of code
• 3 McCabe index (conditional complexity)

def get_gpt4_model():
    """Get the GPT-4 model."""
    # Specify configuration for the AI Dial endpoint
    openai_endpoint = "https://ai-proxy.lab.epam.com"
    openai_deploymet_name = "gpt-4o-2024-05-13"
    openai_api_version = "2024-05-01-preview"

    # Read API key from the environment variables
    azure_api_key = os.environ["AZURE_API_KEY"]
    if not azure_api_key:
        raise ValueError("AZURE_API_KEY environment variable is not set")

    # Define GPT-4-omni model
    model = AzureChatOpenAI(
        temperature=0,  # request deterministic behavior
        azure_endpoint=openai_endpoint,
        azure_deployment=openai_deploymet_name,
        api_version=openai_api_version,
        api_key=azure_api_key,
    )

    return model