langchain 0.3.25__py3-none-any.whl → 0.3.27__py3-none-any.whl

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,
@@ -17,20 +16,22 @@ from langchain_core.prompts.chat import (
 )
 
 __all__ = [
+    "AIMessagePromptTemplate",
+    "BaseChatPromptTemplate",
     "BaseMessagePromptTemplate",
-    "MessagesPlaceholder",
     "BaseStringMessagePromptTemplate",
     "ChatMessagePromptTemplate",
-    "HumanMessagePromptTemplate",
-    "AIMessagePromptTemplate",
-    "SystemMessagePromptTemplate",
-    "BaseChatPromptTemplate",
     "ChatPromptTemplate",
     "ChatPromptValue",
     "ChatPromptValueConcrete",
-    "_convert_to_message",
-    "_create_template_from_message_type",
-    "MessagePromptTemplateT",
+    "HumanMessagePromptTemplate",
     "MessageLike",
     "MessageLikeRepresentation",
+    "MessagePromptTemplateT",
+    "MessagesPlaceholder",
+    "SystemMessagePromptTemplate",
+    "_convert_to_message",
+    "_create_template_from_message_type",
 ]
+
+from langchain_core.prompts.message import BaseMessagePromptTemplate

langchain/prompts/example_selector/__init__.py

@@ -21,7 +21,9 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUPS = {
-    "NGramOverlapExampleSelector": "langchain_community.example_selectors.ngram_overlap"
+    "NGramOverlapExampleSelector": (
+        "langchain_community.example_selectors.ngram_overlap"
+    ),
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUPS)

langchain/prompts/example_selector/semantic_similarity.py

@@ -5,7 +5,7 @@ from langchain_core.example_selectors.semantic_similarity import (
 )
 
 __all__ = [
-    "sorted_values",
-    "SemanticSimilarityExampleSelector",
     "MaxMarginalRelevanceExampleSelector",
+    "SemanticSimilarityExampleSelector",
+    "sorted_values",
 ]

langchain/prompts/few_shot.py

@@ -5,7 +5,7 @@ from langchain_core.prompts.few_shot import (
 )
 
 __all__ = [
-    "FewShotPromptTemplate",
     "FewShotChatMessagePromptTemplate",
+    "FewShotPromptTemplate",
     "_FewShotPromptTemplateMixin",
 ]

langchain/prompts/loading.py

