pygpt-net 2.7.4__py3-none-any.whl → 2.7.6__py3-none-any.whl

pygpt_net/ui/layout/toolbox/model.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2026.01.03 17:00:00                  #
 # ================================================== #
-
-from PySide6.QtWidgets import QVBoxLayout, QWidget
+from PySide6.QtCore import Qt, QSize
+from PySide6.QtGui import QIcon
+from PySide6.QtWidgets import QVBoxLayout, QWidget, QPushButton, QHBoxLayout, QSizePolicy
 
 from pygpt_net.ui.widget.element.labels import TitleLabel
 from pygpt_net.ui.widget.lists.model_combo import ModelCombo
@@ -27,6 +28,7 @@ class Model:
         self.window = window
         self.id = 'prompt.model'
         self.label_key = f'{self.id}.label'
+        self._settings_icon = QIcon(":/icons/settings.svg")
 
     def setup(self) -> QWidget:
         """
@@ -51,12 +53,33 @@ class Model:
         nodes[self.label_key] = label
 
         combo = ModelCombo(self.window, self.id)
+        # Ensure combo takes maximum horizontal space
+        combo.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
         nodes[self.id] = combo
 
+        nodes['prompt.model.settings'] = QPushButton(self._settings_icon, "")
+        # Configure compact, borderless settings button aligned to the right
+        icon_size = 20
+        nodes['prompt.model.settings'].setFlat(True)
+        nodes['prompt.model.settings'].setStyleSheet("QPushButton { border: none; padding: 0; }")
+        nodes['prompt.model.settings'].setIconSize(QSize(icon_size, icon_size))
+        nodes['prompt.model.settings'].setFixedSize(icon_size, icon_size)
+        nodes['prompt.model.settings'].setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
+        nodes['prompt.model.settings'].setFocusPolicy(Qt.NoFocus)
+        nodes['prompt.model.settings'].clicked.connect(self._open_settings)
+
+        model_cols = QHBoxLayout()
+        model_cols.addWidget(combo, 1)  # stretch to take remaining space
+        model_cols.addWidget(nodes['prompt.model.settings'], alignment=Qt.AlignRight)
+        model_cols.setContentsMargins(0, 0, 0, 0)
+
         layout = QVBoxLayout()
         layout.addWidget(label)
-        layout.addWidget(combo)
+        layout.addLayout(model_cols)
         layout.addStretch()
         layout.setContentsMargins(2, 5, 5, 5)
 
-        return layout
+        return layout
+
+    def _open_settings(self):
+        self.window.controller.model.editor.open()

pygpt_net/ui/menu/tools.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2026.01.02 19:00:00                  #
 # ================================================== #
 
 from PySide6.QtGui import QAction, QIcon
@@ -26,8 +26,11 @@ class Tools:
         action = self.window.sender()
         self.window.controller.tools.open_tab(action.data())
 
-    def _toggle_assistant_store(self, checked=False):
-        self.window.controller.assistant.store.toggle_editor()
+    def _toggle_remote_store_openai(self, checked=False):
+        self.window.controller.remote_store.openai.toggle_editor()
+
+    def _toggle_remote_store_google(self, checked=False):
+        self.window.controller.remote_store.google.toggle_editor()
 
     def _rebuild_ipython(self, checked=False):
         self.window.core.plugins.get("cmd_code_interpreter").builder.build_and_restart()
@@ -69,9 +72,14 @@ class Tools:
 
         menu_tools.addSeparator()
         db_icon = QIcon(":/icons/db.svg")
-        ui_menu['menu.tools.openai.stores'] = QAction(db_icon, trans("dialog.assistant.store"), window)
+        ui_menu['menu.tools.openai.stores'] = QAction(db_icon, trans("dialog.remote_store.openai"), window)
         menu_tools.addAction(ui_menu['menu.tools.openai.stores'])
-        ui_menu['menu.tools.openai.stores'].triggered.connect(self._toggle_assistant_store)
+        ui_menu['menu.tools.openai.stores'].triggered.connect(self._toggle_remote_store_openai)
+
+        ui_menu['menu.tools.google.stores'] = QAction(db_icon, trans("dialog.remote_store.google"), window)
+        menu_tools.addAction(ui_menu['menu.tools.google.stores'])
+        ui_menu['menu.tools.google.stores'].triggered.connect(self._toggle_remote_store_google)
+        # ------------------------------------------------- #
 
         menu_tools.addSeparator()
         reload_icon = QIcon(":/icons/reload.svg")

pygpt_net/ui/widget/dialog/remote_store_google.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2026.01.02 20:00:00                  #
+# ================================================== #
+
+from PySide6.QtCore import Qt
+
+from .base import BaseDialog
+
+
+class RemoteStoreGoogleDialog(BaseDialog):
+    def __init__(self, window=None, id=None):
+        """
+        Models dialog (Google File Search)
+
+        :param window: main window
+        :param id: settings id
+        """
+        super(RemoteStoreGoogleDialog, self).__init__(window, id)
+        self.window = window
+        self.id = id
+
+    def closeEvent(self, event):
+        """
+        Close event
+
+        :param event: close event
+        """
+        self.window.controller.remote_store.google.dialog = False
+        self.window.controller.remote_store.google.update()
+        super(RemoteStoreGoogleDialog, self).closeEvent(event)
+
+    def keyPressEvent(self, event):
+        """
+        Key press event
+
+        :param event: key press event
+        """
+        if event.key() == Qt.Key_Escape:
+            self.cleanup()
+            self.close()
+        else:
+            super(RemoteStoreGoogleDialog, self).keyPressEvent(event)
+
+    def cleanup(self):
+        """
+        Cleanup on close
+        """
+        self.window.controller.remote_store.google.dialog = False
+        self.window.controller.remote_store.google.update()

pygpt_net/ui/widget/dialog/{assistant_store.py → remote_store_openai.py}

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.26 23:00:00                  #
+# Updated Date: 2026.01.02 19:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
@@ -14,7 +14,7 @@ from PySide6.QtCore import Qt
 from .base import BaseDialog
 
 
-class AssistantVectorStoreDialog(BaseDialog):
+class RemoteStoreOpenAIDialog(BaseDialog):
     def __init__(self, window=None, id=None):
         """
         Models dialog
