From 04158811646673facb3b29dfb99c5c731c0c3ccc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Tue, 5 Mar 2024 14:18:45 +0800 Subject: [PATCH] fixbug: llm.aask unit test failed --- 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 ca248d861..c58148ead 100644 --- a/metagpt/memory/brain_memory.py +++ b/metagpt/memory/brain_memory.py @@ -186,7 +186,7 @@ class BrainMemory(BaseModel): summaries = [summary, command] msg = "\n".join(summaries) logger.debug(f"title ask:{msg}") - response = await llm.aask(msg=msg, system_msgs=[]) + response = await llm.aask(msg=msg, system_msgs=[], stream=False) logger.debug(f"title rsp: {response}") return response @@ -308,7 +308,7 @@ class BrainMemory(BaseModel): ] if keep_language: system_msgs.append("The generated summary should be in the same language as the original text.") - response = await self.llm.aask(msg=text, system_msgs=system_msgs) + response = await self.llm.aask(msg=text, system_msgs=system_msgs, stream=False) logger.debug(f"{text}\nsummary rsp: {response}") return response