@@ -11,13 +11,13 @@ from langchain_core.prompts.loading import (
 from langchain_core.utils.loading import try_load_from_hub
 
 __all__ = [
-    "load_prompt_from_config",
-    "load_prompt",
-    "try_load_from_hub",
     "_load_examples",
     "_load_few_shot_prompt",
     "_load_output_parser",
     "_load_prompt",
     "_load_prompt_from_file",
     "_load_template",
+    "load_prompt",
+    "load_prompt_from_config",
+    "try_load_from_hub",
 ]

langchain/prompts/prompt.py

@@ -3,4 +3,4 @@ from langchain_core.prompts.prompt import PromptTemplate
 # For backwards compatibility.
 Prompt = PromptTemplate
 
-__all__ = ["PromptTemplate", "Prompt"]
+__all__ = ["Prompt", "PromptTemplate"]

langchain/retrievers/__init__.py

@@ -148,8 +148,8 @@ __all__ = [
     "GoogleDocumentAIWarehouseRetriever",
     "GoogleVertexAIMultiTurnSearchRetriever",
     "GoogleVertexAISearchRetriever",
-    "KayAiRetriever",
     "KNNRetriever",
+    "KayAiRetriever",
     "LlamaIndexGraphRetriever",
     "LlamaIndexRetriever",
     "MergerRetriever",
@@ -157,22 +157,22 @@ __all__ = [
     "MilvusRetriever",
     "MultiQueryRetriever",
     "MultiVectorRetriever",
+    "NeuralDBRetriever",
     "OutlineRetriever",
     "ParentDocumentRetriever",
     "PineconeHybridSearchRetriever",
     "PubMedRetriever",
-    "RemoteLangChainRetriever",
     "RePhraseQueryRetriever",
-    "SelfQueryRetriever",
+    "RemoteLangChainRetriever",
     "SVMRetriever",
-    "TavilySearchAPIRetriever",
+    "SelfQueryRetriever",
     "TFIDFRetriever",
+    "TavilySearchAPIRetriever",
     "TimeWeightedVectorStoreRetriever",
     "VespaRetriever",
     "WeaviateHybridSearchRetriever",
     "WebResearchRetriever",
     "WikipediaRetriever",
     "ZepRetriever",
-    "NeuralDBRetriever",
     "ZillizRetriever",
 ]

langchain/retrievers/bedrock.py

@@ -27,7 +27,7 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "VectorSearchConfig",
-    "RetrievalConfig",
     "AmazonKnowledgeBasesRetriever",
+    "RetrievalConfig",
+    "VectorSearchConfig",
 ]

langchain/retrievers/bm25.py

@@ -23,6 +23,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "default_preprocessing_func",
     "BM25Retriever",
+    "default_preprocessing_func",
 ]

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."""
@@ -42,15 +38,18 @@ class ContextualCompressionRetriever(BaseRetriever):
             Sequence of relevant documents
         """
         docs = self.base_retriever.invoke(
-            query, config={"callbacks": run_manager.get_child()}, **kwargs
+            query,
+            config={"callbacks": run_manager.get_child()},
+            **kwargs,
         )
         if docs:
             compressed_docs = self.base_compressor.compress_documents(
-                docs, query, callbacks=run_manager.get_child()
+                docs,
+                query,
+                callbacks=run_manager.get_child(),
             )
             return list(compressed_docs)
