[ Fix ] for the failing embedchain tests (#1899)

2024-09-25 20:02:53 +05:30
parent 5525c4e6fe
commit 44ee48e924
4 changed files with 71 additions and 29 deletions
--- a/embedchain/tests/llm/test_openai.py
+++ b/embedchain/tests/llm/test_openai.py
@@ -85,7 +85,7 @@ def test_get_llm_model_answer_with_token_usage(config, mocker):
        "prompt_tokens": 1,
        "completion_tokens": 2,
        "total_tokens": 3,
-        "total_cost": 5.5e-06,
+        "total_cost": 1.35e-06,
        "cost_currency": "USD",
    }
    mocked_get_answer.assert_called_once_with("Test query", test_config)