From 282b8d0ae448bf56b29cd6f5dd3f6c0a6b687f38 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 8 May 2024 15:26:53 -0700 Subject: [PATCH] test bedrock pricing --- litellm/tests/test_completion_cost.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_completion_cost.py b/litellm/tests/test_completion_cost.py index fecd53e1937..35e0496fb46 100644 --- a/litellm/tests/test_completion_cost.py +++ b/litellm/tests/test_completion_cost.py @@ -231,14 +231,17 @@ def test_cost_bedrock_pricing(): assert cost == predicted_cost -@pytest.mark.skip(reason="AWS disabled our access") def test_cost_bedrock_pricing_actual_calls(): litellm.set_verbose = True model = "anthropic.claude-instant-v1" messages = [{"role": "user", "content": "Hey, how's it going?"}] - response = litellm.completion(model=model, messages=messages) - assert response._hidden_params["region_name"] is not None + response = litellm.completion( + model=model, messages=messages, mock_response="hello cool one" + ) + + print("response", response) cost = litellm.completion_cost( + model="bedrock/anthropic.claude-instant-v1", completion_response=response, messages=[{"role": "user", "content": "Hey, how's it going?"}], )