def get_openrouter_usage_and_credits()

in generate/utils/open_router_usage.py [0:0]

• 21 lines of code
• 6 McCabe index (conditional complexity)

def get_openrouter_usage_and_credits():
    # Get API key from environment variables
    api_key = os.getenv('OPENROUTER_API_KEY')
    
    if not api_key:
        raise ValueError("OPENROUTER_API_KEY not found in environment variables")

    # API endpoints
    usage_url = "https://openrouter.ai/api/v1/auth/key"
    credits_url = "https://openrouter.ai/api/v1/credits"

    # Headers required for OpenRouter API
    headers = {
        "Authorization": f"Bearer {api_key}",
    }
    
    try:
        # Fetch usage data
        usage_response = requests.get(usage_url, headers=headers)
        usage_response.raise_for_status()  # Raise an exception for bad status codes
        usage_data = usage_response.json()

        # Fetch credits data
        credits_response = requests.get(credits_url, headers=headers)
        print(credits_response.json())
        credits_response.raise_for_status()  # Raise an exception for bad status codes
        credits_data = credits_response.json()

        return usage_data, credits_data
    except requests.exceptions.RequestException as e:
        print(f"Error fetching data: {e}")
        return None, None