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

pygpt_net/controller/ui/mode.py

@@ -6,7 +6,7 @@
 # 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 19:00:00                  #
 # ================================================== #
 
 from pygpt_net.core.types import (
@@ -189,8 +189,12 @@ class Mode:
                 self.window.ui.tabs['input'].setTabVisible(2, False)
                 self.window.ui.tabs['input'].setTabVisible(3, False)
 
-        # toggle chat footer
-        if not self.is_chat_tab():
+        # chat footer
+        self.toggle_chat_footer()
+
+    def toggle_chat_footer(self):
+        """Toggle chat footer"""
+        if self.window.controller.ui.tabs.get_current_type() != Tab.TAB_CHAT:
             self.hide_chat_footer()
         else:
             self.show_chat_footer()
@@ -200,18 +204,12 @@ class Mode:
         Check if vision is allowed
 
         :param mode: current mode
+        :return: True if vision is allowed
         """
-        if mode == MODE_VISION:
-            return True
-
-        allowed = self.window.controller.painter.is_active()
-        if allowed:
+        if self.window.controller.ui.vision.has_vision():
             return True
 
-        if mode in [MODE_IMAGE, MODE_ASSISTANT]:
-            return False
-
-        if self.window.controller.ui.vision.is_vision_model():
+        if self.window.controller.painter.is_active():
             return True
 
         # event: UI: vision
@@ -224,7 +222,12 @@ class Mode:
         return event.data['value']
 
     def are_attachments(self, mode: str) -> bool:
-        """Check if attachments are allowed"""
+        """
+        Check if attachments are allowed
+
+        :param mode: current mode
+        :return: True if attachments are allowed
+        """
         if mode in [MODE_VISION, MODE_ASSISTANT]:
             return True
 
@@ -243,14 +246,6 @@ class Mode:
         self.window.dispatch(event)
         return event.data['value']
 
-    def is_chat_tab(self) -> bool:
-        """
-        Check if current tab is chat
-
-        :return: True if chat tab
-        """
-        return self.window.controller.ui.tabs.get_current_type() == Tab.TAB_CHAT
-
     def show_chat_footer(self):
         """Show chat footer"""
         self.window.ui.nodes['chat.footer'].setVisible(True)

pygpt_net/core/audio/__init__.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.14 08:00:00                  #
+# Updated Date: 2024.12.14 18:00:00                  #
 # ================================================== #
 
 import re
@@ -17,6 +17,8 @@ from bs4 import UnicodeDammit
 from pygpt_net.provider.audio_input.base import BaseProvider as InputBaseProvider
 from pygpt_net.provider.audio_output.base import BaseProvider as OutputBaseProvider
 
+from .whisper import Whisper
+
 
 class Audio:
     def __init__(self, window=None):
@@ -26,6 +28,7 @@ class Audio:
         :param window: Window instance
         """
         self.window = window
+        self.whisper = Whisper(window)
         self.providers = {
             "input": {},
             "output": {},

pygpt_net/core/audio/whisper.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.12.14 18:00:00                  #
+# ================================================== #
+
+from typing import List
+
+class Whisper:
+    def __init__(self, window=None):
+        """
+        Whisper core
+
+        :param window: Window instance
+        """
+        self.window = window
+        self.voices = [
+            "alloy",
+            "echo",
+            "fable",
+            "onyx",
+            "nova",
+            "shimmer",
+        ]
+
+    def get_voices(self) -> List[str]:
+        """
+        Get whisper voices
+
+        :return: whisper voice name
+        """
+        return self.voices

pygpt_net/core/bridge/worker.py

@@ -60,14 +60,14 @@ class BridgeWorker(QObject, QRunnable):
                     extra=self.extra,
                 )
 
-            # Llama-index: chat with files
+            # LlamaIndex: chat with files
             elif self.mode == MODE_LLAMA_INDEX:
                 result = self.window.core.idx.chat.call(
                     context=self.context,
                     extra=self.extra,
                 )
 
-            # Llama-index: agents
+            # LlamaIndex: agents
             elif self.mode == MODE_AGENT_LLAMA:
                 result = self.window.core.agents.runner.call(
                     context=self.context,

pygpt_net/core/debug/events.py

@@ -6,10 +6,17 @@
 # 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 19:00:00                  #
 # ================================================== #
 
-from pygpt_net.core.events import ControlEvent, AppEvent, Event, KernelEvent, RenderEvent
+from pygpt_net.core.events import (
+    ControlEvent,
+    AppEvent,
+    Event,
+    KernelEvent,
+    RenderEvent,
+    BaseEvent
+)
 
 
 class EventsDebug:
@@ -25,19 +32,24 @@ class EventsDebug:
     def update(self):
         """Update debug window."""
         self.window.core.debug.begin(self.id)
-        self.window.core.debug.add(self.id, 'App Events:', str(AppEvent.__dict__))
-        self.window.core.debug.add(self.id, 'Control Events:', str(ControlEvent.__dict__))
-        self.window.core.debug.add(self.id, 'Kernel Events:', str(KernelEvent.__dict__))
-        self.window.core.debug.add(self.id, 'Render Events:', str(RenderEvent.__dict__))
-        self.window.core.debug.add(self.id, 'Plugin Events:', str(self.extract_plugins()))
+        self.window.core.debug.add(self.id, 'App Events:', str(self.extract_events(AppEvent)))
+        self.window.core.debug.add(self.id, 'Control Events:', str(self.extract_events(ControlEvent)))
+        self.window.core.debug.add(self.id, 'Kernel Events:', str(self.extract_events(KernelEvent)))
+        self.window.core.debug.add(self.id, 'Render Events:', str(self.extract_events(RenderEvent)))
+        self.window.core.debug.add(self.id, 'Plugin Events:', str(self.extract_events(Event)))
+        self.window.core.debug.add(self.id, '----', '')
         self.window.core.debug.add(self.id, 'Voice Cmds (all):', str(self.window.core.access.voice.commands))
         self.window.core.debug.add(self.id, 'Voice Cmds (allowed):', str(self.window.core.access.voice.get_commands()))
         self.window.core.debug.end(self.id)
 
-    def extract_plugins(self) -> dict:
-        """Extract plugin events"""
+    def extract_events(self, events: BaseEvent) -> dict:
+        """
+        Extract events
+
+        :param events: Events class
+        """
         result = {}
-        for property, value in vars(Event).items():
+        for property, value in vars(events).items():
             if (isinstance(value, str)
                     or isinstance(value, int)
                     or isinstance(value, float)

pygpt_net/core/debug/tabs.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.12 01:00:00                  #
+# Updated Date: 2024.12.14 17:00:00                  #
 # ================================================== #
 
 class TabsDebug:
@@ -22,10 +22,12 @@ class TabsDebug:
     def update(self):
         """Update debug window."""
         self.window.core.debug.begin(self.id)
-        self.window.core.debug.add(self.id, 'current PID', str(self.window.controller.ui.tabs.get_current_pid()))
+        self.window.core.debug.add(self.id, 'current Col', str(self.window.controller.ui.tabs.get_current_column_idx()))
         self.window.core.debug.add(self.id, 'current IDX', str(self.window.controller.ui.tabs.get_current_idx()))
+        self.window.core.debug.add(self.id, 'current Tab', str(self.window.controller.ui.tabs.get_current_tab()))
+        self.window.core.debug.add(self.id, 'current PID', str(self.window.controller.ui.tabs.get_current_pid()))
         self.window.core.debug.add(self.id, 'current Type', str(self.window.controller.ui.tabs.get_current_type()))
-        self.window.core.debug.add(self.id, 'current Column', str(self.window.controller.ui.tabs.get_current_column_idx()))
+        self.window.core.debug.add(self.id, '----', '')
         self.window.core.debug.add(self.id, 'last_pid', str(self.window.core.tabs.last_pid))
         self.window.core.debug.add(self.id, 'locked', str(self.window.controller.ui.tabs.locked))
         self.window.core.debug.add(self.id, 'col', str(self.window.controller.ui.tabs.col))
@@ -40,6 +42,7 @@ class TabsDebug:
             self.window.core.debug.add(self.id, "PID ["+str(pid)+"]", str(data))
 
         # mapping PID => meta.id
+        self.window.core.debug.add(self.id, '----', '')
         self.window.core.debug.add(self.id, 'PID => meta.id', str(self.window.core.ctx.output.mapping))
         self.window.core.debug.add(self.id, '(last) meta.id => PID', str(self.window.core.ctx.output.last_pids))
         self.window.core.debug.add(self.id, '(last) PID', str(self.window.core.ctx.output.last_pid))

pygpt_net/core/history.py

@@ -6,11 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.01.19 05:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import datetime
 import os
+from typing import List
 
 from pygpt_net.item.ctx import CtxItem
 from pygpt_net.provider.core.history.txt_file import TxtFileProvider
@@ -46,7 +47,7 @@ class History:
         # delete all txt history files from history dir
         self.provider.truncate()
 
-    def remove_items(self, items: list):
+    def remove_items(self, items: List[CtxItem]):
         """
         Remove items from history (txt files)
 

pygpt_net/core/idx/__init__.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.14 08:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import datetime
@@ -406,7 +406,11 @@ class Idx:
             return self.items[store_id]
         return {}
 
-    def append(self, idx: str, files: Dict[str, str]):
+    def append(
+            self,
+            idx: str,
+            files: Dict[str, str]
+    ):
         """
         Append indexed files to index
 
@@ -454,7 +458,11 @@ class Idx:
                 self.items[store_id][idx].items[file_id]["id"] = doc_id
                 self.items[store_id][idx].items[file_id]["indexed_ts"] = ts
 
-    def remove_doc(self, idx: str, doc_id: str):
+    def remove_doc(
+            self,
+            idx: str,
+            doc_id: str
+    ):
         """
         Remove document from index
 
@@ -465,7 +473,11 @@ class Idx:
         if self.storage.remove_document(idx, doc_id):
             self.log("Removed document from index: " + idx + " - " + doc_id)
 
-    def remove_file(self, idx: str, file: str):
+    def remove_file(
+            self,
+            idx: str,
+            file: str
+    ):
         """
         Remove file from index
 

pygpt_net/core/idx/chat.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.14 08:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import json
@@ -360,7 +360,7 @@ class Chat:
             ctx: CtxItem,
             type: str,
             url: str,
-            args: dict,
+            args: Dict[str, Any],
             query: str,
             model: Optional[ModelItem] = None
     ) -> str:
@@ -552,7 +552,11 @@ class Chat:
         ]
         return ChatPromptTemplate(qa_msgs)
 
-    def get_index(self, idx: str, model: ModelItem):
+    def get_index(
+            self,
+            idx: str,
+            model: ModelItem
+    ):
         """
         Get index instance
 
@@ -573,7 +577,10 @@ class Chat:
         index = self.storage.get(idx, service_context=service_context)  # get index
         return index, service_context
 
-    def get_metadata(self, source_nodes: Optional[list]) -> Dict[str, Any]:
+    def get_metadata(
+            self,
+            source_nodes: Optional[list]
+    ) -> Dict[str, Any]:
         """
         Get metadata from source nodes
 
@@ -601,7 +608,10 @@ class Chat:
                             break
         return metadata
 
-    def parse_metadata(self, metadata: Optional[Dict]) -> Dict[str, Any]:
+    def parse_metadata(
+            self,
+            metadata: Optional[Dict]
+    ) -> Dict[str, Any]:
         """
         Parse metadata
 

pygpt_net/core/idx/indexing.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.14 08:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import datetime
@@ -600,7 +600,10 @@ class Indexing:
 
         return indexed, errors
 
-    def get_db_data_from_ts(self, updated_ts: int = 0) -> list:
+    def get_db_data_from_ts(
+            self,
+            updated_ts: int = 0
+    ) -> List[Document]:
         """
         Get data from database from timestamp
 
@@ -640,7 +643,10 @@ class Indexing:
                 documents.append(doc)
         return documents
 
-    def get_db_meta_ids_from_ts(self, updated_ts: int = 0) -> list:
+    def get_db_meta_ids_from_ts(
+            self,
+            updated_ts: int = 0
+    ) -> List[int]:
         """
         Get IDs of meta from database from timestamp
 
@@ -664,7 +670,11 @@ class Indexing:
                 ids.append(data["id"])
         return ids
 
-    def get_db_data_by_id(self, id: int = 0, updated_ts: int = 0) -> list:
+    def get_db_data_by_id(
+            self,
+            id: int = 0,
+            updated_ts: int = 0
+    ) -> List[Document]:
         """
         Get data from database by meta id
 
@@ -726,7 +736,8 @@ class Indexing:
                 self.window.core.idx.log("Indexing documents from database by meta id: {}".format(id))
                 self.remove_old_meta_id(idx, id)
             elif from_ts > 0:
-                self.window.core.idx.log("Indexing documents from database by meta id: {} from timestamp: {}".format(id, from_ts))
+                self.window.core.idx.log("Indexing documents from database by meta id: {} from timestamp: {}".
+                                         format(id, from_ts))
 
             # get items from database
             documents = self.get_db_data_by_id(id, from_ts)
@@ -736,7 +747,8 @@ class Indexing:
 
                 self.index_document(index, d)
                 doc_id = d.id_
-                self.window.core.idx.log("Inserted ctx DB document: {} / {}, id: {}, metadata: {}".format(n+1, len(documents), d.id_, d.metadata))
+                self.window.core.idx.log("Inserted ctx DB document: {} / {}, id: {}, metadata: {}".
+                                         format(n+1, len(documents), d.id_, d.metadata))
                 self.window.core.ctx.idx.set_meta_as_indexed(id, idx, doc_id)  # update ctx
                 n += 1
         except Exception as e:
@@ -850,7 +862,8 @@ class Indexing:
                         idx=idx,
                         doc_id=doc_id,
                     )  # update external index
