langchain 0.3.25__tar.gz → 0.3.26__tar.gz

{langchain-0.3.25 → langchain-0.3.26}/PKG-INFO

@@ -1,15 +1,15 @@
 Metadata-Version: 2.1
 Name: langchain
-Version: 0.3.25
+Version: 0.3.26
 Summary: Building applications with LLMs through composability
 License: MIT
 Project-URL: Source Code, https://github.com/langchain-ai/langchain/tree/master/libs/langchain
 Project-URL: Release Notes, https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain%3D%3D0%22&expanded=true
 Project-URL: repository, https://github.com/langchain-ai/langchain
 Requires-Python: >=3.9
-Requires-Dist: langchain-core<1.0.0,>=0.3.58
+Requires-Dist: langchain-core<1.0.0,>=0.3.66
 Requires-Dist: langchain-text-splitters<1.0.0,>=0.3.8
-Requires-Dist: langsmith<0.4,>=0.1.17
+Requires-Dist: langsmith>=0.1.17
 Requires-Dist: pydantic<3.0.0,>=2.7.4
 Requires-Dist: SQLAlchemy<3,>=1.4
 Requires-Dist: requests<3,>=2

{langchain-0.3.25 → langchain-0.3.26}/langchain/agents/agent_types.py

@@ -15,7 +15,7 @@ from langchain._api.deprecation import AGENT_DEPRECATION_WARNING
 class AgentType(str, Enum):
     """An enum for agent types.
 
-    See documentation: https://python.langchain.com/docs/modules/agents/agent_types/
+    See documentation: https://python.langchain.com/api_reference/langchain/agents/langchain.agents.agent_types.AgentType.html
     """
 
     ZERO_SHOT_REACT_DESCRIPTION = "zero-shot-react-description"

{langchain-0.3.25 → langchain-0.3.26}/langchain/agents/initialize.py

@@ -11,7 +11,8 @@ from langchain_core.tools import BaseTool
 from langchain._api.deprecation import AGENT_DEPRECATION_WARNING
 from langchain.agents.agent import AgentExecutor
 from langchain.agents.agent_types import AgentType
