tests

datastax · Jul 18, 2024 · 9f2bf86 · 9f2bf86
1 parent f124f5e
commit 9f2bf86
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/client/tests/streaming-assistants/test_run_v1.py b/client/tests/streaming-assistants/test_run_v1.py
@@ -1,6 +1,8 @@
 import time
 import logging
 
+import pytest
+
 logger = logging.getLogger(__name__)
 def run_with_assistant(assistant, client):
     user_message = "What's your favorite animal."
@@ -97,6 +99,7 @@ def test_run_claude(streaming_assistants_openai_client):
     )
     run_with_assistant(claude_assistant, streaming_assistants_openai_client)
 
+@pytest.skip(reason="flaky")
 def test_run_gemini(streaming_assistants_openai_client):
     gemini_assistant = streaming_assistants_openai_client.beta.assistants.create(
         name="Gemini Animal Tutor",

diff --git a/client/tests/streaming-assistants/test_streaming_run_v1.py b/client/tests/streaming-assistants/test_streaming_run_v1.py
@@ -1,6 +1,7 @@
 import time
 import logging
 
+import pytest
 from openai.lib.streaming import AssistantEventHandler
 from typing_extensions import override
 
@@ -95,6 +96,7 @@ def test_run_claude(streaming_assistants_openai_client):
     )
     run_with_assistant(claude_assistant, streaming_assistants_openai_client)
 
+@pytest.skip(reason="flaky")
 def test_run_gemini(streaming_assistants_openai_client):
     gemini_assistant = streaming_assistants_openai_client.beta.assistants.create(
         name="Gemini Animal Tutor",