- keep log_llm_stream from effect

This commit is contained in:
Azure Wang 2024-03-27 22:13:52 +08:00
parent 67130fc2d6
commit 5bad744f44

View file

@ -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="")