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

pygpt_net/core/modes.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.26 19:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict, List
+
 from pygpt_net.provider.core.mode.json_file import JsonFileProvider
 from pygpt_net.core.types import (
     MODE_AGENT,
@@ -51,7 +53,7 @@ class Modes:
         ]
         self.items = {}
 
-    def get_by_idx(self, idx):
+    def get_by_idx(self, idx) -> str:
         """
         Return mode by index
 
@@ -61,7 +63,7 @@ class Modes:
         modes = self.get_all()
         return list(modes.keys())[idx]
 
-    def get_idx_by_name(self, name):
+    def get_idx_by_name(self, name) -> int:
         """
         Return mode index by name
 
@@ -71,7 +73,7 @@ class Modes:
         modes = self.get_all()
         return list(modes.keys()).index(name)
 
-    def get_all(self):
+    def get_all(self) -> Dict[str, List[str]]:
         """
         Return modes
 
@@ -79,7 +81,7 @@ class Modes:
         """
         return self.items
 
-    def get_default(self):
+    def get_default(self) -> str:
         """
         Return default mode name
 
@@ -89,7 +91,7 @@ class Modes:
             if self.items[id].default:
                 return id
 
-    def get_next(self, mode: str):
+    def get_next(self, mode: str) -> str:
         """
         Return next mode
 
@@ -103,7 +105,7 @@ class Modes:
             return keys[idx + 1]
         return keys[0]
 
-    def get_prev(self, mode: str):
+    def get_prev(self, mode: str) -> str:
         """
         Return previous mode
 
@@ -118,13 +120,9 @@ class Modes:
         return keys[-1]
 
     def load(self):
-        """
-        Load modes
-        """
+        """Load modes"""
         self.items = self.provider.load()
 
     def save(self):
-        """
-        Save modes
-        """
+        """Save modes"""
         self.provider.save(self.items)

pygpt_net/core/notepad.py

@@ -6,11 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.09 23:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import datetime
 import uuid
+from typing import Optional, Dict
 
 from packaging.version import Version
 
@@ -49,7 +50,7 @@ class Notepad:
         """Reset provider data"""
         self.items = {}
 
-    def get_by_id(self, idx: int) -> NotepadItem or None:
+    def get_by_id(self, idx: int) -> Optional[NotepadItem]:
         """
         Get notepad by idx
 
@@ -60,7 +61,7 @@ class Notepad:
             return self.items[idx]
         return None
 
-    def get_all(self) -> dict:
+    def get_all(self) -> Dict[int, NotepadItem]:
         """
         Get all notepads
 
@@ -136,8 +137,12 @@ class Notepad:
         """Save all notepads"""
         self.provider.save_all(self.items)
 
-    def import_from_db(self) -> dict or None:
-        """Import notepad tabs from database"""
+    def import_from_db(self) -> Optional[Dict[int, dict]]:
+        """
+        Import notepad tabs from database
+
+        :return: dict with tabs
+        """
         self.load_all()
         items = self.get_all()
         if len(items) == 0:

pygpt_net/core/plugins.py

@@ -6,13 +6,14 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.18 21:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import copy
 import configparser
 import io
 import os
+from typing import Optional, Dict, List, Any
 
 from pygpt_net.provider.core.plugin_preset.json_file import JsonFileProvider
 from pygpt_net.plugin.base.plugin import BasePlugin
@@ -48,7 +49,7 @@ class Plugins:
         """
         return id in self.plugins
 
-    def all(self) -> dict:
+    def all(self) -> Dict[str, BasePlugin]:
         """
         Get all plugins
 
@@ -56,7 +57,7 @@ class Plugins:
         """
         return self.plugins
 
-    def get_ids(self) -> list:
+    def get_ids(self) -> List[str]:
         """
         Get all plugins ids
 
@@ -64,7 +65,7 @@ class Plugins:
         """
         return list(self.plugins.keys())
 
-    def get(self, id: str) -> BasePlugin or None:
+    def get(self, id: str) -> Optional[BasePlugin]:
         """
         Get plugin by id
 
@@ -75,7 +76,7 @@ class Plugins:
             return self.plugins[id]
         return None
 
-    def get_option(self, id: str, key: str) -> any:
+    def get_option(self, id: str, key: str) -> Any:
         """
         Get plugin option
 
