pygpt-net 2.6.20__py3-none-any.whl → 2.6.22__py3-none-any.whl

pygpt_net/ui/layout/ctx/ctx_list.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.16 00:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6 import QtCore
@@ -20,7 +20,6 @@ from pygpt_net.ui.widget.element.button import NewCtxButton
 from pygpt_net.ui.widget.element.labels import TitleLabel
 from pygpt_net.ui.widget.lists.context import ContextList, Item, GroupItem, SectionItem
 from pygpt_net.utils import trans
-import pygpt_net.icons_rc
 
 
 class CtxList:
@@ -32,6 +31,8 @@ class CtxList:
         """
         self.window = window
         self.search_input = SearchInput(window)
+        self._group_separators = False
+        self._pinned_separators = False
 
     def setup(self) -> QWidget:
         """
@@ -39,28 +40,40 @@ class CtxList:
 
         :return: QWidget
         """
-        id = 'ctx.list'
-        self.window.ui.nodes['ctx.new'] = NewCtxButton(trans('ctx.new'), self.window)
-        self.window.ui.nodes[id] = ContextList(self.window, id)
-        self.window.ui.nodes[id].selection_locked = self.window.controller.ctx.context_change_locked
-        self.window.ui.nodes['ctx.label'] = TitleLabel(trans("ctx.list.label"))
-        self.window.ui.nodes['ctx.new'].setContentsMargins(0,0,0,0)
+        ctx_id = 'ctx.list'
+        ui = self.window.ui
+        nodes = ui.nodes
+        models = ui.models
+
+        widget = QWidget()
+        widget.setContentsMargins(0, 0, 0, 0)
+
+        new_btn = NewCtxButton(trans('ctx.new'), self.window)
+        new_btn.setContentsMargins(0, 0, 0, 0)
+        nodes['ctx.new'] = new_btn
+
+        ctx_list = ContextList(self.window, ctx_id)
+        ctx_list.selection_locked = self.window.controller.ctx.context_change_locked
+        nodes[ctx_id] = ctx_list
+
+        nodes['ctx.label'] = TitleLabel(trans("ctx.list.label"))
         search_input = self.search_input.setup()
 
-        layout = QVBoxLayout()
-        layout.addWidget(self.window.ui.nodes['ctx.new'])
-        layout.addWidget(search_input)
-        layout.addWidget(self.window.ui.nodes[id])
+        layout = QVBoxLayout(widget)
         layout.setContentsMargins(0, 0, 0, 0)
+        layout.addWidget(new_btn)
+        layout.addWidget(search_input)
+        layout.addWidget(ctx_list)
 
-        self.window.ui.models[id] = self.create_model(self.window)
-        self.window.ui.nodes[id].setModel(self.window.ui.models[id])
-        self.window.ui.nodes[id].selectionModel().selectionChanged.connect(
-            lambda: self.window.controller.ctx.selection_change()
-        )
-        widget = QWidget()
-        widget.setLayout(layout)
-        widget.setContentsMargins(0, 0, 0, 0)
+        model = self.create_model(self.window)
+        models[ctx_id] = model
+        ctx_list.setModel(model)
+
+        ctx = self.window.controller.ctx
+        ctx_list.selectionModel().selectionChanged.connect(lambda *_: ctx.selection_change())
+
+        self._group_separators = self.window.core.config.get("ctx.records.groups.separators")
+        self._pinned_separators = self.window.core.config.get("ctx.records.pinned.separators")
 
         return widget
 
@@ -84,6 +97,9 @@ class CtxList:
         node = self.window.ui.nodes[id]
         node.backup_selection()
 
+        self._group_separators = self.window.core.config.get("ctx.records.groups.separators")
+        self._pinned_separators = self.window.core.config.get("ctx.records.pinned.separators")
+
         model = self.window.ui.models.get(id)
         if model is not None:
             node.setUpdatesEnabled(False)
@@ -109,16 +125,12 @@ class CtxList:
         """
         i = 0
         last_dt_str = None
