diff --git a/libs/partners/ai21/tests/integration_tests/test_standard.py b/libs/partners/ai21/tests/integration_tests/test_standard.py index dcd39cf5b36..6a038a16257 100644 --- a/libs/partners/ai21/tests/integration_tests/test_standard.py +++ b/libs/partners/ai21/tests/integration_tests/test_standard.py @@ -71,3 +71,19 @@ class TestAI21Jamba1_5(BaseTestAI21): return { "model": "jamba-1.5-mini", } + + @pytest.mark.xfail(reason="Prompt doesn't generate tool calls for Jamba 1.5.") + def test_tool_calling(self, model: BaseChatModel) -> None: + super().test_tool_calling(model) + + @pytest.mark.xfail(reason="Prompt doesn't generate tool calls for Jamba 1.5.") + def test_tool_calling_with_no_arguments(self, model: BaseChatModel) -> None: + super().test_tool_calling_with_no_arguments(model) + + @pytest.mark.xfail(reason="Requires tool calling & stream - still WIP") + def test_structured_output(self, model: BaseChatModel) -> None: + super().test_structured_output(model) + + @pytest.mark.xfail(reason="Requires tool calling & stream - still WIP") + def test_structured_output_pydantic_2_v1(self, model: BaseChatModel) -> None: + super().test_structured_output_pydantic_2_v1(model)