fix test

datastax · Mar 19, 2024 · ac16812 · ac16812
1 parent 578c0b8
commit ac16812
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/tests/streaming-assistants/test_file_embedding.py b/tests/streaming-assistants/test_file_embedding.py
@@ -24,18 +24,15 @@ def test_file_embedding_python(patched_openai_client):
     logger.info(file)
 
 def test_file_embedding_invalid(patched_openai_client):
-    try:
-        patched_openai_client.files.create(
-            file=open(
-                "./tests/fixtures/invalid",
-                "rb",
-            ),
-            purpose="assistants",
-            embedding_model="text-embedding-3-large",
-        )
-    except Exception as e:
-        assert e.status_code == 400
-        logger.info(e)
+    file = patched_openai_client.files.create(
+        file=open(
+            "./tests/fixtures/invalid",
+            "rb",
+        ),
+        purpose="assistants",
+        embedding_model="text-embedding-3-large",
+    )
+    logger.info(file)
 
 def test_file_embedding_png(patched_openai_client):
     try: