langchain 0.2.15__py3-none-any.whl → 0.3.0__py3-none-any.whl

langchain/chains/combine_documents/map_rerank.py

@@ -6,9 +6,10 @@ from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import BaseModel, root_validator
 from langchain_core.runnables.config import RunnableConfig
 from langchain_core.runnables.utils import create_model
+from pydantic import BaseModel, ConfigDict, model_validator
+from typing_extensions import Self
 
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 from langchain.chains.llm import LLMChain
@@ -74,9 +75,10 @@ class MapRerankDocumentsChain(BaseCombineDocumentsChain):
     """Return intermediate steps.
     Intermediate steps include the results of calling llm_chain on each document."""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     def get_output_schema(
         self, config: Optional[RunnableConfig] = None
@@ -104,30 +106,31 @@ class MapRerankDocumentsChain(BaseCombineDocumentsChain):
             _output_keys += self.metadata_keys
         return _output_keys
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_llm_output(cls, values: Dict) -> Dict:
+    @model_validator(mode="after")
+    def validate_llm_output(self) -> Self:
         """Validate that the combine chain outputs a dictionary."""
-        output_parser = values["llm_chain"].prompt.output_parser
+        output_parser = self.llm_chain.prompt.output_parser
         if not isinstance(output_parser, RegexParser):
             raise ValueError(
                 "Output parser of llm_chain should be a RegexParser,"
                 f" got {output_parser}"
             )
         output_keys = output_parser.output_keys
-        if values["rank_key"] not in output_keys:
+        if self.rank_key not in output_keys:
             raise ValueError(
-                f"Got {values['rank_key']} as key to rank on, but did not find "
+                f"Got {self.rank_key} as key to rank on, but did not find "
                 f"it in the llm_chain output keys ({output_keys})"
             )
-        if values["answer_key"] not in output_keys:
+        if self.answer_key not in output_keys:
             raise ValueError(
-                f"Got {values['answer_key']} as key to return, but did not find "
+                f"Got {self.answer_key} as key to return, but did not find "
                 f"it in the llm_chain output keys ({output_keys})"
             )
-        return values
+        return self
 
-    @root_validator(pre=True)
-    def get_default_document_variable_name(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_default_document_variable_name(cls, values: Dict) -> Any:
         """Get default document variable name, if not provided."""
         if "llm_chain" not in values:
             raise ValueError("llm_chain must be provided")

langchain/chains/combine_documents/reduce.py

@@ -6,6 +6,7 @@ from typing import Any, Callable, List, Optional, Protocol, Tuple
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
+from pydantic import ConfigDict
 
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 
@@ -204,9 +205,10 @@ class ReduceDocumentsChain(BaseCombineDocumentsChain):
     If None, it will keep trying to collapse documents to fit token_max.
     Otherwise, after it reaches the max number, it will throw an error"""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def _collapse_chain(self) -> BaseCombineDocumentsChain:

langchain/chains/combine_documents/refine.py

@@ -8,7 +8,7 @@ from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.prompts import BasePromptTemplate, format_document
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import Field, root_validator
+from pydantic import ConfigDict, Field, model_validator
 
 from langchain.chains.combine_documents.base import (
     BaseCombineDocumentsChain,
@@ -98,20 +98,23 @@ class RefineDocumentsChain(BaseCombineDocumentsChain):
             _output_keys = _output_keys + ["intermediate_steps"]
         return _output_keys
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def get_return_intermediate_steps(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_return_intermediate_steps(cls, values: Dict) -> Any:
         """For backwards compatibility."""
         if "return_refine_steps" in values:
             values["return_intermediate_steps"] = values["return_refine_steps"]
             del values["return_refine_steps"]
         return values
 
-    @root_validator(pre=True)
-    def get_default_document_variable_name(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_default_document_variable_name(cls, values: Dict) -> Any:
         """Get default document variable name, if not provided."""
         if "initial_llm_chain" not in values:
             raise ValueError("initial_llm_chain must be provided")

langchain/chains/combine_documents/stuff.py

@@ -8,8 +8,8 @@ from langchain_core.documents import Document
 from langchain_core.language_models import LanguageModelLike
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import BasePromptTemplate, format_document
-from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.runnables import Runnable, RunnablePassthrough
+from pydantic import ConfigDict, Field, model_validator
 
 from langchain.chains.combine_documents.base import (
     DEFAULT_DOCUMENT_PROMPT,
@@ -156,12 +156,14 @@ class StuffDocumentsChain(BaseCombineDocumentsChain):
     document_separator: str = "\n\n"
     """The string with which to join the formatted documents"""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def get_default_document_variable_name(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_default_document_variable_name(cls, values: Dict) -> Any:
         """Get default document variable name, if not provided.
 
         If only one variable is present in the llm_chain.prompt,

langchain/chains/constitutional_ai/models.py

@@ -1,6 +1,6 @@
 """Models for the Constitutional AI chain."""
 
-from langchain_core.pydantic_v1 import BaseModel
+from pydantic import BaseModel
 
 
 class ConstitutionalPrinciple(BaseModel):

langchain/chains/conversation/base.py

@@ -1,11 +1,12 @@
 """Chain that carries on a conversation and calls an LLM."""
 
-from typing import Dict, List
+from typing import List
 
 from langchain_core._api import deprecated
 from langchain_core.memory import BaseMemory
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Field, root_validator
+from pydantic import ConfigDict, Field, model_validator
+from typing_extensions import Self
 
 from langchain.chains.conversation.prompt import PROMPT
 from langchain.chains.llm import LLMChain
@@ -110,9 +111,10 @@ class ConversationChain(LLMChain):
     input_key: str = "input"  #: :meta private:
     output_key: str = "response"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @classmethod
     def is_lc_serializable(cls) -> bool:
@@ -123,17 +125,17 @@ class ConversationChain(LLMChain):
         """Use this since so some prompt vars come from history."""
         return [self.input_key]
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_prompt_input_variables(cls, values: Dict) -> Dict:
+    @model_validator(mode="after")
+    def validate_prompt_input_variables(self) -> Self:
         """Validate that prompt input variables are consistent."""
-        memory_keys = values["memory"].memory_variables
-        input_key = values["input_key"]
+        memory_keys = self.memory.memory_variables
+        input_key = self.input_key
         if input_key in memory_keys:
             raise ValueError(
                 f"The input key {input_key} was also found in the memory keys "
                 f"({memory_keys}) - please provide keys that don't overlap."
             )
-        prompt_variables = values["prompt"].input_variables
+        prompt_variables = self.prompt.input_variables
         expected_keys = memory_keys + [input_key]
         if set(expected_keys) != set(prompt_variables):
             raise ValueError(
@@ -141,4 +143,4 @@ class ConversationChain(LLMChain):
                 f"{prompt_variables}, but got {memory_keys} as inputs from "
                 f"memory, and {input_key} as the normal input key."
             )
-        return values
+        return self

langchain/chains/conversational_retrieval/base.py

@@ -18,10 +18,10 @@ from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import BaseMessage
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import RunnableConfig
 from langchain_core.vectorstores import VectorStore
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
@@ -92,14 +92,15 @@ class BaseConversationalRetrievalChain(Chain):
     get_chat_history: Optional[Callable[[List[CHAT_TURN_TYPE]], str]] = None
     """An optional function to get a string of the chat history.
     If None is provided, will use a default."""
-    response_if_no_docs_found: Optional[str]
+    response_if_no_docs_found: Optional[str] = None
     """If specified, the chain will return a fixed response if no docs 
     are found for the question. """
 
-    class Config:
-        allow_population_by_field_name = True
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        populate_by_name=True,
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
@@ -482,8 +483,9 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
     def _chain_type(self) -> str:
         return "chat-vector-db"
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         warnings.warn(
             "`ChatVectorDBChain` is deprecated - "
             "please use `from langchain.chains import ConversationalRetrievalChain`"

langchain/chains/elasticsearch_database/base.py

@@ -9,8 +9,9 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.output_parsers.json import SimpleJsonOutputParser
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import root_validator
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict, model_validator
+from typing_extensions import Self
 
 from langchain.chains.base import Chain
 from langchain.chains.elasticsearch_database.prompts import ANSWER_PROMPT, DSL_PROMPT
@@ -39,7 +40,7 @@ class ElasticsearchDatabaseChain(Chain):
     """Chain for creating the ES query."""
     answer_chain: Runnable
     """Chain for answering the user question."""
-    database: Any
+    database: Any = None
     """Elasticsearch database to connect to of type elasticsearch.Elasticsearch."""
     top_k: int = 10
     """Number of results to return from the query"""
@@ -51,17 +52,18 @@ class ElasticsearchDatabaseChain(Chain):
     return_intermediate_steps: bool = False
     """Whether or not to return the intermediate steps along with the final answer."""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_indices(cls, values: dict) -> dict:
-        if values["include_indices"] and values["ignore_indices"]:
+    @model_validator(mode="after")
+    def validate_indices(self) -> Self:
+        if self.include_indices and self.ignore_indices:
             raise ValueError(
                 "Cannot specify both 'include_indices' and 'ignore_indices'."
             )
-        return values
+        return self
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/flare/base.py

@@ -11,9 +11,9 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import AIMessage
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Field
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import Runnable
+from pydantic import Field
 
 from langchain.chains.base import Chain
 from langchain.chains.flare.prompts import (
@@ -73,7 +73,10 @@ def _low_confidence_spans(
 
 class FlareChain(Chain):
     """Chain that combines a retriever, a question generator,
-    and a response generator."""
+    and a response generator.
+
+    See [Active Retrieval Augmented Generation](https://arxiv.org/abs/2305.06983) paper.
+    """
 
     question_generator_chain: Runnable
     """Chain that generates questions from uncertain spans."""

langchain/chains/hyde/base.py

@@ -14,6 +14,7 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import BasePromptTemplate
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict
 
 from langchain.chains.base import Chain
 from langchain.chains.hyde.prompts import PROMPT_MAP
@@ -29,14 +30,15 @@ class HypotheticalDocumentEmbedder(Chain, Embeddings):
     base_embeddings: Embeddings
     llm_chain: Runnable
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
         """Input keys for Hyde's LLM chain."""
-        return self.llm_chain.input_schema.schema()["required"]
+        return self.llm_chain.input_schema.model_json_schema()["required"]
 
     @property
     def output_keys(self) -> List[str]:

langchain/chains/llm.py

@@ -17,13 +17,11 @@ from langchain_core.language_models import (
     BaseLanguageModel,
     LanguageModelInput,
 )
-from langchain_core.load.dump import dumpd
 from langchain_core.messages import BaseMessage
 from langchain_core.output_parsers import BaseLLMOutputParser, StrOutputParser
 from langchain_core.outputs import ChatGeneration, Generation, LLMResult
 from langchain_core.prompt_values import PromptValue
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
-from langchain_core.pydantic_v1 import Field
 from langchain_core.runnables import (
     Runnable,
     RunnableBinding,
@@ -32,6 +30,7 @@ from langchain_core.runnables import (
 )
 from langchain_core.runnables.configurable import DynamicRunnable
 from langchain_core.utils.input import get_colored_text
+from pydantic import ConfigDict, Field
 
 from langchain.chains.base import Chain
 
@@ -95,9 +94,10 @@ class LLMChain(Chain):
     If false, will return a bunch of extra information about the generation."""
     llm_kwargs: dict = Field(default_factory=dict)
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
@@ -240,7 +240,7 @@ class LLMChain(Chain):
             callbacks, self.callbacks, self.verbose
         )
         run_manager = callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             {"input_list": input_list},
         )
         try:
@@ -260,7 +260,7 @@ class LLMChain(Chain):
             callbacks, self.callbacks, self.verbose
         )
         run_manager = await callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             {"input_list": input_list},
         )
         try:

langchain/chains/llm_checker/base.py

@@ -9,7 +9,7 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import PromptTemplate
-from langchain_core.pydantic_v1 import root_validator
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -100,12 +100,14 @@ class LLMCheckerChain(Chain):
     input_key: str = "query"  #: :meta private:
     output_key: str = "result"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         if "llm" in values:
             warnings.warn(
                 "Directly instantiating an LLMCheckerChain with an llm is deprecated. "

langchain/chains/llm_math/base.py

@@ -14,7 +14,7 @@ from langchain_core.callbacks import (
 )
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import root_validator
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -156,12 +156,14 @@ class LLMMathChain(Chain):
     input_key: str = "question"  #: :meta private:
     output_key: str = "answer"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         try:
             import numexpr  # noqa: F401
         except ImportError:

langchain/chains/llm_summarization_checker/base.py

@@ -10,7 +10,7 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import root_validator
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -105,12 +105,14 @@ class LLMSummarizationCheckerChain(Chain):
     max_checks: int = 2
     """Maximum number of times to check the assertions. Default to double-checking."""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         if "llm" in values:
             warnings.warn(
                 "Directly instantiating an LLMSummarizationCheckerChain with an llm is "

langchain/chains/mapreduce.py

@@ -14,6 +14,7 @@ from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
 from langchain_text_splitters import TextSplitter
+from pydantic import ConfigDict
 
 from langchain.chains import ReduceDocumentsChain
 from langchain.chains.base import Chain
@@ -77,9 +78,10 @@ class MapReduceChain(Chain):
             **kwargs,
         )
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/moderation.py

@@ -6,8 +6,8 @@ from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
-from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.utils import check_package_version, get_from_dict_or_env
+from pydantic import Field, model_validator
 
 from langchain.chains.base import Chain
 
@@ -28,8 +28,8 @@ class OpenAIModerationChain(Chain):
             moderation = OpenAIModerationChain()
     """
 
-    client: Any  #: :meta private:
-    async_client: Any  #: :meta private:
+    client: Any = None  #: :meta private:
+    async_client: Any = None  #: :meta private:
     model_name: Optional[str] = None
     """Moderation model name to use."""
     error: bool = False
@@ -38,10 +38,11 @@ class OpenAIModerationChain(Chain):
     output_key: str = "output"  #: :meta private:
     openai_api_key: Optional[str] = None
     openai_organization: Optional[str] = None
-    _openai_pre_1_0: bool = Field(default=None)
+    openai_pre_1_0: bool = Field(default=None)
 
-    @root_validator(pre=True)
-    def validate_environment(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def validate_environment(cls, values: Dict) -> Any:
         """Validate that api key and python package exists in environment."""
         openai_api_key = get_from_dict_or_env(
             values, "openai_api_key", "OPENAI_API_KEY"
@@ -58,16 +59,17 @@ class OpenAIModerationChain(Chain):
             openai.api_key = openai_api_key
             if openai_organization:
                 openai.organization = openai_organization
-            values["_openai_pre_1_0"] = False
+            values["openai_pre_1_0"] = False
             try:
                 check_package_version("openai", gte_version="1.0")
             except ValueError:
-                values["_openai_pre_1_0"] = True
-            if values["_openai_pre_1_0"]:
+                values["openai_pre_1_0"] = True
+            if values["openai_pre_1_0"]:
                 values["client"] = openai.Moderation
             else:
                 values["client"] = openai.OpenAI()
                 values["async_client"] = openai.AsyncOpenAI()
+
         except ImportError:
             raise ImportError(
                 "Could not import openai python package. "
@@ -92,7 +94,7 @@ class OpenAIModerationChain(Chain):
         return [self.output_key]
 
     def _moderate(self, text: str, results: Any) -> str:
-        if self._openai_pre_1_0:
+        if self.openai_pre_1_0:
             condition = results["flagged"]
         else:
             condition = results.flagged
@@ -110,7 +112,7 @@ class OpenAIModerationChain(Chain):
         run_manager: Optional[CallbackManagerForChainRun] = None,
     ) -> Dict[str, Any]:
         text = inputs[self.input_key]
-        if self._openai_pre_1_0:
+        if self.openai_pre_1_0:
             results = self.client.create(text)
             output = self._moderate(text, results["results"][0])
         else:
@@ -123,7 +125,7 @@ class OpenAIModerationChain(Chain):
         inputs: Dict[str, Any],
         run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
     ) -> Dict[str, Any]:
-        if self._openai_pre_1_0:
+        if self.openai_pre_1_0:
             return await super()._acall(inputs, run_manager=run_manager)
         text = inputs[self.input_key]
         results = await self.async_client.moderations.create(input=text)

langchain/chains/natbot/base.py

@@ -9,8 +9,8 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import StrOutputParser
-from langchain_core.pydantic_v1 import root_validator
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.natbot.prompt import PROMPT
@@ -59,12 +59,14 @@ class NatBotChain(Chain):
     previous_command: str = ""  #: :meta private:
     output_key: str = "command"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         if "llm" in values:
             warnings.warn(
                 "Directly instantiating an NatBotChain with an llm is deprecated. "