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")
# 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_name = "openai_batch_completions.jsonl"
_current_dir = os.path.dirname(os.path.abspath(__file__))
file_path = os.path.join(_current_dir, file_name)
# response = await client.files.create(
# extra_body={"custom_llm_provider": "azure"},
# file=open(file_path, "rb"),
# purpose="fine-tune",
# )
response = await client.files.create(
extra_body={"custom_llm_provider": "azure"},
file=open(file_path, "rb"),
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(
# model="gpt-35-turbo-1106",
# training_file=response.id,
# extra_body={"custom_llm_provider": "azure"},
# )
ft_job = await client.fine_tuning.jobs.create(
model="gpt-35-turbo-1106",
training_file=response.id,
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
# 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"},
# )
# response from example endpoint
assert ft_job.id == "file-abc123"
# list all fine tuning jobs
list_ft_jobs = await client.fine_tuning.jobs.list(
@ -51,3 +45,5 @@ async def test_openai_fine_tuning():
fine_tuning_job_id="123",
extra_body={"custom_llm_provider": "azure"},
)
assert cancel_ft_job.id is not None