command-r

datastax · Apr 1, 2024 · 16fe841 · 16fe841
1 parent 2c461e6
commit 16fe841
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/streaming-assistants/test_streaming_run_retrieval.py b/tests/streaming-assistants/test_streaming_run_retrieval.py
@@ -91,9 +91,9 @@ def test_run_gpt3_5(patched_openai_client):
     )
     run_with_assistant(gpt3_assistant, patched_openai_client)
 
-@pytest.mark.skip(reason="replace with command-r because context window")
+#@pytest.mark.skip(reason="replace with command-r because context window")
 def test_run_cohere(patched_openai_client):
-    model = "cohere/command"
+    model = "command-r"
     name = f"{model} Math Tutor"
 
     cohere_assistant = patched_openai_client.beta.assistants.create(