diff --git a/litellm/llms/bedrock/batches/handler.py b/litellm/llms/bedrock/batches/handler.py index fd4c3dc1659..5931302ac14 100644 --- a/litellm/llms/bedrock/batches/handler.py +++ b/litellm/llms/bedrock/batches/handler.py @@ -20,7 +20,7 @@ if TYPE_CHECKING: _BEDROCK_MIJ_STATUS_TO_OPENAI: Final = { "Submitted": "validating", "Validating": "validating", - "Scheduled": "validating", + "Scheduled": "in_progress", "InProgress": "in_progress", "Stopping": "cancelling", "Stopped": "cancelled", diff --git a/tests/unit/llms/bedrock/batches/test_handler.py b/tests/unit/llms/bedrock/batches/test_handler.py index e69098a460d..eec0cf070cd 100644 --- a/tests/unit/llms/bedrock/batches/test_handler.py +++ b/tests/unit/llms/bedrock/batches/test_handler.py @@ -217,7 +217,7 @@ def test_missing_error_count_maps_to_zero_failed(patched_boto3): [ ("Submitted", "validating"), ("Validating", "validating"), - ("Scheduled", "validating"), + ("Scheduled", "in_progress"), ("InProgress", "in_progress"), ("Stopping", "cancelling"), ("Stopped", "cancelled"),