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

feat support azure ft create endpoint

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-31 13:57:58 -07:00
parent 2cf10a621e
commit ff9bb96217
2 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/fine_tuning/main.py
View file

@ -208,11 +208,15 @@ def create_fine_tuning_job(
seed=seed,
)
create_fine_tuning_job_data_dict = create_fine_tuning_job_data.model_dump(
exclude_none=True
)
response = azure_fine_tuning_apis_instance.create_fine_tuning_job(
api_base=api_base,
api_key=api_key,
api_version=api_version,
create_fine_tuning_job_data=create_fine_tuning_job_data,
create_fine_tuning_job_data=create_fine_tuning_job_data_dict,
timeout=timeout,
max_retries=optional_params.max_retries,
_is_async=_is_async,

4
litellm/llms/fine_tuning_apis/azure.py
View file

@ -21,7 +21,7 @@ class AzureOpenAIFineTuningAPI(BaseLLM):
async def acreate_fine_tuning_job(
self,
create_fine_tuning_job_data: FineTuningJobCreate,
create_fine_tuning_job_data: dict,
openai_client: AsyncAzureOpenAI,
) -> FineTuningJob:
response = await openai_client.fine_tuning.jobs.create(
@ -32,7 +32,7 @@ class AzureOpenAIFineTuningAPI(BaseLLM):
def create_fine_tuning_job(
self,
_is_async: bool,
create_fine_tuning_job_data: FineTuningJobCreate,
create_fine_tuning_job_data: dict,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],