x

libs/langchain/langchain/chains/loading.py

@@ -23,7 +23,6 @@ from langchain.chains.hyde.base import HypotheticalDocumentEmbedder
 from langchain.chains.llm import LLMChain
 from langchain.chains.llm_checker.base import LLMCheckerChain
 from langchain.chains.llm_math.base import LLMMathChain
-from langchain.chains.llm_requests import LLMRequestsChain
 from langchain.chains.qa_with_sources.base import QAWithSourcesChain
 from langchain.chains.qa_with_sources.retrieval import RetrievalQAWithSourcesChain
 from langchain.chains.qa_with_sources.vector_db import VectorDBQAWithSourcesChain
@@ -32,6 +31,8 @@ from langchain.chains.retrieval_qa.base import RetrievalQA, VectorDBQA
 if TYPE_CHECKING:
     from langchain_community.chains.graph_qa.cypher import GraphCypherQAChain
 
+    from langchain.chains.llm_requests import LLMRequestsChain
+
 try:
     from langchain_community.llms.loading import load_llm, load_llm_from_config
 except ImportError:
@@ -598,6 +599,15 @@ def _load_api_chain(config: dict, **kwargs: Any) -> APIChain:
 
 
 def _load_llm_requests_chain(config: dict, **kwargs: Any) -> LLMRequestsChain:
+    try:
+        from langchain.chains.llm_requests import LLMRequestsChain
+    except ImportError:
+        raise ImportError(
+            "To use this LLMRequestsChain functionality you must install the "
+            "langchain package. "
+            "You can install it with `pip install langchain`"
+        )
+
     if "llm_chain" in config:
         llm_chain_config = config.pop("llm_chain")
         llm_chain = load_chain_from_config(llm_chain_config, **kwargs)

libs/langchain/tests/integration_tests/test_document_transformers.py

@@ -1,8 +1,6 @@
 """Integration test for embedding-based redundant doc filtering."""
 import pytest
 
-pytest.importorskip("langchain_community")
-
 from langchain_community.document_transformers.embeddings_redundant_filter import (
     EmbeddingsClusteringFilter,
     EmbeddingsRedundantFilter,

libs/langchain/tests/unit_tests/agents/test_imports.py

@@ -1,5 +1,4 @@
 from langchain import agents
-from tests.unit_tests import assert_all_importable
 
 EXPECTED_ALL = [
     "Agent",
@@ -49,4 +48,3 @@ EXPECTED_ALL = [
 
 def test_all_imports() -> None:
     assert set(agents.__all__) == set(EXPECTED_ALL)
-    assert_all_importable(agents)

libs/langchain/tests/unit_tests/agents/test_serialization.py

@@ -1,12 +1,15 @@
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
+import pytest
 from langchain_core.language_models import FakeListLLM
 from langchain_core.tools import Tool
 
 from langchain.agents.agent_types import AgentType
 from langchain.agents.initialize import initialize_agent, load_agent
 
+pytest.importorskip("langchain_community")
+
 
 def test_mrkl_serialization() -> None:
     agent = initialize_agent(

libs/langchain/tests/unit_tests/llms/test_base.py

@@ -1,7 +1,8 @@
 """Test base LLM functionality."""
+import importlib
+
 import pytest
 from sqlalchemy import Column, Integer, Sequence, String, create_engine
-import importlib
 
 try:
     from sqlalchemy.orm import declarative_base

libs/langchain/tests/unit_tests/test_cache.py

@@ -1,11 +1,11 @@
 """Test caching for LLMs and ChatModels."""
-import sqlite3
 from typing import Dict, Generator, List, Union
 
 import pytest
 
 pytest.importorskip("langchain_community")
 
+import sqlite3
 from _pytest.fixtures import FixtureRequest
 from langchain_community.cache import SQLAlchemyCache
 from langchain_core.caches import InMemoryCache

libs/langchain/tests/unit_tests/test_document_transformers.py

@@ -1,4 +1,7 @@
 """Unit tests for document transformers."""
+import pytest
+pytest.importorskip("langchain_community")
+
 from langchain_community.document_transformers.embeddings_redundant_filter import (
     _filter_similar_embeddings,
 )