From 01f1dfdba38dd0db99bed522c5b902dd8975b92d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 28 Feb 2024 17:51:34 -0800 Subject: [PATCH] (test) supports_function_calling --- litellm/tests/test_utils.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/litellm/tests/test_utils.py b/litellm/tests/test_utils.py index e094afee6d..5da214dbc5 100644 --- a/litellm/tests/test_utils.py +++ b/litellm/tests/test_utils.py @@ -326,9 +326,7 @@ def test_supports_function_calling(): litellm.supports_function_calling(model="azure/gpt-4-1106-preview") == True ) assert ( - litellm.supports_function_calling( - model="bedrock/anthropic.claude-instant-v1" - ) + litellm.supports_function_calling(model="anthropic.claude-instant-v1") == False ) assert litellm.supports_function_calling(model="palm/chat-bison") == False @@ -338,6 +336,5 @@ def test_supports_function_calling(): == False ) assert litellm.supports_function_calling(model="claude-2") == False - except Exception as e: pytest.fail(f"Error occurred: {e}")