diff --git a/libs/community/extended_testing_deps.txt b/libs/community/extended_testing_deps.txt index 0b2cea22981..d9879fd6aa0 100644 --- a/libs/community/extended_testing_deps.txt +++ b/libs/community/extended_testing_deps.txt @@ -34,7 +34,6 @@ hologres-vector==0.0.6 html2text>=2020.1.16 httpx>=0.24.1,<0.25 httpx-sse>=0.4.0,<0.5 -javelin-sdk>=0.1.8,<0.2 jinja2>=3,<4 jq>=1.4.1,<2 jsonschema>1 @@ -60,6 +59,7 @@ pgvector>=0.1.6,<0.2 praw>=7.7.1,<8 premai>=0.3.25,<0.4 psychicapi>=0.8.0,<0.9 +pydantic>=2.7.4,<3 py-trello>=0.19.0,<0.20 pyjwt>=2.8.0,<3 pymupdf>=1.22.3,<2 diff --git a/libs/community/tests/unit_tests/chat_models/test_javelin_ai_gateway.py b/libs/community/tests/integration_tests/chat_models/test_javelin_ai_gateway.py similarity index 91% rename from libs/community/tests/unit_tests/chat_models/test_javelin_ai_gateway.py rename to libs/community/tests/integration_tests/chat_models/test_javelin_ai_gateway.py index f3e7e361917..5a0e91e8fa8 100644 --- a/libs/community/tests/unit_tests/chat_models/test_javelin_ai_gateway.py +++ b/libs/community/tests/integration_tests/chat_models/test_javelin_ai_gateway.py @@ -1,12 +1,10 @@ """Test `Javelin AI Gateway` chat models""" -import pytest from pydantic import SecretStr from langchain_community.chat_models import ChatJavelinAIGateway -@pytest.mark.requires("javelin_sdk") def test_api_key_is_secret_string() -> None: llm = ChatJavelinAIGateway( gateway_uri="", @@ -18,7 +16,6 @@ def test_api_key_is_secret_string() -> None: assert llm.javelin_api_key.get_secret_value() == "secret-api-key" -@pytest.mark.requires("javelin_sdk") def test_api_key_masked_when_passed_via_constructor() -> None: llm = ChatJavelinAIGateway( gateway_uri="", @@ -32,7 +29,6 @@ def test_api_key_masked_when_passed_via_constructor() -> None: assert "secret-api-key" not in repr(llm) -@pytest.mark.requires("javelin_sdk") def test_api_key_alias() -> None: for model in [ ChatJavelinAIGateway(