ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-13 22:02:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix after stuff

Browse Source
This commit is contained in:
Saifeddine ALOUI 2025-04-09 13:05:38 +02:00
parent 7bbc43f8c3
commit 2127a613d1
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lollms/app.py
View File

@ -1090,6 +1090,7 @@ class LollmsApplication(LoLLMsCom):
client.discussion.current_message.nb_tokens = self.model.count_tokens(client.generated_text)
except:
client.discussion.current_message.nb_tokens = None
client.discussion.current_message.update_db()
await self.sio.emit(
"close_message",
{
@ -1108,6 +1109,7 @@ class LollmsApplication(LoLLMsCom):
to=client_id,
)
else:
client.discussion.current_message.update_db()
await self.sio.emit(
"close_message",
{
@ -1738,6 +1740,8 @@ Don't forget encapsulate the code inside a markdown code tag. This is mandatory.
await self.close_message(client_id, True)
client.processing = False
try:
final_ui_update =""
final_text_update = ""
if len(context_details.function_calls)>0:
codes = self.personality.extract_code_blocks(client.generated_text)
for function_call in context_details.function_calls:
@ -1750,9 +1754,13 @@ Don't forget encapsulate the code inside a markdown code tag. This is mandatory.
context_details.ai_output = client.generated_text
output = fc.execute(context_details,**infos["function_parameters"])
if output[0]=="<":
await self.new_message(client_id,self.personality.name,output,message_type=MSG_OPERATION_TYPE.MSG_OPERATION_TYPE_UI, sender_type=SENDER_TYPES.SENDER_TYPES_AI)
final_ui_update+=output+"\n"
else:
await self.new_message(client_id,self.personality.name,output,message_type=MSG_OPERATION_TYPE.MSG_OPERATION_TYPE_SET_CONTENT, sender_type=SENDER_TYPES.SENDER_TYPES_AI)
final_text_update+=output+"\n"
if final_ui_update or final_text_update:
await self.new_message(client_id,"System",final_text_update,message_type=MSG_OPERATION_TYPE.MSG_OPERATION_TYPE_SET_CONTENT, sender_type=SENDER_TYPES.SENDER_TYPES_SYSTEM)
if final_ui_update:
await self.update_message_ui(client_id, final_ui_update)
except Exception as ex:
trace_exception(ex)

2
lollms/databases/discussions_database.py
View File

@ -606,7 +606,7 @@ class Message:
def update_db(self):
self.message_id = self.discussions_db.insert(
"INSERT INTO message (sender, content, metadata, ui, message_type, rank, parent_message_id, binding, model, personality, created_at, started_generating_at, finished_generating_at, nb_tokens, discussion_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
(self.sender, self.content, self.metadata, self.ui, self.message_type, self.rank, self.parent_message_id, self.binding, self.model, self.personality, self.created_at, self.started_generating_at, self.finished_generating_at, nb_tokens, self.discussion_id)
(self.sender, self.content, self.metadata, self.ui, self.message_type, self.rank, self.parent_message_id, self.binding, self.model, self.personality, self.created_at, self.started_generating_at, self.finished_generating_at, self.nb_tokens, self.discussion_id)
)
def update(self, new_content, new_metadata=None, new_ui=None, started_generating_at=None, nb_tokens=None, commit=True):