ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-05-09 20:32:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added some view

Browse Source
This commit is contained in:
saloui 2023-06-27 08:57:32 +02:00
parent 143759e395
commit 1165bd2bc5
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/__init__.py
View File

@ -714,6 +714,13 @@ class LoLLMsAPPI():
1 : a notification message 1 : a notification message
2 : A hidden message 2 : A hidden message
""" """
if message_type == MSG_TYPE.MSG_TYPE_STEP_START:
ASCIIColors.info("--> Step started:"+chunk)
if message_type == MSG_TYPE.MSG_TYPE_STEP_END:
ASCIIColors.success("--> Step ended:"+chunk)
if message_type == MSG_TYPE.MSG_TYPE_EXCEPTION:
ASCIIColors.error("--> Exception from personality:"+chunk)
if message_type == MSG_TYPE.MSG_TYPE_CHUNK: if message_type == MSG_TYPE.MSG_TYPE_CHUNK:
self.current_generated_text += chunk self.current_generated_text += chunk
detected_anti_prompt = False detected_anti_prompt = False
@ -760,10 +767,10 @@ class LoLLMsAPPI():
) )
self.socketio.sleep(0) self.socketio.sleep(0)
return True return True
# Stream the generated text to the main process # Stream the generated text to the frontend
else: else:
self.socketio.emit('message', { self.socketio.emit('message', {
'data': self.current_generated_text, 'data': chunk,
'user_message_id':self.current_user_message_id, 'user_message_id':self.current_user_message_id,
'ai_message_id':self.current_ai_message_id, 'ai_message_id':self.current_ai_message_id,
'discussion_id':self.current_discussion.discussion_id, 'discussion_id':self.current_discussion.discussion_id,