From 57b1bb5e06e60b64c8b76fde491705384d4e15f2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 4 Nov 2024 20:20:19 -0800 Subject: [PATCH] testing fix bedrock deprecated cohere.command-text-v14 --- tests/local_testing/test_completion.py | 1 - tests/local_testing/test_streaming.py | 1 - 2 files changed, 2 deletions(-) diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index a8fbb3dc5..8f28de7b4 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -3543,7 +3543,6 @@ def response_format_tests(response: litellm.ModelResponse): "mistral.mistral-7b-instruct-v0:2", # "bedrock/amazon.titan-tg1-large", "meta.llama3-8b-instruct-v1:0", - "cohere.command-text-v14", ], ) @pytest.mark.parametrize("sync_mode", [True, False]) diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py index b912d98f3..99c506f69 100644 --- a/tests/local_testing/test_streaming.py +++ b/tests/local_testing/test_streaming.py @@ -1430,7 +1430,6 @@ async def test_completion_replicate_llama3_streaming(sync_mode): ["mistral.mistral-7b-instruct-v0:2", None], ["bedrock/amazon.titan-tg1-large", None], ["meta.llama3-8b-instruct-v1:0", None], - ["cohere.command-text-v14", None], ], ) @pytest.mark.asyncio