-        config = self.window.core.config
-        separators = config.get("ctx.records.separators")
-        pinned_separators = config.get("ctx.records.pinned.separators")
         model = self.window.ui.models[id]
-
         for meta_id, meta in data.items():
             gid = meta.group_id
             if (gid is None or gid == 0) and not meta.important:
                 item = self.build_item(meta_id, meta, is_group=False)
-                if separators and (not item.isPinned or pinned_separators):
+                if self._group_separators and (not item.isPinned or self._pinned_separators):
                     if i == 0 or last_dt_str != item.dt:
                         section = self.build_date_section(item.dt, group=False)
                         if section:
@@ -137,15 +149,12 @@ class CtxList:
         i = 0
         last_dt_str = None
         model = self.window.ui.models[id]
-        config = self.window.core.config
-        separators = config.get("ctx.records.separators")
-        pinned_separators = config.get("ctx.records.pinned.separators")
 
         for meta_id, meta in data.items():
             gid = meta.group_id
             if (gid is None or gid == 0) and meta.important:
                 item = self.build_item(meta_id, meta, is_group=False)
-                if separators and pinned_separators:
+                if self._group_separators and self._pinned_separators:
                     if i == 0 or last_dt_str != item.dt:
                         section = self.build_date_section(item.dt, group=False)
                         if section:
@@ -165,10 +174,6 @@ class CtxList:
         model = self.window.ui.models[id]
         groups = self.window.core.ctx.get_groups()
         search_string = self.window.core.ctx.get_search_string()
-        config = self.window.core.config
-        is_group_separators = config.get("ctx.records.groups.separators")
-        is_pinned_separators = config.get("ctx.records.pinned.separators")
-
         grouped = {}
         for meta_id, meta in data.items():
             gid = meta.group_id
@@ -213,7 +218,7 @@ class CtxList:
             i = 0
             for meta_id, meta in items_in_group:
                 item = self.build_item(meta_id, meta, is_group=True)
-                if is_group_separators and (not item.isPinned or is_pinned_separators):
+                if self._group_separators and (not item.isPinned or self._pinned_separators):
                     if i == 0 or last_dt_str != item.dt:
                         section = self.build_date_section(item.dt, group=True)
                         if section:
@@ -259,13 +264,7 @@ class CtxList:
         is_important = data.important
         is_attachment = data.has_additional_ctx()
         in_group = bool(data.group)
-
-        if is_group:
-            if self.window.core.config.get("ctx.records.groups.separators"):
-                append_dt = False
-        else:
-            if self.window.core.config.get("ctx.records.separators"):
-                append_dt = False
+        append_dt = False if (is_group and self._pinned_separators) or ((not is_group) and self._group_separators) else append_dt
 
         dt = self.convert_date(data.updated)
         date_time_str = datetime.fromtimestamp(data.updated).strftime("%Y-%m-%d %H:%M")
@@ -274,15 +273,8 @@ class CtxList:
             title = title[:80] + '...'
         clean_title = title.replace("\n", "")
 
-        if append_dt:
-            name = f"{clean_title} ({dt})"
-        else:
-            name = clean_title
-
-        mode_str = ''
-        if data.last_mode is not None:
-            mode_str = f" ({trans('mode.' + data.last_mode)})"
-
+        name = f"{clean_title} ({dt})" if append_dt else clean_title
+        mode_str = f" ({trans('mode.' + data.last_mode)})" if data.last_mode is not None else ""
         tooltip_text = f"{date_time_str}: {data.name}{mode_str} #{id}"
 
         if is_attachment:
@@ -317,8 +309,7 @@ class CtxList:
         :param group: is group
         :return: SectionItem
         """
-        item = SectionItem(dt, group=group)
-        return item
+        return SectionItem(dt, group=group)
 
     def convert_date(self, timestamp: int) -> str:
         """

