phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test cancel ft jobs

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-31 16:12:52 -07:00
parent 5f1070e47f
commit 729f876f81
1 changed files with 20 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

44
tests/test_openai_fine_tuning.py
View file

@ -10,36 +10,30 @@ async def test_openai_fine_tuning():
""" """
client = AsyncOpenAI(api_key="sk-1234", base_url="http://0.0.0.0:4000") client = AsyncOpenAI(api_key="sk-1234", base_url="http://0.0.0.0:4000")
# file_name = "openai_batch_completions.jsonl" file_name = "openai_batch_completions.jsonl"
# _current_dir = os.path.dirname(os.path.abspath(__file__)) _current_dir = os.path.dirname(os.path.abspath(__file__))
# file_path = os.path.join(_current_dir, file_name) file_path = os.path.join(_current_dir, file_name)
# response = await client.files.create( response = await client.files.create(
# extra_body={"custom_llm_provider": "azure"}, extra_body={"custom_llm_provider": "azure"},
# file=open(file_path, "rb"), file=open(file_path, "rb"),
# purpose="fine-tune", purpose="fine-tune",
# ) )
# print("response from files.create: {}".format(response)) print("response from files.create: {}".format(response))
# # create fine tuning job # create fine tuning job
# ft_job = await client.fine_tuning.jobs.create( ft_job = await client.fine_tuning.jobs.create(
# model="gpt-35-turbo-1106", model="gpt-35-turbo-1106",
# training_file=response.id, training_file=response.id,
# extra_body={"custom_llm_provider": "azure"}, extra_body={"custom_llm_provider": "azure"},
# ) )
# print("response from ft job={}".format(ft_job)) print("response from ft job={}".format(ft_job))
# # response from example endpoint # response from example endpoint
# assert ft_job.id == "file-abc123" assert ft_job.id == "file-abc123"
# get fine tuning job
# specific_ft_job = await client.fine_tuning.jobs.retrieve(
# fine_tuning_job_id="123",
# extra_body={"custom_llm_provider": "azure"},
# )
# list all fine tuning jobs # list all fine tuning jobs
list_ft_jobs = await client.fine_tuning.jobs.list( list_ft_jobs = await client.fine_tuning.jobs.list(
@ -51,3 +45,5 @@ async def test_openai_fine_tuning():
fine_tuning_job_id="123", fine_tuning_job_id="123",
extra_body={"custom_llm_provider": "azure"}, extra_body={"custom_llm_provider": "azure"},
) )
assert cancel_ft_job.id is not None