pygpt-net 2.4.36.post1__py3-none-any.whl → 2.4.41__py3-none-any.whl

pygpt_net/controller/ctx/__init__.py

@@ -6,10 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.24 00:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Optional, List
+
 from PySide6.QtCore import QModelIndex
+from PySide6.QtGui import QStandardItem
 
 from pygpt_net.core.events import Event, AppEvent, RenderEvent
 from pygpt_net.item.ctx import CtxItem
@@ -19,7 +22,7 @@ from .summarizer import Summarizer
 from .extra import Extra
 
 from pygpt_net.utils import trans
-from pygpt_net.core.tabs import Tab
+from pygpt_net.core.tabs.tab import Tab
 
 
 class Ctx:
@@ -60,20 +63,6 @@ class Ctx:
                 # if no ctx then get first ctx
                 self.window.core.ctx.set_current(self.window.core.ctx.get_first())
 
-        # restore previous ctx
-        # 1) try to load from tabs data - if not exists, load from current
-        tab = self.window.core.tabs.get_first_by_type(Tab.TAB_CHAT)
-        loaded = False
-        if tab is not None:
-            meta_id = tab.data_id
-            if meta_id is not None:
-                self.load(meta_id)
-                loaded = True
-
-        # 2) load from current if not loaded yet
-        if not loaded:
-            self.load(self.window.core.ctx.get_current())
-
         # restore search string if exists
         if self.window.core.config.has("ctx.search.string"):
             string = self.window.core.config.get("ctx.search.string")
@@ -89,7 +78,12 @@ class Ctx:
         self.window.ui.nodes['ctx.list'].collapseAll()  # collapse all items at start
         self.restore_expanded_groups()  # restore expanded groups
 
-    def update(self, reload: bool = True, all: bool = True, select: bool = True):
+    def update(
+            self,
+            reload: bool = True,
+            all: bool = True,
+            select: bool = True
+    ):
         """
         Update ctx list
 
@@ -122,7 +116,11 @@ class Ctx:
         # update additional context attachments
         self.window.controller.chat.attachment.update()
 
-    def select(self, id: int, force: bool = False):
+    def select(
+            self,
+            id: int,
+            force: bool = False
+    ):
         """
         Select ctx by id
 
@@ -144,6 +142,19 @@ class Ctx:
         # update additional context attachments
         self.window.controller.chat.attachment.update()
 
+    def select_on_list_only(self, id: int):
+        """
+        Select ctx by id only on list
+
+        :param id: context meta id
+        """
+        self.window.core.ctx.select(id, restore_model=True)
+        self.window.core.ctx.set_current(id)
+        self.update_list(True)
+        self.select_by_current()
+        self.reload_config(all=False)
+        self.update()
+
     def select_by_idx(self, idx: int):
         """
         Select ctx by index
@@ -189,7 +200,7 @@ class Ctx:
         self.set_group(None)
         self.window.ui.nodes['ctx.list'].clearSelection()
 
-    def set_group(self, group_id: int = None):
+    def set_group(self, group_id: Optional[int] = None):
         """
         Set current selected group
 
@@ -207,7 +218,11 @@ class Ctx:
         self.group_id = None
         self.new()
 
-    def new(self, force: bool = False, group_id: int = None):
+    def new(
+            self,
+            force: bool = False,
+            group_id: Optional[int] = None
+    ):
         """
         Create new ctx
 
@@ -326,7 +341,11 @@ class Ctx:
         event = RenderEvent(RenderEvent.CTX_APPEND, data)
         self.window.dispatch(event)
 
-    def load(self, id: int, restore_model: bool = True):
+    def load(
+            self,
+            id: int,
+            restore_model: bool = True
+    ):
         """
         Load ctx data
 
@@ -348,6 +367,21 @@ class Ctx:
             event = RenderEvent(RenderEvent.ON_LOAD, data)
             self.window.dispatch(event)
 
+        self.reload_config()
+
+        # reload ctx list and select current ctx on list, without reloading all
+        self.update(reload=False, all=True)
+
+        # update tab title
+        if meta is not None:
+            self.window.controller.ui.tabs.on_load_ctx(meta)
+
+    def reload_config(self, all: bool = True):
+        """
+        Reload config
+
+        :param all: reload all
+        """
         # get current settings stored in ctx
         thread = self.window.core.ctx.get_thread()
         mode = self.window.core.ctx.get_mode()
