From a1d38549a2b848accca219d258c1314e8c68b504 Mon Sep 17 00:00:00 2001 From: Arnaud Flament Date: Fri, 7 Jun 2024 14:34:41 -0700 Subject: [PATCH] Update test_openai_token_counter.py --- tests/unit/llm/test_openai_token_counter.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/unit/llm/test_openai_token_counter.py b/tests/unit/llm/test_openai_token_counter.py index a5d33778..15668f14 100644 --- a/tests/unit/llm/test_openai_token_counter.py +++ b/tests/unit/llm/test_openai_token_counter.py @@ -16,6 +16,11 @@ def test_token_counter_gpt_4_turbo(self): counter = OpenAITokenCounter.from_model(model) self.assertEqual(counter.token_limit, 128000) + def test_token_counter_gpt_4o(self): + model = "gpt-4o" + counter = OpenAITokenCounter.from_model(model) + self.assertEqual(counter.token_limit, 128000) + def test_token_counter_exception(self): model = "gpt-4" counter = OpenAITokenCounter.from_model(model)