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):