@@ -358,8 +392,9 @@ class Ctx:
         # restore thread from ctx
         self.window.core.config.set('assistant_thread', thread)
 
-        # clear before output and append ctx to output
-        self.refresh_output()
+        if all:
+            # clear before output and append ctx to output
+            self.refresh_output()
 
         # switch mode to ctx mode
         if mode is not None:
@@ -384,16 +419,9 @@ class Ctx:
             if model is not None and self.window.core.models.has_model(mode, model):
                 self.window.controller.model.set(mode, model)
 
-        # reload ctx list and select current ctx on list, without reloading all
-        self.update(reload=False, all=True)
-
         # update current ctx label in UI
         self.common.update_label(mode, assistant_id)
 
-        # update tab title
-        if meta is not None:
-            self.window.controller.ui.tabs.update_title_current(meta.name)
-
     def update_ctx(self):
         """Update current ctx mode if allowed"""
         mode = self.window.core.config.get('mode')
@@ -415,7 +443,11 @@ class Ctx:
         # update ctx label
         self.common.update_label(mode, id)
 
-    def delete(self, id: int, force: int = False):
+    def delete(
+            self,
+            id: int,
+            force: bool = False
+    ):
         """
         Delete ctx by idx
 
@@ -469,7 +501,11 @@ class Ctx:
                 for idx in list(meta.indexes[store_id]):
                     self.window.core.ctx.idx.remove_meta_from_indexed(store_id, id, idx)
 
-    def delete_item(self, id: int, force: bool = False):
+    def delete_item(
+            self,
+            id: int,
+            force: bool = False
+    ):
         """
         Delete ctx item by id
 
@@ -567,7 +603,11 @@ class Ctx:
         self.window.ui.dialog['rename'].show()
         self.update()
 
-    def set_important(self, id: int, value: bool = True):
+    def set_important(
+            self,
+            id: int,
+            value: bool = True
+    ):
         """
         Set as important
 
@@ -593,7 +633,11 @@ class Ctx:
             return meta.important
         return False
 
-    def set_label(self, id: int, label_id: int):
+    def set_label(
+            self,
+            id: int,
+            label_id: int
+    ):
         """
         Set color label for ctx by idx
 
@@ -694,7 +738,7 @@ class Ctx:
         self.window.ui.nodes['ctx.search'].setText(text)
         self.search_string_change(text)  # make search
 
-    def label_filters_changed(self, labels: list):
+    def label_filters_changed(self, labels: List[int]):
         """
         Filters labels change
 
@@ -726,7 +770,7 @@ class Ctx:
         """
         return self.window.controller.chat.input.generating
 
-    def select_index_by_id(self, id):
+    def select_index_by_id(self, id: int):
         """
         Select item by ID on context list
 
@@ -737,7 +781,11 @@ class Ctx:
         self.window.ui.nodes['ctx.list'].setCurrentIndex(index)
         self.window.ui.nodes['ctx.list'].unlocked = False  # tmp allow change if locked (disable)
 
-    def find_index_by_id(self, item, id):
+    def find_index_by_id(
+            self,
+            item: QStandardItem,
+            id: int
+    ) -> QModelIndex:
         """
         Return index of item with given ID, searching recursively through the model.
 
@@ -753,7 +801,11 @@ class Ctx:
                 return found_index
         return QModelIndex()
 
-    def find_parent_index_by_id(self, item, id):
+    def find_parent_index_by_id(
+            self,
+            item: QStandardItem,
+            id: int
+    ) -> QModelIndex:
         """
         Return index of item with given ID, searching recursively through the model.
 
@@ -769,7 +821,7 @@ class Ctx:
                 return found_index
         return QModelIndex()
 
-    def get_parent_index_by_id(self, id):
+    def get_parent_index_by_id(self, id: int) -> QModelIndex:
         """
         Return QModelIndex of parent item based on its ID.
 
@@ -780,7 +832,11 @@ class Ctx:
         root = model.invisibleRootItem()
         return self.find_parent_index_by_id(root, id)
 
-    def get_children_index_by_id(self, parent_id, child_id):
+    def get_children_index_by_id(
+            self,
+            parent_id: int,
+            child_id: int
+    ) -> QModelIndex:
         """
         Return QModelIndex of child item based on its ID and parent ID.
 
@@ -797,7 +853,11 @@ class Ctx:
         parent_item = model.itemFromIndex(parent_index)
         return self.find_index_by_id(parent_item, child_id)
 
-    def find_child_index_by_id(self, root_item, child_id):
+    def find_child_index_by_id(
+            self,
+            root_item: QStandardItem,
+            child_id: int
+    ) -> QModelIndex:
         """
         Find and return QModelIndex of child based on its ID, recursively searching through the model.
 
