Tests: fix skip tests (#60)

* Tests: fix skip tests * fix action * lint
cohere-ai · Apr 26, 2024 · 06aa3d2 · 06aa3d2
1 parent baf1102
commit 06aa3d2
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/src/backend/tests/routers/test_chat.py b/src/backend/tests/routers/test_chat.py
@@ -15,7 +15,10 @@
 from backend.schemas.tool import Category
 from backend.tests.factories import get_factory
 
-is_cohere_env_set = os.environ.get("COHERE_API_KEY") is not None
+is_cohere_env_set = (
+    os.environ.get("COHERE_API_KEY") is not None
+    and os.environ.get("COHERE_API_KEY") != ""
+)
 
 
 @pytest.fixture()

diff --git a/src/backend/tests/tools/retrieval/test_collate.py b/src/backend/tests/tools/retrieval/test_collate.py
@@ -5,7 +5,10 @@
 from backend.chat.custom.model_deployments.cohere_platform import CohereDeployment
 from backend.tools.retrieval import collate
 
-is_cohere_env_set = os.environ.get("COHERE_API_KEY") is not None
+is_cohere_env_set = (
+    os.environ.get("COHERE_API_KEY") is not None
+    and os.environ.get("COHERE_API_KEY") != ""
+)
 
 
 @pytest.mark.skipif(not is_cohere_env_set, reason="Cohere API key not set")

diff --git a/src/backend/tests/tools/retrieval/test_lang_chain.py b/src/backend/tests/tools/retrieval/test_lang_chain.py
@@ -9,7 +9,10 @@
     LangChainWikiRetriever,
 )
 
-is_cohere_env_set = os.environ.get("COHERE_API_KEY") is not None
+is_cohere_env_set = (
+    os.environ.get("COHERE_API_KEY") is not None
+    and os.environ.get("COHERE_API_KEY") != ""
+)
 
 
 def test_wiki_retriever() -> None: