From bd15d0f4ca28d26b9ede310e0a85fae8c7e73b2d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 31 Jul 2024 16:41:14 -0700 Subject: [PATCH] fix mark fine tuning endpoints as enteprrise --- litellm/proxy/fine_tuning_endpoints/endpoints.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/proxy/fine_tuning_endpoints/endpoints.py b/litellm/proxy/fine_tuning_endpoints/endpoints.py index 5439ea9b5..8c5ceef01 100644 --- a/litellm/proxy/fine_tuning_endpoints/endpoints.py +++ b/litellm/proxy/fine_tuning_endpoints/endpoints.py @@ -74,11 +74,13 @@ def get_fine_tuning_provider_config( "/v1/fine_tuning/jobs", dependencies=[Depends(user_api_key_auth)], tags=["fine-tuning"], + summary="✨ (Enterprise) Create Fine-Tuning Job", ) @router.post( "/fine_tuning/jobs", dependencies=[Depends(user_api_key_auth)], tags=["fine-tuning"], + summary="✨ (Enterprise) Create Fine-Tuning Job", ) async def create_fine_tuning_job( request: Request, @@ -199,11 +201,13 @@ async def create_fine_tuning_job( "/v1/fine_tuning/jobs", dependencies=[Depends(user_api_key_auth)], tags=["fine-tuning"], + summary="✨ (Enterprise) List Fine-Tuning Jobs", ) @router.get( "/fine_tuning/jobs", dependencies=[Depends(user_api_key_auth)], tags=["fine-tuning"], + summary="✨ (Enterprise) List Fine-Tuning Jobs", ) async def list_fine_tuning_jobs( request: Request, @@ -306,11 +310,13 @@ async def list_fine_tuning_jobs( "/v1/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel", dependencies=[Depends(user_api_key_auth)], tags=["fine-tuning"], + summary="✨ (Enterprise) Cancel Fine-Tuning Jobs", ) @router.post( "/fine_tuning/jobs/{fine_tuning_job_id:path}/cancel", dependencies=[Depends(user_api_key_auth)], tags=["fine-tuning"], + summary="✨ (Enterprise) Cancel Fine-Tuning Jobs", ) async def retrieve_fine_tuning_job( request: Request,