Skip to content

Commit 7d90734

Browse files
authored
Fix (#841)
1 parent 19cf6e8 commit 7d90734

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

libs/vertexai/tests/integration_tests/test_anthropic_files.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,14 @@
11
"""Test ChatGoogleVertexAI chat model."""
22
import os
33

4+
import pytest
5+
46
from langchain_google_vertexai._image_utils import image_bytes_to_b64_string
57
from langchain_google_vertexai._utils import load_image_from_gcs
68
from langchain_google_vertexai.model_garden import ChatAnthropicVertex
79

810

11+
@pytest.mark.extended
912
def 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
3236
def 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
5762
def test_https_image():
5863
uri = "https://picsum.photos/seed/picsum/200/300.jpg"
5964

0 commit comments

Comments
 (0)