Fix all 130126 tests
This commit is contained in:
parent
d03548b064
commit
7dbf09cb12
@ -3350,7 +3350,8 @@ async def test_bedrock_converse__streaming_passthrough(monkeypatch):
|
||||
mock_callback.assert_called_once()
|
||||
print(mock_callback.call_args.kwargs.keys())
|
||||
assert "response_cost" in mock_callback.call_args.kwargs["kwargs"]
|
||||
assert mock_callback.call_args.kwargs["kwargs"]["response_cost"] > 0
|
||||
response_cost = mock_callback.call_args.kwargs["kwargs"]["response_cost"]
|
||||
assert response_cost is not None and response_cost > 0
|
||||
assert "standard_logging_object" in mock_callback.call_args.kwargs["kwargs"]
|
||||
|
||||
|
||||
|
||||
@ -392,7 +392,6 @@ async def test_langsmith_key_based_logging():
|
||||
"role": "assistant",
|
||||
"tool_calls": None,
|
||||
"function_call": None,
|
||||
"provider_specific_fields": None,
|
||||
},
|
||||
}
|
||||
],
|
||||
|
||||
@ -201,6 +201,10 @@ ignored_keys = [
|
||||
"metadata.usage_object",
|
||||
"metadata.cold_storage_object_key",
|
||||
"metadata.additional_usage_values.prompt_tokens_details.cache_creation_tokens",
|
||||
"metadata.additional_usage_values.completion_tokens_details",
|
||||
"metadata.additional_usage_values.prompt_tokens_details",
|
||||
"metadata.additional_usage_values.cache_creation_input_tokens",
|
||||
"metadata.additional_usage_values.cache_read_input_tokens",
|
||||
"metadata.litellm_overhead_time_ms",
|
||||
"metadata.cost_breakdown",
|
||||
]
|
||||
|
||||
@ -1200,9 +1200,7 @@ async def test_acompletion_streaming_disable_fallbacks_midstream():
|
||||
print("\n=== Test 1: disable_fallbacks=True with original_exception ===")
|
||||
|
||||
# Create an original exception to wrap
|
||||
from litellm.llms.anthropic.chat.anthropic_chat_transformation import (
|
||||
AnthropicError,
|
||||
)
|
||||
from litellm.llms.anthropic.common_utils import AnthropicError
|
||||
|
||||
original_error = AnthropicError(
|
||||
status_code=500,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user