tests

Author: phact

client/tests/astra-assistants/test_run_retreival_v2.py

@@ -154,6 +154,7 @@ def test_run_cohere(patched_openai_client):
     )
     run_with_assistant(cohere_assistant, patched_openai_client, file2_path, embedding_model2)
 
+@pytest.skip
 def test_run_perp(patched_openai_client):
     model="perplexity/llama-3.1-70b-instruct"
     name = f"{model} Math Tutor"

client/tests/astra-assistants/test_streaming_run_retrieval_v2.py

@@ -168,6 +168,7 @@ def test_run_cohere(patched_openai_client):
     )
     run_with_assistant(cohere_assistant, patched_openai_client)
 
+@pytest.skip
 def test_run_perp(patched_openai_client):
     model="perplexity/llama-3.1-70b-instruct"
     name = f"{model} Math Tutor"
@@ -190,6 +191,7 @@ def test_run_claude(patched_openai_client):
     )
     run_with_assistant(claude_assistant, patched_openai_client)
 
+@pytest.skip
 def test_run_gemini(patched_openai_client):
     #model = "gemini/gemini-1.5-pro-latest"
     model = "gemini/gemini-1.5-flash"

client/tests/astra-assistants/test_streaming_run_v2.py

@@ -1,6 +1,7 @@
 import time
 import logging
 
+import pytest
 from openai.lib.streaming import AssistantEventHandler
 from typing_extensions import override
 
@@ -86,6 +87,7 @@ def test_run_cohere(patched_openai_client):
     )
     run_with_assistant(cohere_assistant, patched_openai_client)
 
+@pytest.skip
 def test_run_perp(patched_openai_client):
     perplexity_assistant = patched_openai_client.beta.assistants.create(
         name="Perplexity/Mixtral Animal Tutor",