diff --git a/.github/workflows/scheduled_test.yml b/.github/workflows/scheduled_test.yml index 9ed99e87b00..1e948823b7a 100644 --- a/.github/workflows/scheduled_test.yml +++ b/.github/workflows/scheduled_test.yml @@ -15,7 +15,7 @@ on: env: POETRY_VERSION: "1.8.4" UV_FROZEN: "true" - DEFAULT_LIBS: '["libs/partners/openai", "libs/partners/anthropic", "libs/partners/fireworks", "libs/partners/groq", "libs/partners/mistralai", "libs/partners/google-vertexai", "libs/partners/google-genai", "libs/partners/aws"]' + DEFAULT_LIBS: '["libs/partners/openai", "libs/partners/anthropic", "libs/partners/fireworks", "libs/partners/groq", "libs/partners/mistralai", "libs/partners/xai", "libs/partners/google-vertexai", "libs/partners/google-genai", "libs/partners/aws"]' POETRY_LIBS: ("libs/partners/google-vertexai" "libs/partners/google-genai" "libs/partners/aws") jobs: @@ -139,6 +139,7 @@ jobs: GROQ_API_KEY: ${{ secrets.GROQ_API_KEY }} HUGGINGFACEHUB_API_TOKEN: ${{ secrets.HUGGINGFACEHUB_API_TOKEN }} MISTRAL_API_KEY: ${{ secrets.MISTRAL_API_KEY }} + XAI_API_KEY: ${{ secrets.XAI_API_KEY }} COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }} NVIDIA_API_KEY: ${{ secrets.NVIDIA_API_KEY }} GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }} diff --git a/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py b/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py index edcaf93eebf..0959edc4e91 100644 --- a/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py +++ b/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py @@ -5,7 +5,6 @@ from typing import Optional, Type import pytest # type: ignore[import-not-found] from langchain_core.language_models import BaseChatModel from langchain_core.rate_limiters import InMemoryRateLimiter -from langchain_core.tools import BaseTool from langchain_tests.integration_tests import ( # type: ignore[import-not-found] ChatModelIntegrationTests, # type: ignore[import-not-found] ) @@ -27,7 +26,7 @@ class TestXAIStandard(ChatModelIntegrationTests): @property def chat_model_params(self) -> dict: return { - "model": "grok-beta", + "model": "grok-2", "rate_limiter": rate_limiter, } @@ -39,21 +38,3 @@ class TestXAIStandard(ChatModelIntegrationTests): @pytest.mark.xfail(reason="Not yet supported.") def test_usage_metadata_streaming(self, model: BaseChatModel) -> None: super().test_usage_metadata_streaming(model) - - @pytest.mark.xfail(reason="Can't handle AIMessage with empty content.") - def test_tool_message_error_status( - self, model: BaseChatModel, my_adder_tool: BaseTool - ) -> None: - super().test_tool_message_error_status(model, my_adder_tool) - - @pytest.mark.xfail(reason="Can't handle AIMessage with empty content.") - def test_structured_few_shot_examples( - self, model: BaseChatModel, my_adder_tool: BaseTool - ) -> None: - super().test_structured_few_shot_examples(model, my_adder_tool) - - @pytest.mark.xfail(reason="Can't handle AIMessage with empty content.") - def test_tool_message_histories_string_content( - self, model: BaseChatModel, my_adder_tool: BaseTool - ) -> None: - super().test_tool_message_histories_string_content(model, my_adder_tool)