phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix azure batches test - don't have more quota

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-06 16:44:16 -07:00
parent 42f56e4287
commit e0ac27c00c
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_openai_batches_and_files.py
View file

@ -29,6 +29,9 @@ def test_create_batch(provider):
2. Create Batch Request 2. Create Batch Request
3. Retrieve the specific batch 3. Retrieve the specific batch
""" """
if provider == "azure":
# Don't have anymore Azure Quota
return
file_name = "openai_batch_completions.jsonl" file_name = "openai_batch_completions.jsonl"
_current_dir = os.path.dirname(os.path.abspath(__file__)) _current_dir = os.path.dirname(os.path.abspath(__file__))
file_path = os.path.join(_current_dir, file_name) file_path = os.path.join(_current_dir, file_name)
@ -103,6 +106,9 @@ async def test_async_create_batch(provider):
3. Retrieve the specific batch 3. Retrieve the specific batch
""" """
print("Testing async create batch") print("Testing async create batch")
if provider == "azure":
# Don't have anymore Azure Quota
return
file_name = "openai_batch_completions.jsonl" file_name = "openai_batch_completions.jsonl"
_current_dir = os.path.dirname(os.path.abspath(__file__)) _current_dir = os.path.dirname(os.path.abspath(__file__))