diff --git a/litellm/tests/test_fine_tuning_api.py b/litellm/tests/test_fine_tuning_api.py index b7e3c957cce..d35cd661ab8 100644 --- a/litellm/tests/test_fine_tuning_api.py +++ b/litellm/tests/test_fine_tuning_api.py @@ -12,6 +12,7 @@ from openai import APITimeoutError as Timeout import litellm litellm.num_retries = 0 +import asyncio import logging from litellm import create_fine_tuning_job @@ -112,3 +113,58 @@ async def test_create_fine_tune_jobs_async(): assert response.status == "cancelled" assert response.id == create_fine_tuning_response.id pass + + +@pytest.mark.asyncio +async def test_azure_create_fine_tune_jobs_async(): + verbose_logger.setLevel(logging.DEBUG) + file_name = "azure_fine_tune.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(file_path, "rb"), + purpose="fine-tune", + custom_llm_provider="azure", + api_key=os.getenv("AZURE_SWEDEN_API_KEY"), + api_base="https://my-endpoint-sweden-berri992.openai.azure.com/", + ) + print("Response from creating file=", file_obj) + + await asyncio.sleep(5) + + create_fine_tuning_response = await litellm.acreate_fine_tuning_job( + model="gpt-35-turbo-1106", + training_file=file_obj.id, + custom_llm_provider="azure", + api_key=os.getenv("AZURE_SWEDEN_API_KEY"), + api_base="https://my-endpoint-sweden-berri992.openai.azure.com/", + ) + + print("response from litellm.create_fine_tuning_job=", create_fine_tuning_response) + + assert create_fine_tuning_response.id is not None + assert create_fine_tuning_response.model == "gpt-35-turbo-1106" + + # # list fine tuning jobs + # print("listing ft jobs") + # ft_jobs = await litellm.alist_fine_tuning_jobs(limit=2) + # print("response from litellm.list_fine_tuning_jobs=", ft_jobs) + # assert len(list(ft_jobs)) > 0 + + # # delete file + + # await litellm.afile_delete( + # file_id=file_obj.id, + # ) + + # # cancel ft job + # response = await litellm.acancel_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