btl143/project_main
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
bmy
2024-07-02 10:53:54 +08:00
parent 605d218b15
commit c968e28b1f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
utils.py
View File

@@ -264,7 +264,8 @@ class LLM:
self.prompt += '''你无需回复我''' self.prompt += '''你无需回复我'''
self.messages = [] self.messages = []
self.resp = None self.resp = None
self.reset() worker = threading.Thread(target=self.reset, daemon=True)
worker.start()
def reset(self): def reset(self):
self.messages = [self.make_message(self.prompt)] self.messages = [self.make_message(self.prompt)]
self.resp = erniebot.ChatCompletion.create( self.resp = erniebot.ChatCompletion.create(
@@ -272,6 +273,8 @@ class LLM:
messages=self.messages, messages=self.messages,
) )
self.messages.append(self.resp.to_message()) self.messages.append(self.resp.to_message())
self.init_done_flag = True
logger.info("LLM init done")
def make_message(self,content): def make_message(self,content):
return {'role': 'user', 'content': content} return {'role': 'user', 'content': content}
def get_command_json(self,chat): def get_command_json(self,chat):