add tests to check ai21 models cost is calculated correct

2024-08-30 07:30:16 -07:00 · 2024-08-30 07:30:16 -07:00 · fdaaea9b0d
commit fdaaea9b0d
parent 6bd6c956a5
1 changed files with 4 additions and 0 deletions
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@ -2555,3 +2555,7 @@ async def test_partner_models_httpx_ai21():
        assert response.usage.total_tokens == 194

        print(f"response: {response}")
+
+        print("hidden params from response=", response._hidden_params)
+
+        assert response._hidden_params["response_cost"] > 0