@@ -22,7 +22,7 @@ class AssistantVectorStoreDialog(BaseDialog):
         :param window: main window
         :param id: settings id
         """
-        super(AssistantVectorStoreDialog, self).__init__(window, id)
+        super(RemoteStoreOpenAIDialog, self).__init__(window, id)
         self.window = window
         self.id = id
 
@@ -32,9 +32,9 @@ class AssistantVectorStoreDialog(BaseDialog):
 
         :param event: close event
         """
-        self.window.controller.assistant.store.dialog = False
-        self.window.controller.assistant.store.update()
-        super(AssistantVectorStoreDialog, self).closeEvent(event)
+        self.window.controller.remote_store.openai.dialog = False
+        self.window.controller.remote_store.openai.update()
+        super(RemoteStoreOpenAIDialog, self).closeEvent(event)
 
     def keyPressEvent(self, event):
         """
@@ -46,11 +46,11 @@ class AssistantVectorStoreDialog(BaseDialog):
             self.cleanup()
             self.close()  # close dialog when the Esc key is pressed.
         else:
-            super(AssistantVectorStoreDialog, self).keyPressEvent(event)
+            super(RemoteStoreOpenAIDialog, self).keyPressEvent(event)
 
     def cleanup(self):
         """
         Cleanup on close
         """
-        self.window.controller.assistant.store.dialog = False
-        self.window.controller.assistant.store.update()
+        self.window.controller.remote_store.openai.dialog = False
+        self.window.controller.remote_store.openai.update()

pygpt_net/ui/widget/element/button.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2026.01.02 19:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
@@ -87,7 +87,7 @@ class SyncButton(QPushButton):
         super().__init__(title)
         self.window = window
         self.clicked.connect(
-            lambda: self.window.controller.assistant.batch.import_files_current()
+            lambda: self.window.controller.remote_store.openai.batch.import_files_current()
         )
 
     @classmethod
@@ -112,11 +112,11 @@ class SyncButton(QPushButton):
         menu = QMenu(parent)
         act_current = menu.addAction(type(self)._icon_download, trans('attachments_uploaded.btn.sync.current'))
         act_current.triggered.connect(
-            lambda checked=False: self.window.controller.assistant.batch.import_files_current()
+            lambda checked=False: self.window.controller.remote_store.openai.batch.import_files_current()
         )
         act_all = menu.addAction(type(self)._icon_download, trans('attachments_uploaded.btn.sync.all'))
         act_all.triggered.connect(
-            lambda checked=False: self.window.controller.assistant.batch.import_files()
+            lambda checked=False: self.window.controller.remote_store.openai.batch.import_files()
         )
         menu.exec_(parent.mapToGlobal(pos))
         menu.deleteLater()

pygpt_net/ui/widget/image/display.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2026.01.03 17:00:00                  #
 # ================================================== #
 
 from PySide6.QtGui import QAction, QIcon
@@ -27,6 +27,17 @@ class ImageLabel(QLabel):
         self.window = window
         self.path = path
 
+    def _get_window(self):
+        """
+        Get main window instance
+
+        :return: Window instance
+        """
+        if self.window is not None:
+            if hasattr(self.window, 'window'):
+                return self.window.window
+        return self.window
+
     def contextMenuEvent(self, event):
         """
         Context menu event
