Skip to content

Commit ede20fc

Browse files
authored
Add tests 2 (#217)
* add * add req * fix api key * refactor * lint * max tokens 256 * use api key
1 parent 98bfa0a commit ede20fc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/public_models/test_public_models_predicts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
)
4141

4242
# New constant for the OpenAI compatible endpoint test
43-
API_KEY = os.environ.get("CLARIFAI_PAT_KEY", os.environ.get("CLARIFAI_PAT"))
43+
API_KEY = os.environ.get("CLARIFAI_API_KEY", os.environ.get("CLARIFAI_API"))
4444
MAX_RETRY_ATTEMPTS = 3
4545

4646

@@ -476,7 +476,7 @@ async def _call_openai_model_async(model_identifier, session):
476476
async def test_openai_compatible_endpoint_on_featured_models_async():
477477
"""Tests the OpenAI compatible endpoint concurrently with featured models."""
478478
if not API_KEY:
479-
pytest.skip("Skipping test: CLARIFAI_PAT environment variable not set.")
479+
pytest.skip("Skipping test: CLARIFAI_API_KEY environment variable not set.")
480480

481481
featured_models = _list_featured_models()
482482
tasks = []

0 commit comments

Comments
 (0)