x

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
 
 

libs/langchain/langchain/memory/__init__.py

@@ -103,4 +103,8 @@ __all__ = [
     "ZepChatMessageHistory",
     "ZepMemory",
     "UpstashRedisChatMessageHistory",
+    "RedisEntityStore",
+    "SQLiteEntityStore",
+    "UpstashRedisEntityStore",
+    "InMemoryEntityStore",
 ]