pygpt_net/ui/layout/status.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2023.12.25 21:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
@@ -32,20 +32,29 @@ class Status:
 
         :return: QHBoxLayout
         """
-        self.window.ui.nodes['status'] = QLabel(trans('status.started'))
-        self.window.ui.nodes['status.agent'] = HelpLabel("")
-        self.window.ui.nodes['status.agent'].setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
-        self.window.ui.nodes['global.stop'] = QPushButton("STOP")
-        self.window.ui.nodes['global.stop'].setVisible(False)
-        self.window.ui.nodes['global.stop'].clicked.connect(self.window.controller.ui.on_global_stop)
-        self.window.ui.nodes['anim.loading.status'] = Loader()
-        self.window.ui.nodes['anim.loading.status'].hide()
+        nodes = self.window.ui.nodes
+
+        nodes['status'] = QLabel(trans('status.started'))
+        nodes['status'].setParent(self.window)
+
+        nodes['status.agent'] = HelpLabel("")
+        nodes['status.agent'].setParent(self.window)
+        nodes['status.agent'].setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
+
+        nodes['global.stop'] = QPushButton("STOP")
+        nodes['global.stop'].setParent(self.window)
+        nodes['global.stop'].hide()
+        nodes['global.stop'].clicked.connect(self.window.controller.ui.on_global_stop)
+
+        nodes['anim.loading.status'] = Loader()
+        nodes['anim.loading.status'].setParent(self.window)
+        nodes['anim.loading.status'].hide()
 
         layout = QHBoxLayout()
-        layout.addWidget(self.window.ui.nodes['anim.loading.status'])
-        layout.addWidget(self.window.ui.nodes['status.agent'])
-        layout.addWidget(self.window.ui.nodes['status'])
-        layout.addWidget(self.window.ui.nodes['global.stop'])
+        layout.addWidget(nodes['anim.loading.status'])
+        layout.addWidget(nodes['status.agent'])
+        layout.addWidget(nodes['status'])
+        layout.addWidget(nodes['global.stop'])
         layout.setAlignment(Qt.AlignLeft)
 
-        return layout
+        return layout

pygpt_net/ui/layout/toolbox/agent.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.24 22:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget, QCheckBox
@@ -31,49 +31,38 @@ class Agent:
 
         :return: QWidget
         """
-        # iterations
-        option = self.window.controller.agent.legacy.options["agent.iterations"]
-        self.window.ui.nodes['agent.iterations.label'] = QLabel(trans("toolbox.agent.iterations.label"))
-        self.window.ui.nodes['agent.iterations'] = \
-            OptionSlider(
-                self.window,
-                'global',
-                'agent.iterations',
-                option,
-            )
-        self.window.ui.config['global']['agent.iterations'] = self.window.ui.nodes['agent.iterations']
+        window = self.window
+        ui = window.ui
+        nodes = ui.nodes
+        cfg = ui.config['global']
+        common = window.controller.agent.common
 
-        # auto stop
-        self.window.ui.nodes['agent.auto_stop'] = ToggleLabel(trans("toolbox.agent.auto_stop.label"), parent=self.window)
-        self.window.ui.nodes['agent.auto_stop'].box.stateChanged.connect(
-            lambda:
-            self.window.controller.agent.common.toggle_auto_stop(
-                self.window.ui.config['global']['agent.auto_stop'].box.isChecked())
-        )
-        self.window.ui.config['global']['agent.auto_stop'] = self.window.ui.nodes['agent.auto_stop']
+        container = QWidget(window)
 
-        # continue more
-        self.window.ui.nodes['agent.continue'] = ToggleLabel(trans("toolbox.agent.continue.label"), parent=self.window)
-        self.window.ui.nodes['agent.continue'].box.stateChanged.connect(
-            lambda:
-            self.window.controller.agent.common.toggle_continue(
-                self.window.ui.config['global']['agent.continue'].box.isChecked())
-        )
-        self.window.ui.config['global']['agent.continue'] = self.window.ui.nodes['agent.continue']
+        option = window.controller.agent.legacy.options["agent.iterations"]
+        nodes['agent.iterations.label'] = QLabel(trans("toolbox.agent.iterations.label"), parent=container)
+        nodes['agent.iterations'] = OptionSlider(window, 'global', 'agent.iterations', option)
+        cfg['agent.iterations'] = nodes['agent.iterations']
+
+        nodes['agent.auto_stop'] = ToggleLabel(trans("toolbox.agent.auto_stop.label"), parent=window)
+        nodes['agent.auto_stop'].box.toggled.connect(common.toggle_auto_stop)
+        cfg['agent.auto_stop'] = nodes['agent.auto_stop']
+
+        nodes['agent.continue'] = ToggleLabel(trans("toolbox.agent.continue.label"), parent=window)
+        nodes['agent.continue'].box.toggled.connect(common.toggle_continue)
+        cfg['agent.continue'] = nodes['agent.continue']
 
