From 5bad744f444a1805fd2f5dd6b6fd2f05e4ca2a1f Mon Sep 17 00:00:00 2001
From: Azure Wang <azurewtl@hotmail.com>
Date: Wed, 27 Mar 2024 22:13:52 +0800
Subject: [PATCH] - keep log_llm_stream from effect

---
 metagpt/logs.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/metagpt/logs.py b/metagpt/logs.py
index ebb741c28..2fb80ba0b 100644
--- a/metagpt/logs.py
+++ b/metagpt/logs.py
@@ -35,8 +35,7 @@ logger = define_log_level()
 
 
 def log_llm_stream(msg):
-  if _print_level in ["DEBUG"]:
-        _llm_stream_log(msg)
+    _llm_stream_log(msg)
 
 
 def set_llm_stream_logfunc(func):
@@ -44,4 +43,6 @@ def set_llm_stream_logfunc(func):
     _llm_stream_log = func
 
 
-_llm_stream_log = partial(print, end="")
+def _llm_stream_log(msg):
+    if _print_level in ["DEBUG"]:
+        print(msg, end="")
\ No newline at end of file