_is_model_gemini_gemini_spec_model

This commit is contained in:
Ishaan Jaff 2025-03-26 10:53:23 -07:00
parent fb31006cd8
commit bbe69a47a9
4 changed files with 19 additions and 7 deletions

View file

@ -3380,8 +3380,7 @@ def test_gemini_fine_tuned_model_request_consistency():
with patch.object(client, "post", new=MagicMock()) as mock_post_1:
try:
response_1 = completion(
model="vertex_ai/ft-uuid",
base_model="vertex_ai/gemini-2.0-flash-001",
model="vertex_ai/gemini/ft-uuid",
messages=messages,
tools=tools,
tool_choice="auto",