mirror of
https://github.com/ParisNeo/lollms.git
synced 2025-03-03 21:00:39 +00:00
63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
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_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 = 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 = 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 = 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)
|
|
|
|
#Commands
|
|
MSG_TYPE_NEW_MESSAGE = 15# A new message
|
|
MSG_TYPE_FINISHED_MESSAGE = 17# End of current message
|
|
|
|
|
|
|
|
class SENDER_TYPES(Enum):
|
|
SENDER_TYPES_USER = 0 # Sent by user
|
|
SENDER_TYPES_AI = 1 # Sent by ai
|
|
SENDER_TYPES_SYSTEM = 2 # Sent by athe system
|
|
|
|
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} |