pygpt-net 2.4.37__py3-none-any.whl → 2.4.42__py3-none-any.whl

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
 

pygpt_net/controller/dialogs/confirm.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.26 02:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Optional
+
+
 class Confirm:
     def __init__(self, window=None):
         """
@@ -18,7 +21,12 @@ class Confirm:
         """
         self.window = window
 
-    def accept(self, type: str, id: any = None, parent_object=None):
+    def accept(
+            self,
+            type: str,
+            id: Optional[str] = None,
+            parent_object: Any = None
+    ):
         """
         Confirm dialog accept
 
@@ -154,7 +162,7 @@ class Confirm:
 
         # tab close all
         elif type == 'tab.close_all':
-            self.window.controller.ui.tabs.close_all(id, True)  # by type
+            self.window.controller.ui.tabs.close_all(id, 0, True)  # by type
 
         # editor
         elif type == 'editor.changed.clear':
@@ -264,7 +272,11 @@ class Confirm:
         elif type == 'prompt.custom.delete':
             self.window.controller.presets.delete_prompt(id, True)
 
-    def dismiss(self, type: str, id: any):
+    def dismiss(
+            self,
+            type: str,
+            id: Any
+    ):
         """
         Confirm dialog dismiss
 
@@ -283,7 +295,12 @@ class Confirm:
 
         self.window.ui.dialog['confirm'].close()
 
-    def accept_rename(self, type: str, id: any, name: str):
+    def accept_rename(
+            self,
+            type: str,
+            id: Any,
+            name: str
+    ):
         """
         Update name of object
 
@@ -325,7 +342,12 @@ class Confirm:
         elif type == 'prompt.custom.new':
             self.window.controller.presets.save_prompt(name, True)
 
-    def accept_create(self, type: str, id: any, name: str):
+    def accept_create(
+            self,
+            type: str,
+            id: any,
+            name: str
+    ):
         """
         Create new object
 
@@ -349,7 +371,12 @@ class Confirm:
         elif type == 'ctx.group':
             self.window.controller.ctx.create_group(name, id)
 
-    def accept_url(self, type: str, id: any, url: str):
+    def accept_url(
+            self,
+            type: str,
+            id: any,
+            url: str
+    ):
         """
         Update URL provided
 

pygpt_net/controller/dialogs/debug.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.05 23:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
+from typing import Any
+
 from PySide6.QtCore import Qt
 from PySide6.QtGui import QStandardItemModel
 
@@ -95,7 +97,7 @@ class Debug:
         self.initialized[id] = True
         self.window.ui.debug[id].on_data_end()
 
-    def add(self, id: str, k: str, v: any):
+    def add(self, id: str, k: str, v: Any):
         """
         Append debug entry
 

pygpt_net/controller/dialogs/info.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.19 01:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import webbrowser
@@ -32,7 +32,12 @@ class Info:
     def setup(self):
         pass
 
-    def toggle(self, id: str, width: int = 400, height: int = 400):
+    def toggle(
+            self,
+            id: str,
+            width: int = 400,
+            height: int = 400
+    ):
         """
         Toggle info dialog
 

pygpt_net/controller/files.py

@@ -6,12 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.30 04:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import datetime
 import os
 import shutil
+from typing import Optional
 
 from PySide6.QtCore import QUrl
 from PySide6.QtGui import QDesktopServices
@@ -38,7 +39,11 @@ class Files:
         # TODO: implement this
         pass
 
-    def delete_recursive(self, path: str, force: bool = False):
+    def delete_recursive(
+            self,
+            path: str,
+            force: bool = False
+    ):
         """
         Delete directory with all files
 
@@ -62,7 +67,12 @@ class Files:
             self.window.core.debug.log(e)
             print("Error deleting directory: {} - {}".format(path, e))
 