@@ -36,6 +47,8 @@ class ImageLabel(QLabel):
         if not self.path:
             return
 
+        win = self._get_window()
+
         actions = {}
         use_actions = []
         actions['open'] = QAction(QIcon(":/icons/fullscreen.svg"), trans('img.action.open'), self)
@@ -64,13 +77,13 @@ class ImageLabel(QLabel):
             self,
         )
         actions['use_attachment'].triggered.connect(
-            lambda: self.window.controller.files.use_attachment(self.path),
+            lambda: win.controller.files.use_attachment(self.path),
         )
         use_actions.append(actions['use_attachment'])
 
         # use by filetype
-        if self.window.core.filesystem.actions.has_use(self.path):
-            extra_use_actions = self.window.core.filesystem.actions.get_use(self, self.path)
+        if win.core.filesystem.actions.has_use(self.path):
+            extra_use_actions = win.core.filesystem.actions.get_use(self, self.path)
             for action in extra_use_actions:
                 use_actions.append(action)
 
@@ -97,7 +110,8 @@ class ImageLabel(QLabel):
 
         :param event: mouse event
         """
-        self.window.tools.get("viewer").open(self.path)
+        win = self._get_window()
+        win.tools.get("viewer").open(self.path)
 
     def action_open_dir(self, event):
         """
@@ -105,7 +119,8 @@ class ImageLabel(QLabel):
 
         :param event: mouse event
         """
-        self.window.tools.get("viewer").open_dir(self.path)
+        win = self._get_window()
+        win.tools.get("viewer").open_dir(self.path)
 
     def action_save(self, event):
         """
@@ -113,7 +128,8 @@ class ImageLabel(QLabel):
 
         :param event: mouse event
         """
-        self.window.tools.get("viewer").save(self.path)
+        win = self._get_window()
+        win.tools.get("viewer").save(self.path)
 
     def action_delete(self, event):
         """
