From 0c21aa810f64743ac3a484d53c005bf654cbf3bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Thu, 7 Sep 2023 19:49:51 +0800
Subject: [PATCH] refactor: brain memory

---
 metagpt/memory/brain_memory.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/metagpt/memory/brain_memory.py b/metagpt/memory/brain_memory.py
index adb1f0114..596928a4c 100644
--- a/metagpt/memory/brain_memory.py
+++ b/metagpt/memory/brain_memory.py
@@ -10,7 +10,7 @@
 import json
 import re
 from enum import Enum
-from typing import Dict, List
+from typing import Dict, List, Optional
 
 import openai
 import pydantic
@@ -40,7 +40,7 @@ class BrainMemory(pydantic.BaseModel):
     last_history_id: str = ""
     is_dirty: bool = False
     last_talk: str = None
-    llm_type: str
+    llm_type: Optional[str] = None
 
     def add_talk(self, msg: Message):
         msg.add_tag(MessageType.Talk.value)