File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
libs/vertexai/tests/integration_tests Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change 11"""Test ChatGoogleVertexAI chat model."""
22import os
33
4+ import pytest
5+
46from langchain_google_vertexai ._image_utils import image_bytes_to_b64_string
57from langchain_google_vertexai ._utils import load_image_from_gcs
68from langchain_google_vertexai .model_garden import ChatAnthropicVertex
79
810
11+ @pytest .mark .extended
912def test_pdf_gcs_uri ():
1013 gcs_uri = "gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf"
1114 llm = ChatAnthropicVertex (
@@ -29,6 +32,7 @@ def test_pdf_gcs_uri():
2932 assert len (res .content ) > 100
3033
3134
35+ @pytest .mark .extended
3236def test_pdf_byts ():
3337 gcs_uri = "gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf"
3438 llm = ChatAnthropicVertex (
@@ -54,6 +58,7 @@ def test_pdf_byts():
5458 assert len (res .content ) > 100
5559
5660
61+ @pytest .mark .extended
5762def test_https_image ():
5863 uri = "https://picsum.photos/seed/picsum/200/300.jpg"
5964
You can’t perform that action at this time.
0 commit comments