Fix unit test after merge from main

andrewyng · Oct 3, 2024 · 34c1af0 · 34c1af0
1 parent a49c100
commit 34c1af0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/client/test_client.py b/tests/client/test_client.py
@@ -61,7 +61,7 @@ def test_client_chat_completions(
                 "region": "us-west4",
                 "application_credentials": "test_google_application_credentials",
             },
-            ProviderNames.FIREWORKS: {
+            "fireworks": {
                 "api_key": "fireworks-api-key",
             },
         }
@@ -127,7 +127,7 @@ def test_client_chat_completions(
         mock_google.assert_called_once()
 
         # Test Fireworks model
-        fireworks_model = ProviderNames.FIREWORKS + ":" + "fireworks-model"
+        fireworks_model = "fireworks" + ":" + "fireworks-model"
         fireworks_response = client.chat.completions.create(
             fireworks_model, messages=messages
         )