From fae24fd38144cc6b2c709dedd703ca11921092aa Mon Sep 17 00:00:00 2001
From: seehi <6580@pm.me>
Date: Tue, 20 Feb 2024 17:08:50 +0800
Subject: [PATCH] fix by pre-commit hooks

---
 metagpt/memory/memory2.py        | 9 ++++++---
 metagpt/memory/memory_network.py | 8 +++++---
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/metagpt/memory/memory2.py b/metagpt/memory/memory2.py
index f33b740de..74f848278 100644
--- a/metagpt/memory/memory2.py
+++ b/metagpt/memory/memory2.py
@@ -2,8 +2,9 @@
 # -*- coding: utf-8 -*-
 # @Desc   : memory mechanism including store/retrieval/rank
 
-from typing import Union, Optional
-from pydantic import Field, BaseModel
+from typing import Optional
+
+from pydantic import BaseModel, Field
 
 from metagpt.memory.memory_network import MemoryNetwork
 from metagpt.memory.schema import MemoryNode
@@ -11,7 +12,9 @@ from metagpt.schema import Message
 
 
 class Memory(BaseModel):
-    mem_network: Optional[MemoryNetwork] = Field(default_factory=MemoryNetwork, description="the network to store memory")
+    mem_network: Optional[MemoryNetwork] = Field(
+        default_factory=MemoryNetwork, description="the network to store memory"
+    )
 
     def add_msg(self, message: Message):
         mem_node = MemoryNode.create_mem_node_from_message(message)
diff --git a/metagpt/memory/memory_network.py b/metagpt/memory/memory_network.py
index 00bc2ba78..f8f2244ed 100644
--- a/metagpt/memory/memory_network.py
+++ b/metagpt/memory/memory_network.py
@@ -2,13 +2,15 @@
 # -*- coding: utf-8 -*-
 # @Desc   : the memory network to store memory segment
 
-from pydantic import Field, BaseModel
+from pydantic import BaseModel, Field
 
-from metagpt.memory.schema import MemorySegment, MemoryNode
+from metagpt.memory.schema import MemoryNode, MemorySegment
 
 
 class MemoryNetwork(BaseModel):
-    mem_seg: MemorySegment = Field(default_factory=MemorySegment, description="the memory segment to store memory nodes")
+    mem_seg: MemorySegment = Field(
+        default_factory=MemorySegment, description="the memory segment to store memory nodes"
+    )
 
     def add_mem(self, mem_node: MemoryNode):
         self.mem_seg.add_mem_node(mem_node)