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

Merge branch 'main' of https://github.com/ParisNeo/lollms

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-07-07 09:59:56 +02:00
commit a7dc7b8014
2 changed files with 63 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lollms/personality.py
View File

@ -1098,6 +1098,27 @@ class APScript:
if callback:
callback(full_text, MSG_TYPE.MSG_TYPE_FULL)
def full_invisible_to_ai(self, full_text:str, callback=None):
"""This sends full text to front end (INVISIBLE to AI)
Args:
step_text (dict): The step text
callback (callable, optional): A callable with this signature (str, MSG_TYPE) to send the text to. Defaults to None.
"""
if callback:
callback(full_text, MSG_TYPE.MSG_TYPE_FULL_INVISIBLE_TO_AI)
def full_invisible_to_user(self, full_text:str, callback=None):
"""This sends full text to front end (INVISIBLE to user)
Args:
step_text (dict): The step text
callback (callable, optional): A callable with this signature (str, MSG_TYPE) to send the text to. Defaults to None.
"""
if callback:
callback(full_text, MSG_TYPE.MSG_TYPE_FULL_INVISIBLE_TO_USER)
def info(self, info_text:str, callback=None):
"""This sends info text to front end

55
lollms/types.py
View File

@ -2,22 +2,51 @@ from enum import Enum
class MSG_TYPE(Enum):
# Messaging
MSG_TYPE_CHUNK=0 # A chunk of a message (used for classical chat)
MSG_TYPE_FULL=1 # A full message (for some personality the answer is sent in bulk)
MSG_TYPE_CHUNK = 0 # A chunk of a message (used for classical chat)
MSG_TYPE_FULL = 1 # A full message (for some personality the answer is sent in bulk)
MSG_TYPE_FULL_INVISIBLE_TO_AI = 2 # A full message (for some personality the answer is sent in bulk)
MSG_TYPE_FULL_INVISIBLE_TO_USER = 3 # A full message (for some personality the answer is sent in bulk)
# Conditionning
# Informations
MSG_TYPE_EXCEPTION=2 # An exception occured
MSG_TYPE_WARNING=3 # A warning occured
MSG_TYPE_INFO=4 # An information to be shown to user
MSG_TYPE_EXCEPTION = 4 # An exception occured
MSG_TYPE_WARNING = 5 # A warning occured
MSG_TYPE_INFO = 6 # An information to be shown to user
# Steps
MSG_TYPE_STEP=5 # An instant step (a step that doesn't need time to be executed)
MSG_TYPE_STEP_START=6 # A step has started (the text contains an explanation of the step done by he personality)
MSG_TYPE_STEP_PROGRESS=7 # The progress value (the text contains a percentage and can be parsed by the reception)
MSG_TYPE_STEP_END=8 # A step has been done (the text contains an explanation of the step done by he personality)
MSG_TYPE_STEP = 7 # An instant step (a step that doesn't need time to be executed)
MSG_TYPE_STEP_START = 8 # A step has started (the text contains an explanation of the step done by he personality)
MSG_TYPE_STEP_PROGRESS = 9 # The progress value (the text contains a percentage and can be parsed by the reception)
MSG_TYPE_STEP_END = 10# A step has been done (the text contains an explanation of the step done by he personality)
#Extra
MSG_TYPE_JSON_INFOS=9 # A JSON output that is useful for summarizing the process of generation used by personalities like chain of thoughts and tree of thooughts
MSG_TYPE_REF=10 # References (in form of [text](path))
MSG_TYPE_CODE=11 # A javascript code to execute
MSG_TYPE_UI=12 # A vue.js component to show (we need to build some and parse the text to show it)
MSG_TYPE_JSON_INFOS = 11# A JSON output that is useful for summarizing the process of generation used by personalities like chain of thoughts and tree of thooughts
MSG_TYPE_REF = 12# References (in form of [text](path))
MSG_TYPE_CODE = 13# A javascript code to execute
MSG_TYPE_UI = 14# A vue.js component to show (we need to build some and parse the text to show it)
class GenerationPresets:
"""
Class containing various generation presets.
"""
@staticmethod
def deterministic_preset():
"""
Preset for deterministic output with low temperature and top_k, and high top_p.
"""
return {'temperature': 0.2, 'top_k': 10, 'top_p': 0.8}
@staticmethod
def creative_preset():
"""
Preset for creative output with high temperature, top_k, and top_p.
"""
return {'temperature': 0.8, 'top_k': 50, 'top_p': 0.9}
@staticmethod
def default_preset():
"""
Default preset with moderate temperature, top_k, and top_p.
"""
return {'temperature': 0.5, 'top_k': 20, 'top_p': 0.85}