src/gui/pages/agents.py - master - githubmirror/AgentPilot

src/gui/pages/agents.py
75 строк3 KB
0.4.3
30 янв 2025, 08:55
30 янв 2025, 08:554e272e2

from src.members.workflow import WorkflowSettings
from src.gui.config import ConfigDBTree

from PySide6.QtWidgets import QPushButton

class Page_Entities(ConfigDBTree):
    def __init__(self, parent):
        super().__init__(
            parent=parent,
            table_name='entities',
            query="""
                SELECT
                    COALESCE(json_extract(config, '$."info.name"'), name) AS name,
                    id,
                    config,
                    '' AS chat_button,
                    folder_id
                FROM entities
                WHERE kind = "{{kind}}"
                ORDER BY pinned DESC, ordr, name""",
            schema=[
                {
                    'text': 'Name',
                    'key': 'name',
                    'type': str,
                    'stretch': True,
                    'image_key': 'config',
                },
                {
                    'text': 'id',
                    'key': 'id',
                    'type': int,
                    'visible': False,
                },
                {
                    'text': 'config',
                    'type': str,
                    'visible': False,
                },
                {
                    'text': '',
                    'type': QPushButton,
                    'icon': ':/resources/icon-chat.png',
                    'func': self.on_chat_btn_clicked,
                    'width': 45,
                },
            ],
            kind='AGENT',
            # kind_list=['AGENT', 'CONTACT'],
            add_item_options={'title': 'Add Agent', 'prompt': 'Enter a name for the agent:'},
            del_item_options={'title': 'Delete Agent', 'prompt': 'Are you sure you want to delete this agent?'},
            layout_type='vertical',
            config_widget=self.Entity_Config_Widget(parent=self),
            tree_header_hidden=True,
            folder_key='agents',
            filterable=True,
            searchable=True,
        )
        self.icon_path = ":/resources/icon-agent.png"
        self.tree.itemDoubleClicked.connect(self.on_chat_btn_clicked)
        self.try_add_breadcrumb_widget(root_title='Agents')
        self.splitter.setSizes([500, 500])

    def load(self, select_id=None, silent_select_id=None, append=False):
        self.config_widget.set_edit_mode(False)
        super().load(select_id, silent_select_id, append)

    def on_chat_btn_clicked(self, _):
        run_btn = getattr(self.tree_buttons, 'btn_run', None)
        if run_btn:
            run_btn.click()

    class Entity_Config_Widget(WorkflowSettings):
        def __init__(self, parent):
            super().__init__(parent=parent, compact_mode=True)