code: def test_image_generation_openai():
      try:
          litellm.set_verbose = True
          response = litellm.image_generation(
              prompt="A cute baby sea otter", model="dall-e-3"
          )
          print(f"response: {response}")
          assert len(response.data) > 0
      except litellm.RateLimitError as e:
          pass
      except litellm.ContentPolicyViolationError:
          pass  # OpenAI randomly raises these errors - skip when they occur
      except Exception as e:
          pytest.fail(f"An exception occurred - {str(e)}")
  
file_path: litellm/litellm/tests/test_image_generation.py
imports: import sys, os
  import traceback
  from dotenv import load_dotenv
  import logging
  import os
  import asyncio
  import pytest
  import litellm
label_desc: Use image node_id 1 for calling a global function with example usage: test_image_generation_openai() without return types