phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix - test openai batches

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-28 17:29:13 -07:00
parent be38c3142d
commit 0aa2d883e5
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_openai_batches.py
View file

@ -24,8 +24,12 @@ def test_create_batch():
2. Create Batch Request 2. Create Batch Request
3. Retrieve the specific batch 3. Retrieve the specific batch
""" """
file_name = "openai_batch_completions.jsonl"
_current_dir = os.path.dirname(os.path.abspath(__file__))
file_path = os.path.join(_current_dir, file_name)
file_obj = litellm.create_file( file_obj = litellm.create_file(
file=open("openai_batch_completions.jsonl", "rb"), file=open(file_path, "rb"),
purpose="batch", purpose="batch",
custom_llm_provider="openai", custom_llm_provider="openai",
) )
@ -77,8 +81,12 @@ async def test_async_create_batch():
3. Retrieve the specific batch 3. Retrieve the specific batch
""" """
print("Testing async create batch") print("Testing async create batch")
file_name = "openai_batch_completions.jsonl"
_current_dir = os.path.dirname(os.path.abspath(__file__))
file_path = os.path.join(_current_dir, file_name)
file_obj = await litellm.acreate_file( file_obj = await litellm.acreate_file(
file=open("openai_batch_completions.jsonl", "rb"), file=open(file_path, "rb"),
purpose="batch", purpose="batch",
custom_llm_provider="openai", custom_llm_provider="openai",
) )