diff --git a/.github/workflows/scheduled_test.yml b/.github/workflows/scheduled_test.yml index 1b6522385de..e67d3afff42 100644 --- a/.github/workflows/scheduled_test.yml +++ b/.github/workflows/scheduled_test.yml @@ -19,11 +19,11 @@ jobs: working-directory: - "libs/partners/openai" - "libs/partners/anthropic" - # - "libs/partners/ai21" # standard-tests broken + - "libs/partners/ai21" - "libs/partners/fireworks" - # - "libs/partners/groq" # rate-limited + - "libs/partners/groq" - "libs/partners/mistralai" - # - "libs/partners/together" # rate-limited + - "libs/partners/together" name: Python ${{ matrix.python-version }} - ${{ matrix.working-directory }} steps: - uses: actions/checkout@v4 diff --git a/libs/partners/ai21/tests/integration_tests/test_standard.py b/libs/partners/ai21/tests/integration_tests/test_standard.py index 5c62b025999..2070ba3a701 100644 --- a/libs/partners/ai21/tests/integration_tests/test_standard.py +++ b/libs/partners/ai21/tests/integration_tests/test_standard.py @@ -19,3 +19,25 @@ class TestAI21Standard(ChatModelIntegrationTests): return { "model": "j2-ultra", } + + @pytest.mark.xfail(reason="Emits AIMessage instead of AIMessageChunk.") + def test_stream( + self, + chat_model_class: Type[BaseChatModel], + chat_model_params: dict, + ) -> None: + super().test_stream( + chat_model_class, + chat_model_params, + ) + + @pytest.mark.xfail(reason="Emits AIMessage instead of AIMessageChunk.") + async def test_astream( + self, + chat_model_class: Type[BaseChatModel], + chat_model_params: dict, + ) -> None: + await super().test_astream( + chat_model_class, + chat_model_params, + ) diff --git a/libs/partners/groq/tests/integration_tests/test_standard.py b/libs/partners/groq/tests/integration_tests/test_standard.py index 83ca841caa2..4048f7e8f6a 100644 --- a/libs/partners/groq/tests/integration_tests/test_standard.py +++ b/libs/partners/groq/tests/integration_tests/test_standard.py @@ -13,3 +13,14 @@ class TestMistralStandard(ChatModelIntegrationTests): @pytest.fixture def chat_model_class(self) -> Type[BaseChatModel]: return ChatGroq + + @pytest.mark.xfail(reason="Not yet implemented.") + def test_tool_message_histories_list_content( + self, + chat_model_class: Type[BaseChatModel], + chat_model_params: dict, + chat_model_has_tool_calling: bool, + ) -> None: + super().test_tool_message_histories_list_content( + chat_model_class, chat_model_params, chat_model_has_tool_calling + )