add xfails

langchain-ai · Jul 11, 2024 · cfeed21 · cfeed21
1 parent 94868d9
commit cfeed21
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/libs/partners/ai21/tests/integration_tests/test_standard.py b/libs/partners/ai21/tests/integration_tests/test_standard.py
@@ -33,14 +33,18 @@ def chat_model_params(self) -> dict:
             "model": "j2-ultra",
         }
 
-    @pytest.mark.xfail(reason="Emits AIMessage instead of AIMessageChunk.")
+    @pytest.mark.xfail(reason="Streaming is not supported for Jurassic models.")
     def test_stream(self, model: BaseChatModel) -> None:
         super().test_stream(model)
 
-    @pytest.mark.xfail(reason="Emits AIMessage instead of AIMessageChunk.")
+    @pytest.mark.xfail(reason="Streaming is not supported for Jurassic models.")
     async def test_astream(self, model: BaseChatModel) -> None:
         await super().test_astream(model)
 
+    @pytest.mark.xfail(reason="Streaming is not supported for Jurassic models.")
+    def test_usage_metadata_streaming(self, model: BaseChatModel) -> None:
+        super().test_usage_metadata_streaming(model)
+
 
 class TestAI21Jamba(BaseTestAI21):
     @property

diff --git a/libs/partners/together/tests/integration_tests/test_chat_models_standard.py b/libs/partners/together/tests/integration_tests/test_chat_models_standard.py
@@ -23,3 +23,7 @@ def chat_model_params(self) -> dict:
     @pytest.mark.xfail(reason=("May not call a tool."))
     def test_tool_calling_with_no_arguments(self, model: BaseChatModel) -> None:
         super().test_tool_calling_with_no_arguments(model)
+
+    @pytest.mark.xfail(reason="Not yet supported.")
+    def test_usage_metadata_streaming(self, model: BaseChatModel) -> None:
+        super().test_usage_metadata_streaming(model)