fix bug in test

This commit is contained in:
geekan 2023-12-26 19:19:32 +08:00
parent 0435b1321f
commit 9531dbf3ff
3 changed files with 4 additions and 12 deletions

View file

@ -33,6 +33,9 @@ class BrainMemory(BaseModel):
cacheable: bool = True
llm: Optional[BaseLLM] = None
class Config:
arbitrary_types_allowed = True
def add_talk(self, msg: Message):
"""
Add message from user.