-    def touch_file(self, path: str, name: str = None, force: bool = False):
+    def touch_file(
+            self,
+            path: str,
+            name: Optional[str] = None,
+            force: bool = False
+    ):
         """
         Touch empty file
 
@@ -90,7 +100,11 @@ class Files:
             self.window.core.debug.log(e)
             print("Error creating file: {} - {}".format(path, e))
 
-    def delete(self, path: str, force: bool = False):
+    def delete(
+            self,
+            path: str,
+            force: bool = False
+    ):
         """
         Delete file or directory
 
@@ -123,7 +137,12 @@ class Files:
                 self.window.core.debug.log(e)
                 print("Error deleting file: {} - {}".format(path, e))
 
-    def duplicate_local(self, path: str, name: str, force: bool = False):
+    def duplicate_local(
+            self,
+            path: str,
+            name: str,
+            force: bool = False
+    ):
         """
         Duplicate file or directory
 
@@ -198,7 +217,10 @@ class Files:
                     self.window.core.debug.log(e)
                     print("Error downloading file: {} - {}".format(path, e))
 
-    def upload_local(self, parent_path: str = None):
+    def upload_local(
+            self,
+            parent_path: Optional[str] = None
+    ):
         """
         Upload local file(s) to directory
 
@@ -263,7 +285,11 @@ class Files:
         self.window.ui.dialog['rename'].show()
         self.window.ui.dialog['rename'].input.setFocus()
 
-    def update_name(self, path: str, name: str):
+    def update_name(
+            self,
+            path: str,
+            name: str
+    ):
         """
         Update name of file or directory
 
@@ -283,7 +309,11 @@ class Files:
         self.window.ui.dialog['rename'].close()
         self.update_explorer()
 
-    def open_dir(self, path: str, select: bool = False):
+    def open_dir(
+            self,
+            path: str,
+            select: bool = False
+    ):
         """
         Open file or directory in file manager
 
@@ -308,7 +338,11 @@ class Files:
             else:
                 subprocess.run(['xdg-open', path])
 
-    def open_in_file_manager(self, path: str, select: bool = False):
+    def open_in_file_manager(
+            self,
+            path: str,
+            select: bool = False
+    ):
         """
         Open in file manager
 
@@ -341,7 +375,11 @@ class Files:
         self.window.ui.dialog['create'].show()
         self.window.ui.dialog['create'].input.setFocus()
 
-    def make_dir(self, path: str, name: str = None):
+    def make_dir(
+            self,
+            path: str,
+            name: Optional[str] = None
+    ):
         """
         Make directory
 

pygpt_net/controller/finder.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.20 06:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any
+
 
 class Finder:
     def __init__(self, window=None):
@@ -86,7 +88,7 @@ class Finder:
         """
         return self.window.ui.nodes['dialog.find.input'].text()
 
-    def set(self, parent):
+    def set(self, parent: Any):
         """
         Set current parent for the finder
 
@@ -94,7 +96,7 @@ class Finder:
         """
         self.parent = parent
 
-    def unset(self, parent):
+    def unset(self, parent: Any):
         """
         Unset parent from the finder
 
@@ -103,7 +105,7 @@ class Finder:
         if self.parent == parent:
             self.parent = None
 
-    def open(self, parent):
+    def open(self, parent: Any):
         """
         Open finder dialog
 
@@ -134,7 +136,11 @@ class Finder:
         self.window.ui.dialog['find'].hide()
         self.opened = False
 
-    def clear(self, restore: bool = False, to_end: bool = True):
+    def clear(
+            self,
+            restore: bool = False,
+            to_end: bool = True
+    ):
         """
         Clear current parent highlights
 

pygpt_net/controller/idx/__init__.py

@@ -6,13 +6,15 @@
 # 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                  #
 # ================================================== #
 
 import datetime
+from typing import Optional
 
 from pygpt_net.item.ctx import CtxItem
 from pygpt_net.utils import trans
+
 from .common import Common
 from .indexer import Indexer
 from .settings import Settings
@@ -227,7 +229,12 @@ class Idx:
                 return True
         return False
 
