From c968e28b1f0365b4cb39450be3baa0be5860a999 Mon Sep 17 00:00:00 2001 From: bmy <2583236812@qq.com> Date: Tue, 2 Jul 2024 10:53:54 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E8=A1=A5=E5=85=A8=E4=B8=8A=E6=AC=A1?= =?UTF-8?q?=E6=8F=90=E4=BA=A4=E6=9C=AA=E5=AE=8C=E5=85=A8=E7=9A=84=E5=85=A7?= =?UTF-8?q?=E5=AE=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/utils.py b/utils.py index c3e84d2..d1c9504 100644 --- a/utils.py +++ b/utils.py @@ -264,7 +264,8 @@ class LLM: self.prompt += '''你无需回复我''' self.messages = [] self.resp = None - self.reset() + worker = threading.Thread(target=self.reset, daemon=True) + worker.start() def reset(self): self.messages = [self.make_message(self.prompt)] self.resp = erniebot.ChatCompletion.create( @@ -272,6 +273,8 @@ class LLM: messages=self.messages, ) self.messages.append(self.resp.to_message()) + self.init_done_flag = True + logger.info("LLM init done") def make_message(self,content): return {'role': 'user', 'content': content} def get_command_json(self,chat):