@@ -142,7 +143,7 @@ class Plugins:
         if removed:
             self.window.core.config.save()
 
-    def register_options(self, id: str, options: dict):
+    def register_options(self, id: str, options: Dict[str, dict]):
         """
         Register plugin options
 
@@ -171,7 +172,7 @@ class Plugins:
 
     def enable(self, id: str):
         """
-        Enable plugin
+        Enable plugin by ID
 
         :param id: plugin id
         """
@@ -182,7 +183,7 @@ class Plugins:
 
     def disable(self, id: str):
         """
-        Disable plugin
+        Disable plugin by ID
 
         :param id: plugin id
         """
@@ -276,7 +277,7 @@ class Plugins:
             tooltip = trans('plugin.description', domain=domain)
         return tooltip
 
-    def dump_locale(self, plugin, path: str):
+    def dump_locale(self, plugin: BasePlugin, path: str):
         """
         Dump locale
 
@@ -317,7 +318,7 @@ class Plugins:
         """
         return id in self.presets
 
-    def get_preset(self, id: str) -> dict:
+    def get_preset(self, id: str) -> Dict[str, Any]:
         """
         Get preset by id
 
@@ -327,7 +328,7 @@ class Plugins:
         if self.has_preset(id):
             return self.presets[id]
 
-    def set_preset(self, id: str, preset: dict):
+    def set_preset(self, id: str, preset: Dict[str, Any]):
         """
         Set config value
 
@@ -336,7 +337,7 @@ class Plugins:
         """
         self.presets[id] = preset
 
-    def replace_presets(self, presets: dict):
+    def replace_presets(self, presets: Dict[str, Any]):
         """
         Replace presets
 
@@ -348,7 +349,7 @@ class Plugins:
         """Load presets"""
         self.presets = self.provider.load()
 
-    def get_presets(self) -> dict:
+    def get_presets(self) -> Dict[str, Any]:
         """
         Return all presets
 
@@ -356,7 +357,11 @@ class Plugins:
         """
         return self.presets
 
-    def reset_options(self, plugin_id: str, keys: list):
+    def reset_options(
+            self,
+            plugin_id: str,
+            keys: List[str]
+    ):
         """
         Reset plugin options
 
@@ -407,7 +412,11 @@ class Plugins:
             self.save_presets()
             print("[FIX] Removed invalid keys from plugin presets.")
 
-    def remove_preset_values(self, plugin_id:str, key: str):
+    def remove_preset_values(
+            self,
+            plugin_id: str,
+            key: str
+    ):
         """
         Update preset value
 
@@ -430,7 +439,12 @@ class Plugins:
         if updated:
             self.save_presets()
 
-    def update_preset_values(self, plugin_id:str, key: str, value: any):
+    def update_preset_values(
+            self,
+            plugin_id: str,
+            key: str,
+            value: Any
+    ):
         """
         Update preset value
 
@@ -470,9 +484,7 @@ class Plugins:
             self.dump_locale(self.plugins[id], path)
 
     def dump_locales(self):
-        """
-        Dump all locales
-        """
+        """Dump all locales"""
         langs = ['en', 'pl']
         for id in self.plugins:
             domain = 'plugin.' + id

pygpt_net/core/presets.py

@@ -6,11 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.26 19:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import copy
 import uuid