-    def on_ctx_end(self, ctx: CtxItem = None, mode: str = None, sync: bool = False):
+    def on_ctx_end(
+            self,
+            ctx: Optional[CtxItem] = None,
+            mode: Optional[str] = None,
+            sync: bool = False
+    ):
         """
         After context item updated (request + response received)
 
@@ -255,7 +262,7 @@ class Idx:
                 if meta is not None:
                     self.indexer.index_ctx_realtime(meta, idx, sync=sync)
 
-    def after_index(self, idx: str = None):
+    def after_index(self, idx: Optional[str] = None):
         """
         Called after index (update things, etc...)
 

pygpt_net/controller/idx/common.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.08.25 04:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import List, Dict
+
 
 class Common:
     def __init__(self, window=None):
@@ -23,7 +25,7 @@ class Common:
         """Set up UI"""
         pass
 
-    def get_loaders_choices(self) -> list:
+    def get_loaders_choices(self) -> List[Dict[str, str]]:
         """
         Get available loaders choices
 

pygpt_net/controller/idx/indexer.py

@@ -6,11 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.14 01:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import datetime
 import os
+from typing import Any, List, Dict
 
 from PySide6.QtCore import Slot, QObject
 from PySide6.QtWidgets import QApplication
@@ -218,8 +219,8 @@ class Indexer(QObject):
             self,
             path: str,
             idx: str = "base",
-            replace: bool = None,
-            recursive: bool = None
+            replace: bool = False,
+            recursive: bool = False
     ):
         """
         Index all files in path (threaded)
@@ -228,7 +229,6 @@ class Indexer(QObject):
         :param idx: index name
         :param replace: replace index
         :param recursive: recursive indexing
-        :param silent: silent mode
         """
         self.window.update_status(trans('idx.status.indexing'))
         self.worker = IndexWorker()
@@ -245,10 +245,10 @@ class Indexer(QObject):
 
     def index_paths(
             self,
-            paths: list,
+            paths: List[str],
             idx: str = "base",
-            replace: bool = None,
-            recursive: bool = None
+            replace: bool = False,
+            recursive: bool = False
     ):
         """
         Index all files in path (threaded)
@@ -380,8 +380,8 @@ class Indexer(QObject):
             self,
             idx: str,
             loader: str,
-            params: dict,
-            config: dict,
+            params: Dict[str, Any],
+            config: Dict[str, Any],
             replace: bool = True,
     ):
         """
@@ -456,7 +456,11 @@ class Indexer(QObject):
         idx_name = self.window.core.idx.get_by_idx(idx)
         self.clear(idx_name)
 
-    def clear(self, idx: str, force: bool = False):
+    def clear(
+            self,
+            idx: str,
+            force: bool = False
+    ):
         """
         Clear index data
 
@@ -497,7 +501,11 @@ class Indexer(QObject):
 
         self.window.tools.get("indexer").refresh()
 
-    def truncate(self, idx: str, force: bool = False):
+    def truncate(
+            self,
+            idx: str,
+            force: bool = False
+    ):
         """
         Truncate index data
 
@@ -542,7 +550,7 @@ class Indexer(QObject):
         self.window.tools.get("indexer").refresh()
 
     @Slot(object)
-    def handle_error(self, err: any):
+    def handle_error(self, err: Any):
         """
         Handle thread error signal
 
@@ -560,7 +568,7 @@ class Indexer(QObject):
             self,
             idx: str,
             num: int,
-            errors: list,
+            errors: List[str],
             silent: bool = False
     ):
         """
@@ -600,7 +608,7 @@ class Indexer(QObject):
             self,
             idx: str,
             num: int,
-            errors: list,
+            errors: List[str],
             silent: bool = False
     ):
         """
@@ -631,8 +639,8 @@ class Indexer(QObject):
     def handle_finished_file(
             self,
             idx: str,
-            files: dict,
-            errors: list,
+            files: Dict[str, str],
+            errors: List[str],
             silent: bool = False
     ):
         """
@@ -667,7 +675,7 @@ class Indexer(QObject):
             self,
             idx: str,
             num: int,
-            errors: list,
+            errors: List[str],
             silent: bool = False
     ):
         """