mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-18 03:31:23 +00:00
Sameer Kankute
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
# What this tests?
|
|
## This tests the litellm support for the openai /generations endpoint
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
import traceback
|
|
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system path
|
|
|
|
from dotenv import load_dotenv
|
|
from openai.types.image import Image
|
|
from litellm.caching import InMemoryCache
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
load_dotenv()
|
|
import asyncio
|
|
import os
|
|
import pytest
|
|
|
|
import litellm
|
|
import json
|
|
import tempfile
|
|
from base_image_generation_test import BaseImageGenTest
|
|
import logging
|
|
from litellm._logging import verbose_logger
|
|
from io import BytesIO
|
|
from PIL import Image as PILImage
|
|
|
|
verbose_logger.setLevel(logging.DEBUG)
|
|
|
|
|
|
@pytest.fixture
|
|
def image_url():
|
|
# DALL-E 2 image variations require a square PNG (less than 4MB)
|
|
# Generate a 1024x1024 square PNG programmatically to avoid network dependency
|
|
# and the non-square aspect ratio of the old LiteLLM logo URL
|
|
img = PILImage.new("RGBA", (1024, 1024), color=(128, 128, 128, 255))
|
|
image_file = BytesIO()
|
|
img.save(image_file, format="PNG")
|
|
image_file.seek(0)
|
|
# openai>=2.24.0 requires BytesIO to have .name for MIME type detection in multipart uploads
|
|
image_file.name = "litellm_logo.png"
|
|
|
|
return image_file
|
|
|
|
|
|
# Commented out: OpenAI /images/variations endpoint deprecated (DALL-E 2 shutdown May 12, 2026)
|
|
# def test_openai_image_variation_openai_sdk(image_url):
|
|
# from openai import OpenAI
|
|
#
|
|
# client = OpenAI()
|
|
# response = client.images.create_variation(image=image_url, n=2, size="1024x1024")
|
|
# print(response)
|
|
#
|
|
#
|
|
# @pytest.mark.parametrize("sync_mode", [True, False])
|
|
# @pytest.mark.asyncio
|
|
# async def test_openai_image_variation_litellm_sdk(image_url, sync_mode):
|
|
# from litellm import image_variation, aimage_variation
|
|
#
|
|
# if sync_mode:
|
|
# image_variation(image=image_url, n=2, size="1024x1024")
|
|
# else:
|
|
# await aimage_variation(image=image_url, n=2, size="1024x1024")
|
|
#
|
|
#
|
|
# def test_topaz_image_variation(image_url):
|
|
# from litellm import image_variation, aimage_variation
|
|
# from litellm.llms.custom_httpx.http_handler import HTTPHandler
|
|
# from unittest.mock import patch
|
|
#
|
|
# client = HTTPHandler()
|
|
# with patch.object(client, "post") as mock_post:
|
|
# try:
|
|
# image_variation(
|
|
# model="topaz/Standard V2",
|
|
# image=image_url,
|
|
# n=2,
|
|
# size="1024x1024",
|
|
# client=client,
|
|
# )
|
|
# except Exception as e:
|
|
# print(e)
|
|
# mock_post.assert_called_once()
|
|
|
|
|
|
def test_image_variation_placeholder():
|
|
"""Placeholder: variation tests commented out - OpenAI /images/variations deprecated (DALL-E 2 shutdown May 12, 2026)."""
|
|
pass
|