From 8fdfeb7897bfa8a0806bfc1eef5b7cbad3f3ded0 Mon Sep 17 00:00:00 2001 From: raspawar Date: Fri, 12 Jul 2024 19:37:05 +0530 Subject: [PATCH] skip test cases for mistralai --- .../tests/integration_tests/test_chat_models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/libs/ai-endpoints/tests/integration_tests/test_chat_models.py b/libs/ai-endpoints/tests/integration_tests/test_chat_models.py index 4940c5b5..e65341f8 100644 --- a/libs/ai-endpoints/tests/integration_tests/test_chat_models.py +++ b/libs/ai-endpoints/tests/integration_tests/test_chat_models.py @@ -125,6 +125,12 @@ def test_messages( ) -> None: if not system and not exchange: pytest.skip("No messages to test") + if ( + chat_model == "mistralai/mixtral-8x7b-instruct-v0.1" + and exchange + and isinstance(exchange[0], AIMessage) + ): + pytest.skip("mistralai does not support system=>AIMessage") chat = ChatNVIDIA(model=chat_model, max_tokens=36, **mode) response = chat.invoke(system + exchange) assert isinstance(response, BaseMessage)