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

fix mark fine tuning endpoints as enteprrise

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-31 16:41:14 -07:00
parent 424a70c331
commit bd15d0f4ca
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -74,11 +74,13 @@ def get_fine_tuning_provider_config(
"/v1/fine_tuning/jobs", "/v1/fine_tuning/jobs",
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"], tags=["fine-tuning"],
summary="✨ (Enterprise) Create Fine-Tuning Job",
) )
@router.post( @router.post(
"/fine_tuning/jobs", "/fine_tuning/jobs",
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"], tags=["fine-tuning"],
summary="✨ (Enterprise) Create Fine-Tuning Job",
) )
async def create_fine_tuning_job( async def create_fine_tuning_job(
request: Request, request: Request,
@ -199,11 +201,13 @@ async def create_fine_tuning_job(
"/v1/fine_tuning/jobs", "/v1/fine_tuning/jobs",
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"], tags=["fine-tuning"],
summary="✨ (Enterprise) List Fine-Tuning Jobs",
) )
@router.get( @router.get(
"/fine_tuning/jobs", "/fine_tuning/jobs",
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"], tags=["fine-tuning"],
summary="✨ (Enterprise) List Fine-Tuning Jobs",
) )
async def list_fine_tuning_jobs( async def list_fine_tuning_jobs(
request: Request, request: Request,
@ -306,11 +310,13 @@ async def list_fine_tuning_jobs(
"/v1/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel", "/v1/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel",
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"], tags=["fine-tuning"],
summary="✨ (Enterprise) Cancel Fine-Tuning Jobs",
) )
@router.post( @router.post(
"/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel", "/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel",
dependencies=[Depends(user_api_key_auth)], dependencies=[Depends(user_api_key_auth)],
tags=["fine-tuning"], tags=["fine-tuning"],
summary="✨ (Enterprise) Cancel Fine-Tuning Jobs",
) )
async def retrieve_fine_tuning_job( async def retrieve_fine_tuning_job(
request: Request, request: Request,