From 16d595c4ff0287400b220bab479faccbf3ffbf34 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 29 Jul 2024 19:22:41 -0700 Subject: [PATCH] test cancel cancel_fine_tuning_job --- litellm/tests/test_fine_tuning_api.py | 22 +++++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) diff --git a/litellm/tests/test_fine_tuning_api.py b/litellm/tests/test_fine_tuning_api.py index eec81d0073..d4935cb0d5 100644 --- a/litellm/tests/test_fine_tuning_api.py +++ b/litellm/tests/test_fine_tuning_api.py @@ -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