@@ -814,7 +874,7 @@ class Ctx:
                 return child_index
         return QModelIndex()
 
-    def get_child_index_by_id(self, child_id):
+    def get_child_index_by_id(self, child_id: int) -> QModelIndex:
         """
         Return QModelIndex of child item based on its ID.
 
@@ -848,7 +908,12 @@ class Ctx:
         """Save visible ctx list items"""
         self.store_expanded_groups()
 
-    def move_to_group(self, meta_id, group_id, update: bool = True):
+    def move_to_group(
+            self,
+            meta_id: int,
+            group_id: int,
+            update: bool = True
+    ):
         """
         Move ctx to group
 
@@ -871,7 +936,10 @@ class Ctx:
         self.group_id = None
         self.update()
 
-    def new_group(self, meta_id = None):
+    def new_group(
+            self,
+            meta_id: Optional[int] = None
+    ):
         """
         Open new group dialog
 
@@ -883,7 +951,11 @@ class Ctx:
         self.window.ui.dialog['create'].show()
         self.window.ui.dialog['create'].input.setFocus()
 
-    def create_group(self, name: str = None, meta_id = None):
+    def create_group(
+            self,
+            name: Optional[str] = None,
+            meta_id: Optional[int] = None
+    ):
         """
         Make directory
 
@@ -910,7 +982,11 @@ class Ctx:
             self.select_group(id)
             self.group_id = id
 
-    def rename_group(self, id: int, force: bool = False):
+    def rename_group(
+            self,
+            id: int,
+            force: bool = False
+    ):
         """
         Rename group
 
@@ -926,7 +1002,12 @@ class Ctx:
             self.window.ui.dialog['rename'].current = id
             self.window.ui.dialog['rename'].show()
 
-    def update_group_name(self, id: int, name: str, close: bool = True):
+    def update_group_name(
+            self,
+            id: int,
+            name: str,
+            close: bool = True
+    ):
         """
         Update group name
 
@@ -967,7 +1048,11 @@ class Ctx:
         self.window.ui.nodes['ctx.list'].setCurrentIndex(index)
         self.window.ui.nodes['ctx.list'].unlocked = False  # tmp allow change if locked (disable)
 
-    def delete_group(self, id: int, force: bool = False):
+    def delete_group(
+            self,
+            id: int,
+            force: bool = False
+    ):
         """
         Delete group only
 
@@ -988,7 +1073,11 @@ class Ctx:
                 self.group_id = None
             self.update()
 
-    def delete_group_all(self, id: int, force: bool = False):
+    def delete_group_all(
+            self,
+            id: int,
+            force: bool = False
+    ):
         """
         Delete group with all items
 

pygpt_net/controller/ctx/common.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.23 00:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Optional
+
 from PySide6.QtWidgets import QApplication
 
 from pygpt_net.core.tabs.tab import Tab
@@ -53,7 +55,11 @@ class Common:
         # update ctx label
         self.window.controller.ui.update_ctx_label(label)
 
-    def update_label(self, mode: str, assistant_id: str = None):
+    def update_label(
+            self,
+            mode: str,
+            assistant_id: Optional[str] = None
+    ):
         """
         Update ctx label
 
@@ -91,10 +97,11 @@ class Common:
 
     def focus_chat(self):
         """Focus chat"""
+        tabs = self.window.ui.layout.get_active_tabs()
         if self.window.controller.ui.tabs.get_current_type() != Tab.TAB_CHAT:
             idx = self.window.core.tabs.get_min_idx_by_type(Tab.TAB_CHAT)
             if idx is not None:
-                self.window.ui.tabs['output'].setCurrentIndex(idx)
+                tabs.setCurrentIndex(idx)
 
     def restore_display_filter(self):
         """Restore display filter"""
@@ -159,7 +166,11 @@ class Common:
         self.window.controller.chat.common.append_to_input(value, separator=" ")
         QApplication.clipboard().setText(value)
 
-    def reset(self, meta_id: int, force: bool = False):
+    def reset(
+            self,
+            meta_id: int,
+            force: bool = False
+    ):
         """
         Reset by meta id
 