-                self.window.core.idx.log("Inserted web document: {} / {}, id: {}, metadata: {}".format(n+1, len(documents), d.id_, d.metadata))
+                self.window.core.idx.log("Inserted web document: {} / {}, id: {}, metadata: {}".
+                                         format(n+1, len(documents), d.id_, d.metadata))
                 n += 1
         except Exception as e:
             errors.append(str(e))
@@ -1044,7 +1057,10 @@ class Indexing:
             model = self.window.core.models.from_defaults()
 
         service_context = self.window.core.idx.llm.get_service_context(model=model)
-        index = self.window.core.idx.storage.get_ctx_idx(index_path, service_context=service_context)  # get or create ctx index
+        index = self.window.core.idx.storage.get_ctx_idx(
+            index_path,
+            service_context=service_context
+        )  # get or create ctx index
 
         idx = "tmp:{}".format(index_path)  # tmp index id
         self.window.core.idx.log("Indexing to context attachment index: {}...".format(idx))

pygpt_net/core/installer.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2023.12.31 04:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 from pathlib import Path
@@ -23,9 +23,7 @@ class Installer:
         self.window = window
 
     def install(self):
-        """
-        Install db, config data and directories
-        """
+        """Install db, config data and directories"""
         try:
             # create user config directory
             path = Path(self.window.core.config.path)

