/
githubmirror
/
AgentPilot
Обзор
Документация
Войти
/
githubmirror
/
AgentPilot
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
v0.4.0
src/members/agent.py
353 строки
13 KB
jb
0.4.0
16 янв 2025, 05:27
16 янв 2025, 05:27
d57d8e9
Код
Авторство
О чём код?
from abc import abstractmethod from PySide6.QtGui import Qt from src.gui.config import ConfigPages, ConfigFields, ConfigTabs, ConfigJsonTree, \ ConfigJoined, ConfigJsonFileTree, ConfigJsonDBTree from src.gui.widgets import find_main_widget from src.members.base import LlmMember class Agent(LlmMember): def __init__(self, **kwargs): super().__init__(**kwargs, model_config_key='chat.model') self.name = self.config.get('info.name', 'Assistant') # self.parameters = { todo # 'System message': 'chat.sys_msg', # 'Max messages': 'chat.max_messages', # 'Max turns': 'chat.max_turns', # } def system_message(self, msgs_in_system=None, response_instruction='', msgs_in_system_len=0): raw_sys_msg = self.config.get('chat.sys_msg', '') builtin_blocks = { 'char_name': self.name, 'full_name': self.name, 'response_type': 'response', 'verb': '', } if self.member_id == '4': pass formatted_sys_msg = self.workflow.system.blocks.format_string( raw_sys_msg, ref_workflow=self.workflow, additional_blocks=builtin_blocks, ) message_str = '' if msgs_in_system: if msgs_in_system_len > 0: msgs_in_system = msgs_in_system[-msgs_in_system_len:] message_str = "\n".join( f"""{msg['role']}: \"{msg['content'].strip().strip('"')}\"""" for msg in msgs_in_system) message_str = f"\n\nCONVERSATION:\n\n{message_str}\nassistant: " if response_instruction != '': response_instruction = f"\n\n{response_instruction}\n\n" return formatted_sys_msg + response_instruction + message_str class StreamSpeaker: def __init__(self, member): self.member = member self.previous_blocks = [] # list of tuple(block_text, audio_file_id) self.chunk_chars = ['.', '?', '!', '\n', ': ', ';'] # , ','] self.current_block = '' def stream_chunk(self, chunk): if chunk is None or chunk == '': return self.current_block += chunk if any(c in chunk for c in self.chunk_chars): self.push_block() def finish_stream(self): self.push_block() def push_block(self): if self.current_block == '': return self.generate_voices(self.msg_uuid, self.current_block, '') self.current_block = '' class AgentSettings(ConfigPages): def __init__(self, parent): super().__init__(parent=parent) self.main = find_main_widget(parent) self.layout.addSpacing(10) self.pages = { 'Info': self.Info_Settings(self), 'Chat': self.Chat_Settings(self), # 'Files': self.File_Settings(self), 'Tools': self.Tool_Settings(self), } @abstractmethod def save_config(self): """Saves the config to database when modified""" pass class Info_Settings(ConfigJoined): def __init__(self, parent): super().__init__(parent=parent, layout_type='vertical') self.widgets = [ self.Info_Fields(parent=self), ] class Info_Fields(ConfigFields): def __init__(self, parent): super().__init__(parent=parent) self.conf_namespace = 'info' self.alignment = Qt.AlignHCenter self.schema = [ { 'text': 'Avatar', 'key': 'avatar_path', 'type': 'CircularImageLabel', 'default': '', 'label_position': None, }, { 'text': 'Name', 'type': str, 'default': 'Assistant', 'stretch_x': True, 'text_size': 15, 'text_alignment': Qt.AlignCenter, 'label_position': None, 'transparent': True, }, { 'text': 'Plugin', 'key': 'use_plugin', 'type': 'PluginComboBox', 'label_position': None, 'plugin_type': 'Agent', 'centered': True, 'default': '', } ] class Chat_Settings(ConfigTabs): def __init__(self, parent): super().__init__(parent=parent) self.pages = { 'Messages': self.Page_Chat_Messages(parent=self), 'Preload': self.Page_Chat_Preload(parent=self), 'Group': self.Page_Chat_Group(parent=self), # 'Voice': self.Page_Chat_Voice(parent=self), } class Page_Chat_Messages(ConfigFields): def __init__(self, parent): super().__init__(parent=parent) self.conf_namespace = 'chat' self.schema = [ { 'text': 'Model', 'type': 'ModelComboBox', 'default': '', 'row_key': 0, }, { 'text': 'Display markdown', 'type': bool, 'default': True, 'row_key': 0, }, { 'text': 'System message', 'key': 'sys_msg', 'type': str, 'num_lines': 12, 'default': '', 'stretch_x': True, 'gen_block_folder_name': 'Enhance system msg', 'stretch_y': True, 'label_position': 'top', }, { 'text': 'Max messages', 'type': int, 'minimum': 1, 'maximum': 99, 'default': 10, 'width': 60, 'has_toggle': True, 'row_key': 1, }, { 'text': 'Max turns', 'type': int, 'minimum': 1, 'maximum': 99, 'default': 7, 'width': 60, 'has_toggle': True, 'row_key': 1, }, ] class Page_Chat_Preload(ConfigJsonTree): def __init__(self, parent): super().__init__(parent=parent, add_item_prompt=('NA', 'NA'), del_item_prompt=('NA', 'NA')) self.conf_namespace = 'chat.preload' self.schema = [ { 'text': 'Role', 'type': 'RoleComboBox', 'width': 120, 'default': 'assistant', }, { 'text': 'Content', 'type': str, 'stretch': True, 'wrap_text': True, 'default': '', }, { 'text': 'Type', 'type': ('Normal', 'Context', 'Welcome'), 'width': 90, 'default': 'Normal', }, ] class Page_Chat_Variables(ConfigJsonTree): def __init__(self, parent): super().__init__(parent=parent, add_item_prompt=('NA', 'NA'), del_item_prompt=('NA', 'NA')) self.conf_namespace = 'blocks' self.schema = [ { 'text': 'Placeholder', 'type': str, 'width': 120, 'default': '< Placeholder >', }, { 'text': 'Value', 'type': str, 'stretch': True, 'wrap_text': True, 'default': '', }, ] class Page_Chat_Group(ConfigFields): def __init__(self, parent): super().__init__(parent=parent) self.conf_namespace = 'group' self.label_width = 220 self.schema = [ { 'text': 'Output role', 'type': 'RoleComboBox', 'width': 90, 'tooltip': 'Set the primary output role for this member', 'default': 'assistant', }, { 'text': 'Output placeholder', 'type': str, # 'stretch_x': True, 'tooltip': 'A tag to use this member\'s output from other members system messages', 'default': '', }, { 'text': 'Hide bubbles', 'type': bool, 'tooltip': 'When checked, the responses from this member will not be shown in the chat', 'default': False, }, # { # 'text': 'On multiple message inputs', # 'type': ('Use all', 'Use only sender'), # Append to system msg', 'Merged user message'), # todo this needs implementing into workflow # 'tooltip': 'How to handle multiple inputs from the user (Not implemented yet)', # # 'width': 175, # 'default': 'Merged user message', # }, { 'text': 'Member description', 'type': str, 'num_lines': 4, 'label_position': 'top', 'stretch_x': True, 'tooltip': 'A description of the member that can be used by other members (Not implemented yet)', 'default': '', } ] # class Page_Chat_Voice(ConfigVoiceTree): # def __init__(self, parent): # super().__init__(parent=parent) class File_Settings(ConfigJsonFileTree): def __init__(self, parent): self.IS_DEV_MODE = True super().__init__(parent=parent, add_item_prompt=('NA', 'NA'), del_item_prompt=('NA', 'NA'), tree_header_hidden=True, readonly=True) self.parent = parent self.conf_namespace = 'files' self.schema = [ { 'text': 'Filename', 'type': str, 'width': 175, 'default': '', }, { 'text': 'Location', 'type': str, # 'visible': False, 'stretch': True, 'default': '', }, { 'text': 'is_dir', 'type': bool, 'visible': False, 'default': False, }, ] class Tool_Settings(ConfigJsonDBTree): def __init__(self, parent): super().__init__(parent=parent, add_item_prompt=('NA', 'NA'), del_item_prompt=('NA', 'NA'), tree_header_hidden=True, table_name='tools', key_field='uuid', item_icon_path=':/resources/icon-tool-small.png', show_fields=[ 'name', 'uuid', # ID ALWAYS LAST ], readonly=True) self.parent = parent self.conf_namespace = 'tools' self.schema = [ { 'text': 'Tool', 'type': str, 'width': 175, 'default': '', }, { 'text': 'id', 'visible': False, 'default': '', }, ]