fix: 补全上次提交未完全的內容
This commit is contained in:
5
utils.py
5
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):
|
||||
|
||||
Reference in New Issue
Block a user