-        # options
         cols = QHBoxLayout()
-        cols.addWidget(self.window.ui.config['global']['agent.auto_stop'])
-        cols.addWidget(self.window.ui.config['global']['agent.continue'])
+        cols.addWidget(cfg['agent.auto_stop'])
+        cols.addWidget(cfg['agent.continue'])
 
-        # rows
         rows = QVBoxLayout()
-        rows.addWidget(self.window.ui.nodes['agent.iterations.label'])
-        rows.addWidget(self.window.ui.config['global']['agent.iterations'])
+        rows.addWidget(nodes['agent.iterations.label'])
+        rows.addWidget(cfg['agent.iterations'])
         rows.addLayout(cols)
 
-        self.window.ui.nodes['agent.options'] = QWidget()
-        self.window.ui.nodes['agent.options'].setLayout(rows)
-        self.window.ui.nodes['agent.options'].setContentsMargins(0, 0, 0, 0)
+        nodes['agent.options'] = container
+        nodes['agent.options'].setLayout(rows)
+        nodes['agent.options'].setContentsMargins(0, 0, 0, 0)
 
-        return self.window.ui.nodes['agent.options']
+        return nodes['agent.options']

pygpt_net/ui/layout/toolbox/agent_llama.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.07.22 15:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget, QCheckBox
@@ -32,56 +32,53 @@ class AgentLlama:
 
         :return: QWidget
         """
-        # loop score
-        option = self.window.controller.agent.llama.options["agent.llama.loop.score"]
-        self.window.ui.nodes['agent.llama.loop.score.label'] = QLabel(trans("toolbox.agent.llama.loop.score.label"))
-        self.window.ui.nodes['agent.llama.loop.score'] = \
-            OptionSlider(
-                self.window,
-                'global',
-                'agent.llama.loop.score',
-                option,
-            )
-        self.window.ui.nodes['agent.llama.loop.score'].setToolTip(trans("toolbox.agent.llama.loop.score.tooltip"))
-        self.window.ui.config['global']['agent.llama.loop.score'] = self.window.ui.nodes['agent.llama.loop.score']
-
-        option_mode = self.window.controller.agent.llama.options["agent.llama.loop.mode"]
-        self.window.ui.nodes['agent.llama.loop.mode.label'] = QLabel(trans("toolbox.agent.llama.loop.mode.label"))
-        self.window.ui.nodes['agent.llama.loop.mode'] = \
-            OptionCombo(
-                self.window,
-                'global',
-                'agent.llama.loop.mode',
-                option_mode,
-            )
-        self.window.ui.nodes['agent.llama.loop.mode'].setToolTip(trans("toolbox.agent.llama.loop.mode.tooltip"))
-        self.window.ui.config['global']['agent.llama.loop.mode'] = self.window.ui.nodes['agent.llama.loop.mode']
-
-        # loop enabled
-        self.window.ui.nodes['agent.llama.loop.enabled'] = ToggleLabel(trans("toolbox.agent.llama.loop.enabled.label"), parent=self.window)
-        self.window.ui.nodes['agent.llama.loop.enabled'].box.stateChanged.connect(
-            lambda:
-            self.window.controller.agent.common.toggle_loop(
-                self.window.ui.config['global']['agent.llama.loop.enabled'].box.isChecked())
+        win = self.window
+        ui = win.ui
+        nodes = ui.nodes
+        config_global = ui.config.setdefault('global', {})
+        llama_opts = win.controller.agent.llama.options
+
+        container = QWidget(win)
+
+        nodes['agent.llama.loop.score.label'] = QLabel(trans("toolbox.agent.llama.loop.score.label"), parent=container)
+        nodes['agent.llama.loop.score'] = OptionSlider(
+            win,
+            'global',
+            'agent.llama.loop.score',
+            llama_opts["agent.llama.loop.score"],
+        )
+        nodes['agent.llama.loop.score'].setToolTip(trans("toolbox.agent.llama.loop.score.tooltip"))
+        config_global['agent.llama.loop.score'] = nodes['agent.llama.loop.score']
+
+        nodes['agent.llama.loop.mode.label'] = QLabel(trans("toolbox.agent.llama.loop.mode.label"), parent=container)
+        nodes['agent.llama.loop.mode'] = OptionCombo(
+            win,
+            'global',
+            'agent.llama.loop.mode',
+            llama_opts["agent.llama.loop.mode"],
+        )
+        nodes['agent.llama.loop.mode'].setToolTip(trans("toolbox.agent.llama.loop.mode.tooltip"))
+        config_global['agent.llama.loop.mode'] = nodes['agent.llama.loop.mode']
+
+        nodes['agent.llama.loop.enabled'] = ToggleLabel(trans("toolbox.agent.llama.loop.enabled.label"), parent=win)
+        nodes['agent.llama.loop.enabled'].box.toggled.connect(
+            win.controller.agent.common.toggle_loop
         )
-        self.window.ui.config['global']['agent.llama.loop.enabled'] = self.window.ui.nodes['agent.llama.loop.enabled']
+        config_global['agent.llama.loop.enabled'] = nodes['agent.llama.loop.enabled']
 
-        # label
-        self.window.ui.nodes['agent.llama.loop.label'] = QLabel(trans("toolbox.agent.llama.loop.label"))
+        nodes['agent.llama.loop.label'] = QLabel(trans("toolbox.agent.llama.loop.label"), parent=container)
 
-        # options
         cols = QHBoxLayout()
-        cols.addWidget(self.window.ui.config['global']['agent.llama.loop.enabled'])
-        cols.addWidget(self.window.ui.config['global']['agent.llama.loop.score'])
+        cols.addWidget(config_global['agent.llama.loop.enabled'])
+        cols.addWidget(config_global['agent.llama.loop.score'])
 
-        # rows
         rows = QVBoxLayout()
-        rows.addWidget(self.window.ui.nodes['agent.llama.loop.label'])
+        rows.addWidget(nodes['agent.llama.loop.label'])
         rows.addLayout(cols)
-        rows.addWidget(self.window.ui.nodes['agent.llama.loop.mode'])
+        rows.addWidget(nodes['agent.llama.loop.mode'])
 
-        self.window.ui.nodes['agent_llama.options'] = QWidget()
-        self.window.ui.nodes['agent_llama.options'].setLayout(rows)
-        self.window.ui.nodes['agent_llama.options'].setContentsMargins(0, 0, 0, 0)
+        container.setLayout(rows)
+        container.setContentsMargins(0, 0, 0, 0)
 
-        return self.window.ui.nodes['agent_llama.options']
+        nodes['agent_llama.options'] = container
+        return nodes['agent_llama.options']

pygpt_net/ui/layout/toolbox/assistants.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.17 03:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6 import QtCore
@@ -16,7 +16,6 @@ from PySide6.QtWidgets import QVBoxLayout, QHBoxLayout, QPushButton, QWidget
 from pygpt_net.ui.widget.element.labels import HelpLabel, TitleLabel
 from pygpt_net.ui.widget.lists.assistant import AssistantList
 from pygpt_net.utils import trans
-import pygpt_net.icons_rc
 
 class Assistants:
     def __init__(self, window=None):
@@ -48,49 +47,50 @@ class Assistants:
 
         :return: QVBoxLayout
         """
