apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix by pre-commit hooks

Browse source
This commit is contained in:
seehi 2024-02-20 17:08:50 +08:00
parent aca3d1a0cb
commit fae24fd381
2 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
metagpt/memory/memory2.py
View file

@ -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)

8
metagpt/memory/memory_network.py
View file

@ -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)