fix: 补全上次提交未完全的內容

This commit is contained in:
bmy
2024-07-02 10:53:54 +08:00
parent 605d218b15
commit c968e28b1f

View File

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