feat: +brain memory

This commit is contained in:
莘权 马 2023-08-26 19:54:49 +08:00
parent 5dc352bf2f
commit 2c83921aee

View file

@ -28,6 +28,7 @@ from metagpt.schema import Message
DEFAULT_MAX_TOKENS = 1500
COMMAND_TOKENS = 500
BRAIN_MEMORY = "BRAIN_MEMORY"
class Assistant(Role):
@ -37,7 +38,8 @@ class Assistant(Role):
constraints="Talk in {language}", desc="", *args, **kwargs):
super(Assistant, self).__init__(options=options, cost_manager=cost_manager, name=name, profile=profile,
goal=goal, constraints=constraints, desc=desc, *args, **kwargs)
self.memory = BrainMemory()
brain_memory = options.get(BRAIN_MEMORY)
self.memory = BrainMemory(**brain_memory) if brain_memory else BrainMemory()
self.skills = SkillLoader()
async def think(self) -> bool: