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

fix cancel ft job route

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-31 16:19:15 -07:00
parent 577fbcd770
commit 6d88f4f31b
2 changed files with 19 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/fine_tuning_endpoints/endpoints.py
View file

@ -197,6 +197,11 @@ async def create_fine_tuning_job(
dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"],
)
@router.get(
"/fine_tuning/jobs",
dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"],
)
async def list_fine_tuning_jobs(
request: Request,
fastapi_response: Response,
@ -299,10 +304,14 @@ async def list_fine_tuning_jobs(
dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"],
)
@router.post(
"/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel",
dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"],
)
async def retrieve_fine_tuning_job(
request: Request,
fastapi_response: Response,
custom_llm_provider: Literal["openai", "azure"],
fine_tuning_job_id: str,
user_api_key_dict: dict = Depends(user_api_key_auth),
):
@ -336,6 +345,10 @@ async def retrieve_fine_tuning_job(
proxy_config=proxy_config,
)
request_body = await request.json()
custom_llm_provider = request_body.get("custom_llm_provider", None)
# get configs for custom_llm_provider
llm_provider_config = get_fine_tuning_provider_config(
custom_llm_provider=custom_llm_provider