pygpt_net/core/models.py

@@ -6,11 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.14 00:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import copy
-from typing import Optional
+from typing import Optional, List, Dict
 
 from packaging.version import Version
 
@@ -92,7 +92,7 @@ class Models:
         if key in self.items:
             return self.items[key]
 
-    def get_ids(self) -> list:
+    def get_ids(self) -> List[str]:
         """
         Return models ids
 
@@ -118,7 +118,11 @@ class Models:
         """
         return model in self.items
 
-    def is_allowed(self, model: str, mode: str) -> bool:
+    def is_allowed(
+            self,
+            model: str,
+            mode: str
+    ) -> bool:
         """
         Check if model is allowed for mode
 
@@ -130,7 +134,10 @@ class Models:
             return mode in self.items[model].mode
         return False
 
-    def get_id(self, key: str) -> str:
+    def get_id(
+            self,
+            key: str
+    ) -> str:
         """
         Return model internal ID
 
@@ -140,7 +147,11 @@ class Models:
         if key in self.items:
             return self.items[key].id
 
-    def get_by_idx(self, idx: int, mode: str) -> str:
+    def get_by_idx(
+            self,
+            idx: int,
+            mode: str
+    ) -> str:
         """
         Return model by index
 
@@ -151,7 +162,10 @@ class Models:
         items = self.get_by_mode(mode)
         return list(items.keys())[idx]
 
-    def get_by_mode(self, mode: str) -> dict:
+    def get_by_mode(
+            self,
+            mode: str
+    ) -> Dict[str, ModelItem]:
         """
         Return models for mode
 
