Skip to content

Commit fd189c8

Browse files
committed
Add gpt 4o support
1 parent d0c0f5c commit fd189c8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_modelhelper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_get_token_limit_default(caplog):
4848
system_message_with_name,
4949
],
5050
)
51-
def test_count_tokens_for_message(model: str, count_key: str, message: dict):
51+
def test_count_tokens_for_message(model, count_key, message):
5252
assert count_tokens_for_message(model, message["message"]) == message[count_key]
5353

5454

@@ -59,7 +59,7 @@ def test_count_tokens_for_message(model: str, count_key: str, message: dict):
5959
("gpt-4o", "count_omni"),
6060
],
6161
)
62-
def test_count_tokens_for_message_list(model: str, count_key: str):
62+
def test_count_tokens_for_message_list(model, count_key):
6363
assert count_tokens_for_message(model, text_and_image_message["message"]) == text_and_image_message[count_key]
6464

6565

0 commit comments

Comments
 (0)