scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

fix(ci): image variation openai sdk 2.24.0 compat + swap bedrock nova-premier to nova-pro (#23066)

Browse source 
* fix(ci): fix image variation test for openai sdk 2.24.0 and swap nova-premier to nova-pro

image_gen_tests: openai==2.24.0 (bumped Feb 25) requires BytesIO objects to have
a .name attribute for MIME type detection in multipart uploads. Add .name to the
fixture so create_variation works. Also guard with OPENAI_API_KEY skipif.

proxy_e2e_anthropic_messages_tests: nova-premier requires provisioned throughput
not available via standard on-demand cross-region inference on the CI account.
Swap to nova-pro which uses standard inference profiles.

* fix: remove skipif, keep only .name fix for openai sdk compat
This commit is contained in:
Ishaan Jaff 2026-03-07 16:41:54 -08:00 • committed by GitHub
parent e8a7116899
commit 66c822435e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
tests/image_gen_tests/test_image_variation.py
View file

@ -45,6 +45,7 @@ def image_url():
# Load the image into a file-like object # Load the image into a file-like object
image_file = BytesIO(response.content) image_file = BytesIO(response.content)
image_file.name = "litellm_logo.png"
return image_file return image_file

10
tests/proxy_e2e_anthropic_messages_tests/test_claude_agent_sdk.py
View file

@ -4,7 +4,7 @@ E2E tests for Claude Agent SDK with LiteLLM Proxy using Bedrock models.
Tests streaming messages across different Bedrock models: Tests streaming messages across different Bedrock models:
- Regular Bedrock Claude Sonnet 4.5 - Regular Bedrock Claude Sonnet 4.5
- Bedrock Converse Claude Sonnet 4.5 - Bedrock Converse Claude Sonnet 4.5
- AWS Nova Premier - AWS Nova Pro
""" """
import os import os
@ -14,14 +14,14 @@ from claude_agent_sdk import ClaudeSDKClient, ClaudeAgentOptions
# Test models from test_config.yaml # Test models from test_config.yaml
# Note: bedrock-converse-claude-sonnet-4.5 removed temporarily as the Bedrock Converse API # Note: bedrock-converse-claude-sonnet-4.5 removed temporarily as the Bedrock Converse API
# for Claude Sonnet 4.5 may not be available in all regions/accounts # for Claude Sonnet 4.5 may not be available in all regions/accounts
# Note: bedrock-nova-premier requires an inference profile for on-demand throughput # Note: bedrock-nova-premier requires provisioned throughput (not standard cross-region
# https://docs.aws.amazon.com/bedrock/latest/userguide/inference-profiles.html # inference profile) and is not reliably available in CI accounts. Using nova-pro instead.
TEST_MODELS = [ TEST_MODELS = [
("bedrock-claude-sonnet-4.5", "Bedrock Invoke API"), ("bedrock-claude-sonnet-4.5", "Bedrock Invoke API"),
("bedrock-converse-claude-sonnet-4.5", "Bedrock Converse API"), ("bedrock-converse-claude-sonnet-4.5", "Bedrock Converse API"),
("bedrock-nova-premier", "AWS Nova Premier"), ("bedrock-nova-pro", "AWS Nova Pro"),
] ]

4
tests/proxy_e2e_anthropic_messages_tests/test_config.yaml
View file

@ -24,9 +24,9 @@ model_list:
model: "bedrock/us.anthropic.claude-opus-4-5-20251101-v1:0" model: "bedrock/us.anthropic.claude-opus-4-5-20251101-v1:0"
aws_region_name: "us-east-1" aws_region_name: "us-east-1"
- model_name: bedrock-nova-premier - model_name: bedrock-nova-pro
litellm_params: litellm_params:
model: "bedrock/us.amazon.nova-premier-v1:0" model: "bedrock/us.amazon.nova-pro-v1:0"
aws_region_name: "us-east-1" aws_region_name: "us-east-1"
# Converse API models # Converse API models