@@ -164,7 +178,11 @@ class Models:
                 items[key] = self.items[key]
         return items
 
-    def get_next(self, model: str, mode: str) -> str:
+    def get_next(
+            self,
+            model: str,
+            mode: str
+    ) -> str:
         """
         Return next model
 
@@ -179,7 +197,11 @@ class Models:
             return keys[idx + 1]
         return keys[0]
 
-    def get_prev(self, model: str, mode: str) -> str:
+    def get_prev(
+            self,
+            model: str,
+            mode: str
+    ) -> str:
         """
         Return previous model
 
@@ -218,7 +240,7 @@ class Models:
         self.items[id] = model
         return model
 
-    def get_all(self) -> dict:
+    def get_all(self) -> Dict[str, ModelItem]:
         """
         Return all models
 
@@ -248,7 +270,11 @@ class Models:
         if model in self.items:
             del self.items[model]
 
-    def has_model(self, mode: str, model: str) -> bool:
+    def has_model(
+            self,
+            mode: str,
+            model: str
+    ) -> bool:
         """
         Check if model exists for mode
 
@@ -296,8 +322,15 @@ class Models:
             return self.items[model].ctx
         return 4096
 
-    def restore_default(self, model: Optional[str] = None):
-        """Restore default models"""
+    def restore_default(
+            self,
+            model: Optional[str] = None
+    ):
+        """
+        Restore default models
+
+        :param model: model name
+        """
         # restore all models
         if model is None:
             self.load_base()
@@ -308,8 +341,12 @@ class Models:
         if model in items:
             self.items[model] = items[model]
 
-    def get_base(self) -> dict:
-        """Get base models"""
+    def get_base(self) -> Dict[str, ModelItem]:
+        """
+        Get base models
+
+        :return: base models
+        """
         return self.provider.load_base()
 
     def load_base(self):
@@ -330,7 +367,11 @@ class Models:
         """Save models"""
         self.provider.save(self.items)
 
-    def get_supported_mode(self, model: ModelItem, mode: str) -> str:
+    def get_supported_mode(
+            self,
+            model: ModelItem,
+            mode: str
+    ) -> str:
         """
         Get supported mode
 
@@ -353,5 +394,9 @@ class Models:
         return mode
 
     def get_version(self) -> str:
-        """Get config version"""
+        """
+        Get config version
+
+        :return: config version
+        """
         return self.provider.get_version()