langchain 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

langchain/chains/qa_with_sources/vector_db.py

@@ -61,7 +61,7 @@ class VectorDBQAWithSourcesChain(BaseQAWithSourcesChain):
     ) -> List[Document]:
         raise NotImplementedError("VectorDBQAWithSourcesChain does not support async")
 
-    @root_validator()
+    @root_validator(pre=True)
     def raise_deprecation(cls, values: Dict) -> Dict:
         warnings.warn(
             "`VectorDBQAWithSourcesChain` is deprecated - "

langchain/chains/query_constructor/base.py

@@ -1,4 +1,5 @@
 """LLM Chain for turning a user text query into a structured query."""
+
 from __future__ import annotations
 
 import json

langchain/chains/query_constructor/ir.py

@@ -1,4 +1,5 @@
 """Internal representation of a structured query language."""
+
 from langchain_core.structured_query import (
     Comparator,
     Comparison,

langchain/chains/question_answering/chain.py

@@ -1,4 +1,5 @@
 """Load question answering chains."""
+
 from typing import Any, Mapping, Optional, Protocol
 
 from langchain_core.callbacks import BaseCallbackManager, Callbacks

langchain/chains/retrieval_qa/base.py

@@ -1,4 +1,5 @@
 """Chain for question-answering against a vector database."""
+
 from __future__ import annotations
 
 import inspect
@@ -284,7 +285,7 @@ class VectorDBQA(BaseRetrievalQA):
     search_kwargs: Dict[str, Any] = Field(default_factory=dict)
     """Extra search args."""
 
-    @root_validator()
+    @root_validator(pre=True)
     def raise_deprecation(cls, values: Dict) -> Dict:
         warnings.warn(
             "`VectorDBQA` is deprecated - "
@@ -292,7 +293,7 @@ class VectorDBQA(BaseRetrievalQA):
         )
         return values
 
-    @root_validator()
+    @root_validator(pre=True)
     def validate_search_type(cls, values: Dict) -> Dict:
         """Validate search type."""
         if "search_type" in values:

langchain/chains/router/base.py

@@ -1,4 +1,5 @@
 """Base classes for chain routing."""
+
 from __future__ import annotations
 
 from abc import ABC

langchain/chains/router/llm_router.py

@@ -1,4 +1,5 @@
 """Base classes for LLM-powered router chains."""
+
 from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Type, cast
@@ -24,7 +25,7 @@ class LLMRouterChain(RouterChain):
     llm_chain: LLMChain
     """LLM chain used to perform routing"""
 
-    @root_validator()
+    @root_validator(pre=False, skip_on_failure=True)
     def validate_prompt(cls, values: dict) -> dict:
         prompt = values["llm_chain"].prompt
         if prompt.output_parser is None:

langchain/chains/router/multi_prompt.py

@@ -1,4 +1,5 @@
 """Use a single chain to route an input to one of multiple llm chains."""
+
 from __future__ import annotations
 
 from typing import Any, Dict, List, Optional

langchain/chains/router/multi_retrieval_qa.py

@@ -1,4 +1,5 @@
 """Use a single chain to route an input to one of multiple retrieval qa chains."""
+
 from __future__ import annotations
 
 from typing import Any, Dict, List, Mapping, Optional

langchain/chains/sequential.py

@@ -1,4 +1,5 @@
 """Chain pipeline where the outputs of one step feed directly into next."""
+
 from typing import Any, Dict, List, Optional
 
 from langchain_core.callbacks import (
@@ -152,7 +153,7 @@ class SimpleSequentialChain(Chain):
         """
         return [self.output_key]
 
-    @root_validator()
+    @root_validator(pre=False, skip_on_failure=True)
     def validate_chains(cls, values: Dict) -> Dict:
         """Validate that chains are all single input/output."""
         for chain in values["chains"]:

langchain/chains/structured_output/base.py

@@ -32,13 +32,13 @@ from langchain_core.utils.function_calling import (
         "LangChain has introduced a method called `with_structured_output` that "
         "is available on ChatModels capable of tool calling. "
         "You can read more about the method here: "
-        "https://python.langchain.com/docs/modules/model_io/chat/structured_output/ "
+        "<https://python.langchain.com/docs/modules/model_io/chat/structured_output/>. "
         "Please follow our extraction use case documentation for more guidelines "
         "on how to do information extraction with LLMs. "
-        "https://python.langchain.com/docs/use_cases/extraction/. "
+        "<https://python.langchain.com/docs/use_cases/extraction/>. "
         "If you notice other issues, please provide "
         "feedback here: "
-        "https://github.com/langchain-ai/langchain/discussions/18154"
+        "<https://github.com/langchain-ai/langchain/discussions/18154>"
     ),
     removal="0.3.0",
     alternative=(
@@ -150,13 +150,13 @@ def create_openai_fn_runnable(
         "LangChain has introduced a method called `with_structured_output` that "
         "is available on ChatModels capable of tool calling. "
         "You can read more about the method here: "
-        "https://python.langchain.com/docs/modules/model_io/chat/structured_output/ "
+        "<https://python.langchain.com/docs/modules/model_io/chat/structured_output/>."
         "Please follow our extraction use case documentation for more guidelines "
         "on how to do information extraction with LLMs. "
-        "https://python.langchain.com/docs/use_cases/extraction/. "
+        "<https://python.langchain.com/docs/use_cases/extraction/>. "
         "If you notice other issues, please provide "
         "feedback here: "
-        "https://github.com/langchain-ai/langchain/discussions/18154"
+        "<https://github.com/langchain-ai/langchain/discussions/18154>"
     ),
     removal="0.3.0",
     alternative=(
@@ -466,9 +466,9 @@ def _get_openai_tool_output_parser(
     first_tool_only: bool = False,
 ) -> Union[BaseOutputParser, BaseGenerationOutputParser]:
     if isinstance(tool, type) and issubclass(tool, BaseModel):
-        output_parser: Union[
-            BaseOutputParser, BaseGenerationOutputParser
-        ] = PydanticToolsParser(tools=[tool], first_tool_only=first_tool_only)
+        output_parser: Union[BaseOutputParser, BaseGenerationOutputParser] = (
+            PydanticToolsParser(tools=[tool], first_tool_only=first_tool_only)
+        )
     else:
         key_name = convert_to_openai_tool(tool)["function"]["name"]
         output_parser = JsonOutputKeyToolsParser(
@@ -500,9 +500,9 @@ def get_openai_output_parser(
             }
         else:
             pydantic_schema = functions[0]
-        output_parser: Union[
-            BaseOutputParser, BaseGenerationOutputParser
-        ] = PydanticOutputFunctionsParser(pydantic_schema=pydantic_schema)
+        output_parser: Union[BaseOutputParser, BaseGenerationOutputParser] = (
+            PydanticOutputFunctionsParser(pydantic_schema=pydantic_schema)
+        )
     else:
         output_parser = JsonOutputFunctionsParser(args_only=len(functions) <= 1)
     return output_parser

langchain/chains/summarize/chain.py

@@ -1,4 +1,5 @@
 """Load summarizing chains."""
+
 from typing import Any, Mapping, Optional, Protocol
 
 from langchain_core.callbacks import Callbacks

langchain/chains/transform.py

@@ -1,4 +1,5 @@
 """Chain that runs an arbitrary python function."""
+
 import functools
 import logging
 from typing import Any, Awaitable, Callable, Dict, List, Optional
@@ -31,9 +32,9 @@ class TransformChain(Chain):
     """The keys returned by the transform's output dictionary."""
     transform_cb: Callable[[Dict[str, str]], Dict[str, str]] = Field(alias="transform")
     """The transform function."""
-    atransform_cb: Optional[
-        Callable[[Dict[str, Any]], Awaitable[Dict[str, Any]]]
-    ] = Field(None, alias="atransform")
+    atransform_cb: Optional[Callable[[Dict[str, Any]], Awaitable[Dict[str, Any]]]] = (
+        Field(None, alias="atransform")
+    )
     """The async coroutine transform function."""
 
     @staticmethod

langchain/chat_models/__init__.py

@@ -16,6 +16,7 @@ an interface where "chat messages" are the inputs and outputs.
 
     AIMessage, BaseMessage, HumanMessage
 """  # noqa: E501
+
 import warnings
 
 from langchain_core._api import LangChainDeprecationWarning

langchain/chat_models/base.py

@@ -94,7 +94,7 @@ def init_chat_model(
         _check_pkg("langchain_anthropic")
         from langchain_anthropic import ChatAnthropic
 
-        return ChatAnthropic(model=model, **kwargs)
+        return ChatAnthropic(model=model, **kwargs)  # type: ignore[call-arg]
     elif model_provider == "azure_openai":
         _check_pkg("langchain_openai")
         from langchain_openai import AzureChatOpenAI
@@ -134,7 +134,7 @@ def init_chat_model(
         _check_pkg("langchain_mistralai")
         from langchain_mistralai import ChatMistralAI
 
-        return ChatMistralAI(model=model, **kwargs)
+        return ChatMistralAI(model=model, **kwargs)  # type: ignore[call-arg]
     elif model_provider == "huggingface":
         _check_pkg("langchain_huggingface")
         from langchain_huggingface import ChatHuggingFace

langchain/docstore/__init__.py

@@ -14,6 +14,7 @@ The **Docstore** is a simplified version of the Document Loader.
 
     Document, AddableMixin
 """
+
 from typing import TYPE_CHECKING, Any
 
 from langchain._api import create_importer

langchain/document_loaders/__init__.py

@@ -14,6 +14,7 @@
 
     Document, <name>TextSplitter
 """
+
 from typing import TYPE_CHECKING, Any
 
 from langchain._api import create_importer

langchain/document_transformers/__init__.py

@@ -14,6 +14,7 @@
 
     Document
 """  # noqa: E501
+
 from typing import TYPE_CHECKING, Any
 
 from langchain._api import create_importer

langchain/embeddings/__init__.py

@@ -10,7 +10,6 @@ from different APIs and services.
     Embeddings --> <name>Embeddings  # Examples: OpenAIEmbeddings, HuggingFaceEmbeddings
 """
 
-
 import logging
 from typing import TYPE_CHECKING, Any
 

langchain/evaluation/__init__.py

@@ -38,7 +38,7 @@ name of the dataset to load.
 - Comparing the output of two models: :class:`PairwiseStringEvalChain <langchain.evaluation.comparison.eval_chain.PairwiseStringEvalChain>` or :class:`LabeledPairwiseStringEvalChain <langchain.evaluation.comparison.eval_chain.LabeledPairwiseStringEvalChain>` when there is additionally a reference label.
 - Judging the efficacy of an agent's tool usage: :class:`TrajectoryEvalChain <langchain.evaluation.agents.trajectory_eval_chain.TrajectoryEvalChain>`
 - Checking whether an output complies with a set of criteria: :class:`CriteriaEvalChain <langchain.evaluation.criteria.eval_chain.CriteriaEvalChain>` or :class:`LabeledCriteriaEvalChain <langchain.evaluation.criteria.eval_chain.LabeledCriteriaEvalChain>` when there is additionally a reference label.
-- Computing semantic difference between a prediction and reference: :class:`EmbeddingDistanceEvalChain <langchain.evaluation.embedding_distance.base.EmbeddingDistanceEvalChain>` or between two predictions: :class:`PairwiseEmbeddingDistanceEvalChain <langchain.evaluation.embedding_distance.base.PairwiseEmbeddingDistanceEvalChain>` 
+- Computing semantic difference between a prediction and reference: :class:`EmbeddingDistanceEvalChain <langchain.evaluation.embedding_distance.base.EmbeddingDistanceEvalChain>` or between two predictions: :class:`PairwiseEmbeddingDistanceEvalChain <langchain.evaluation.embedding_distance.base.PairwiseEmbeddingDistanceEvalChain>`
 - Measuring the string distance between a prediction and reference :class:`StringDistanceEvalChain <langchain.evaluation.string_distance.base.StringDistanceEvalChain>` or between two predictions :class:`PairwiseStringDistanceEvalChain <langchain.evaluation.string_distance.base.PairwiseStringDistanceEvalChain>`
 
 **Low-level API**
@@ -52,6 +52,7 @@ These evaluators implement one of the following interfaces:
 These interfaces enable easier composability and usage within a higher level evaluation framework.
 
 """  # noqa: E501
+
 from langchain.evaluation.agents import TrajectoryEvalChain
 from langchain.evaluation.comparison import (
     LabeledPairwiseStringEvalChain,

langchain/evaluation/agents/__init__.py

@@ -1,4 +1,5 @@
 """Chains for evaluating ReAct style agents."""
+
 from langchain.evaluation.agents.trajectory_eval_chain import TrajectoryEvalChain
 
 __all__ = ["TrajectoryEvalChain"]

langchain/evaluation/agents/trajectory_eval_prompt.py

@@ -1,4 +1,5 @@
 """Prompt for trajectory evaluation chain."""
+
 # flake8: noqa
 from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
 

langchain/evaluation/comparison/__init__.py

@@ -27,6 +27,7 @@ Example:
     # .     " by explaining what the formula means.\\n[[B]]"
     # }
 """
+
 from langchain.evaluation.comparison.eval_chain import (
     LabeledPairwiseStringEvalChain,
     PairwiseStringEvalChain,

langchain/evaluation/comparison/eval_chain.py

@@ -1,4 +1,5 @@
 """Base classes for comparing the output of two models."""
+
 from __future__ import annotations
 
 import logging

langchain/evaluation/comparison/prompt.py

@@ -4,6 +4,7 @@ This prompt is used to compare two responses and evaluate which one best follows
 and answers the question. The prompt is based on the paper from
 Zheng, et. al. https://arxiv.org/abs/2306.05685
 """
+
 # flake8: noqa
 from langchain_core.prompts.chat import ChatPromptTemplate
 

langchain/evaluation/embedding_distance/__init__.py

@@ -1,4 +1,5 @@
 """Evaluators that measure embedding distances."""
+
 from langchain.evaluation.embedding_distance.base import (
     EmbeddingDistance,
     EmbeddingDistanceEvalChain,

langchain/evaluation/embedding_distance/base.py

@@ -1,4 +1,5 @@
 """A chain for comparing the output of two models using embeddings."""
+
 from enum import Enum
 from typing import Any, Dict, List, Optional
 

langchain/evaluation/loading.py

@@ -1,4 +1,5 @@
 """Loading datasets and evaluators."""
+
 from typing import Any, Dict, List, Optional, Sequence, Type, Union
 
 from langchain_core.language_models import BaseLanguageModel

langchain/evaluation/parsing/base.py

@@ -1,4 +1,5 @@
 """Evaluators for parsing strings."""
+
 import json
 from operator import eq
 from typing import Any, Callable, Optional, Union, cast

langchain/evaluation/qa/__init__.py

@@ -1,4 +1,5 @@
 """Chains and utils related to evaluating question answering functionality."""
+
 from langchain.evaluation.qa.eval_chain import (
     ContextQAEvalChain,
     CotQAEvalChain,

langchain/evaluation/qa/eval_chain.py

@@ -1,4 +1,5 @@
 """LLM Chains for evaluating question answering."""
+
 from __future__ import annotations
 
 import re

langchain/evaluation/qa/generate_chain.py

@@ -1,4 +1,5 @@
 """LLM Chain for generating examples for question answering."""
+
 from __future__ import annotations
 
 from typing import Any

langchain/evaluation/schema.py

@@ -1,4 +1,5 @@
 """Interfaces to be implemented by general evaluators."""
+
 from __future__ import annotations
 
 import logging

langchain/evaluation/scoring/__init__.py

@@ -22,6 +22,7 @@ Example:
     #    "However, it does not provide an explanation of what the formula means."
     # }
 """
+
 from langchain.evaluation.scoring.eval_chain import (
     LabeledScoreStringEvalChain,
     ScoreStringEvalChain,

langchain/evaluation/scoring/eval_chain.py

@@ -1,4 +1,5 @@
 """Base classes for scoring the output of a model on a scale of 1-10."""
+
 from __future__ import annotations
 
 import logging

langchain/evaluation/scoring/prompt.py

@@ -4,6 +4,7 @@ This prompt is used to score the responses and evaluate how it follows the instr
 and answers the question. The prompt is based on the paper from
 Zheng, et. al. https://arxiv.org/abs/2306.05685
 """
+
 # flake8: noqa
 from langchain_core.prompts.chat import ChatPromptTemplate
 

langchain/evaluation/string_distance/__init__.py

@@ -1,4 +1,5 @@
 """String distance evaluators."""
+
 from langchain.evaluation.string_distance.base import (
     PairwiseStringDistanceEvalChain,
     StringDistance,

langchain/example_generator.py

@@ -1,4 +1,5 @@
 """Keep here for backwards compatibility."""
+
 from langchain.chains.example_generator import generate_example
 
 __all__ = ["generate_example"]

langchain/formatting.py

@@ -1,4 +1,5 @@
 """DEPRECATED: Kept for backwards compatibility."""
+
 from langchain_core.utils.formatting import StrictFormatter, formatter
 
 __all__ = ["StrictFormatter", "formatter"]

langchain/globals/__init__.py

@@ -1,4 +1,5 @@
 """Global values and configuration that apply to all of LangChain."""
+
 import warnings
 from typing import TYPE_CHECKING, Optional
 

langchain/graphs/__init__.py

@@ -1,4 +1,5 @@
 """**Graphs** provide a natural language interface to graph databases."""
+
 from typing import TYPE_CHECKING, Any
 
 from langchain._api import create_importer

langchain/indexes/__init__.py

@@ -11,6 +11,7 @@ Importantly, Index keeps on working even if the content being written is derived
 via a set of transformations from some source content (e.g., indexing children
 documents that were derived from parent documents by chunking.)
 """
+
 from typing import TYPE_CHECKING, Any
 
 from langchain_core.indexing.api import IndexingResult, aindex, index

langchain/indexes/_sql_record_manager.py

@@ -13,6 +13,7 @@ allow it to work with a variety of SQL as a backend.
 * Keys can be listed based on the updated at field.
 * Keys can be deleted.
 """
+
 import contextlib
 import decimal
 import uuid
@@ -20,9 +21,7 @@ from typing import Any, AsyncGenerator, Dict, Generator, List, Optional, Sequenc
 
 from langchain_core.indexing import RecordManager
 from sqlalchemy import (
-    URL,
     Column,
-    Engine,
     Float,
     Index,
     String,
@@ -33,14 +32,19 @@ from sqlalchemy import (
     select,
     text,
 )
+from sqlalchemy.engine import URL, Engine
 from sqlalchemy.ext.asyncio import (
     AsyncEngine,
     AsyncSession,
-    async_sessionmaker,
     create_async_engine,
 )
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import Query, Session, sessionmaker
+from sqlalchemy.orm import Query, Session, declarative_base, sessionmaker
+
+try:
+    from sqlalchemy.ext.asyncio import async_sessionmaker
+except ImportError:
+    # dummy for sqlalchemy < 2
+    async_sessionmaker = type("async_sessionmaker", (type,), {})  # type: ignore
 
 Base = declarative_base()
 

langchain/indexes/graph.py

@@ -1,4 +1,5 @@
 """**Graphs** provide a natural language interface to graph databases."""
+
 from typing import TYPE_CHECKING, Any
 
 from langchain._api import create_importer

langchain/indexes/prompts/__init__.py

@@ -1,4 +1,5 @@
 """Relevant prompts for constructing indexes."""
+
 from langchain_core._api import warn_deprecated
 
 warn_deprecated(

langchain/input.py

@@ -1,4 +1,5 @@
 """DEPRECATED: Kept for backwards compatibility."""
+
 from langchain_core.utils.input import (
     get_bolded_text,
     get_color_mapping,

langchain/llms/__init__.py

@@ -17,6 +17,7 @@ access to the large language model (**LLM**) APIs and services.
     CallbackManager, AsyncCallbackManager,
     AIMessage, BaseMessage
 """  # noqa: E501
+
 import warnings
 from typing import Any, Callable, Dict, Type
 

langchain/load/__init__.py

@@ -1,4 +1,5 @@
 """Serialization and deserialization."""
+
 from langchain_core.load.dump import dumpd, dumps
 from langchain_core.load.load import load, loads
 

langchain/memory/__init__.py

@@ -26,6 +26,7 @@
 
     AIMessage, BaseMessage, HumanMessage
 """  # noqa: E501
+
 from typing import TYPE_CHECKING, Any
 
 from langchain._api import create_importer
@@ -48,6 +49,9 @@ from langchain.memory.summary import ConversationSummaryMemory
 from langchain.memory.summary_buffer import ConversationSummaryBufferMemory
 from langchain.memory.token_buffer import ConversationTokenBufferMemory
 from langchain.memory.vectorstore import VectorStoreRetrieverMemory
+from langchain.memory.vectorstore_token_buffer_memory import (
+    ConversationVectorStoreTokenBufferMemory,  # avoid circular import
+)
 
 if TYPE_CHECKING:
     from langchain_community.chat_message_histories import (
@@ -122,6 +126,7 @@ __all__ = [
     "ConversationSummaryBufferMemory",
     "ConversationSummaryMemory",
     "ConversationTokenBufferMemory",
+    "ConversationVectorStoreTokenBufferMemory",
     "CosmosDBChatMessageHistory",
     "DynamoDBChatMessageHistory",
     "ElasticsearchChatMessageHistory",