diff --git a/libs/core/langchain_core/memory.py b/libs/core/langchain_core/memory.py
index 3eef0404c3..7b9cab94a9 100644
--- a/libs/core/langchain_core/memory.py
+++ b/libs/core/langchain_core/memory.py
@@ -4,9 +4,8 @@ from __future__ import annotations
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel
-
 from langchain_core.load.serializable import Serializable
+from langchain_core.pydantic_v1 import BaseModel
 from langchain_core.runnables import run_in_executor
 
 
diff --git a/libs/langchain/langchain/memory/__init__.py b/libs/langchain/langchain/memory/__init__.py
index 9652083349..16f8a5c0a7 100644
--- a/libs/langchain/langchain/memory/__init__.py
+++ b/libs/langchain/langchain/memory/__init__.py
@@ -103,4 +103,8 @@ __all__ = [
     "ZepChatMessageHistory",
     "ZepMemory",
     "UpstashRedisChatMessageHistory",
+    "RedisEntityStore",
+    "SQLiteEntityStore",
+    "UpstashRedisEntityStore",
+    "InMemoryEntityStore",
 ]