phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat FT cancel and LIST endpoints for Azure

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-30 16:03:31 -07:00
parent c6bff3286c
commit 02736ac8b5
3 changed files with 132 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -91,13 +91,15 @@ class AzureOpenAIFineTuningAPI(BaseLLM):
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: Optional[int],
organization: Optional[str],
organization: Optional[str] = None,
api_version: Optional[str] = None,
client: Optional[Union[AzureOpenAI, AsyncAzureOpenAI]] = None,
):
openai_client: Optional[Union[AzureOpenAI, AsyncAzureOpenAI]] = (
get_azure_openai_client(
api_key=api_key,
api_base=api_base,
api_version=api_version,
timeout=timeout,
max_retries=max_retries,
organization=organization,
@ -141,8 +143,9 @@ class AzureOpenAIFineTuningAPI(BaseLLM):
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: Optional[int],
organization: Optional[str],
organization: Optional[str] = None,
client: Optional[Union[AzureOpenAI, AsyncAzureOpenAI]] = None,
api_version: Optional[str] = None,
after: Optional[str] = None,
limit: Optional[int] = None,
):
@ -150,6 +153,7 @@ class AzureOpenAIFineTuningAPI(BaseLLM):
get_azure_openai_client(
api_key=api_key,
api_base=api_base,
api_version=api_version,
timeout=timeout,
max_retries=max_retries,
organization=organization,
@ -175,4 +179,3 @@ class AzureOpenAIFineTuningAPI(BaseLLM):
verbose_logger.debug("list fine tuning job, after= %s, limit= %s", after, limit)
response = openai_client.fine_tuning.jobs.list(after=after, limit=limit) # type: ignore
return response
pass