langchain 0.2.16__py3-none-any.whl → 0.3.0.dev1__py3-none-any.whl

langchain/chains/router/embedding_router.py

@@ -9,6 +9,7 @@ from langchain_core.callbacks import (
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VectorStore
+from pydantic import ConfigDict
 
 from langchain.chains.router.base import RouterChain
 
@@ -19,9 +20,10 @@ class EmbeddingRouterChain(RouterChain):
     vectorstore: VectorStore
     routing_keys: List[str] = ["query"]
 
-    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/router/llm_router.py

@@ -13,8 +13,9 @@ from langchain_core.exceptions import OutputParserException
 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 root_validator
 from langchain_core.utils.json import parse_and_check_json_markdown
+from pydantic import model_validator
+from typing_extensions import Self
 
 from langchain.chains import LLMChain
 from langchain.chains.router.base import RouterChain
@@ -100,9 +101,9 @@ class LLMRouterChain(RouterChain):
     llm_chain: LLMChain
     """LLM chain used to perform routing"""
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_prompt(cls, values: dict) -> dict:
-        prompt = values["llm_chain"].prompt
+    @model_validator(mode="after")
+    def validate_prompt(self) -> Self:
+        prompt = self.llm_chain.prompt
         if prompt.output_parser is None:
             raise ValueError(
                 "LLMRouterChain requires base llm_chain prompt to have an output"
@@ -110,7 +111,7 @@ class LLMRouterChain(RouterChain):
                 " 'destination' and 'next_inputs'. Received a prompt with no output"
                 " parser."
             )
-        return values
+        return self
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/sequential.py

@@ -6,8 +6,9 @@ from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
-from langchain_core.pydantic_v1 import root_validator
 from langchain_core.utils.input import get_color_mapping
+from pydantic import ConfigDict, model_validator
+from typing_extensions import Self
 
 from langchain.chains.base import Chain
 
@@ -20,9 +21,10 @@ class SequentialChain(Chain):
     output_variables: List[str]  #: :meta private:
     return_all: bool = False
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
@@ -40,8 +42,9 @@ class SequentialChain(Chain):
         """
         return self.output_variables
 
-    @root_validator(pre=True)
-    def validate_chains(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def validate_chains(cls, values: Dict) -> Any:
         """Validate that the correct inputs exist for all chains."""
         chains = values["chains"]
         input_variables = values["input_variables"]
@@ -129,9 +132,10 @@ class SimpleSequentialChain(Chain):
     input_key: str = "input"  #: :meta private:
     output_key: str = "output"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
@@ -149,10 +153,10 @@ class SimpleSequentialChain(Chain):
         """
         return [self.output_key]
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_chains(cls, values: Dict) -> Dict:
+    @model_validator(mode="after")
+    def validate_chains(self) -> Self:
         """Validate that chains are all single input/output."""
-        for chain in values["chains"]:
+        for chain in self.chains:
             if len(chain.input_keys) != 1:
                 raise ValueError(
                     "Chains used in SimplePipeline should all have one input, got "
@@ -163,7 +167,7 @@ class SimpleSequentialChain(Chain):
                     "Chains used in SimplePipeline should all have one output, got "
                     f"{chain} with {len(chain.output_keys)} outputs."
                 )
-        return values
+        return self
 
     def _call(
         self,

langchain/chains/structured_output/base.py

@@ -18,13 +18,13 @@ from langchain_core.output_parsers.openai_tools import (
     PydanticToolsParser,
 )
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel
 from langchain_core.runnables import Runnable
 from langchain_core.utils.function_calling import (
     convert_to_openai_function,
     convert_to_openai_tool,
 )
 from langchain_core.utils.pydantic import is_basemodel_subclass
+from pydantic import BaseModel
 
 
 @deprecated(

langchain/chains/transform.py

@@ -8,7 +8,7 @@ from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
-from langchain_core.pydantic_v1 import Field
+from pydantic import Field
 
 from langchain.chains.base import Chain
 

langchain/chat_models/base.py

@@ -30,11 +30,11 @@ from langchain_core.language_models.chat_models import (
     generate_from_stream,
 )
 from langchain_core.messages import AnyMessage, BaseMessage
-from langchain_core.pydantic_v1 import BaseModel
 from langchain_core.runnables import Runnable, RunnableConfig
 from langchain_core.runnables.schema import StreamEvent
 from langchain_core.tools import BaseTool
 from langchain_core.tracers import RunLog, RunLogPatch
+from pydantic import BaseModel
 from typing_extensions import TypeAlias
 
 __all__ = [

langchain/evaluation/agents/trajectory_eval_chain.py

@@ -28,8 +28,8 @@ 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 Field
 from langchain_core.tools import BaseTool
+from pydantic import ConfigDict, Field
 
 from langchain.chains.llm import LLMChain
 from langchain.evaluation.agents.trajectory_eval_prompt import (
@@ -156,8 +156,9 @@ class TrajectoryEvalChain(AgentTrajectoryEvaluator, LLMEvalChain):
     return_reasoning: bool = False  # :meta private:
     """DEPRECATED. Reasoning always returned."""
 
-    class Config:
-        extra = "ignore"
+    model_config = ConfigDict(
+        extra="ignore",
+    )
 
     @property
     def requires_reference(self) -> bool:

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 Field
+from pydantic import ConfigDict, Field
 
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 from langchain.chains.llm import LLMChain
@@ -191,8 +191,9 @@ class PairwiseStringEvalChain(PairwiseStringEvaluator, LLMEvalChain, LLMChain):
     def is_lc_serializable(cls) -> bool:
         return False
 
-    class Config:
-        extra = "ignore"
+    model_config = ConfigDict(
+        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 Field
+from pydantic import ConfigDict, Field
 
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 from langchain.chains.llm import LLMChain
@@ -236,8 +236,9 @@ class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):
     def is_lc_serializable(cls) -> bool:
         return False
 
-    class Config:
-        extra = "ignore"
+    model_config = ConfigDict(
+        extra="ignore",
+    )
 
     @property
     def requires_reference(self) -> bool:

langchain/evaluation/embedding_distance/base.py

@@ -10,8 +10,8 @@ from langchain_core.callbacks.manager import (
     Callbacks,
 )
 from langchain_core.embeddings import Embeddings
-from langchain_core.pydantic_v1 import Field
 from langchain_core.utils import pre_init
+from pydantic import ConfigDict, Field
 
 from langchain.chains.base import Chain
 from langchain.evaluation.schema import PairwiseStringEvaluator, StringEvaluator
@@ -113,8 +113,9 @@ class _EmbeddingDistanceChainMixin(Chain):
                 )
         return values
 
-    class Config:
-        arbitrary_types_allowed: bool = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
 
     @property
     def output_keys(self) -> List[str]:

langchain/evaluation/qa/eval_chain.py

@@ -9,6 +9,7 @@ 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 pydantic import ConfigDict
 
 from langchain.chains.llm import LLMChain
 from langchain.evaluation.qa.eval_prompt import CONTEXT_PROMPT, COT_PROMPT, PROMPT
@@ -72,8 +73,9 @@ class QAEvalChain(LLMChain, StringEvaluator, LLMEvalChain):
 
     output_key: str = "results"  #: :meta private:
 
-    class Config:
-        extra = "ignore"
+    model_config = ConfigDict(
+        extra="ignore",
+    )
 
     @classmethod
     def is_lc_serializable(cls) -> bool:
@@ -220,8 +222,9 @@ class ContextQAEvalChain(LLMChain, StringEvaluator, LLMEvalChain):
         """Whether the chain requires an input string."""
         return True
 
-    class Config:
-        extra = "ignore"
+    model_config = ConfigDict(
+        extra="ignore",
+    )
 
     @classmethod
     def _validate_input_vars(cls, prompt: PromptTemplate) -> None:

langchain/evaluation/qa/generate_chain.py

@@ -6,7 +6,7 @@ from typing import Any
 
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseLLMOutputParser
-from langchain_core.pydantic_v1 import Field
+from pydantic import Field
 
 from langchain.chains.llm import LLMChain
 from langchain.evaluation.qa.generate_prompt import PROMPT

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 Field
+from pydantic import ConfigDict, Field
 
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 from langchain.chains.llm import LLMChain
@@ -179,8 +179,9 @@ class ScoreStringEvalChain(StringEvaluator, LLMEvalChain, LLMChain):
     criterion_name: str
     """The name of the criterion being evaluated."""
 
-    class Config:
-        extra = "ignore"
+    model_config = ConfigDict(
+        extra="ignore",
+    )
 
     @classmethod
     def is_lc_serializable(cls) -> bool:

langchain/evaluation/string_distance/base.py

@@ -8,8 +8,8 @@ from langchain_core.callbacks.manager import (
     CallbackManagerForChainRun,
     Callbacks,
 )
-from langchain_core.pydantic_v1 import Field
 from langchain_core.utils import pre_init
+from pydantic import Field
 
 from langchain.chains.base import Chain
 from langchain.evaluation.schema import PairwiseStringEvaluator, StringEvaluator

langchain/indexes/vectorstore.py

@@ -4,9 +4,9 @@ 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, Field
 from langchain_core.vectorstores import VectorStore
 from langchain_text_splitters import RecursiveCharacterTextSplitter, TextSplitter
+from pydantic import BaseModel, ConfigDict, Field
 
 from langchain.chains.qa_with_sources.retrieval import RetrievalQAWithSourcesChain
 from langchain.chains.retrieval_qa.base import RetrievalQA
@@ -21,9 +21,10 @@ class VectorStoreIndexWrapper(BaseModel):
 
     vectorstore: VectorStore
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     def query(
         self,
@@ -142,9 +143,10 @@ class VectorstoreIndexCreator(BaseModel):
     text_splitter: TextSplitter = Field(default_factory=_get_default_text_splitter)
     vectorstore_kwargs: dict = Field(default_factory=dict)
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     def from_loaders(self, loaders: List[BaseLoader]) -> VectorStoreIndexWrapper:
         """Create a vectorstore index from loaders."""

langchain/memory/chat_memory.py

@@ -8,7 +8,7 @@ from langchain_core.chat_history import (
 )
 from langchain_core.memory import BaseMemory
 from langchain_core.messages import AIMessage, HumanMessage
-from langchain_core.pydantic_v1 import Field
+from pydantic import Field
 
 from langchain.memory.utils import get_prompt_input_key
 

langchain/memory/combined.py

@@ -2,7 +2,7 @@ import warnings
 from typing import Any, Dict, List, Set
 
 from langchain_core.memory import BaseMemory
-from langchain_core.pydantic_v1 import validator
+from pydantic import validator
 
 from langchain.memory.chat_memory import BaseChatMemory
 

langchain/memory/entity.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, Iterable, List, Optional
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from langchain.chains.llm import LLMChain
 from langchain.memory.chat_memory import BaseChatMemory
@@ -245,8 +245,9 @@ class SQLiteEntityStore(BaseEntityStore):
     table_name: str = "memory_store"
     conn: Any = None
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
 
     def __init__(
         self,

langchain/memory/summary.py

@@ -7,8 +7,8 @@ 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
 from langchain_core.utils import pre_init
+from pydantic import BaseModel
 
 from langchain.chains.llm import LLMChain
 from langchain.memory.chat_memory import BaseChatMemory

langchain/memory/vectorstore.py

@@ -3,8 +3,8 @@
 from typing import Any, Dict, List, Optional, Sequence, Union
 
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import Field
 from langchain_core.vectorstores import VectorStoreRetriever
+from pydantic import Field
 
 from langchain.memory.chat_memory import BaseMemory
 from langchain.memory.utils import get_prompt_input_key

langchain/memory/vectorstore_token_buffer_memory.py

@@ -13,8 +13,8 @@ from typing import Any, Dict, List
 
 from langchain_core.messages import BaseMessage
 from langchain_core.prompts.chat import SystemMessagePromptTemplate
-from langchain_core.pydantic_v1 import Field, PrivateAttr
 from langchain_core.vectorstores import VectorStoreRetriever
+from pydantic import Field, PrivateAttr
 
 from langchain.memory import ConversationTokenBufferMemory, VectorStoreRetrieverMemory
 from langchain.memory.chat_memory import BaseChatMemory

langchain/output_parsers/fix.py

@@ -1,11 +1,12 @@
 from __future__ import annotations
 
-from typing import Any, TypeVar, Union
+from typing import Annotated, Any, TypeVar, Union
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import BasePromptTemplate
 from langchain_core.runnables import Runnable, RunnableSerializable
+from pydantic import SkipValidation
 from typing_extensions import TypedDict
 
 from langchain.output_parsers.prompts import NAIVE_FIX_PROMPT
@@ -26,7 +27,7 @@ class OutputFixingParser(BaseOutputParser[T]):
     def is_lc_serializable(cls) -> bool:
         return True
 
-    parser: BaseOutputParser[T]
+    parser: Annotated[BaseOutputParser[T], SkipValidation()]
     """The parser to use to parse the output."""
     # Should be an LLMChain but we want to avoid top-level imports from langchain.chains
     retry_chain: Union[

langchain/output_parsers/pandas_dataframe.py

@@ -3,7 +3,7 @@ from typing import Any, Dict, List, Tuple, Union
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers.base import BaseOutputParser
-from langchain_core.pydantic_v1 import validator
+from pydantic import validator
 
 from langchain.output_parsers.format_instructions import (
     PANDAS_DATAFRAME_FORMAT_INSTRUCTIONS,

langchain/output_parsers/retry.py

@@ -8,7 +8,8 @@ from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompt_values import PromptValue
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
 from langchain_core.runnables import RunnableSerializable
-from typing_extensions import TypedDict
+from pydantic import SkipValidation
+from typing_extensions import Annotated, TypedDict
 
 NAIVE_COMPLETION_RETRY = """Prompt:
 {prompt}
@@ -53,7 +54,7 @@ class RetryOutputParser(BaseOutputParser[T]):
     LLM, and telling it the completion did not satisfy criteria in the prompt.
     """
 
-    parser: BaseOutputParser[T]
+    parser: Annotated[BaseOutputParser[T], SkipValidation()]
     """The parser to use to parse the output."""
     # Should be an LLMChain but we want to avoid top-level imports from langchain.chains
     retry_chain: Union[RunnableSerializable[RetryOutputParserRetryChainInput, str], Any]
@@ -183,7 +184,7 @@ class RetryWithErrorOutputParser(BaseOutputParser[T]):
     LLM, which in theory should give it more information on how to fix it.
     """
 
-    parser: BaseOutputParser[T]
+    parser: Annotated[BaseOutputParser[T], SkipValidation()]
     """The parser to use to parse the output."""
     # Should be an LLMChain but we want to avoid top-level imports from langchain.chains
     retry_chain: Union[

langchain/output_parsers/structured.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, List
 
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.output_parsers.json import parse_and_check_json_markdown
-from langchain_core.pydantic_v1 import BaseModel
+from pydantic import BaseModel
 
 from langchain.output_parsers.format_instructions import (
     STRUCTURED_FORMAT_INSTRUCTIONS,

langchain/output_parsers/yaml.py

@@ -5,7 +5,7 @@ from typing import Type, TypeVar
 import yaml
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers import BaseOutputParser
-from langchain_core.pydantic_v1 import BaseModel, ValidationError
+from pydantic import BaseModel, ValidationError
 
 from langchain.output_parsers.format_instructions import YAML_FORMAT_INSTRUCTIONS
 

langchain/retrievers/contextual_compression.py

@@ -6,6 +6,7 @@ from langchain_core.callbacks import (
 )
 from langchain_core.documents import Document
 from langchain_core.retrievers import BaseRetriever, RetrieverLike
+from pydantic import ConfigDict
 
 from langchain.retrievers.document_compressors.base import (
     BaseDocumentCompressor,
@@ -21,8 +22,9 @@ class ContextualCompressionRetriever(BaseRetriever):
     base_retriever: RetrieverLike
     """Base Retriever to use for getting relevant documents."""
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
 
     def _get_relevant_documents(
         self,

langchain/retrievers/document_compressors/base.py

@@ -7,6 +7,7 @@ from langchain_core.documents import (
     BaseDocumentTransformer,
     Document,
 )
+from pydantic import ConfigDict
 
 
 class DocumentCompressorPipeline(BaseDocumentCompressor):
@@ -15,8 +16,9 @@ class DocumentCompressorPipeline(BaseDocumentCompressor):
     transformers: List[Union[BaseDocumentTransformer, BaseDocumentCompressor]]
     """List of document filters that are chained together and run in sequence."""
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
 
     def compress_documents(
         self,

langchain/retrievers/document_compressors/chain_extract.py

@@ -11,6 +11,7 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import PromptTemplate
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict
 
 from langchain.chains.llm import LLMChain
 from langchain.retrievers.document_compressors.base import BaseDocumentCompressor
@@ -56,8 +57,9 @@ class LLMChainExtractor(BaseDocumentCompressor):
     get_input: Callable[[str, Document], dict] = default_get_input
     """Callable for constructing the chain input from the query and a Document."""
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
 
     def compress_documents(
         self,

langchain/retrievers/document_compressors/chain_filter.py

@@ -9,6 +9,7 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
 from langchain_core.runnables import Runnable
 from langchain_core.runnables.config import RunnableConfig
+from pydantic import ConfigDict
 
 from langchain.chains import LLMChain
 from langchain.output_parsers.boolean import BooleanOutputParser
@@ -41,8 +42,9 @@ class LLMChainFilter(BaseDocumentCompressor):
     get_input: Callable[[str, Document], dict] = default_get_input
     """Callable for constructing the chain input from the query and a Document."""
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+    )
 
     def compress_documents(
         self,

langchain/retrievers/document_compressors/cohere_rerank.py

@@ -6,8 +6,8 @@ from typing import Any, Dict, List, Optional, Sequence, Union
 from langchain_core._api.deprecation import deprecated
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import root_validator
 from langchain_core.utils import get_from_dict_or_env
+from pydantic import ConfigDict, model_validator
 
 from langchain.retrievers.document_compressors.base import BaseDocumentCompressor
 
@@ -30,12 +30,14 @@ class CohereRerank(BaseDocumentCompressor):
     user_agent: str = "langchain"
     """Identifier for the application making the request."""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @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."""
         if not values.get("client"):
             try: