ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-19 04:37:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #24 from tobitege/main

Browse Source 
Personality: allow lua, cs and pas files
This commit is contained in:
Saifeddine ALOUI 2024-04-12 22:15:05 +02:00 committed by GitHub
commit ec9736cd6b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 202 additions and 195 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

377
lollms/personality.py
View File

File diff suppressed because it is too large Load Diff

20
lollms/server/events/lollms_personality_events.py
View File

@ -1,8 +1,8 @@
"""
project: lollms
file: lollms_files_events.py
file: lollms_files_events.py
author: ParisNeo
description:
description:
Events related to socket io personality events
"""
@ -31,7 +31,7 @@ lollmsElfServer = LOLLMSElfServer.get_instance()
# ----------------------------------- events -----------------------------------------
def add_events(sio:socketio):
@sio.on('get_personality_files')
def get_personality_files(sid, data):
client_id = sid
@ -39,19 +39,23 @@ def add_events(sio:socketio):
client.generated_text = ""
client.cancel_generation = False
try:
lollmsElfServer.personality.setCallback(partial(lollmsElfServer.process_chunk,client_id = client_id))
except Exception as ex:
trace_exception(ex)
trace_exception(ex)
import os
import imghdr
import mimetypes
ALLOWED_EXTENSIONS = {
'txt', 'csv', 'py', 'html', 'js', 'jpg', 'jpeg', 'png', 'gif', 'bmp', 'tiff', 'ico', 'svg', 'mp4', 'mp3', 'avi', 'mov',
'doc', 'docx', 'ppt', 'pptx', 'xls', 'xlsx', 'pdf', 'js', "md", "json"
'asm', 'avi', 'bat', 'bmp', 'c', 'cpp', 'cs', 'csproj', 'css', 'csv', 'doc', 'docx',
'gif', 'h', 'hh', 'hpp', 'html', 'ico', 'inc', 'ini', 'java', 'jpeg', 'jpg',
'js', 'json', 'log', 'lua', 'map', 'md', 'mov', 'mp3', 'mp4', 'pas', 'pdf',
'php', 'png', 'ppt', 'pptx', 'ps1', 'py', 'rb', 'rtf', 's', 'se', 'sh', 'sln', 'snippet',
'snippets', 'sql', 'svg', 'sym', 'tif', 'tiff', 'ts', 'txt', 'wav', 'webp',
'xlsx', 'xls', 'xml', 'yaml', 'yml'
}
def allowed_file(filename):
@ -101,7 +105,7 @@ def add_events(sio:socketio):
if is_last_chunk:
lollmsElfServer.success('File received and saved successfully')
lollmsElfServer.ShowBlockingMessage(f"File received {file_path.name}.\nProcessing ...")
if lollmsElfServer.personality.processor:
result = lollmsElfServer.personality.processor.add_file(file_path, client, partial(lollmsElfServer.process_chunk, client_id=client_id))
else: