tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-07 02:25:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix - test openai batches

Browse Source
This commit is contained in:
Ishaan Jaff
2024-05-28 17:29:13 -07:00
parent 215f19440c
commit 5e9d024c2c
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_openai_batches.py
+10 -2
View File
@@ -24,8 +24,12 @@ def test_create_batch():
2. Create Batch Request
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=open("openai_batch_completions.jsonl", "rb"),
file=open(file_path, "rb"),
purpose="batch",
custom_llm_provider="openai",
)
@@ -77,8 +81,12 @@ async def test_async_create_batch():
3. Retrieve the specific 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=open("openai_batch_completions.jsonl", "rb"),
file=open(file_path, "rb"),
purpose="batch",
custom_llm_provider="openai",
)