From 8be37bee043af315d5567abb3f640d47e58cf347 Mon Sep 17 00:00:00 2001 From: James Braza Date: Tue, 11 Jun 2024 18:18:21 -0700 Subject: [PATCH] Added simple test of get_supported_openai_params with no custom provider --- litellm/tests/test_utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/tests/test_utils.py b/litellm/tests/test_utils.py index 8f99bd665c..9f2e99a805 100644 --- a/litellm/tests/test_utils.py +++ b/litellm/tests/test_utils.py @@ -23,6 +23,7 @@ from litellm.utils import ( create_pretrained_tokenizer, create_tokenizer, get_max_tokens, + get_supported_openai_params, ) # Assuming your trim_messages, shorten_message_to_fit_limit, and get_token_count functions are all in a module named 'message_utils' @@ -386,3 +387,7 @@ def test_get_max_token_unit_test(): ) # Returns a number instead of throwing an Exception assert isinstance(max_tokens, int) + + +def test_get_supported_openai_params() -> None: + assert isinstance(get_supported_openai_params("gpt-4"), list)