From 017ae731d4992aa970dabf846111d87eda304f05 Mon Sep 17 00:00:00 2001
From: Eugene Yurtsev <eyurtsev@gmail.com>
Date: Tue, 23 Apr 2024 16:34:50 -0400
Subject: [PATCH] x

---
 libs/core/langchain_core/legacy/memory/__init__.py       | 2 +-
 libs/core/langchain_core/legacy/memory/base.py           | 2 +-
 libs/core/langchain_core/legacy/memory/buffer_window.py  | 3 +--
 libs/core/langchain_core/legacy/memory/entity.py         | 2 +-
 libs/core/langchain_core/legacy/memory/summary.py        | 7 +++----
 libs/core/langchain_core/legacy/memory/summary_buffer.py | 5 ++---
 libs/core/langchain_core/legacy/memory/token_buffer.py   | 3 +--
 7 files changed, 10 insertions(+), 14 deletions(-)

diff --git a/libs/core/langchain_core/legacy/memory/__init__.py b/libs/core/langchain_core/legacy/memory/__init__.py
index 52589110cf6..0c4aed94b68 100644
--- a/libs/core/langchain_core/legacy/memory/__init__.py
+++ b/libs/core/langchain_core/legacy/memory/__init__.py
@@ -1,10 +1,10 @@
 from langchain_core.legacy.memory.base import BaseMemory
-from langchain_core.legacy.memory.chat_memory import BaseChatMemory
 from langchain_core.legacy.memory.buffer import (
     ConversationBufferMemory,
     ConversationStringBufferMemory,
 )
 from langchain_core.legacy.memory.buffer_window import ConversationBufferWindowMemory
+from langchain_core.legacy.memory.chat_memory import BaseChatMemory
 from langchain_core.legacy.memory.combined import CombinedMemory
 from langchain_core.legacy.memory.entity import (
     ConversationEntityMemory,
diff --git a/libs/core/langchain_core/legacy/memory/base.py b/libs/core/langchain_core/legacy/memory/base.py
index 3139787bb4e..f7960cc3093 100644
--- a/libs/core/langchain_core/legacy/memory/base.py
+++ b/libs/core/langchain_core/legacy/memory/base.py
@@ -69,7 +69,7 @@ class BaseMemory(Serializable, ABC):
         """Save the context of this chain run to memory."""
 
     async def asave_context(
-            self, inputs: Dict[str, Any], outputs: Dict[str, str]
+        self, inputs: Dict[str, Any], outputs: Dict[str, str]
     ) -> None:
         """Save the context of this chain run to memory."""
         await run_in_executor(None, self.save_context, inputs, outputs)
diff --git a/libs/core/langchain_core/legacy/memory/buffer_window.py b/libs/core/langchain_core/legacy/memory/buffer_window.py
index ad21e6899bb..ab02b2f292d 100644
--- a/libs/core/langchain_core/legacy/memory/buffer_window.py
+++ b/libs/core/langchain_core/legacy/memory/buffer_window.py
@@ -1,8 +1,7 @@
 from typing import Any, Dict, List, Union
 
-from langchain_core.messages import BaseMessage, get_buffer_string
-
 from langchain_core.legacy.memory.chat_memory import BaseChatMemory
+from langchain_core.messages import BaseMessage, get_buffer_string
 
 
 class ConversationBufferWindowMemory(BaseChatMemory):
diff --git a/libs/core/langchain_core/legacy/memory/entity.py b/libs/core/langchain_core/legacy/memory/entity.py
index 208b365d518..25ca289b899 100644
--- a/libs/core/langchain_core/legacy/memory/entity.py
+++ b/libs/core/langchain_core/legacy/memory/entity.py
@@ -1,6 +1,6 @@
 import logging
 from abc import ABC, abstractmethod
-from typing import Dict, Optional, List, Any
+from typing import Any, Dict, List, Optional
 
 from pydantic import Field
 
diff --git a/libs/core/langchain_core/legacy/memory/summary.py b/libs/core/langchain_core/legacy/memory/summary.py
index c3f95c29be7..db335646859 100644
--- a/libs/core/langchain_core/legacy/memory/summary.py
+++ b/libs/core/langchain_core/legacy/memory/summary.py
@@ -4,13 +4,12 @@ from typing import Any, Dict, List, Type
 
 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, root_validator
-
 from langchain_core.legacy.chains.llm import LLMChain
 from langchain_core.legacy.memory.chat_memory import BaseChatMemory
 from langchain_core.legacy.memory.prompt import SUMMARY_PROMPT
+from langchain_core.messages import BaseMessage, SystemMessage, get_buffer_string
+from langchain_core.prompts import BasePromptTemplate
+from langchain_core.pydantic_v1 import BaseModel, root_validator
 
 
 class SummarizerMixin(BaseModel):
diff --git a/libs/core/langchain_core/legacy/memory/summary_buffer.py b/libs/core/langchain_core/legacy/memory/summary_buffer.py
index c32638e5ff8..f92261c8ae3 100644
--- a/libs/core/langchain_core/legacy/memory/summary_buffer.py
+++ b/libs/core/langchain_core/legacy/memory/summary_buffer.py
@@ -1,10 +1,9 @@
 from typing import Any, Dict, List
 
-from langchain_core.messages import BaseMessage, get_buffer_string
-from langchain_core.pydantic_v1 import root_validator
-
 from langchain_core.legacy.memory.chat_memory import BaseChatMemory
 from langchain_core.legacy.memory.summary import SummarizerMixin
+from langchain_core.messages import BaseMessage, get_buffer_string
+from langchain_core.pydantic_v1 import root_validator
 
 
 class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):
diff --git a/libs/core/langchain_core/legacy/memory/token_buffer.py b/libs/core/langchain_core/legacy/memory/token_buffer.py
index 0f3ac27b2bd..fe2dc6c8cd1 100644
--- a/libs/core/langchain_core/legacy/memory/token_buffer.py
+++ b/libs/core/langchain_core/legacy/memory/token_buffer.py
@@ -1,9 +1,8 @@
 from typing import Any, Dict, List
 
 from langchain_core.language_models import BaseLanguageModel
-from langchain_core.messages import BaseMessage, get_buffer_string
-
 from langchain_core.legacy.memory.chat_memory import BaseChatMemory
+from langchain_core.messages import BaseMessage, get_buffer_string
 
 
 class ConversationTokenBufferMemory(BaseChatMemory):