From d588ce1f296de7b26bd7165317f330584aabd4c7 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Mon, 9 Sep 2024 16:10:10 -0700
Subject: [PATCH] community[patch]: fix extended deps (#26238)

---
 libs/community/extended_testing_deps.txt                      | 2 +-
 .../chat_models/test_javelin_ai_gateway.py                    | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)
 rename libs/community/tests/{unit_tests => integration_tests}/chat_models/test_javelin_ai_gateway.py (91%)

diff --git a/libs/community/extended_testing_deps.txt b/libs/community/extended_testing_deps.txt
index 0b2cea22981a6..d9879fd6aa07c 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 f3e7e361917ca..5a0e91e8fa8f6 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="<javelin-ai-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="<javelin-ai-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(