skip test cases for mistralai

langchain-ai · Jul 12, 2024 · 8fdfeb7 · 8fdfeb7
1 parent 88c51cc
commit 8fdfeb7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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)