diff --git a/.github/workflows/_integration_test.yml b/.github/workflows/_integration_test.yml
index d2157eb495630..7f9ea8ddf57c6 100644
--- a/.github/workflows/_integration_test.yml
+++ b/.github/workflows/_integration_test.yml
@@ -79,6 +79,7 @@ jobs:
           VOYAGE_API_KEY: ${{ secrets.VOYAGE_API_KEY }}
           COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
           UPSTAGE_API_KEY: ${{ secrets.UPSTAGE_API_KEY }}
+          XAI_API_KEY: ${{ secrets.XAI_API_KEY }}
         run: |
           make integration_tests
 
diff --git a/libs/partners/xai/tests/unit_tests/__init__.py b/libs/partners/xai/tests/unit_tests/__init__.py
index e69de29bb2d1d..7e2e6c70d591b 100644
--- a/libs/partners/xai/tests/unit_tests/__init__.py
+++ b/libs/partners/xai/tests/unit_tests/__init__.py
@@ -0,0 +1,3 @@
+import os
+
+os.environ["XAI_API_KEY"] = "test"
diff --git a/libs/partners/xai/tests/unit_tests/test_chat_models.py b/libs/partners/xai/tests/unit_tests/test_chat_models.py
index dee5ddd5ea3f2..981d0d91411ff 100644
--- a/libs/partners/xai/tests/unit_tests/test_chat_models.py
+++ b/libs/partners/xai/tests/unit_tests/test_chat_models.py
@@ -1,5 +1,6 @@
 import json
 
+import pytest  # type: ignore[import-not-found]
 from langchain_core.messages import (
     AIMessage,
     FunctionMessage,