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

refactor: brain memory

Browse source
This commit is contained in:
莘权 马 2023-09-07 21:30:15 +08:00
parent 8e30dfd84a
commit 12b2fcd4be
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
metagpt/memory/brain_memory.py
View file

@ -171,16 +171,17 @@ class BrainMemory(pydantic.BaseModel):
total_length = 0
msgs = []
for m in reversed(self.history):
for i in reversed(self.history):
m = Message(**i)
delta = len(m.content)
if total_length + delta > max_words:
left = max_words - total_length
if left == 0:
break
m.content = m.content[0:left]
msgs.append(m)
msgs.append(m.dict())
break
msgs.append(m)
msgs.append(m.dict())
total_length += delta
self.history = msgs
self.is_dirty = True
@ -198,7 +199,8 @@ class BrainMemory(pydantic.BaseModel):
mmsg.append(r)
return json.dumps(mmsg)
async def _get_summary(self, text: str, llm, max_words=20, keep_language: bool = False):
@staticmethod
async def _get_summary(text: str, llm, max_words=20, keep_language: bool = False):
"""Generate text summary"""
if len(text) < max_words:
return text
@ -214,7 +216,7 @@ class BrainMemory(pydantic.BaseModel):
async def get_title(self, text: str, llm, max_words=5, **kwargs) -> str:
"""Generate text title"""
summary = await self.get_summary(text, max_words=500)
summary = await self.summarize(text, max_words=500)
language = CONFIG.language or DEFAULT_LANGUAGE
command = f"Translate the above summary into a {language} title of less than {max_words} words."