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

langchain/chains/qa_with_sources/base.py

@@ -15,7 +15,7 @@ from langchain_core.callbacks import (
 from langchain_core.documents import Document
 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 import ReduceDocumentsChain
 from langchain.chains.base import Chain
@@ -97,9 +97,10 @@ class BaseQAWithSourcesChain(Chain, ABC):
         )
         return cls(combine_documents_chain=combine_documents_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]:
@@ -120,8 +121,9 @@ class BaseQAWithSourcesChain(Chain, ABC):
             _output_keys = _output_keys + ["source_documents"]
         return _output_keys
 
-    @root_validator(pre=True)
-    def validate_naming(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def validate_naming(cls, values: Dict) -> Any:
         """Fix backwards compatibility in naming."""
         if "combine_document_chain" in values:
             values["combine_documents_chain"] = values.pop("combine_document_chain")

langchain/chains/qa_with_sources/retrieval.py

@@ -7,8 +7,8 @@ from langchain_core.callbacks import (
     CallbackManagerForChainRun,
 )
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import Field
 from langchain_core.retrievers import BaseRetriever
+from pydantic import Field
 
 from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.qa_with_sources.base import BaseQAWithSourcesChain

langchain/chains/qa_with_sources/vector_db.py

@@ -8,8 +8,8 @@ from langchain_core.callbacks import (
     CallbackManagerForChainRun,
 )
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.vectorstores import VectorStore
+from pydantic import Field, model_validator
 
 from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.qa_with_sources.base import BaseQAWithSourcesChain
@@ -61,8 +61,9 @@ class VectorDBQAWithSourcesChain(BaseQAWithSourcesChain):
     ) -> List[Document]:
         raise NotImplementedError("VectorDBQAWithSourcesChain does not support async")
 
-    @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(
             "`VectorDBQAWithSourcesChain` is deprecated - "
             "please use `from langchain.chains import RetrievalQAWithSourcesChain`"

langchain/chains/query_constructor/schema.py

@@ -1,4 +1,4 @@
-from langchain_core.pydantic_v1 import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 
 class AttributeInfo(BaseModel):
@@ -8,6 +8,7 @@ class AttributeInfo(BaseModel):
     description: str
     type: str
 
-    class Config:
-        arbitrary_types_allowed = True
-        frozen = True
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        frozen=True,
+    )

langchain/chains/retrieval_qa/base.py

@@ -16,9 +16,9 @@ from langchain_core.callbacks import (
 from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import PromptTemplate
-from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.vectorstores import VectorStore
+from pydantic import ConfigDict, Field, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
@@ -47,10 +47,11 @@ class BaseRetrievalQA(Chain):
     return_source_documents: bool = False
     """Return the source documents or not."""
 
-    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]:
@@ -309,16 +310,18 @@ class VectorDBQA(BaseRetrievalQA):
     search_kwargs: Dict[str, Any] = Field(default_factory=dict)
     """Extra search args."""
 
-    @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(
             "`VectorDBQA` is deprecated - "
             "please use `from langchain.chains import RetrievalQA`"
         )
         return values
 
-    @root_validator(pre=True)
-    def validate_search_type(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def validate_search_type(cls, values: Dict) -> Any:
         """Validate search type."""
         if "search_type" in values:
             search_type = values["search_type"]

langchain/chains/router/base.py

@@ -10,6 +10,7 @@ from langchain_core.callbacks import (
     CallbackManagerForChainRun,
     Callbacks,
 )
+from pydantic import ConfigDict
 
 from langchain.chains.base import Chain
 
@@ -60,9 +61,10 @@ class MultiRouteChain(Chain):
     """If True, use default_chain when an invalid destination name is provided. 
     Defaults to 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]:

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(
@@ -44,7 +44,7 @@ from langchain_core.utils.pydantic import is_basemodel_subclass
     removal="1.0",
     alternative=(
         """
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
             from langchain_anthropic import ChatAnthropic
     
             class Joke(BaseModel):
@@ -108,7 +108,7 @@ def create_openai_fn_runnable(
 
                 from langchain.chains.structured_output import create_openai_fn_runnable
                 from langchain_openai import ChatOpenAI
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
 
                 class RecordPerson(BaseModel):
@@ -162,7 +162,7 @@ def create_openai_fn_runnable(
     removal="1.0",
     alternative=(
         """
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
             from langchain_anthropic import ChatAnthropic
 
             class Joke(BaseModel):
@@ -237,7 +237,7 @@ def create_structured_output_runnable(
 
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai import ChatOpenAI
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
 
                 class RecordDog(BaseModel):
@@ -318,7 +318,7 @@ def create_structured_output_runnable(
 
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai import ChatOpenAI
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
                 class Dog(BaseModel):
                     '''Identifying information about a dog.'''
@@ -340,7 +340,7 @@ def create_structured_output_runnable(
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai import ChatOpenAI
                 from langchain_core.prompts import ChatPromptTemplate
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
                 class Dog(BaseModel):
                     '''Identifying information about a dog.'''
@@ -366,7 +366,7 @@ def create_structured_output_runnable(
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai import ChatOpenAI
                 from langchain_core.prompts import ChatPromptTemplate
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
                 class Dog(BaseModel):
                     '''Identifying information about a dog.'''

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__ = [
@@ -241,7 +241,7 @@ def init_chat_model(
 
             # pip install langchain langchain-openai langchain-anthropic
             from langchain.chat_models import init_chat_model
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
 
             class GetWeather(BaseModel):
                 '''Get the current weather in a given location'''

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 field_validator
 
 from langchain.memory.chat_memory import BaseChatMemory
 
@@ -13,7 +13,8 @@ class CombinedMemory(BaseMemory):
     memories: List[BaseMemory]
     """For tracking all the memories that should be accessed."""
 
-    @validator("memories")
+    @field_validator("memories")
+    @classmethod
     def check_repeated_memory_variable(
         cls, value: List[BaseMemory]
     ) -> List[BaseMemory]:
@@ -29,7 +30,8 @@ class CombinedMemory(BaseMemory):
 
         return value
 
-    @validator("memories")
+    @field_validator("memories")
+    @classmethod
     def check_input_key(cls, value: List[BaseMemory]) -> List[BaseMemory]:
         """Check that if memories are of type BaseChatMemory that input keys exist."""
         for val in value:

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,