mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-13 23:11:40 +00:00
fix(ci): image variation openai sdk 2.24.0 compat + swap bedrock nova-premier to nova-pro (#23066)
* 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:
parent
e8a7116899
commit
66c822435e
3 changed files with 8 additions and 7 deletions
|
|
@ -45,6 +45,7 @@ def image_url():
|
|||
|
||||
# Load the image into a file-like object
|
||||
image_file = BytesIO(response.content)
|
||||
image_file.name = "litellm_logo.png"
|
||||
|
||||
return image_file
|
||||
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@ E2E tests for Claude Agent SDK with LiteLLM Proxy using Bedrock models.
|
|||
Tests streaming messages across different Bedrock models:
|
||||
- Regular Bedrock Claude Sonnet 4.5
|
||||
- Bedrock Converse Claude Sonnet 4.5
|
||||
- AWS Nova Premier
|
||||
- AWS Nova Pro
|
||||
"""
|
||||
|
||||
import os
|
||||
|
|
@ -14,14 +14,14 @@ from claude_agent_sdk import ClaudeSDKClient, ClaudeAgentOptions
|
|||
|
||||
|
||||
# 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
|
||||
# Note: bedrock-nova-premier requires an inference profile for on-demand throughput
|
||||
# https://docs.aws.amazon.com/bedrock/latest/userguide/inference-profiles.html
|
||||
# Note: bedrock-nova-premier requires provisioned throughput (not standard cross-region
|
||||
# inference profile) and is not reliably available in CI accounts. Using nova-pro instead.
|
||||
TEST_MODELS = [
|
||||
("bedrock-claude-sonnet-4.5", "Bedrock Invoke API"),
|
||||
("bedrock-converse-claude-sonnet-4.5", "Bedrock Converse API"),
|
||||
("bedrock-nova-premier", "AWS Nova Premier"),
|
||||
("bedrock-nova-pro", "AWS Nova Pro"),
|
||||
]
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -24,9 +24,9 @@ model_list:
|
|||
model: "bedrock/us.anthropic.claude-opus-4-5-20251101-v1:0"
|
||||
aws_region_name: "us-east-1"
|
||||
|
||||
- model_name: bedrock-nova-premier
|
||||
- model_name: bedrock-nova-pro
|
||||
litellm_params:
|
||||
model: "bedrock/us.amazon.nova-premier-v1:0"
|
||||
model: "bedrock/us.amazon.nova-pro-v1:0"
|
||||
aws_region_name: "us-east-1"
|
||||
|
||||
# Converse API models
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue