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

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
     ):
         """

pygpt_net/controller/idx/settings.py

@@ -6,12 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.02.28 22:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import datetime
+from typing import Dict, Any
 
-from PySide6.QtWidgets import QHBoxLayout, QLabel, QMenu
+from PySide6.QtWidgets import QHBoxLayout, QLabel, QMenu, QLayout
 
 from pygpt_net.ui.widget.element.button import ContextMenuButton
 from pygpt_net.ui.widget.element.labels import HelpLabel, TitleLabel
@@ -35,7 +36,12 @@ class Settings:
         """
         return ["update"]
 
-    def append(self, content: dict, widgets: dict, options: dict):
+    def append(
+            self,
+            content: Dict[str, QLayout],
+            widgets: Dict[str, Any],
+            options: Dict[str, Any],
+    ):
         """
         Append extra settings to settings dialog (section: llama-index)
 

pygpt_net/controller/kernel/__init__.py

@@ -6,10 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.21 20:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import time
+from typing import Any, Dict, Optional, Union, List
+
 from PySide6.QtCore import QObject, Slot
 
 from pygpt_net.core.types import (
@@ -117,14 +119,18 @@ class Kernel(QObject):
 
         event.data["response"] = response  # update response
 
-    def input(self, context: BridgeContext, extra: dict, event: KernelEvent) -> any:
+    def input(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any],
+            event: KernelEvent
+    ):
         """
         Input message to kernel
 
         :param context: bridge context
         :param extra: extra data
         :param event: kernel event
-        :return: response
         """
         if self.stopped() and event.name != KernelEvent.INPUT_USER:
             return
@@ -135,7 +141,12 @@ class Kernel(QObject):
         ]:
             return self.window.controller.chat.input.send(context, extra)
 
-    def queue(self, context: BridgeContext, extra: dict, event: KernelEvent) -> any:
+    def queue(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any],
+            event: KernelEvent
+    ) -> Optional[Union[bool, str, List[Dict]]]:
         """
         Queue messages to kernel
 
@@ -170,7 +181,12 @@ class Kernel(QObject):
         elif event.name == KernelEvent.CALL:
             return self.call(context, extra, event)
 
-    def call(self, context: BridgeContext, extra: dict, event: KernelEvent) -> any:
+    def call(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any],
+            event: KernelEvent
+    ) -> Optional[Union[bool, str]]:
         """
         Execute message
 
@@ -189,7 +205,12 @@ class Kernel(QObject):
         elif event.name == KernelEvent.CALL:
             return self.window.core.bridge.call(context, extra)
 
-    def reply(self, context: BridgeContext, extra: dict, event: KernelEvent) -> any:
+    def reply(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any],
+            event: KernelEvent
+    ) -> Optional[List[Dict]]:
         """
         Queue reply message
 
@@ -206,14 +227,18 @@ class Kernel(QObject):
         elif event.name == KernelEvent.REPLY_RETURN:
             return self.input(context, extra, KernelEvent(KernelEvent.INPUT_SYSTEM))
 
-    def output(self, context: BridgeContext, extra: dict, event: KernelEvent) -> any:
+    def output(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any],
+            event: KernelEvent
+    ):
         """
         Handle output from kernel
 
         :param context: bridge context
         :param extra: extra data
         :param event: kernel event
-        :return: response
         """
         if self.stopped():
             return
@@ -251,6 +276,7 @@ class Kernel(QObject):
         """
         self.halt = True
         self.window.controller.chat.common.stop(exit=exit)  # it stops legacy agent also
+        self.window.controller.audio.stop_audio()
         if not exit:
             self.window.dispatch(KernelEvent(KernelEvent.STOP))
             self.set_state(KernelEvent(KernelEvent.STATE_IDLE, {"msg": trans("status.stopped")}))

pygpt_net/controller/kernel/reply.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.25 01:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import json
+from typing import Optional, Dict, Any, List
 
 from pygpt_net.core.events import KernelEvent, RenderEvent
 from pygpt_net.core.bridge import BridgeContext
@@ -28,7 +29,11 @@ class Reply:
         self.last_result = None
         self.reply_idx = -1
 
-    def add(self, context, extra) -> list:
+    def add(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any] = None
+    ) -> List[Dict]:
         """
         Send reply from plugins to model
 
@@ -129,7 +134,11 @@ class Reply:
         })
         self.window.dispatch(event)
 
-    def run_post_response(self, ctx: CtxItem, extra_data: dict = None):
+    def run_post_response(
+            self,
+            ctx: CtxItem,
+            extra_data: Optional[Dict[str, Any]] = None
+    ):
         """
         Run post-response operations