+        nodes = self.window.ui.nodes
+        ctrl = self.window.controller
+
         # new
-        self.window.ui.nodes['assistants.new'] = QPushButton(QIcon(":/icons/add.svg"), "")
-        self.window.ui.nodes['assistants.new'].clicked.connect(
-            lambda: self.window.controller.assistant.editor.edit())
+        nodes['assistants.new'] = QPushButton(QIcon(":/icons/add.svg"), "")
+        _cb_new = ctrl.assistant.editor.edit
+        nodes['assistants.new'].clicked.connect(lambda checked=False, cb=_cb_new: cb())
 
         # import
-        self.window.ui.nodes['assistants.import'] = QPushButton(trans('assistant.import'))
-        self.window.ui.nodes['assistants.import'].clicked.connect(
-            lambda: self.window.controller.assistant.batch.import_assistants()
-        )
+        nodes['assistants.import'] = QPushButton(trans('assistant.import'))
+        _cb_import = ctrl.assistant.batch.import_assistants
+        nodes['assistants.import'].clicked.connect(lambda checked=False, cb=_cb_import: cb())
 
         # label
-        self.window.ui.nodes['assistants.label'] = TitleLabel(trans("toolbox.assistants.label"))
+        nodes['assistants.label'] = TitleLabel(trans("toolbox.assistants.label"))
 
         # header
         header = QHBoxLayout()
