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

fix test fine tuning api azure

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-05 11:08:13 -07:00
parent bf28b893d8
commit 35792ccc1f
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_fine_tuning_api.py
View file

@ -158,8 +158,7 @@ async def test_azure_create_fine_tune_jobs_async():
model="gpt-35-turbo-1106", model="gpt-35-turbo-1106",
training_file=file_id, training_file=file_id,
custom_llm_provider="azure", custom_llm_provider="azure",
api_key=os.getenv("AZURE_SWEDEN_API_KEY"), api_base="https://exampleopenaiendpoint-production.up.railway.app",
api_base="https://my-endpoint-sweden-berri992.openai.azure.com/",
) )
print( print(
@ -167,15 +166,16 @@ async def test_azure_create_fine_tune_jobs_async():
) )
assert create_fine_tuning_response.id is not None assert create_fine_tuning_response.id is not None
assert create_fine_tuning_response.model == "gpt-35-turbo-1106"
# response from Example/mocked endpoint
assert create_fine_tuning_response.model == "davinci-002"
# list fine tuning jobs # list fine tuning jobs
print("listing ft jobs") print("listing ft jobs")
ft_jobs = await litellm.alist_fine_tuning_jobs( ft_jobs = await litellm.alist_fine_tuning_jobs(
limit=2, limit=2,
custom_llm_provider="azure", custom_llm_provider="azure",
api_key=os.getenv("AZURE_SWEDEN_API_KEY"), api_base="https://exampleopenaiendpoint-production.up.railway.app",
api_base="https://my-endpoint-sweden-berri992.openai.azure.com/",
) )
print("response from litellm.list_fine_tuning_jobs=", ft_jobs) print("response from litellm.list_fine_tuning_jobs=", ft_jobs)
@ -184,7 +184,7 @@ async def test_azure_create_fine_tune_jobs_async():
fine_tuning_job_id=create_fine_tuning_response.id, fine_tuning_job_id=create_fine_tuning_response.id,
custom_llm_provider="azure", custom_llm_provider="azure",
api_key=os.getenv("AZURE_SWEDEN_API_KEY"), api_key=os.getenv("AZURE_SWEDEN_API_KEY"),
api_base="https://my-endpoint-sweden-berri992.openai.azure.com/", api_base="https://exampleopenaiendpoint-production.up.railway.app",
) )
print("response from litellm.cancel_fine_tuning_job=", response) print("response from litellm.cancel_fine_tuning_job=", response)