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

pygpt_net/controller/config/placeholder.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.21 21:00:00                  #
+# Updated Date: 2024.12.14 18:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any, List
+
 from pygpt_net.core.types import (
     MODE_AGENT_LLAMA,
     MODE_LANGCHAIN,
@@ -26,7 +28,7 @@ class Placeholder:
         """
         self.window = window
 
-    def apply(self, option: dict):
+    def apply(self, option: Dict[str, Any]):
         """
         Apply placeholders to option
 
@@ -52,7 +54,7 @@ class Placeholder:
             if "use" in option and option["use"] is not None:
                 option["keys"] = self.apply_by_id(option["use"])
 
-    def apply_by_id(self, id: str) -> list:
+    def apply_by_id(self, id: str) -> List[Dict[str, str]]:
         """
         Apply placeholders by id
 
@@ -86,6 +88,8 @@ class Placeholder:
             return self.get_agent_providers()
         elif id == "syntax_styles":
             return self.get_syntax_styles()
+        elif id == "styles":
+            return self.get_styles()
         elif id == "idx":
             return self.get_idx()
         elif id == "keys":
@@ -98,10 +102,38 @@ class Placeholder:
             return self.get_speech_synthesis_actions()
         elif id == "voice_control_actions":
             return self.get_voice_control_actions()
+        elif id == "audio_input_devices":
+            return self.get_audio_input_devices()
+        elif id == "audio_tts_whisper_voices":
+            return self.get_audio_tts_whisper_voices()
         else:
             return []
 
-    def get_langchain_providers(self) -> list:
+    def get_audio_tts_whisper_voices(self) -> List[Dict[str, str]]:
+        """
+        Get audio TTS whisper voices list
+
+        :return: placeholders list
+        """
+        voices = self.window.core.audio.whisper.get_voices()
+        data = []
+        for voice in voices:
+            data.append({voice: voice})
+        return data
+
+    def get_audio_input_devices(self) -> List[Dict[str, str]]:
+        """
+        Get audio input devices list
+
+        :return: placeholders list
+        """
+        devices = self.window.core.audio.get_input_devices()
+        data = []
+        for device in devices:
+            data.append({str(device[0]): device[1]})
+        return data
+
+    def get_langchain_providers(self) -> List[Dict[str, str]]:
         """
         Get Langchain LLM provider placeholders list
 
@@ -113,7 +145,7 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_llama_index_providers(self) -> list:
+    def get_llama_index_providers(self) -> List[Dict[str, str]]:
         """
         Get Llama-index LLM provider placeholders list
 
@@ -125,7 +157,7 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_agent_providers(self) -> list:
+    def get_agent_providers(self) -> List[Dict[str, str]]:
         """
         Get Llama-index agent provider placeholders list
 
@@ -137,7 +169,7 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_embeddings_providers(self) -> list:
+    def get_embeddings_providers(self) -> List[Dict[str, str]]:
         """
         Get embeddings placeholders list
 
@@ -149,7 +181,7 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_llama_index_chat_modes(self) -> list:
+    def get_llama_index_chat_modes(self) -> List[Dict[str, str]]:
         """
         Get llama chat modes list
 
@@ -165,7 +197,7 @@ class Placeholder:
             {"openai": "openai"},
         ]
 
-    def get_llama_index_loaders(self, type: str = "all") -> list:
+    def get_llama_index_loaders(self, type: str = "all") -> List[Dict[str, str]]:
         """
         Get data loaders placeholders list
 
@@ -187,7 +219,7 @@ class Placeholder:
                     data.append(choice)
         return data
 
-    def get_vector_storage(self) -> list:
+    def get_vector_storage(self) -> List[Dict[str, str]]:
         """
         Get vector storage placeholders list
 
@@ -199,7 +231,7 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_var_types(self) -> list:
+    def get_var_types(self) -> List[Dict[str, str]]:
         """
         Get langchain placeholders list
 
@@ -211,7 +243,7 @@ class Placeholder:
             data.append({type: type})
         return data
 
-    def get_presets(self) -> list:
+    def get_presets(self) -> List[Dict[str, str]]:
         """
         Get presets placeholders list
 
@@ -226,7 +258,7 @@ class Placeholder:
             data.append({id: id})  # TODO: name
         return data
 
-    def get_models(self) -> list:
+    def get_models(self) -> List[Dict[str, str]]:
         """
         Get models placeholders list
 
@@ -238,7 +270,7 @@ class Placeholder:
             data.append({id: id})  # TODO: name
         return data
 
-    def get_agent_modes(self) -> list:
+    def get_agent_modes(self) -> List[Dict[str, str]]:
         """
         Get agent/expert modes placeholders list
 
@@ -250,7 +282,7 @@ class Placeholder:
             data.append({id: id})  # TODO: name
         return data
 
-    def get_idx(self) -> list:
+    def get_idx(self) -> List[Dict[str, str]]:
         """
         Get indexes placeholders list
 
@@ -263,7 +295,7 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_syntax_styles(self) -> list:
+    def get_syntax_styles(self) -> List[Dict[str, str]]:
         """
         Get highlighter styles list
 
@@ -276,7 +308,20 @@ class Placeholder:
             data.append({id: id})
         return data
 
-    def get_keys(self) -> list:
+    def get_styles(self) -> List[Dict[str, str]]:
+        """
+        Get styles list
+
+        :return: placeholders list
+        """
+        styles = self.window.controller.theme.common.get_styles_list()
+        styles.sort()
+        data = []
+        for id in styles:
+            data.append({id: id})
+        return data
+
+    def get_keys(self) -> List[Dict[str, str]]:
         """
         Get keys
 
@@ -284,7 +329,7 @@ class Placeholder:
         """
         return self.window.core.access.shortcuts.get_keys_choices()
 
-    def get_modifiers(self) -> list:
+    def get_modifiers(self) -> List[Dict[str, str]]:
         """
         Get modifiers
 
@@ -292,7 +337,7 @@ class Placeholder:
         """
         return self.window.core.access.shortcuts.get_modifiers_choices()
 
-    def get_access_actions(self) -> list:
+    def get_access_actions(self) -> List[Dict[str, str]]:
         """
         Get access actions list
 
@@ -307,7 +352,7 @@ class Placeholder:
         translated_choices.sort(key=lambda x: list(x.values())[0])
         return translated_choices
 
-    def get_speech_synthesis_actions(self) -> list:
+    def get_speech_synthesis_actions(self) -> List[Dict[str, str]]:
         """
         Get speech actions list
 
@@ -322,7 +367,7 @@ class Placeholder:
         translated_choices.sort(key=lambda x: list(x.values())[0])
         return translated_choices
 
-    def get_voice_control_actions(self) -> list:
+    def get_voice_control_actions(self) -> List[Dict[str, str]]:
         """
         Get voice control actions list
 

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