@@ -121,4 +137,5 @@ class ImageLabel(QLabel):
 
         :param event: mouse event
         """
-        self.window.tools.get("viewer").delete(self.path)
+        win = self._get_window()
+        win.tools.get("viewer").delete(self.path)

pygpt_net/ui/widget/lists/remote_store_google.py

@@ -0,0 +1,248 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2026.01.02 20:00:00                  #
+# ================================================== #
+
+from PySide6.QtGui import QAction, QIcon
+from PySide6.QtCore import Qt, QItemSelectionModel
+from PySide6.QtWidgets import QMenu, QAbstractItemView
+
+from pygpt_net.ui.widget.lists.base import BaseList
+from pygpt_net.utils import trans
+import pygpt_net.icons_rc
+
+
+class RemoteStoreGoogleEditorList(BaseList):
+    def __init__(self, window=None, id=None):
+        """
+        Store select menu (in editor) - Google File Search
+
+        :param window: main window
+        :param id: parent id
+        """
+        super(RemoteStoreGoogleEditorList, self).__init__(window)
+        self.window = window
+        self.id = id
+
+        self._suppress_item_click = False
+        self._ctrl_multi_active = False
+        self._ctrl_multi_index = None
+        self._was_shift_click = False
+
+        self._backup_selection = None
+        self.restore_after_ctx_menu = True
+
+        self.setSelectionBehavior(QAbstractItemView.SelectRows)
+        self.setSelectionMode(QAbstractItemView.ExtendedSelection)
+
+        # Disable BaseList click handler; business action handled manually
+        self.clicked.disconnect(self.click)
+
+    def _selected_rows(self) -> list[int]:
+        try:
+            return sorted([ix.row() for ix in self.selectionModel().selectedRows()])
+        except Exception:
+            return []
+
+    def _has_multi_selection(self) -> bool:
+        try:
+            return len(self.selectionModel().selectedRows()) > 1
+        except Exception:
+            return False
+
+    def mousePressEvent(self, event):
+        if event.button() == Qt.LeftButton and (event.modifiers() & Qt.ControlModifier):
+            idx = self.indexAt(event.pos())
+            if idx.isValid():
+                self._ctrl_multi_active = True
+                self._ctrl_multi_index = idx
+                self._suppress_item_click = True
+                event.accept()
+                return
+            self._suppress_item_click = True
+            event.accept()
+            return
+
+        if event.button() == Qt.LeftButton and (event.modifiers() & Qt.ShiftModifier):
+            idx = self.indexAt(event.pos())
+            self._suppress_item_click = True
+            self._was_shift_click = True
+            if idx.isValid():
+                super(RemoteStoreGoogleEditorList, self).mousePressEvent(event)
+            else:
+                event.accept()
+            return
+
+        if event.button() == Qt.LeftButton:
+            idx = self.indexAt(event.pos())
+            if self._has_multi_selection():
+                sel_model = self.selectionModel()
+                sel_model.clearSelection()
+                if not idx.isValid():
+                    event.accept()
+                    return
+            super(RemoteStoreGoogleEditorList, self).mousePressEvent(event)
+            return
+
+        if event.button() == Qt.RightButton:
+            idx = self.indexAt(event.pos())
+            sel_model = self.selectionModel()
+            selected_rows = [ix.row() for ix in sel_model.selectedRows()]
+            multi = len(selected_rows) > 1
+
+            if idx.isValid():
+                if multi and idx.row() in selected_rows:
+                    self._backup_selection = None
+                else:
+                    self._backup_selection = list(sel_model.selectedIndexes())
+                    sel_model.clearSelection()
+                    sel_model.select(idx, QItemSelectionModel.Select | QItemSelectionModel.Rows)
+            event.accept()
+            return
+
+        super(RemoteStoreGoogleEditorList, self).mousePressEvent(event)
+
+    def mouseReleaseEvent(self, event):
+        if event.button() == Qt.LeftButton and self._was_shift_click:
+            self._was_shift_click = False
+            self._suppress_item_click = False
+            super(RemoteStoreGoogleEditorList, self).mouseReleaseEvent(event)
+            return
+
+        if event.button() == Qt.LeftButton and self._ctrl_multi_active:
+            try:
+                idx = self.indexAt(event.pos())
+                if idx.isValid() and self._ctrl_multi_index and idx == self._ctrl_multi_index:
+                    sel_model = self.selectionModel()
+                    sel_model.select(idx, QItemSelectionModel.Toggle | QItemSelectionModel.Rows)
+            finally:
+                self._ctrl_multi_active = False
+                self._ctrl_multi_index = None
+                self._suppress_item_click = False
+            event.accept()
+            return
+
+        if event.button() == Qt.LeftButton:
+            idx = self.indexAt(event.pos())
+            if not self._has_multi_selection():
+                if idx.isValid() and not self._suppress_item_click:
+                    self.window.controller.remote_store.google.select(idx.row())
+            self._suppress_item_click = False
+            super(RemoteStoreGoogleEditorList, self).mouseReleaseEvent(event)
+            return
+
+        super(RemoteStoreGoogleEditorList, self).mouseReleaseEvent(event)
+
+    def click(self, val):
+        pass
+
+    def contextMenuEvent(self, event):
+        actions = {}
+        actions['refresh'] = QAction(
+            QIcon(":/icons/reload.svg"),
+            trans('dialog.remote_store.menu.current.refresh_store'),
+            self
+        )
+        actions['delete'] = QAction(QIcon(":/icons/delete.svg"), trans('action.delete'), self)
+        actions['clear'] = QAction(
+            QIcon(":/icons/close.svg"),
+            trans('dialog.remote_store.menu.current.clear_files'),
+            self
+        )
+        actions['truncate'] = QAction(
+            QIcon(":/icons/delete.svg"),
+            trans('dialog.remote_store.menu.current.truncate_files'),
+            self
+        )
+
+        menu = QMenu(self)
+        menu.addAction(actions['refresh'])
+        menu.addAction(actions['delete'])
+        menu.addAction(actions['clear'])
+        menu.addAction(actions['truncate'])
+
+        index = self.indexAt(event.pos())
+        idx = index.row() if index.isValid() else -1
+
+        selected_rows = self._selected_rows()
+        multi = len(selected_rows) > 1
+
+        if not index.isValid() and not multi:
+            if self._backup_selection is not None and self.restore_after_ctx_menu:
+                sel_model = self.selectionModel()
+                sel_model.clearSelection()
+                for i in self._backup_selection:
+                    sel_model.select(i, QItemSelectionModel.Select | QItemSelectionModel.Rows)
+                self._backup_selection = None
+            return
+
+        if multi:
+            actions['refresh'].triggered.connect(lambda: self.action_refresh(list(selected_rows)))
+            actions['delete'].triggered.connect(lambda: self.action_delete(list(selected_rows)))
+            actions['clear'].triggered.connect(lambda: self.action_clear(list(selected_rows)))
+            actions['truncate'].triggered.connect(lambda: self.action_truncate(list(selected_rows)))
+        else:
+            actions['refresh'].triggered.connect(lambda: self.action_refresh(idx))
+            actions['delete'].triggered.connect(lambda: self.action_delete(idx))
+            actions['clear'].triggered.connect(lambda: self.action_clear(idx))
+            actions['truncate'].triggered.connect(lambda: self.action_truncate(idx))
+
+        menu.exec_(event.globalPos())
+
+        if self.restore_after_ctx_menu and self._backup_selection is not None:
+            sel_model = self.selectionModel()
+            sel_model.clearSelection()
+            for i in self._backup_selection:
+                sel_model.select(i, QItemSelectionModel.Select | QItemSelectionModel.Rows)
+            self._backup_selection = None
+        self.restore_after_ctx_menu = True
+
+    def action_delete(self, item):
+        if isinstance(item, (list, tuple)):
+            if item:
+                self.restore_after_ctx_menu = False
+                self.window.controller.remote_store.google.delete_by_idx(list(item))
+            return
+        idx = int(item)
+        if idx >= 0:
+            self.restore_after_ctx_menu = False
+            self.window.controller.remote_store.google.delete_by_idx(idx)
+
+    def action_clear(self, item):
+        if isinstance(item, (list, tuple)):
+            if item:
+                self.restore_after_ctx_menu = False
+                self.window.controller.remote_store.google.batch.clear_store_files_by_idx(list(item))
+            return
+        idx = int(item)
+        if idx >= 0:
+            self.restore_after_ctx_menu = False
+            self.window.controller.remote_store.google.batch.clear_store_files_by_idx(idx)
+
+    def action_truncate(self, item):
+        if isinstance(item, (list, tuple)):
+            if item:
+                self.restore_after_ctx_menu = False
+                self.window.controller.remote_store.google.batch.truncate_store_files_by_idx(list(item))
+            return
+        idx = int(item)
+        if idx >= 0:
+            self.restore_after_ctx_menu = False
+            self.window.controller.remote_store.google.batch.truncate_store_files_by_idx(idx)
+
+    def action_refresh(self, item):
+        if isinstance(item, (list, tuple)):
+            if item:
+                self.restore_after_ctx_menu = False
+                self.window.controller.remote_store.google.refresh_by_idx(list(item))
+            return
+        idx = int(item)
+        if idx >= 0:
+            self.restore_after_ctx_menu = False
+            self.window.controller.remote_store.google.refresh_by_idx(idx)