-        header.addWidget(self.window.ui.nodes['assistants.label'])
+        header.addWidget(nodes['assistants.label'])
         header.addStretch(1)
-
-        header.addWidget(self.window.ui.nodes['assistants.new'], alignment=Qt.AlignRight)
-        header.addWidget(self.window.ui.nodes['assistants.import'], alignment=Qt.AlignRight)
+        header.addWidget(nodes['assistants.new'], alignment=Qt.AlignRight)
+        header.addWidget(nodes['assistants.import'], alignment=Qt.AlignRight)
         header.setContentsMargins(5, 0, 0, 0)
         header_widget = QWidget()
         header_widget.setLayout(header)
 
         # list
-        self.window.ui.nodes[self.id] = AssistantList(self.window, self.id)
-        self.window.ui.nodes[self.id].selection_locked = self.window.controller.assistant.change_locked
-        self.window.ui.nodes[self.id].setMinimumWidth(40)
+        nodes[self.id] = AssistantList(self.window, self.id)
+        nodes[self.id].selection_locked = ctrl.assistant.change_locked
+        nodes[self.id].setMinimumWidth(40)
 
-        self.window.ui.nodes['tip.toolbox.assistants'] = HelpLabel(trans('tip.toolbox.assistants'), self.window)
-        self.window.ui.nodes['tip.toolbox.assistants'].setAlignment(Qt.AlignCenter)
+        nodes['tip.toolbox.assistants'] = HelpLabel(trans('tip.toolbox.assistants'), self.window)
+        nodes['tip.toolbox.assistants'].setAlignment(Qt.AlignCenter)
 
         # rows
         layout = QVBoxLayout()
         layout.addWidget(header_widget)
-        layout.addWidget(self.window.ui.nodes[self.id])
-        layout.addWidget(self.window.ui.nodes['tip.toolbox.assistants'])
+        layout.addWidget(nodes[self.id])
+        layout.addWidget(nodes['tip.toolbox.assistants'])
         layout.setContentsMargins(2, 5, 5, 5)
 
         # model
-        self.window.ui.models[self.id] = self.create_model(self.window)
-        self.window.ui.nodes[self.id].setModel(self.window.ui.models[self.id])
+        self.window.ui.models[self.id] = self.create_model(nodes[self.id])
+        nodes[self.id].setModel(self.window.ui.models[self.id])
 
         return layout
 
