Skip to content

Commit 02e46e6

Browse files
authored
fix google multi-modal (#594)
1 parent 526753c commit 02e46e6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

libs/e2e-tests/e2e_tests/langchain/test_compatibility_rag.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -372,7 +372,7 @@ def vertex_gemini_multimodal_embedding():
372372

373373
@pytest.fixture()
374374
def vertex_gemini_pro_vision_llm():
375-
return ChatVertexAI(model_name="gemini-1.5-flash")
375+
return ChatVertexAI(model_name="gemini-pro-vision")
376376

377377

378378
@pytest.fixture()
@@ -381,9 +381,9 @@ def vertex_gemini_pro_llm():
381381

382382

383383
@pytest.fixture()
384-
def gemini_pro_vision_llm():
384+
def gemini_flash_llm():
385385
return ChatGoogleGenerativeAI(
386-
model="gemini-pro-vision", google_api_key=get_required_env("GOOGLE_API_KEY")
386+
model="gemini-1.5-flash", google_api_key=get_required_env("GOOGLE_API_KEY")
387387
)
388388

389389

@@ -404,7 +404,7 @@ def gemini_pro_llm():
404404
# disable due to this bug:
405405
# https://github.com/googleapis/python-aiplatform/issues/3227
406406
# ("vertex_gemini_multimodal_embedding", "vertex_gemini_pro_vision_llm"),
407-
("vertex_gemini_multimodal_embedding", "gemini_pro_vision_llm"),
407+
("vertex_gemini_multimodal_embedding", "gemini_flash_llm"),
408408
],
409409
)
410410
def test_multimodal(vector_store, embedding, llm, request, record_property):

0 commit comments

Comments
 (0)