ai21: integration test fixes

2026-02-05 00:30:18 +00:00 · 2024-08-26 15:54:21 -07:00
1 changed files with 5 additions and 0 deletions
--- a/libs/partners/ai21/tests/integration_tests/test_standard.py
+++ b/libs/partners/ai21/tests/integration_tests/test_standard.py
@@ -71,3 +71,8 @@ class TestAI21Jamba1_5(BaseTestAI21):
        return {
            "model": "jamba-1.5-mini",
        }
+
+    # tool calling xfail
+    @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)