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

test ft response vertex ai

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-02 18:02:34 -07:00
parent ac6224c2b1
commit 960de8a3d0
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_fine_tuning_api.py
View file

@ -202,5 +202,10 @@ async def test_create_vertex_fine_tune_jobs():
vertex_location="us-central1", vertex_location="us-central1",
vertex_credentials=vertex_credentials, 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: except:
pass pass