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

src/gui/pages/tools.py
104 строки4 KB
0.5.1
13 май 2025, 22:07
13 май 2025, 22:07577d300

from src.gui.config import ConfigFields, ConfigDBTree, ConfigTabs
from src.gui.widgets import find_main_widget
from src.members.workflow import WorkflowSettings


class Page_Tool_Settings(ConfigDBTree):
    def __init__(self, parent):
        super().__init__(
            parent=parent,
            table_name='tools',
            query="""
                SELECT
                    name,
                    id,
                    uuid,
                    COALESCE(json_extract(config, '$.type'), ''),
                    -- COALESCE(json_extract(config, '$.environment'), 'Local'),
                    folder_id
                FROM tools
                ORDER BY pinned DESC, ordr, name""",
            schema=[
                {
                    'text': 'Name',
                    'key': 'name',
                    'type': str,
                    'stretch': True,
                },
                {
                    'text': 'id',
                    'key': 'id',
                    'type': int,
                    'visible': False,
                },
                {
                    'text': 'uuid',
                    'key': 'uuid',
                    'type': str,
                    'visible': False,
                },
                {
                    'text': 'Type',
                    'key': 'type',
                    'type': str,
                    'is_config_field': True,
                    'width': 150,
                },
                # {
                #     'text': 'Type',
                #     'key': 'type',
                #     'type': ('Function', 'computer_20241022',),  # , 'Prompt based',),
                #     'is_config_field': True,
                #     'width': 175,
                # },
            ],
            add_item_options={'title': 'Add Tool', 'prompt': 'Enter a name for the tool:'},
            del_item_options={'title': 'Delete Tool', 'prompt': 'Are you sure you want to delete this tool?'},
            readonly=False,
            layout_type='vertical',
            folder_key='tools',
            config_widget=self.ToolWorkflowSettings(parent=self),
            tree_header_resizable=True,
            default_item_icon=':/resources/icon-tool-small.png',
        )
        self.icon_path = ":/resources/icon-tool.png"
        self.main = find_main_widget(self)
        self.try_add_breadcrumb_widget(root_title='Tools')
        self.splitter.setSizes([500, 500])

        self.extra_config = self.ToolWorkflowExtraConfig(parent=self.config_widget)
        self.config_widget.workflow_extras = self.extra_config
        self.config_widget.workflow_extras.build_schema()
        self.config_widget.workflow_panel_layout.insertWidget(1, self.extra_config)

    def on_edited(self):
        self.main.system.tools.load()

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

    class ToolWorkflowExtraConfig(ConfigTabs):
        def __init__(self, parent):
            super().__init__(parent=parent)
            self.provider = None
            self.pages = {
                'Description': self.Tab_Description(parent=self),
                # 'Prompt': self.Tab_System_Prompt(parent=self),
            }

        class Tab_Description(ConfigFields):
            def __init__(self, parent):
                super().__init__(parent=parent)
                self.setFixedHeight(90)
                self.schema = [
                    {
                        'text': 'Description',
                        'type': str,
                        'num_lines': 4,
                        'label_position': None,
                        'stretch_x': True,
                        'tooltip': 'A description of the tool, this is required and used by the LLM',
                        'default': '',
                    },
                ]