From 960de8a3d094c56e9829120b6811e2c7c52fd7c8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 2 Aug 2024 18:02:34 -0700 Subject: [PATCH] test ft response vertex ai --- litellm/tests/test_fine_tuning_api.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/tests/test_fine_tuning_api.py b/litellm/tests/test_fine_tuning_api.py index 8f49271da5..4d8c0257ab 100644 --- a/litellm/tests/test_fine_tuning_api.py +++ b/litellm/tests/test_fine_tuning_api.py @@ -202,5 +202,10 @@ async def test_create_vertex_fine_tune_jobs(): vertex_location="us-central1", vertex_credentials=vertex_credentials, ) + print("vertex ai create fine tuning response=", create_fine_tuning_response) + + assert create_fine_tuning_response.id is not None + assert create_fine_tuning_response.model == "gemini-1.0-pro-002" + assert create_fine_tuning_response.object == "fine_tuning.job" except: pass