lint

langchain-ai · Nov 12, 2024 · fdd656a · fdd656a
1 parent 0eb67b7
commit fdd656a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git 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
@@ -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
@@ -1,5 +1,6 @@
 import json
 
+import pytest  # type: ignore[import-not-found]
 from langchain_core.messages import (
     AIMessage,
     FunctionMessage,