@@ -109,18 +109,22 @@ class Assistants:
 
         :param data: Data to update
         """
-        # store previous selection
-        self.window.ui.nodes[self.id].backup_selection()
-
-        self.window.ui.models[self.id].removeRows(0, self.window.ui.models[self.id].rowCount())
-        i = 0
-        for n in data:
-            self.window.ui.models[self.id].insertRow(i)
-            name = data[n].name
-            index = self.window.ui.models[self.id].index(i, 0)
-            self.window.ui.models[self.id].setData(index, "ID: " + data[n].id, QtCore.Qt.ToolTipRole)
-            self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 0), name)
-            i += 1
-
-        # restore previous selection
-        self.window.ui.nodes[self.id].restore_selection()
+        view = self.window.ui.nodes[self.id]
+        model = self.window.ui.models[self.id]
+
+        view.backup_selection()
+        view.setUpdatesEnabled(False)
+        try:
+            with QtCore.QSignalBlocker(model):
+                model.setRowCount(0)
+                count = len(data)
+                if count:
+                    model.setRowCount(count)
+                    for i, item in enumerate(data.values()):
+                        index = model.index(i, 0)
+                        model.setData(index, "ID: " + item.id, QtCore.Qt.ToolTipRole)
+                        model.setData(index, item.name)
+        finally:
+            view.setUpdatesEnabled(True)
+
+        view.restore_selection()

pygpt_net/ui/layout/toolbox/computer_env.py

@@ -6,14 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.07.26 18:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
-from PySide6.QtWidgets import QVBoxLayout, QHBoxLayout, QWidget, QComboBox
+from PySide6.QtWidgets import QHBoxLayout, QWidget, QComboBox
 
 from pygpt_net.ui.widget.element.labels import TitleLabel
 from pygpt_net.utils import trans
-import pygpt_net.icons_rc
 
 
 class ComputerEnv:
@@ -33,36 +32,46 @@ class ComputerEnv:
         :return: QWidget
         """
         layout = self.setup_env()
-        self.window.ui.nodes['env.widget'] = QWidget()
-        self.window.ui.nodes['env.widget'].setLayout(layout)
-        return self.window.ui.nodes['env.widget']
+        nodes = self.window.ui.nodes
+        widget = QWidget()
+        widget.setLayout(layout)
+        nodes['env.widget'] = widget
+        return widget
 
-    def setup_env(self) -> QVBoxLayout:
+    def _on_env_index_changed(self, index: int) -> None:
+        nodes = self.window.ui.nodes
+        cbox = nodes.get(self.id)
+        if cbox is None:
+            return
+        data = cbox.itemData(index)
+        self.window.controller.ui.on_computer_env_changed(data)
+
+    def setup_env(self) -> QHBoxLayout:
         """
         Setup list of environments
 
         :return: QVBoxLayout
         """
 
-        # label
-        self.window.ui.nodes['env.label'] = TitleLabel(trans("toolbox.env.label"))
+        nodes = self.window.ui.nodes
+        label = TitleLabel(trans("toolbox.env.label"))
+        nodes['env.label'] = label
 
-        # list
-        self.window.ui.nodes[self.id] = QComboBox()
-        self.window.ui.nodes[self.id].addItem("Browser", "browser")
-        self.window.ui.nodes[self.id].addItem("Windows", "windows")
-        self.window.ui.nodes[self.id].addItem("Linux", "linux")
-        self.window.ui.nodes[self.id].addItem("Mac", "mac")
-        self.window.ui.nodes[self.id].setMinimumWidth(40)
+        cbox = QComboBox()
+        for text, data in (
+            ("Browser", "browser"),
+            ("Windows", "windows"),
+            ("Linux", "linux"),
+            ("Mac", "mac"),
+        ):
+            cbox.addItem(text, data)
+        cbox.setMinimumWidth(40)
 
-        # on change signal
-        self.window.ui.nodes[self.id].currentIndexChanged.connect(
-            lambda index: self.window.controller.ui.on_computer_env_changed(
-                self.window.ui.nodes[self.id].itemData(index)))
+        cbox.currentIndexChanged.connect(self._on_env_index_changed)
+        nodes[self.id] = cbox
 
-        # rows
         layout = QHBoxLayout()
-        layout.addWidget(self.window.ui.nodes['env.label'], 0)  # label
-        layout.addWidget(self.window.ui.nodes[self.id], 1)  # combobox
+        layout.addWidget(label, 0)
+        layout.addWidget(cbox, 1)
         layout.setContentsMargins(2, 5, 5, 5)
         return layout