+from typing import Optional, Tuple, Dict
 
 from packaging.version import Version
 from pygpt_net.core.types import (
@@ -47,7 +48,11 @@ class Presets:
         self.provider.install()
 
     def patch(self, app_version: Version):
-        """Patch provider data"""
+        """
+        Patch provider data
+
+        :param app_version: app version
+        """
         self.provider.patch(app_version)
 
     def build(self) -> PresetItem:
@@ -188,6 +193,7 @@ class Presets:
         """
         if id in self.items:
             self.items[id].enabled = True
+
     def disable(self, id: str):
         """
         Disable preset
@@ -197,7 +203,7 @@ class Presets:
         if id in self.items:
             self.items[id].enabled = False
 
-    def get_first_mode(self, id: str) -> str or None:
+    def get_first_mode(self, id: str) -> Optional[str]:
         """
         Return first mode for preset
 
@@ -253,7 +259,7 @@ class Presets:
         presets = self.get_by_mode(mode)
         return list(presets.keys())[idx]
 
-    def get_by_id(self, mode: str, id: str) -> PresetItem or None:
+    def get_by_id(self, mode: str, id: str) -> Optional[PresetItem]:
         """
         Return preset by id
 
@@ -266,7 +272,7 @@ class Presets:
             return presets[id]
         return None
 
-    def get_by_uuid(self, uuid: str) -> PresetItem or None:
+    def get_by_uuid(self, uuid: str) -> Optional[PresetItem]:
         """
         Return preset by UUID
 
@@ -278,7 +284,7 @@ class Presets:
                 return self.items[id]
         return None
 
-    def get_by_mode(self, mode: str) -> dict:
+    def get_by_mode(self, mode: str) -> Dict[str, PresetItem]:
         """
         Return presets for mode
 
@@ -317,7 +323,7 @@ class Presets:
             i += 1
         return 0
 
-    def get_default(self, mode: str) -> str or None:
+    def get_default(self, mode: str) -> Optional[str]:
         """
         Return default preset for mode
 
@@ -329,7 +335,7 @@ class Presets:
             return None
         return list(presets.keys())[0]
 
-    def get_duplicate_name(self, id: str) -> (str, str):
+    def get_duplicate_name(self, id: str) -> Tuple[str, str]:
         """
         Prepare name for duplicated preset
 
@@ -361,7 +367,11 @@ class Presets:
         self.sort_by_name()
         return id
 
-    def remove(self, id: str, remove_file: bool = True):
+    def remove(
+            self,
+            id: str,
+            remove_file: bool = True
+    ):
         """
         Delete preset
 
@@ -376,9 +386,7 @@ class Presets:
             # self.load_presets()
 
     def sort_by_name(self):
-        """
-        Sort presets by name
-        """
+        """Sort presets by name"""
         self.items = dict(
             sorted(
                 self.items.items(),
@@ -403,7 +411,7 @@ class Presets:
         self.items[preset.filename] = preset
         self.save(preset.filename)
 
-    def get_all(self) -> dict:
+    def get_all(self) -> Dict[str, PresetItem]:
         """
         Return all presets
 
@@ -423,8 +431,12 @@ class Presets:
             self.items[id] = base[id]
             self.save(id)
 
-    def load_base(self):
-        """Load base presets"""
+    def load_base(self) -> Dict[str, PresetItem]:
+        """
+        Load base presets
+
+        :return: base presets
+        """
         return self.provider.load_base()
 
     def load(self):
@@ -451,7 +463,11 @@ class Presets:
         """Save all presets"""
         self.provider.save_all(self.items)
 
-    def add_expert(self, agent_uuid: str, expert_uuid: str):
+    def add_expert(
+            self,
+            agent_uuid: str,
+            expert_uuid: str
+    ):
         """
         Add expert to agent
 
@@ -465,7 +481,11 @@ class Presets:
             agent.experts.append(expert_uuid)
         self.save(agent.filename)
 
-    def remove_expert(self, agent_uuid: str, expert_uuid: str):
+    def remove_expert(
+            self,
+            agent_uuid: str,
+            expert_uuid: str
+    ):
         """
         Remove expert from agent
         :param agent_uuid: agent uuid

pygpt_net/core/profile.py

@@ -6,12 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.09 23:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import json
 import os
 from pathlib import Path
+from typing import Optional, Dict
 from uuid import uuid4
 
 
@@ -44,7 +45,7 @@ class Profile:
             self.load()
             self.update_path()
 
-    def set_base_workdir(self, workdir):
+    def set_base_workdir(self, workdir: str):
         """
         Set base workdir
 
@@ -119,7 +120,11 @@ class Profile:
         with open(f, 'w', encoding='utf-8') as f:
             f.write(json_data)
 
-    def add(self, name: str, workdir: str) -> str:
+    def add(
+            self,
+            name: str,
+            workdir: str
+    ) -> str:
         """
         Add new profile
 
@@ -162,7 +167,12 @@ class Profile:
             return True
         return False
 
-    def update_profile(self, uuid: str, name: str, workdir: str) -> bool:
+    def update_profile(
+            self,
+            uuid: str,
+            name: str,
+            workdir: str
+    ) -> bool:
         """
         Update profile
 
@@ -188,7 +198,7 @@ class Profile:
         """
         return self.current
 
-    def get(self, uuid: str) -> dict or None:
+    def get(self, uuid: str) -> Optional[dict]:
         """
         Get profile by uuid
 
@@ -199,7 +209,7 @@ class Profile:
             return self.profiles[uuid]
         return None
 
-    def get_all(self) -> dict:
+    def get_all(self) -> Dict[str, Dict]:
         """
         Get all profiles
 
@@ -222,7 +232,11 @@ class Profile:
                 return tmp_dir
         return self.base_workdir
 
-    def append(self, uuid: str, profile: dict):
+    def append(
+            self,
+            uuid: str,
+            profile: Dict
+    ):
         """
         Append profile
 

pygpt_net/core/prompt/__init__.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 pygpt_net.core.events import Event
@@ -41,6 +41,12 @@ class Prompt:
         return ""
 
     def build_final_system_prompt(self, prompt: str) -> str:
+        """
+        Build final system prompt
+
+        :param prompt: prompt
+        :return: final system prompt
+        """
         # tmp dispatch event: system prompt
         event = Event(Event.SYSTEM_PROMPT, {
             'mode': self.window.core.config.get('mode'),
@@ -50,7 +56,8 @@ class Prompt:
         self.window.dispatch(event)
         prompt = event.data['value']
 
-        if self.window.core.config.get('cmd') or self.window.controller.plugins.is_type_enabled("cmd.inline"):
+        if (self.window.core.config.get('cmd')
+                or self.window.controller.plugins.is_type_enabled("cmd.inline")):
 
             # abort if native func call enabled
             if self.window.core.command.is_native_enabled():
@@ -88,7 +95,7 @@ class Prompt:
             internal: bool,
             is_expert: bool = False,
             disable_native_tool_calls: bool = False,
-    ):
+    ) -> str:
         """
         Prepare system prompt
 

pygpt_net/core/prompt/custom.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.08.29 05:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
 import copy
+from typing import Optional
 
 from PySide6.QtGui import QAction, QIcon
 from packaging.version import Version
@@ -70,7 +71,7 @@ class Custom:
         """
         return list(self.items.keys())
 
-    def get_id_by_idx(self, idx: int) -> str or None:
+    def get_id_by_idx(self, idx: int) -> Optional[str]:
         """
         Get ID by index
 
@@ -83,7 +84,7 @@ class Custom:
                 return id
             i += 1
 
-    def get_by_id(self, id: str) -> PromptItem or None:
+    def get_by_id(self, id: str) -> Optional[PromptItem]:
         """
         Return prompt by ID
 
@@ -93,7 +94,7 @@ class Custom:
         if id in self.items:
             return self.items[id]
 
-    def get_by_idx(self, idx: int) -> PromptItem or None:
+    def get_by_idx(self, idx: int) -> Optional[PromptItem]:
         """
         Return item by index
 
@@ -114,8 +115,8 @@ class Custom:
 
     def new(
             self,
-            name: str = None,
-            content: str = None,
+            name: Optional[str] = None,
+            content: Optional[str] = None,
             auto_save: bool = True
     ) -> PromptItem:
         """

pygpt_net/core/prompt/template.py

@@ -6,11 +6,12 @@
 # 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 00:00:00                  #
 # ================================================== #
 
 import os
 import csv
+from typing import Optional
 
 from PySide6.QtGui import QAction
 from pygpt_net.utils import trans
@@ -62,7 +63,12 @@ class Template:
         self.prompts = dict(sorted(self.prompts.items(), key=lambda item: item[1]['name']))
 
     def to_menu_options(self, menu, parent: str = "global"):
-        """Convert prompts to menu options"""
+        """
+        Convert prompts to menu options
+
+        :param menu: Menu
+        :param parent: Parent menu
+        """
         self.init()
         menu.addSeparator()
         submenu = menu.addMenu(trans("preset.prompt.paste_template"))
@@ -82,7 +88,7 @@ class Template:
             action.setToolTip(value['prompt'])
             letter_submenus[letter].addAction(action)
 
-    def get_by_id(self, id: int) -> dict or None:
+    def get_by_id(self, id: int) -> Optional[dict]:
         """
         Get prompt by id