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

test cancel cancel_fine_tuning_job

Browse Source
This commit is contained in:
Ishaan Jaff
2024-07-29 19:22:41 -07:00
parent 3802eaa6b5
commit 16d595c4ff
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_fine_tuning_api.py
+17 -5
View File
@@ -31,17 +31,29 @@ def test_create_fine_tune_job():
)
print("Response from creating file=", file_obj)
response = litellm.create_fine_tuning_job(
model="gpt-3.5-turbo",
create_fine_tuning_response = litellm.create_fine_tuning_job(
model="gpt-3.5-turbo-0125",
training_file=file_obj.id,
)
print("response from litellm.create_fine_tuning_job=", response)
print("response from litellm.create_fine_tuning_job=", create_fine_tuning_response)
assert response.id is not None
assert response.model == "gpt-3.5-turbo"
assert create_fine_tuning_response.id is not None
assert create_fine_tuning_response.model == "gpt-3.5-turbo-0125"
# delete file
litellm.file_delete(
file_id=file_obj.id,
)
# cancel ft job
response = litellm.cancel_fine_tuning_job(
fine_tuning_job_id=create_fine_tuning_response.id,
)
print("response from litellm.cancel_fine_tuning_job=", response)
assert response.status == "cancelled"
assert response.id == create_fine_tuning_response.id
pass