mark double ai / human message as known issue

langchain-ai · May 14, 2024 · e0c3188 · e0c3188
1 parent 99616e4
commit e0c3188
Showing 1 changed file with 2 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
@@ -120,6 +120,8 @@ def test_messages(
 ) -> None:
     if not system and not exchange:
         pytest.skip("No messages to test")
+    if len(exchange) == 2 and type(exchange[0]) == type(exchange[1]):
+        pytest.skip("Known issue, messages types must alternate")
     chat = ChatNVIDIA(model=chat_model, max_tokens=36).mode(**mode)
     response = chat.invoke(system + exchange)
     assert isinstance(response, BaseMessage)