-from langchain.agents.loading import AGENT_TO_CLASS, load_agent
+from langchain.agents.loading import load_agent
+from langchain.agents.types import AGENT_TO_CLASS
 
 
 @deprecated(

{langchain-0.3.25 → langchain-0.3.26}/langchain/agents/openai_functions_agent/base.py

@@ -13,11 +13,11 @@ from langchain_core.messages import (
 )
 from langchain_core.prompts import BasePromptTemplate
 from langchain_core.prompts.chat import (
-    BaseMessagePromptTemplate,
     ChatPromptTemplate,
     HumanMessagePromptTemplate,
     MessagesPlaceholder,
 )
+from langchain_core.prompts.message import BaseMessagePromptTemplate
 from langchain_core.runnables import Runnable, RunnablePassthrough
 from langchain_core.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_function

{langchain-0.3.25 → langchain-0.3.26}/langchain/agents/openai_functions_multi_agent/base.py

@@ -17,11 +17,11 @@ from langchain_core.messages import (
 )
 from langchain_core.prompts import BasePromptTemplate
 from langchain_core.prompts.chat import (
-    BaseMessagePromptTemplate,
     ChatPromptTemplate,
     HumanMessagePromptTemplate,
     MessagesPlaceholder,
 )
+from langchain_core.prompts.message import BaseMessagePromptTemplate
 from langchain_core.tools import BaseTool
 from pydantic import model_validator
 from typing_extensions import Self

{langchain-0.3.25 → langchain-0.3.26}/langchain/agents/output_parsers/json.py

@@ -5,7 +5,7 @@ from typing import Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException
-from langchain_core.output_parsers.json import parse_json_markdown
+from langchain_core.utils.json import parse_json_markdown
 
 from langchain.agents.agent import AgentOutputParser
 

{langchain-0.3.25 → langchain-0.3.26}/langchain/callbacks/manager.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any
 
+from langchain_core.callbacks import Callbacks
 from langchain_core.callbacks.manager import (
     AsyncCallbackManager,
     AsyncCallbackManagerForChainGroup,
@@ -18,7 +19,6 @@ from langchain_core.callbacks.manager import (
     CallbackManagerForLLMRun,
     CallbackManagerForRetrieverRun,
     CallbackManagerForToolRun,
-    Callbacks,
     ParentRunManager,
     RunManager,
     ahandle_event,

langchain-0.3.26/langchain/callbacks/tracers/base.py

@@ -0,0 +1,6 @@
+"""Base interfaces for tracing runs."""
+
+from langchain_core.exceptions import TracerException
+from langchain_core.tracers.base import BaseTracer
+
+__all__ = ["BaseTracer", "TracerException"]

{langchain-0.3.25 → langchain-0.3.26}/langchain/chains/base.py

@@ -27,7 +27,7 @@ from langchain_core.runnables import (
     ensure_config,
     run_in_executor,
 )
-from langchain_core.runnables.utils import create_model
+from langchain_core.utils.pydantic import create_model
 from pydantic import (
     BaseModel,
     ConfigDict,

{langchain-0.3.25 → langchain-0.3.26}/langchain/chains/combine_documents/base.py

@@ -11,7 +11,7 @@ from langchain_core.callbacks import (
 from langchain_core.documents import Document
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
 from langchain_core.runnables.config import RunnableConfig
-from langchain_core.runnables.utils import create_model
+from langchain_core.utils.pydantic import create_model
 from langchain_text_splitters import RecursiveCharacterTextSplitter, TextSplitter
 from pydantic import BaseModel, Field
 

{langchain-0.3.25 → langchain-0.3.26}/langchain/chains/combine_documents/map_reduce.py

@@ -8,7 +8,7 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.runnables.config import RunnableConfig
-from langchain_core.runnables.utils import create_model
+from langchain_core.utils.pydantic import create_model
 from pydantic import BaseModel, ConfigDict, model_validator
 
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain

{langchain-0.3.25 → langchain-0.3.26}/langchain/chains/combine_documents/map_rerank.py

@@ -9,7 +9,7 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.runnables.config import RunnableConfig
-from langchain_core.runnables.utils import create_model
+from langchain_core.utils.pydantic import create_model
 from pydantic import BaseModel, ConfigDict, model_validator
 from typing_extensions import Self
 

{langchain-0.3.25 → langchain-0.3.26}/langchain/chains/conversation/base.py

@@ -13,10 +13,7 @@ from langchain.memory.buffer import ConversationBufferMemory
 
 @deprecated(
     since="0.2.7",
-    alternative=(
-        "RunnableWithMessageHistory: "
-        "https://python.langchain.com/v0.2/api_reference/core/runnables/langchain_core.runnables.history.RunnableWithMessageHistory.html"  # noqa: E501
-    ),
+    alternative="langchain_core.runnables.history.RunnableWithMessageHistory",
     removal="1.0",
 )
 class ConversationChain(LLMChain):

{langchain-0.3.25 → langchain-0.3.26}/langchain/chains/loading.py

@@ -219,33 +219,13 @@ def _load_reduce_documents_chain(config: dict, **kwargs: Any) -> ReduceDocuments
 
 
 def _load_llm_bash_chain(config: dict, **kwargs: Any) -> Any:
-    from langchain_experimental.llm_bash.base import LLMBashChain
-
-    llm_chain = None
-    if "llm_chain" in config:
-        llm_chain_config = config.pop("llm_chain")
-        llm_chain = load_chain_from_config(llm_chain_config, **kwargs)
-    elif "llm_chain_path" in config:
-        llm_chain = load_chain(config.pop("llm_chain_path"), **kwargs)
-    # llm attribute is deprecated in favor of llm_chain, here to support old configs
-    elif "llm" in config:
-        llm_config = config.pop("llm")
-        llm = load_llm_from_config(llm_config, **kwargs)
-    # llm_path attribute is deprecated in favor of llm_chain_path,
-    # its to support old configs
-    elif "llm_path" in config:
-        llm = load_llm(config.pop("llm_path"), **kwargs)
-    else:
-        raise ValueError("One of `llm_chain` or `llm_chain_path` must be present.")
-    if "prompt" in config:
-        prompt_config = config.pop("prompt")
-        prompt = load_prompt_from_config(prompt_config)
-    elif "prompt_path" in config:
-        prompt = load_prompt(config.pop("prompt_path"))
-    if llm_chain:
-        return LLMBashChain(llm_chain=llm_chain, prompt=prompt, **config)
-    else:
-        return LLMBashChain(llm=llm, prompt=prompt, **config)
+    """Load LLM Bash chain from config dict"""
+    raise NotImplementedError(
+        "LLMBash Chain is not available through LangChain anymore. "
+        "The relevant code can be found in langchain_experimental, "
+        "but it is not appropriate for production usage due to security "
+        "concerns. Please refer to langchain-experimental repository for more details."
+    )
 
 
 def _load_llm_checker_chain(config: dict, **kwargs: Any) -> LLMCheckerChain:
@@ -336,16 +316,12 @@ def _load_map_rerank_documents_chain(
 
 
 def _load_pal_chain(config: dict, **kwargs: Any) -> Any:
-    from langchain_experimental.pal_chain import PALChain
-
-    if "llm_chain" in config:
-        llm_chain_config = config.pop("llm_chain")
-        llm_chain = load_chain_from_config(llm_chain_config, **kwargs)
-    elif "llm_chain_path" in config:
-        llm_chain = load_chain(config.pop("llm_chain_path"), **kwargs)
-    else:
-        raise ValueError("One of `llm_chain` or `llm_chain_path` must be present.")
-    return PALChain(llm_chain=llm_chain, **config)
+    raise NotImplementedError(
+        "PALChain is not available through LangChain anymore. "
+        "The relevant code can be found in langchain_experimental, "
+        "but it is not appropriate for production usage due to security "
+        "concerns. Please refer to langchain-experimental repository for more details."
+    )
 
 
 def _load_refine_documents_chain(config: dict, **kwargs: Any) -> RefineDocumentsChain:
@@ -399,30 +375,15 @@ def _load_qa_with_sources_chain(config: dict, **kwargs: Any) -> QAWithSourcesCha
 
 
 def _load_sql_database_chain(config: dict, **kwargs: Any) -> Any:
-    from langchain_experimental.sql import SQLDatabaseChain
-
-    if "database" in kwargs:
-        database = kwargs.pop("database")
-    else:
-        raise ValueError("`database` must be present.")
-    if "llm_chain" in config:
-        llm_chain_config = config.pop("llm_chain")
-        chain = load_chain_from_config(llm_chain_config, **kwargs)
-        return SQLDatabaseChain(llm_chain=chain, database=database, **config)
-    if "llm" in config:
-        llm_config = config.pop("llm")
-        llm = load_llm_from_config(llm_config, **kwargs)
-    elif "llm_path" in config:
-        llm = load_llm(config.pop("llm_path"), **kwargs)
-    else:
-        raise ValueError("One of `llm` or `llm_path` must be present.")
-    if "prompt" in config:
-        prompt_config = config.pop("prompt")
-        prompt = load_prompt_from_config(prompt_config)
-    else:
-        prompt = None
-
-    return SQLDatabaseChain.from_llm(llm, database, prompt=prompt, **config)
+    """Load SQL Database chain from config dict."""
+    raise NotImplementedError(
+        "SQLDatabaseChain is not available through LangChain anymore. "
+        "The relevant code can be found in langchain_experimental, "
+        "but it is not appropriate for production usage due to security "
+        "concerns. Please refer to langchain-experimental repository for more details, "
+        "or refer to this tutorial for best practices: "
+        "https://python.langchain.com/docs/tutorials/sql_qa/"
+    )
 
 
 def _load_vector_db_qa_with_sources_chain(

{langchain-0.3.25 → langchain-0.3.26}/langchain/embeddings/cache.py

@@ -12,9 +12,9 @@ from __future__ import annotations
 import hashlib
 import json
 import uuid
+import warnings
 from collections.abc import Sequence
-from functools import partial
-from typing import Callable, Optional, Union, cast
+from typing import Callable, Literal, Optional, Union, cast
 
 from langchain_core.embeddings import Embeddings
 from langchain_core.stores import BaseStore, ByteStore
@@ -25,20 +25,51 @@ from langchain.storage.encoder_backed import EncoderBackedStore
 NAMESPACE_UUID = uuid.UUID(int=1985)
 
 
-def _hash_string_to_uuid(input_string: str) -> uuid.UUID:
-    """Hash a string and returns the corresponding UUID."""
-    hash_value = hashlib.sha1(input_string.encode("utf-8")).hexdigest()
-    return uuid.uuid5(NAMESPACE_UUID, hash_value)
+def _sha1_hash_to_uuid(text: str) -> uuid.UUID:
+    """Return a UUID derived from *text* using SHA‑1 (deterministic).
 
+    Deterministic and fast, **but not collision‑resistant**.
 
-def _key_encoder(key: str, namespace: str) -> str:
-    """Encode a key."""
-    return namespace + str(_hash_string_to_uuid(key))
+    A malicious attacker could try to create two different texts that hash to the same
+    UUID. This may not necessarily be an issue in the context of caching embeddings,
+    but new applications should swap this out for a stronger hash function like
+    xxHash, BLAKE2 or SHA‑256, which are collision-resistant.
+    """
+    sha1_hex = hashlib.sha1(text.encode("utf-8")).hexdigest()
+    # Embed the hex string in `uuid5` to obtain a valid UUID.
+    return uuid.uuid5(NAMESPACE_UUID, sha1_hex)
+
+
+def _make_default_key_encoder(namespace: str, algorithm: str) -> Callable[[str], str]:
+    """Create a default key encoder function.
+
+    Args:
+       namespace: Prefix that segregates keys from different embedding models.
+       algorithm:
+           * `sha1` - fast but not collision‑resistant
+           * `blake2b` - cryptographically strong, faster than SHA‑1
+           * `sha256` - cryptographically strong, slower than SHA‑1
+           * `sha512` - cryptographically strong, slower than SHA‑1
 
+    Returns:
+        A function that encodes a key using the specified algorithm.
+    """
+    if algorithm == "sha1":
+        _warn_about_sha1_encoder()
+
+    def _key_encoder(key: str) -> str:
+        """Encode a key using the specified algorithm."""
+        if algorithm == "sha1":
+            return f"{namespace}{_sha1_hash_to_uuid(key)}"
+        if algorithm == "blake2b":
+            return f"{namespace}{hashlib.blake2b(key.encode('utf-8')).hexdigest()}"
+        if algorithm == "sha256":
+            return f"{namespace}{hashlib.sha256(key.encode('utf-8')).hexdigest()}"
+        if algorithm == "sha512":
+            return f"{namespace}{hashlib.sha512(key.encode('utf-8')).hexdigest()}"
+        raise ValueError(f"Unsupported algorithm: {algorithm}")
 
-def _create_key_encoder(namespace: str) -> Callable[[str], str]:
-    """Create an encoder for a key."""
-    return partial(_key_encoder, namespace=namespace)
+    return _key_encoder
 
 
 def _value_serializer(value: Sequence[float]) -> bytes:
@@ -51,6 +82,28 @@ def _value_deserializer(serialized_value: bytes) -> list[float]:
     return cast(list[float], json.loads(serialized_value.decode()))
 
 
+# The warning is global; track emission, so it appears only once.
+_warned_about_sha1: bool = False
+
+
+def _warn_about_sha1_encoder() -> None:
+    """Emit a one‑time warning about SHA‑1 collision weaknesses."""
+    global _warned_about_sha1
+    if not _warned_about_sha1:
+        warnings.warn(
+            "Using default key encoder: SHA‑1 is *not* collision‑resistant. "
+            "While acceptable for most cache scenarios, a motivated attacker "
+            "can craft two different payloads that map to the same cache key. "
+            "If that risk matters in your environment, supply a stronger "
+            "encoder (e.g. SHA‑256 or BLAKE2) via the `key_encoder` argument. "
+            "If you change the key encoder, consider also creating a new cache, "
+            "to avoid (the potential for) collisions with existing keys.",
+            category=UserWarning,
+            stacklevel=2,
+        )
+        _warned_about_sha1 = True
+
+
 class CacheBackedEmbeddings(Embeddings):
     """Interface for caching results from embedding models.
 
@@ -234,6 +287,9 @@ class CacheBackedEmbeddings(Embeddings):
         namespace: str = "",
         batch_size: Optional[int] = None,
         query_embedding_cache: Union[bool, ByteStore] = False,
+        key_encoder: Union[
+            Callable[[str], str], Literal["sha1", "blake2b", "sha256", "sha512"]
+        ] = "sha1",
     ) -> CacheBackedEmbeddings:
         """On-ramp that adds the necessary serialization and encoding to the store.
 
@@ -248,9 +304,39 @@ class CacheBackedEmbeddings(Embeddings):
             query_embedding_cache: The cache to use for storing query embeddings.
                 True to use the same cache as document embeddings.
                 False to not cache query embeddings.
+            key_encoder: Optional callable to encode keys. If not provided,
+                a default encoder using SHA‑1 will be used. SHA-1 is not
+                collision-resistant, and a motivated attacker could craft two
+                different texts that hash to the same cache key.
+
+                New applications should use one of the alternative encoders
+                or provide a custom and strong key encoder function to avoid this risk.
+
+                If you change a key encoder in an existing cache, consider
+                just creating a new cache, to avoid (the potential for)
+                collisions with existing keys or having duplicate keys
+                for the same text in the cache.
+
+        Returns:
+            An instance of CacheBackedEmbeddings that uses the provided cache.
         """
-        namespace = namespace
-        key_encoder = _create_key_encoder(namespace)
+        if isinstance(key_encoder, str):
+            key_encoder = _make_default_key_encoder(namespace, key_encoder)
+        elif callable(key_encoder):
+            # If a custom key encoder is provided, it should not be used with a
+            # namespace.
+            # A user can handle namespacing in directly their custom key encoder.
+            if namespace:
+                raise ValueError(
+                    "Do not supply `namespace` when using a custom key_encoder; "
+                    "add any prefixing inside the encoder itself."
+                )
+        else:
+            raise ValueError(
+                "key_encoder must be either 'blake2b', 'sha1', 'sha256', 'sha512' "
+                "or a callable that encodes keys."
+            )
+
         document_embedding_store = EncoderBackedStore[str, list[float]](
             document_embedding_cache,
             key_encoder,

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/agents/trajectory_eval_chain.py

@@ -16,10 +16,10 @@ from typing import (
 )
 
 from langchain_core.agents import AgentAction
+from langchain_core.callbacks import Callbacks
 from langchain_core.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
-    Callbacks,
 )
 from langchain_core.exceptions import OutputParserException
 from langchain_core.language_models import BaseLanguageModel

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/comparison/eval_chain.py

@@ -6,7 +6,7 @@ import logging
 import re
 from typing import Any, Optional, Union
 
-from langchain_core.callbacks.manager import Callbacks
+from langchain_core.callbacks import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts.prompt import PromptTemplate

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/criteria/eval_chain.py

@@ -5,7 +5,7 @@ from collections.abc import Mapping
 from enum import Enum
 from typing import Any, Optional, Union
 
-from langchain_core.callbacks.manager import Callbacks
+from langchain_core.callbacks import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts import BasePromptTemplate

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/embedding_distance/base.py

@@ -6,10 +6,10 @@ from enum import Enum
 from importlib import util
 from typing import Any, Optional
 
+from langchain_core.callbacks import Callbacks
 from langchain_core.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
-    Callbacks,
 )
 from langchain_core.embeddings import Embeddings
 from langchain_core.utils import pre_init

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/qa/eval_chain.py

@@ -7,7 +7,7 @@ import string
 from collections.abc import Sequence
 from typing import Any, Optional
 
-from langchain_core.callbacks.manager import Callbacks
+from langchain_core.callbacks import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import PromptTemplate
 from pydantic import ConfigDict

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/scoring/eval_chain.py

@@ -6,7 +6,7 @@ import logging
 import re
 from typing import Any, Optional, Union
 
-from langchain_core.callbacks.manager import Callbacks
+from langchain_core.callbacks import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts.prompt import PromptTemplate

{langchain-0.3.25 → langchain-0.3.26}/langchain/evaluation/string_distance/base.py

@@ -3,10 +3,10 @@
 from enum import Enum
 from typing import Any, Callable, Optional
 
+from langchain_core.callbacks import Callbacks
 from langchain_core.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
-    Callbacks,
 )
 from langchain_core.utils import pre_init
 from pydantic import Field

{langchain-0.3.25 → langchain-0.3.26}/langchain/memory/buffer.py

@@ -1,10 +1,11 @@
 from typing import Any, Optional
 
 from langchain_core._api import deprecated
+from langchain_core.memory import BaseMemory
 from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_core.utils import pre_init
 
-from langchain.memory.chat_memory import BaseChatMemory, BaseMemory
+from langchain.memory.chat_memory import BaseChatMemory
 from langchain.memory.utils import get_prompt_input_key
 
 

{langchain-0.3.25 → langchain-0.3.26}/langchain/memory/vectorstore.py

@@ -5,10 +5,10 @@ from typing import Any, Optional, Union
 
 from langchain_core._api import deprecated
 from langchain_core.documents import Document
+from langchain_core.memory import BaseMemory
 from langchain_core.vectorstores import VectorStoreRetriever
 from pydantic import Field
 
-from langchain.memory.chat_memory import BaseMemory
 from langchain.memory.utils import get_prompt_input_key
 
 

{langchain-0.3.25 → langchain-0.3.26}/langchain/prompts/chat.py

@@ -2,7 +2,6 @@ from langchain_core.prompt_values import ChatPromptValue, ChatPromptValueConcret
 from langchain_core.prompts.chat import (
     AIMessagePromptTemplate,
     BaseChatPromptTemplate,
-    BaseMessagePromptTemplate,
     BaseStringMessagePromptTemplate,
     ChatMessagePromptTemplate,
     ChatPromptTemplate,
@@ -34,3 +33,5 @@ __all__ = [
     "MessageLike",
     "MessageLikeRepresentation",
 ]
+
+from langchain_core.prompts.message import BaseMessagePromptTemplate

{langchain-0.3.25 → langchain-0.3.26}/langchain/retrievers/contextual_compression.py

@@ -4,14 +4,10 @@ from langchain_core.callbacks import (
     AsyncCallbackManagerForRetrieverRun,
     CallbackManagerForRetrieverRun,
 )
-from langchain_core.documents import Document
+from langchain_core.documents import BaseDocumentCompressor, Document
 from langchain_core.retrievers import BaseRetriever, RetrieverLike
 from pydantic import ConfigDict
 
-from langchain.retrievers.document_compressors.base import (
-    BaseDocumentCompressor,
-)
-
 
 class ContextualCompressionRetriever(BaseRetriever):
     """Retriever that wraps a base retriever and compresses the results."""

{langchain-0.3.25 → langchain-0.3.26}/langchain/retrievers/document_compressors/__init__.py

@@ -34,9 +34,10 @@ def __getattr__(name: str) -> Any:
 __all__ = [
     "DocumentCompressorPipeline",
     "EmbeddingsFilter",
+    "FlashrankRerank",
     "LLMListwiseRerank",
     "LLMChainExtractor",
     "LLMChainFilter",
     "CohereRerank",
     "CrossEncoderReranker",
-] + list(_module_lookup.keys())
+]

{langchain-0.3.25 → langchain-0.3.26}/langchain/retrievers/document_compressors/base.py

@@ -2,7 +2,7 @@ from collections.abc import Sequence
 from inspect import signature
 from typing import Optional, Union
 
-from langchain_core.callbacks.manager import Callbacks
+from langchain_core.callbacks import Callbacks
 from langchain_core.documents import (
     BaseDocumentCompressor,
     BaseDocumentTransformer,

{langchain-0.3.25 → langchain-0.3.26}/langchain/retrievers/document_compressors/chain_extract.py

@@ -5,8 +5,8 @@ from __future__ import annotations
 from collections.abc import Sequence
 from typing import Any, Callable, Optional, cast
 
-from langchain_core.callbacks.manager import Callbacks
-from langchain_core.documents import Document
+from langchain_core.callbacks import Callbacks
+from langchain_core.documents import BaseDocumentCompressor, Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import PromptTemplate
@@ -14,7 +14,6 @@ from langchain_core.runnables import Runnable
 from pydantic import ConfigDict
 
 from langchain.chains.llm import LLMChain
-from langchain.retrievers.document_compressors.base import BaseDocumentCompressor
 from langchain.retrievers.document_compressors.chain_extract_prompt import (
     prompt_template,
 )

{langchain-0.3.25 → langchain-0.3.26}/langchain/retrievers/document_compressors/chain_filter.py

@@ -3,8 +3,8 @@
 from collections.abc import Sequence
 from typing import Any, Callable, Optional
 
-from langchain_core.callbacks.manager import Callbacks
-from langchain_core.documents import Document
+from langchain_core.callbacks import Callbacks
+from langchain_core.documents import BaseDocumentCompressor, Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
@@ -14,7 +14,6 @@ from pydantic import ConfigDict
 
 from langchain.chains import LLMChain
 from langchain.output_parsers.boolean import BooleanOutputParser
-from langchain.retrievers.document_compressors.base import BaseDocumentCompressor
 from langchain.retrievers.document_compressors.chain_filter_prompt import (
     prompt_template,
 )

{langchain-0.3.25 → langchain-0.3.26}/langchain/retrievers/document_compressors/cohere_rerank.py

@@ -5,13 +5,11 @@ from copy import deepcopy
 from typing import Any, Optional, Union
 
 from langchain_core._api.deprecation import deprecated
-from langchain_core.callbacks.manager import Callbacks
-from langchain_core.documents import Document
+from langchain_core.callbacks import Callbacks
+from langchain_core.documents import BaseDocumentCompressor, Document
 from langchain_core.utils import get_from_dict_or_env
 from pydantic import ConfigDict, model_validator
 
-from langchain.retrievers.document_compressors.base import BaseDocumentCompressor
-
 
 @deprecated(
     since="0.0.30", removal="1.0", alternative_import="langchain_cohere.CohereRerank"