pygpt_net/controller/ctx/extra.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.20 21:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QApplication
@@ -144,7 +144,11 @@ class Extra:
         self.window.controller.ctx.edit_meta_id = None
         self.edit_hide()
 
-    def replay_item(self, item_id: int, force: bool = False):
+    def replay_item(
+            self,
+            item_id: int,
+            force: bool = False
+    ):
         """
         Regenerate ctx item
 
@@ -203,7 +207,11 @@ class Extra:
                     self.window.controller.audio.on_stop()
                     self.audio_play_id = None
 
-    def join_item(self, item_id: int, force: bool = False):
+    def join_item(
+            self,
+            item_id: int,
+            force: bool = False
+    ):
         """
         Join ctx items
 

pygpt_net/controller/ctx/summarizer.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.01.25 19:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import QObject, Signal, Slot
@@ -27,7 +27,11 @@ class Summarizer:
         """
         self.window = window
 
-    def summarize(self, id: int, ctx: CtxItem):
+    def summarize(
+            self,
+            id: int,
+            ctx: CtxItem
+    ):
         """
         Summarize context
 
@@ -39,7 +43,13 @@ class Summarizer:
         ctx_copy.from_dict(ctx.to_dict())
         self.start_worker(id, ctx_copy)
 
-    def summarizer(self, id: int, ctx: CtxItem, window, updated_signal: Signal):
+    def summarizer(
+            self,
+            id: int,
+            ctx: CtxItem,
+            window,
+            updated_signal: Signal
+    ):
         """
         Summarize worker callback
 
@@ -52,7 +62,11 @@ class Summarizer:
         if title:
             updated_signal.emit(id, ctx, title)
 
-    def start_worker(self, id: int, ctx: CtxItem):
+    def start_worker(
+            self,
+            id: int,
+            ctx: CtxItem
+    ):
         """
         Handle worker thread
 
@@ -69,7 +83,12 @@ class Summarizer:
         self.window.threadpool.start(worker)
 
     @Slot(int, object, str)
-    def handle_update(self, id: int, ctx: CtxItem, title: str):
+    def handle_update(
+            self,
+            id: int,
+            ctx: CtxItem,
+            title: str
+    ):
         """
         Handle update signal (make update)
 

pygpt_net/controller/debug/__init__.py

@@ -6,15 +6,18 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.20 03:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 from datetime import datetime
 from logging import ERROR, WARNING, INFO, DEBUG
+from typing import Any
 
 from PySide6.QtCore import Slot, QObject
 from PySide6.QtGui import QTextCursor
 
+from pygpt_net.core.events import RenderEvent
+
 
 class Debug(QObject):
     def __init__(self, window=None):
@@ -53,8 +56,17 @@ class Debug(QObject):
 
     def toggle_menu(self):
         """Toggle debug menu"""
-        stage = self.window.core.config.get('debug')
-        self.window.ui.menu['menu.debug'].menuAction().setVisible(stage)
+        state = self.window.core.config.get('debug')
+        self.window.ui.menu['menu.debug'].menuAction().setVisible(state)
+
+    def toggle_render(self):
+        """Toggle render debug"""
+        value = self.window.ui.menu['debug.render'].isChecked()
+        self.window.core.config.set('debug.render', value)
+        self.window.core.config.save()
+        event = RenderEvent(RenderEvent.ON_THEME_CHANGE)
+        self.window.dispatch(event)
+        self.window.controller.ctx.refresh()
 
     def set_log_level(self, level: str = 'error'):
         """
@@ -122,7 +134,12 @@ class Debug(QObject):
             signals.js_message.connect(self.handle_js_message)
 
     @Slot(int, str, str)
-    def handle_js_message(self, line_number: int, message: str, source_id: str):
+    def handle_js_message(
+            self,
+            line_number: int,
+            message: str,
+            source_id: str
+    ):
         """
         Handle JS message
 
@@ -134,7 +151,7 @@ class Debug(QObject):
         self.log(data, window=True)
 
     @Slot(object)
-    def handle_log(self, data: any):
+    def handle_log(self, data: Any):
         """
         Handle log message
 
@@ -142,7 +159,11 @@ class Debug(QObject):
         """
         self.log(data)
 
-    def log(self, data: any, window: bool = True):
+    def log(
+            self,
+            data: Any,
+            window: bool = True
+    ):
         """
         Log message to console or logger window