langchain 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

langchain/chat_models/base.py

@@ -102,6 +102,13 @@ def init_chat_model(
 
     .. versionchanged:: 0.2.8
 
+        Support for ``configurable_fields`` and ``config_prefix`` added.
+
+    .. versionchanged:: 0.2.12
+
+        Support for Ollama via langchain-ollama package added. Previously
+        langchain-community version of Ollama (now deprecated) was installed by default.
+
     Args:
         model: The name of the model, e.g. "gpt-4o", "claude-3-opus-20240229".
         model_provider: The model provider. Supported model_provider values and the
@@ -118,7 +125,7 @@ def init_chat_model(
                 - mistralai (langchain-mistralai)
                 - huggingface (langchain-huggingface)
                 - groq (langchain-groq)
-                - ollama (langchain-community)
+                - ollama (langchain-ollama)  [support added in langchain==0.2.12]
 
             Will attempt to infer model_provider from model if not specified. The
             following providers will be inferred based on these model prefixes:
@@ -336,8 +343,18 @@ def _init_chat_model_helper(
 
         return ChatFireworks(model=model, **kwargs)
     elif model_provider == "ollama":
-        _check_pkg("langchain_community")
-        from langchain_community.chat_models import ChatOllama
+        try:
+            _check_pkg("langchain_ollama")
+            from langchain_ollama import ChatOllama
+        except ImportError:
+            # For backwards compatibility
+            try:
+                _check_pkg("langchain_community")
+                from langchain_community.chat_models import ChatOllama
+            except ImportError:
+                # If both langchain-ollama and langchain-community aren't available,
+                # raise an error related to langchain-ollama
+                _check_pkg("langchain_ollama")
 
         return ChatOllama(model=model, **kwargs)
     elif model_provider == "together":
@@ -366,6 +383,11 @@ def _init_chat_model_helper(
 
         # TODO: update to use model= once ChatBedrock supports
         return ChatBedrock(model_id=model, **kwargs)
+    elif model_provider == "bedrock_converse":
+        _check_pkg("langchain_aws")
+        from langchain_aws import ChatBedrockConverse
+
+        return ChatBedrockConverse(model=model, **kwargs)
     else:
         supported = ", ".join(_SUPPORTED_PROVIDERS)
         raise ValueError(
@@ -388,6 +410,7 @@ _SUPPORTED_PROVIDERS = {
     "huggingface",
     "groq",
     "bedrock",
+    "bedrock_converse",
 }
 
 

langchain/evaluation/agents/trajectory_eval_chain.py

@@ -28,7 +28,7 @@ from langchain_core.exceptions import OutputParserException
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.output_parsers import BaseOutputParser
-from langchain_core.pydantic_v1 import Extra, Field
+from langchain_core.pydantic_v1 import Field
 from langchain_core.tools import BaseTool
 
 from langchain.chains.llm import LLMChain
@@ -157,9 +157,7 @@ class TrajectoryEvalChain(AgentTrajectoryEvaluator, LLMEvalChain):
     """DEPRECATED. Reasoning always returned."""
 
     class Config:
-        """Configuration for the QAEvalChain."""
-
-        extra = Extra.ignore
+        extra = "ignore"
 
     @property
     def requires_reference(self) -> bool:
@@ -283,8 +281,7 @@ The following is the expected answer. Use this to measure correctness:
 
     def prep_inputs(self, inputs: Union[Dict[str, Any], Any]) -> Dict[str, str]:
         """Validate and prep inputs."""
-        if "reference" not in inputs:
-            inputs["reference"] = self._format_reference(inputs.get("reference"))
+        inputs["reference"] = self._format_reference(inputs.get("reference"))
         return super().prep_inputs(inputs)
 
     def _call(

langchain/evaluation/comparison/eval_chain.py

@@ -10,7 +10,7 @@ from langchain_core.callbacks.manager import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import Extra, Field
+from langchain_core.pydantic_v1 import Field
 
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 from langchain.chains.llm import LLMChain
@@ -192,9 +192,7 @@ class PairwiseStringEvalChain(PairwiseStringEvaluator, LLMEvalChain, LLMChain):
         return False
 
     class Config:
-        """Configuration for the PairwiseStringEvalChain."""
-
-        extra = Extra.ignore
+        extra = "ignore"
 
     @property
     def requires_reference(self) -> bool:

langchain/evaluation/criteria/eval_chain.py

@@ -8,7 +8,7 @@ from langchain_core.callbacks.manager import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Extra, Field
+from langchain_core.pydantic_v1 import Field
 
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 from langchain.chains.llm import LLMChain
@@ -237,9 +237,7 @@ class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):
         return False
 
     class Config:
-        """Configuration for the QAEvalChain."""
-
-        extra = Extra.ignore
+        extra = "ignore"
 
     @property
     def requires_reference(self) -> bool:

langchain/evaluation/embedding_distance/base.py

@@ -10,7 +10,8 @@ from langchain_core.callbacks.manager import (
     Callbacks,
 )
 from langchain_core.embeddings import Embeddings
-from langchain_core.pydantic_v1 import Field, root_validator
+from langchain_core.pydantic_v1 import Field
+from langchain_core.utils import pre_init
 
 from langchain.chains.base import Chain
 from langchain.evaluation.schema import PairwiseStringEvaluator, StringEvaluator
@@ -68,7 +69,7 @@ class _EmbeddingDistanceChainMixin(Chain):
     embeddings: Embeddings = Field(default_factory=_embedding_factory)
     distance_metric: EmbeddingDistance = Field(default=EmbeddingDistance.COSINE)
 
-    @root_validator(pre=False)
+    @pre_init
     def _validate_tiktoken_installed(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         """Validate that the TikTok library is installed.
 
@@ -113,8 +114,6 @@ class _EmbeddingDistanceChainMixin(Chain):
         return values
 
     class Config:
-        """Permit embeddings to go unvalidated."""
-
         arbitrary_types_allowed: bool = True
 
     @property

langchain/evaluation/parsing/json_schema.py

@@ -36,7 +36,7 @@ class JsonSchemaEvaluator(StringEvaluator):
         """Initializes the JsonSchemaEvaluator.
 
         Args:
-            **kwargs: Additional keyword arguments.
+            kwargs: Additional keyword arguments.
 
         Raises:
             ImportError: If the jsonschema package is not installed.

langchain/evaluation/qa/eval_chain.py

@@ -9,7 +9,6 @@ from typing import Any, List, Optional, Sequence, Tuple
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import PromptTemplate
-from langchain_core.pydantic_v1 import Extra
 
 from langchain.chains.llm import LLMChain
 from langchain.evaluation.qa.eval_prompt import CONTEXT_PROMPT, COT_PROMPT, PROMPT
@@ -74,9 +73,7 @@ class QAEvalChain(LLMChain, StringEvaluator, LLMEvalChain):
     output_key: str = "results"  #: :meta private:
 
     class Config:
-        """Configuration for the QAEvalChain."""
-
-        extra = Extra.ignore
+        extra = "ignore"
 
     @classmethod
     def is_lc_serializable(cls) -> bool:
@@ -224,9 +221,7 @@ class ContextQAEvalChain(LLMChain, StringEvaluator, LLMEvalChain):
         return True
 
     class Config:
-        """Configuration for the QAEvalChain."""
-
-        extra = Extra.ignore
+        extra = "ignore"
 
     @classmethod
     def _validate_input_vars(cls, prompt: PromptTemplate) -> None:

langchain/evaluation/schema.py

@@ -158,7 +158,7 @@ class StringEvaluator(_EvalArgsMixin, ABC):
             prediction (str): The LLM or chain prediction to evaluate.
             reference (Optional[str], optional): The reference label to evaluate against.
             input (Optional[str], optional): The input to consider during evaluation.
-            **kwargs: Additional keyword arguments, including callbacks, tags, etc.
+            kwargs: Additional keyword arguments, including callbacks, tags, etc.
         Returns:
             dict: The evaluation results containing the score or value.
                 It is recommended that the dictionary contain the following keys:
@@ -181,7 +181,7 @@ class StringEvaluator(_EvalArgsMixin, ABC):
             prediction (str): The LLM or chain prediction to evaluate.
             reference (Optional[str], optional): The reference label to evaluate against.
             input (Optional[str], optional): The input to consider during evaluation.
-            **kwargs: Additional keyword arguments, including callbacks, tags, etc.
+            kwargs: Additional keyword arguments, including callbacks, tags, etc.
         Returns:
             dict: The evaluation results containing the score or value.
                 It is recommended that the dictionary contain the following keys:
@@ -212,7 +212,7 @@ class StringEvaluator(_EvalArgsMixin, ABC):
             prediction (str): The LLM or chain prediction to evaluate.
             reference (Optional[str], optional): The reference label to evaluate against.
             input (Optional[str], optional): The input to consider during evaluation.
-            **kwargs: Additional keyword arguments, including callbacks, tags, etc.
+            kwargs: Additional keyword arguments, including callbacks, tags, etc.
         Returns:
             dict: The evaluation results containing the score or value.
         """  # noqa: E501
@@ -235,7 +235,7 @@ class StringEvaluator(_EvalArgsMixin, ABC):
             prediction (str): The LLM or chain prediction to evaluate.
             reference (Optional[str], optional): The reference label to evaluate against.
             input (Optional[str], optional): The input to consider during evaluation.
-            **kwargs: Additional keyword arguments, including callbacks, tags, etc.
+            kwargs: Additional keyword arguments, including callbacks, tags, etc.
         Returns:
             dict: The evaluation results containing the score or value.
         """  # noqa: E501
@@ -265,7 +265,7 @@ class PairwiseStringEvaluator(_EvalArgsMixin, ABC):
             prediction_b (str): The output string from the second model.
             reference (Optional[str], optional): The expected output / reference string.
             input (Optional[str], optional): The input string.
-            **kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
+            kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
         Returns:
             dict: A dictionary containing the preference, scores, and/or other information.
         """  # noqa: E501
@@ -286,7 +286,7 @@ class PairwiseStringEvaluator(_EvalArgsMixin, ABC):
             prediction_b (str): The output string from the second model.
             reference (Optional[str], optional): The expected output / reference string.
             input (Optional[str], optional): The input string.
-            **kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
+            kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
         Returns:
             dict: A dictionary containing the preference, scores, and/or other information.
         """  # noqa: E501
@@ -316,7 +316,7 @@ class PairwiseStringEvaluator(_EvalArgsMixin, ABC):
             prediction_b (str): The output string from the second model.
             reference (Optional[str], optional): The expected output / reference string.
             input (Optional[str], optional): The input string.
-            **kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
+            kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
         Returns:
             dict: A dictionary containing the preference, scores, and/or other information.
         """  # noqa: E501
@@ -345,7 +345,7 @@ class PairwiseStringEvaluator(_EvalArgsMixin, ABC):
             prediction_b (str): The output string from the second model.
             reference (Optional[str], optional): The expected output / reference string.
             input (Optional[str], optional): The input string.
-            **kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
+            kwargs: Additional keyword arguments, such as callbacks and optional reference strings.
         Returns:
             dict: A dictionary containing the preference, scores, and/or other information.
         """  # noqa: E501

langchain/evaluation/scoring/eval_chain.py

@@ -10,7 +10,7 @@ from langchain_core.callbacks.manager import Callbacks
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import Extra, Field
+from langchain_core.pydantic_v1 import Field
 
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 from langchain.chains.llm import LLMChain
@@ -180,9 +180,7 @@ class ScoreStringEvalChain(StringEvaluator, LLMEvalChain, LLMChain):
     """The name of the criterion being evaluated."""
 
     class Config:
-        """Configuration for the ScoreStringEvalChain."""
-
-        extra = Extra.ignore
+        extra = "ignore"
 
     @classmethod
     def is_lc_serializable(cls) -> bool:

langchain/evaluation/string_distance/base.py

@@ -8,7 +8,8 @@ from langchain_core.callbacks.manager import (
     CallbackManagerForChainRun,
     Callbacks,
 )
-from langchain_core.pydantic_v1 import Field, root_validator
+from langchain_core.pydantic_v1 import Field
+from langchain_core.utils import pre_init
 
 from langchain.chains.base import Chain
 from langchain.evaluation.schema import PairwiseStringEvaluator, StringEvaluator
@@ -63,7 +64,7 @@ class _RapidFuzzChainMixin(Chain):
     """Whether to normalize the score to a value between 0 and 1.
     Applies only to the Levenshtein and Damerau-Levenshtein distances."""
 
-    @root_validator
+    @pre_init
     def validate_dependencies(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         """
         Validate that the rapidfuzz library is installed.
@@ -277,7 +278,7 @@ class StringDistanceEvalChain(StringEvaluator, _RapidFuzzChainMixin):
             reference (Optional[str], optional): The reference string.
             input (Optional[str], optional): The input string.
             callbacks (Callbacks, optional): The callbacks to use.
-            **kwargs: Additional keyword arguments.
+            kwargs: Additional keyword arguments.
 
         Returns:
             dict: The evaluation results containing the score.
@@ -313,7 +314,7 @@ class StringDistanceEvalChain(StringEvaluator, _RapidFuzzChainMixin):
             reference (Optional[str], optional): The reference string.
             input (Optional[str], optional): The input string.
             callbacks (Callbacks, optional): The callbacks to use.
-            **kwargs: Additional keyword arguments.
+            kwargs: Additional keyword arguments.
 
         Returns:
             dict: The evaluation results containing the score.
@@ -411,7 +412,7 @@ class PairwiseStringDistanceEvalChain(PairwiseStringEvaluator, _RapidFuzzChainMi
             callbacks (Callbacks, optional): The callbacks to use.
             tags (List[str], optional): Tags to apply to traces.
             metadata (Dict[str, Any], optional): Metadata to apply to traces.
-            **kwargs: Additional keyword arguments.
+            kwargs: Additional keyword arguments.
 
         Returns:
             dict: The evaluation results containing the score.
@@ -445,7 +446,7 @@ class PairwiseStringDistanceEvalChain(PairwiseStringEvaluator, _RapidFuzzChainMi
             callbacks (Callbacks, optional): The callbacks to use.
             tags (List[str], optional): Tags to apply to traces.
             metadata (Dict[str, Any], optional): Metadata to apply to traces.
-            **kwargs: Additional keyword arguments.
+            kwargs: Additional keyword arguments.
 
         Returns:
             dict: The evaluation results containing the score.

langchain/hub.py

@@ -3,27 +3,37 @@
 from __future__ import annotations
 
 import json
-from typing import TYPE_CHECKING, Any, Optional
+from typing import Any, Optional, Sequence
 
 from langchain_core.load.dump import dumps
 from langchain_core.load.load import loads
 from langchain_core.prompts import BasePromptTemplate
 
-if TYPE_CHECKING:
-    from langchainhub import Client
 
-
-def _get_client(api_url: Optional[str] = None, api_key: Optional[str] = None) -> Client:
+def _get_client(
+    api_key: Optional[str] = None,
+    api_url: Optional[str] = None,
+) -> Any:
     try:
-        from langchainhub import Client
-    except ImportError as e:
-        raise ImportError(
-            "Could not import langchainhub, please install with `pip install "
-            "langchainhub`."
-        ) from e
+        from langsmith import Client as LangSmithClient
+
+        ls_client = LangSmithClient(api_url, api_key=api_key)
+        if hasattr(ls_client, "push_prompt") and hasattr(ls_client, "pull_prompt"):
+            return ls_client
+        else:
+            from langchainhub import Client as LangChainHubClient
 
-    # Client logic will also attempt to load URL/key from environment variables
-    return Client(api_url, api_key=api_key)
+            return LangChainHubClient(api_url, api_key=api_key)
+    except ImportError:
+        try:
+            from langchainhub import Client as LangChainHubClient
+
+            return LangChainHubClient(api_url, api_key=api_key)
+        except ImportError as e:
+            raise ImportError(
+                "Could not import langsmith or langchainhub (deprecated),"
+                "please install with `pip install langsmith`."
+            ) from e
 
 
 def push(
@@ -32,27 +42,43 @@ def push(
     *,
     api_url: Optional[str] = None,
     api_key: Optional[str] = None,
-    parent_commit_hash: Optional[str] = "latest",
-    new_repo_is_public: bool = True,
-    new_repo_description: str = "",
+    parent_commit_hash: Optional[str] = None,
+    new_repo_is_public: bool = False,
+    new_repo_description: Optional[str] = None,
+    readme: Optional[str] = None,
+    tags: Optional[Sequence[str]] = None,
 ) -> str:
     """
     Push an object to the hub and returns the URL it can be viewed at in a browser.
 
-    :param repo_full_name: The full name of the repo to push to in the format of
-        `owner/repo`.
+    :param repo_full_name: The full name of the prompt to push to in the format of
+        `owner/prompt_name` or `prompt_name`.
     :param object: The LangChain to serialize and push to the hub.
     :param api_url: The URL of the LangChain Hub API. Defaults to the hosted API service
         if you have an api key set, or a localhost instance if not.
     :param api_key: The API key to use to authenticate with the LangChain Hub API.
     :param parent_commit_hash: The commit hash of the parent commit to push to. Defaults
         to the latest commit automatically.
-    :param new_repo_is_public: Whether the repo should be public. Defaults to
-        True (Public by default).
-    :param new_repo_description: The description of the repo. Defaults to an empty
+    :param new_repo_is_public: Whether the prompt should be public. Defaults to
+        False (Private by default).
+    :param new_repo_description: The description of the prompt. Defaults to an empty
         string.
     """
-    client = _get_client(api_url=api_url, api_key=api_key)
+    client = _get_client(api_key=api_key, api_url=api_url)
+
+    # Then it's langsmith
+    if hasattr(client, "push_prompt"):
+        return client.push_prompt(
+            repo_full_name,
+            object=object,
+            parent_commit_hash=parent_commit_hash,
+            is_public=new_repo_is_public,
+            description=new_repo_description,
+            readme=readme,
+            tags=tags,
+        )
+
+    # Then it's langchainhub
     manifest_json = dumps(object)
     message = client.push(
         repo_full_name,
@@ -67,20 +93,28 @@ def push(
 def pull(
     owner_repo_commit: str,
     *,
+    include_model: Optional[bool] = None,
     api_url: Optional[str] = None,
     api_key: Optional[str] = None,
 ) -> Any:
     """
     Pull an object from the hub and returns it as a LangChain object.
 
-    :param owner_repo_commit: The full name of the repo to pull from in the format of
-        `owner/repo:commit_hash`.
+    :param owner_repo_commit: The full name of the prompt to pull from in the format of
+        `owner/prompt_name:commit_hash` or `owner/prompt_name`
+        or just `prompt_name` if it's your own prompt.
     :param api_url: The URL of the LangChain Hub API. Defaults to the hosted API service
         if you have an api key set, or a localhost instance if not.
     :param api_key: The API key to use to authenticate with the LangChain Hub API.
     """
-    client = _get_client(api_url=api_url, api_key=api_key)
+    client = _get_client(api_key=api_key, api_url=api_url)
+
+    # Then it's langsmith
+    if hasattr(client, "pull_prompt"):
+        response = client.pull_prompt(owner_repo_commit, include_model=include_model)
+        return response
 
+    # Then it's langchainhub
     if hasattr(client, "pull_repo"):
         # >= 0.1.15
         res_dict = client.pull_repo(owner_repo_commit)
@@ -93,6 +127,6 @@ def pull(
             obj.metadata["lc_hub_commit_hash"] = res_dict["commit_hash"]
         return obj
 
-    # Then it's < 0.1.15
+    # Then it's < 0.1.15 langchainhub
     resp: str = client.pull(owner_repo_commit)
     return loads(resp)

langchain/indexes/vectorstore.py

@@ -4,7 +4,7 @@ from langchain_core.document_loaders import BaseLoader
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.language_models import BaseLanguageModel
-from langchain_core.pydantic_v1 import BaseModel, Extra, Field
+from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.vectorstores import VectorStore
 from langchain_text_splitters import RecursiveCharacterTextSplitter, TextSplitter
 
@@ -22,10 +22,8 @@ class VectorStoreIndexWrapper(BaseModel):
     vectorstore: VectorStore
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     def query(
         self,
@@ -145,10 +143,8 @@ class VectorstoreIndexCreator(BaseModel):
     vectorstore_kwargs: dict = Field(default_factory=dict)
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     def from_loaders(self, loaders: List[BaseLoader]) -> VectorStoreIndexWrapper:
         """Create a vectorstore index from loaders."""

langchain/memory/buffer.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, List, Optional
 
 from langchain_core.messages import BaseMessage, get_buffer_string
-from langchain_core.pydantic_v1 import root_validator
+from langchain_core.utils import pre_init
 
 from langchain.memory.chat_memory import BaseChatMemory, BaseMemory
 from langchain.memory.utils import get_prompt_input_key
@@ -82,7 +82,7 @@ class ConversationStringBufferMemory(BaseMemory):
     input_key: Optional[str] = None
     memory_key: str = "history"  #: :meta private:
 
-    @root_validator()
+    @pre_init
     def validate_chains(cls, values: Dict) -> Dict:
         """Validate that return messages is not True."""
         if values.get("return_messages", False):

langchain/memory/entity.py

@@ -246,8 +246,6 @@ class SQLiteEntityStore(BaseEntityStore):
     conn: Any = None
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
     def __init__(

langchain/memory/summary.py

@@ -2,17 +2,27 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Type
 
+from langchain_core._api import deprecated
 from langchain_core.chat_history import BaseChatMessageHistory
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import BaseMessage, SystemMessage, get_buffer_string
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, root_validator
+from langchain_core.pydantic_v1 import BaseModel
+from langchain_core.utils import pre_init
 
 from langchain.chains.llm import LLMChain
 from langchain.memory.chat_memory import BaseChatMemory
 from langchain.memory.prompt import SUMMARY_PROMPT
 
 
+@deprecated(
+    since="0.2.12",
+    removal="1.0",
+    message=(
+        "Refer here for how to incorporate summaries of conversation history: "
+        "https://langchain-ai.github.io/langgraph/how-tos/memory/add-summary-conversation-history/"  # noqa: E501
+    ),
+)
 class SummarizerMixin(BaseModel):
     """Mixin for summarizer."""
 
@@ -85,7 +95,7 @@ class ConversationSummaryMemory(BaseChatMemory, SummarizerMixin):
             buffer = self.buffer
         return {self.memory_key: buffer}
 
-    @root_validator()
+    @pre_init
     def validate_prompt_input_variables(cls, values: Dict) -> Dict:
         """Validate that prompt input variables are consistent."""
         prompt_variables = values["prompt"].input_variables

langchain/memory/summary_buffer.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, List, Union
 
 from langchain_core.messages import BaseMessage, get_buffer_string
-from langchain_core.pydantic_v1 import root_validator
+from langchain_core.utils import pre_init
 
 from langchain.memory.chat_memory import BaseChatMemory
 from langchain.memory.summary import SummarizerMixin
@@ -64,7 +64,7 @@ class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):
             )
         return {self.memory_key: final_buffer}
 
-    @root_validator()
+    @pre_init
     def validate_prompt_input_variables(cls, values: Dict) -> Dict:
         """Validate that prompt input variables are consistent."""
         prompt_variables = values["prompt"].input_variables

langchain/output_parsers/combining.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from typing import Any, Dict, List
 
 from langchain_core.output_parsers import BaseOutputParser
-from langchain_core.pydantic_v1 import root_validator
+from langchain_core.utils import pre_init
 
 
 class CombiningOutputParser(BaseOutputParser[Dict[str, Any]]):
@@ -15,7 +15,7 @@ class CombiningOutputParser(BaseOutputParser[Dict[str, Any]]):
     def is_lc_serializable(cls) -> bool:
         return True
 
-    @root_validator()
+    @pre_init
     def validate_parsers(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         """Validate the parsers."""
         parsers = values["parsers"]

langchain/output_parsers/enum.py

@@ -3,7 +3,7 @@ from typing import Dict, List, Type
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers import BaseOutputParser
-from langchain_core.pydantic_v1 import root_validator
+from langchain_core.utils import pre_init
 
 
 class EnumOutputParser(BaseOutputParser[Enum]):
@@ -12,7 +12,7 @@ class EnumOutputParser(BaseOutputParser[Enum]):
     enum: Type[Enum]
     """The enum to parse. Its values must be strings."""
 
-    @root_validator()
+    @pre_init
     def raise_deprecation(cls, values: Dict) -> Dict:
         enum = values["enum"]
         if not all(isinstance(e.value, str) for e in enum):