-        else:
-            return []
+        return []
 
     async def _aget_relevant_documents(
         self,
@@ -68,12 +67,15 @@ class ContextualCompressionRetriever(BaseRetriever):
             List of relevant documents
         """
         docs = await self.base_retriever.ainvoke(
-            query, config={"callbacks": run_manager.get_child()}, **kwargs
+            query,
+            config={"callbacks": run_manager.get_child()},
+            **kwargs,
         )
         if docs:
             compressed_docs = await self.base_compressor.acompress_documents(
-                docs, query, callbacks=run_manager.get_child()
+                docs,
+                query,
+                callbacks=run_manager.get_child(),
             )
             return list(compressed_docs)
-        else:
-            return []
+        return []

langchain/retrievers/docarray.py

@@ -23,6 +23,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "SearchType",
     "DocArrayRetriever",
+    "SearchType",
 ]

langchain/retrievers/document_compressors/__init__.py

@@ -28,15 +28,17 @@ def __getattr__(name: str) -> Any:
     if name in _module_lookup:
         module = importlib.import_module(_module_lookup[name])
         return getattr(module, name)
-    raise AttributeError(f"module {__name__} has no attribute {name}")
+    msg = f"module {__name__} has no attribute {name}"
+    raise AttributeError(msg)
 
 
 __all__ = [
+    "CohereRerank",
+    "CrossEncoderReranker",
     "DocumentCompressorPipeline",
     "EmbeddingsFilter",
-    "LLMListwiseRerank",
+    "FlashrankRerank",
     "LLMChainExtractor",
     "LLMChainFilter",
-    "CohereRerank",
-    "CrossEncoderReranker",
-] + list(_module_lookup.keys())
+    "LLMListwiseRerank",
+]

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,
@@ -32,20 +32,23 @@ class DocumentCompressorPipeline(BaseDocumentCompressor):
             if isinstance(_transformer, BaseDocumentCompressor):
                 accepts_callbacks = (
                     signature(_transformer.compress_documents).parameters.get(
-                        "callbacks"
+                        "callbacks",
                     )
                     is not None
                 )
                 if accepts_callbacks:
                     documents = _transformer.compress_documents(
-                        documents, query, callbacks=callbacks
+                        documents,
+                        query,
+                        callbacks=callbacks,
                     )
                 else:
                     documents = _transformer.compress_documents(documents, query)
             elif isinstance(_transformer, BaseDocumentTransformer):
                 documents = _transformer.transform_documents(documents)
             else:
-                raise ValueError(f"Got unexpected transformer type: {_transformer}")
+                msg = f"Got unexpected transformer type: {_transformer}"
+                raise ValueError(msg)  # noqa: TRY004
         return documents
 
     async def acompress_documents(
@@ -59,18 +62,21 @@ class DocumentCompressorPipeline(BaseDocumentCompressor):
             if isinstance(_transformer, BaseDocumentCompressor):
                 accepts_callbacks = (
                     signature(_transformer.acompress_documents).parameters.get(
-                        "callbacks"
+                        "callbacks",
                     )
                     is not None
                 )
                 if accepts_callbacks:
                     documents = await _transformer.acompress_documents(
-                        documents, query, callbacks=callbacks
+                        documents,
+                        query,
+                        callbacks=callbacks,
                     )
                 else:
                     documents = await _transformer.acompress_documents(documents, query)
             elif isinstance(_transformer, BaseDocumentTransformer):
                 documents = await _transformer.atransform_documents(documents)
             else:
-                raise ValueError(f"Got unexpected transformer type: {_transformer}")
+                msg = f"Got unexpected transformer type: {_transformer}"
+                raise ValueError(msg)  # noqa: TRY004
         return documents

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,
 )
@@ -81,7 +80,7 @@ class LLMChainExtractor(BaseDocumentCompressor):
             if len(output) == 0:
                 continue
             compressed_docs.append(
-                Document(page_content=cast(str, output), metadata=doc.metadata)
+                Document(page_content=cast(str, output), metadata=doc.metadata),
             )
         return compressed_docs
 
@@ -99,7 +98,7 @@ class LLMChainExtractor(BaseDocumentCompressor):
             if len(outputs[i]) == 0:
                 continue
             compressed_docs.append(
-                Document(page_content=outputs[i], metadata=doc.metadata)
+                Document(page_content=outputs[i], metadata=doc.metadata),
             )
         return compressed_docs
 

langchain/retrievers/document_compressors/chain_extract_prompt.py

@@ -1,5 +1,4 @@
-# flake8: noqa
-prompt_template = """Given the following question and context, extract any part of the context *AS IS* that is relevant to answer the question. If none of the context is relevant return {no_output_str}. 
+prompt_template = """Given the following question and context, extract any part of the context *AS IS* that is relevant to answer the question. If none of the context is relevant return {no_output_str}.
 
 Remember, *DO NOT* edit the extracted parts of the context.
 
@@ -8,4 +7,4 @@ Remember, *DO NOT* edit the extracted parts of the context.
 >>>
 {{context}}
 >>>
-Extracted relevant parts:"""
+Extracted relevant parts:"""  # noqa: E501

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,
 )
@@ -37,7 +36,7 @@ class LLMChainFilter(BaseDocumentCompressor):
     """Filter that drops documents that aren't relevant to the query."""
 
     llm_chain: Runnable
-    """LLM wrapper to use for filtering documents. 
+    """LLM wrapper to use for filtering documents.
     The chain prompt is expected to have a BooleanOutputParser."""
 
     get_input: Callable[[str, Document], dict] = default_get_input
@@ -59,7 +58,8 @@ class LLMChainFilter(BaseDocumentCompressor):
         config = RunnableConfig(callbacks=callbacks)
         outputs = zip(
             self.llm_chain.batch(
-                [self.get_input(query, doc) for doc in documents], config=config
+                [self.get_input(query, doc) for doc in documents],
+                config=config,
             ),
             documents,
         )
@@ -70,9 +70,8 @@ class LLMChainFilter(BaseDocumentCompressor):
                 output = output_[self.llm_chain.output_key]
                 if self.llm_chain.prompt.output_parser is not None:
                     include_doc = self.llm_chain.prompt.output_parser.parse(output)
-            else:
-                if isinstance(output_, bool):
-                    include_doc = output_
+            elif isinstance(output_, bool):
+                include_doc = output_
             if include_doc:
                 filtered_docs.append(doc)
 
@@ -90,7 +89,8 @@ class LLMChainFilter(BaseDocumentCompressor):
         config = RunnableConfig(callbacks=callbacks)
         outputs = zip(
             await self.llm_chain.abatch(
-                [self.get_input(query, doc) for doc in documents], config=config
+                [self.get_input(query, doc) for doc in documents],
+                config=config,
             ),
             documents,
         )
@@ -100,9 +100,8 @@ class LLMChainFilter(BaseDocumentCompressor):
                 output = output_[self.llm_chain.output_key]
                 if self.llm_chain.prompt.output_parser is not None:
                     include_doc = self.llm_chain.prompt.output_parser.parse(output)
-            else:
-                if isinstance(output_, bool):
-                    include_doc = output_
+            elif isinstance(output_, bool):
+                include_doc = output_
             if include_doc:
                 filtered_docs.append(doc)
 

langchain/retrievers/document_compressors/chain_filter_prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 prompt_template = """Given the following question and context, return YES if the context is relevant to the question and NO if it isn't.
 
 > Question: {question}
@@ -6,4 +5,4 @@ prompt_template = """Given the following question and context, return YES if the
 >>>
 {context}
 >>>
-> Relevant (YES / NO):"""
+> Relevant (YES / NO):"""  # noqa: E501

langchain/retrievers/document_compressors/cohere_rerank.py

@@ -5,16 +5,16 @@ 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"
+    since="0.0.30",
+    removal="1.0",
+    alternative_import="langchain_cohere.CohereRerank",
 )
 class CohereRerank(BaseDocumentCompressor):
     """Document compressor that uses `Cohere Rerank API`."""
@@ -26,7 +26,7 @@ class CohereRerank(BaseDocumentCompressor):
     model: str = "rerank-english-v2.0"
     """Model to use for reranking."""
     cohere_api_key: Optional[str] = None
-    """Cohere API key. Must be specified directly or via environment variable 
+    """Cohere API key. Must be specified directly or via environment variable
         COHERE_API_KEY."""
     user_agent: str = "langchain"
     """Identifier for the application making the request."""
@@ -43,13 +43,16 @@ class CohereRerank(BaseDocumentCompressor):
         if not values.get("client"):
             try:
                 import cohere
-            except ImportError:
-                raise ImportError(
+            except ImportError as e:
+                msg = (
                     "Could not import cohere python package. "
                     "Please install it with `pip install cohere`."
                 )
+                raise ImportError(msg) from e
             cohere_api_key = get_from_dict_or_env(
-                values, "cohere_api_key", "COHERE_API_KEY"
+                values,
+                "cohere_api_key",
+                "COHERE_API_KEY",
             )
             client_name = values.get("user_agent", "langchain")
             values["client"] = cohere.Client(cohere_api_key, client_name=client_name)
@@ -89,16 +92,11 @@ class CohereRerank(BaseDocumentCompressor):
             max_chunks_per_doc=max_chunks_per_doc,
         )
         if hasattr(results, "results"):
-            results = getattr(results, "results")
-        result_dicts = []
-        for res in results:
-            result_dicts.append(
-                {
-                    "index": res.index,
-                    "relevance_score": res.relevance_score,
-                }
-            )
-        return result_dicts
+            results = results.results
+        return [
+            {"index": res.index, "relevance_score": res.relevance_score}
+            for res in results
+        ]
 
     def compress_documents(
         self,

langchain/retrievers/document_compressors/embeddings_filter.py

@@ -1,25 +1,22 @@
 from collections.abc import Sequence
 from typing import 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.embeddings import Embeddings
 from langchain_core.utils import pre_init
 from pydantic import ConfigDict, Field
 
-from langchain.retrievers.document_compressors.base import (
-    BaseDocumentCompressor,
-)
-
 
 def _get_similarity_function() -> Callable:
     try:
         from langchain_community.utils.math import cosine_similarity
-    except ImportError:
-        raise ImportError(
+    except ImportError as e:
+        msg = (
             "To use please install langchain-community "
             "with `pip install langchain-community`."
         )
+        raise ImportError(msg) from e
     return cosine_similarity
 
 
@@ -49,7 +46,8 @@ class EmbeddingsFilter(BaseDocumentCompressor):
     def validate_params(cls, values: dict) -> dict:
         """Validate similarity parameters."""
         if values["k"] is None and values["similarity_threshold"] is None:
-            raise ValueError("Must specify one of `k` or `similarity_threshold`.")
+            msg = "Must specify one of `k` or `similarity_threshold`."
+            raise ValueError(msg)
         return values
 
     def compress_documents(
@@ -64,21 +62,22 @@ class EmbeddingsFilter(BaseDocumentCompressor):
                 _get_embeddings_from_stateful_docs,
                 get_stateful_documents,
             )
-        except ImportError:
-            raise ImportError(
+        except ImportError as e:
+            msg = (
                 "To use please install langchain-community "
                 "with `pip install langchain-community`."
             )
+            raise ImportError(msg) from e
 
         try:
             import numpy as np
         except ImportError as e:
-            raise ImportError(
-                "Could not import numpy, please install with `pip install numpy`."
-            ) from e
+            msg = "Could not import numpy, please install with `pip install numpy`."
+            raise ImportError(msg) from e
         stateful_documents = get_stateful_documents(documents)
         embedded_documents = _get_embeddings_from_stateful_docs(
-            self.embeddings, stateful_documents
+            self.embeddings,
+            stateful_documents,
         )
         embedded_query = self.embeddings.embed_query(query)
         similarity = self.similarity_fn([embedded_query], embedded_documents)[0]
@@ -87,7 +86,7 @@ class EmbeddingsFilter(BaseDocumentCompressor):
             included_idxs = np.argsort(similarity)[::-1][: self.k]
         if self.similarity_threshold is not None:
             similar_enough = np.where(
-                similarity[included_idxs] > self.similarity_threshold
+                similarity[included_idxs] > self.similarity_threshold,
             )
             included_idxs = included_idxs[similar_enough]
         for i in included_idxs:
@@ -106,21 +105,22 @@ class EmbeddingsFilter(BaseDocumentCompressor):
                 _aget_embeddings_from_stateful_docs,
                 get_stateful_documents,
             )
-        except ImportError:
-            raise ImportError(
+        except ImportError as e:
+            msg = (
                 "To use please install langchain-community "
                 "with `pip install langchain-community`."
             )
+            raise ImportError(msg) from e
 
         try:
             import numpy as np
         except ImportError as e:
-            raise ImportError(
-                "Could not import numpy, please install with `pip install numpy`."
-            ) from e
+            msg = "Could not import numpy, please install with `pip install numpy`."
+            raise ImportError(msg) from e
         stateful_documents = get_stateful_documents(documents)
         embedded_documents = await _aget_embeddings_from_stateful_docs(
-            self.embeddings, stateful_documents
+            self.embeddings,
+            stateful_documents,
         )
         embedded_query = await self.embeddings.aembed_query(query)
         similarity = self.similarity_fn([embedded_query], embedded_documents)[0]
@@ -129,7 +129,7 @@ class EmbeddingsFilter(BaseDocumentCompressor):
             included_idxs = np.argsort(similarity)[::-1][: self.k]
         if self.similarity_threshold is not None:
             similar_enough = np.where(
-                similarity[included_idxs] > self.similarity_threshold
+                similarity[included_idxs] > self.similarity_threshold,
             )
             included_idxs = included_idxs[similar_enough]
         for i in included_idxs:

langchain/retrievers/document_compressors/flashrank_rerank.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "FlashrankRerank": "langchain_community.document_compressors.flashrank_rerank"
+    "FlashrankRerank": "langchain_community.document_compressors.flashrank_rerank",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)