elif self.api_key is not API_KEY_SENTINEL:
    if headers.get("api-key") is None:
        headers["api-key"] = self.api_key  # ← Uses stale/empty value!

#!/usr/bin/env python3
"""Test script to verify callable api_key functionality in OpenAI and Azure OpenAI clients"""

import os
from openai import OpenAI, AsyncOpenAI
from openai.lib.azure import AzureOpenAI, AsyncAzureOpenAI

def test_openai_callable():
    """Test callable api_key with regular OpenAI client"""
    
    call_count = 0
    
    def get_api_key():
        nonlocal call_count
        call_count += 1
        print(f"OpenAI: get_api_key called (call #{call_count})")
        return os.environ.get("OPENAI_API_KEY", "test-key")
    
    client = OpenAI(api_key=get_api_key)
    
    # Check that callable is stored properly
    print(f"OpenAI client api_key: '{client.api_key}'")
    print(f"OpenAI client _api_key_provider: {client._api_key_provider}")
    
    # Try to make a request (will fail without real key, but we can see if callable is invoked)
    try:
        # This should trigger the callable
        client.models.list()
    except Exception as e:
        print(f"OpenAI request failed (expected): {e}")
    
    print(f"OpenAI: Total calls to get_api_key: {call_count}")
    print()

def test_azure_callable():
    """Test callable api_key with Azure OpenAI client"""
    
    call_count = 0
    
    def get_api_key():
        nonlocal call_count
        call_count += 1
        print(f"Azure: get_api_key called (call #{call_count})")
        return os.environ.get("AZURE_OPENAI_API_KEY", "test-key")
    
    client = AzureOpenAI(
        api_key=get_api_key,
        api_version="2024-02-01",
        azure_endpoint="https://test.openai.azure.com"
    )
    
    # Check that callable is stored properly  
    print(f"Azure client api_key: '{client.api_key}'")
    # Azure client inherits from OpenAI, so it should have _api_key_provider
    print(f"Azure client _api_key_provider: {getattr(client, '_api_key_provider', 'NOT FOUND')}")
    
    # Try to make a request (will fail without real key, but we can see if callable is invoked)
    try:
        # This should trigger the callable
        client.models.list()
    except Exception as e:
        print(f"Azure request failed (expected): {e}")
    
    print(f"Azure: Total calls to get_api_key: {call_count}")
    print()

def test_azure_prepare_options():
    """Test to see what happens in _prepare_options for Azure"""
    
    def get_api_key():
        print("Azure _prepare_options test: get_api_key called")
        return "dynamic-key-12345"
    
    client = AzureOpenAI(
        api_key=get_api_key,
        api_version="2024-02-01", 
        azure_endpoint="https://test.openai.azure.com"
    )
    
    print(f"Initial api_key value: '{client.api_key}'")
    
    # Simulate what happens when a request is made
    # The OpenAI parent class should call _refresh_api_key in its _prepare_options
    # But Azure overrides _prepare_options and doesn't call the parent
    
    from openai._models import FinalRequestOptions
    options = FinalRequestOptions(
        method="GET",
        url="/models",
    )
    
    # This should call Azure's _prepare_options
    prepared = client._prepare_options(options)
    
    print(f"After _prepare_options, api_key value: '{client.api_key}'")
    print()

if __name__ == "__main__":
    print("Testing Callable API Key Support\n")
    print("=" * 50)
    
    test_openai_callable()
    test_azure_callable()
    test_azure_prepare_options()