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

pygpt_net/provider/core/ctx/db_sqlite/__init__.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.08.29 04:00:00                  #
+# Updated Date: 2024.12.09 00:00:00                  #
 # ================================================== #
 
 import time
@@ -118,6 +118,15 @@ class DbSqliteProvider(BaseProvider):
         """
         return self.storage.get_last_meta_id()
 
+    def get_item_by_id(self, id: int) -> CtxItem:
+        """
+        Get ctx item by ID
+
+        :param id: ctx item ID
+        :return: ctx item
+        """
+        return self.storage.get_item_by_id(id)
+
     def load(self, id: int) -> list:
         """
         Load items for ctx ID

pygpt_net/provider/core/ctx/db_sqlite/storage.py

@@ -6,7 +6,7 @@
 # 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.09 00:00:00                  #
 # ================================================== #
 
 from datetime import datetime
@@ -189,6 +189,27 @@ class Storage:
                 items[meta.id] = meta
         return items
 
+    def get_item_by_id(self, id: int) -> CtxItem:
+        """
+        Return ctx item by ID
+
+        :return: CtxItem
+        """
+        stmt = text("""
+            SELECT * FROM ctx_item WHERE id = :id
+        """).bindparams(id=id)
+        db = self.window.core.db.get_db()
+        with db.connect() as conn:
+            result = conn.execute(stmt)
+            row = result.fetchone()
+            if row:
+                item = CtxItem()
+                unpack_item(item, row._asdict())
+                meta = self.get_meta_by_id(item.meta_id)  # append meta
+                item.meta = meta
+                return item
+        return None
+
     def get_meta_by_root_id_and_preset_id(self, root_id: int, preset_id: str) -> dict:
         """
         Return dict with indexed CtxMeta objects, indexed by ID

pygpt_net/provider/gpt/assistants.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.30 16:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
 import json
+from typing import Optional
 
 from pygpt_net.item.assistant import AssistantItem
 from pygpt_net.item.ctx import CtxItem
@@ -37,7 +38,7 @@ class Assistants:
         """
         return self.window.core.gpt.get_client()
 
-    def log(self, msg: str, callback: callable = None):
+    def log(self, msg: str, callback: Optional[callable] = None):
         """
         Log message
 
@@ -140,7 +141,7 @@ class Assistants:
         thread_messages = client.beta.threads.messages.list(thread_id)
         return thread_messages.data
 
-    def msg_send(self, id: str, text: str, file_ids: list = None):
+    def msg_send(self, id: str, text: str, file_ids: Optional[list] = None):
         """
         Send message to thread
 
@@ -226,8 +227,8 @@ class Assistants:
             self,
             thread_id: str,
             assistant_id: str,
-            model: str = None,
-            instructions: str = None
+            model: Optional[str] = None,
+            instructions: Optional[str] = None
     ):
         """
         Create assistant run
@@ -259,8 +260,8 @@ class Assistants:
             ctx: CtxItem,
             thread_id: str,
             assistant_id: str,
-            model: str = None,
-            instructions: str = None
+            model: Optional[str] = None,
+            instructions: Optional[str] = None
     ):
         """
         Create assistant run (stream)
@@ -377,8 +378,8 @@ class Assistants:
             items: dict,
             order: str = "asc",
             limit: int = 100,
-            after: str = None,
-            callback: callable = None
+            after: Optional[str] = None,
+            callback: Optional[callable] = None
     ) -> dict:
         """
         Import assistants from API (all, paginated)

pygpt_net/provider/gpt/audio.py

@@ -6,10 +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 00:00:00                  #
 # ================================================== #
 
 import base64
+from typing import Optional
 
 from pygpt_net.core.bridge.context import MultimodalContext
 
@@ -26,7 +27,7 @@ class Audio:
     def build_content(
             self,
             content,
-            multimodal_ctx: MultimodalContext = None,
+            multimodal_ctx: Optional[MultimodalContext] = None,
     ) -> list:
         """
         Build audio content

pygpt_net/provider/gpt/chat.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 00:00:00                  #
 # ================================================== #
 
 import json
 import time
+from typing import Optional
 
 from pygpt_net.core.types import (
     MODE_CHAT,
@@ -36,7 +37,7 @@ class Chat:
     def send(
             self,
             context: BridgeContext,
-            extra: dict = None
+            extra: Optional[dict] = None
     ):
         """
         Call OpenAI API for chat
@@ -151,11 +152,11 @@ class Chat:
             prompt: str,
             system_prompt: str,
             model: ModelItem,
-            history: list = None,
-            attachments: dict = None,
-            ai_name: str = None,
-            user_name: str = None,
-            multimodal_ctx: MultimodalContext = None,
+            history: Optional[list] = None,
+            attachments: Optional[dict] = None,
+            ai_name: Optional[str] = None,
+            user_name: Optional[str] = None,
+            multimodal_ctx: Optional[MultimodalContext] = None,
     ) -> list:
         """
         Build list of chat messages

pygpt_net/provider/gpt/completion.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 00:00:00                  #
 # ================================================== #
 
+from typing import Optional
+
 from pygpt_net.core.types import (
     MODE_COMPLETION,
 )
@@ -98,9 +100,9 @@ class Completion:
             prompt: str,
             system_prompt: str,
             model: ModelItem,
-            history: list = None,
-            ai_name: str = None,
-            user_name: str = None,
+            history: Optional[list] = None,
+            ai_name: Optional[str] = None,
+            user_name: Optional[str] = None,
     ) -> str:
         """
         Build completion string

pygpt_net/provider/gpt/image.py

@@ -6,11 +6,13 @@
 # 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 00:00:00                  #
 # ================================================== #
 
 import datetime
 import os
+from typing import Optional
+
 import requests
 
 from PySide6.QtCore import QObject, Signal, QRunnable, Slot
@@ -31,7 +33,12 @@ class Image:
         self.window = window
         self.worker = None
 
-    def generate(self, context: BridgeContext, extra: dict = None, sync: bool = True):
+    def generate(
+            self,
+            context: BridgeContext,
+            extra: Optional[dict] = None,
+            sync: bool = True
+    ):
         """
         Call images API
 

pygpt_net/provider/gpt/store.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.30 16:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
 import os
+from typing import Optional
 
 from pygpt_net.item.assistant import AssistantStoreItem
 
@@ -31,7 +32,7 @@ class Store:
         """
         return self.window.core.gpt.get_client()
 
-    def log(self, msg: str, callback: callable = None):
+    def log(self, msg: str, callback: Optional[callable] = None):
         """
         Log message
 
@@ -57,7 +58,7 @@ class Store:
             self,
             path: str,
             purpose: str = "assistants"
-    ) -> str or None:
+    ) -> Optional[str]:
         """
         Upload file to assistant
 
@@ -98,7 +99,7 @@ class Store:
             items: list,
             order: str = "asc",
             limit: int = 100,
-            after: str = None,
+            after: Optional[str] = None,
     ) -> list:
         """
         Get all vector store IDs
@@ -148,7 +149,7 @@ class Store:
                     items.append(id)
         return items
 
-    def remove_files(self, callback: callable = None) -> int:
+    def remove_files(self, callback: Optional[callable] = None) -> int:
         """
         Remove all files
 
@@ -167,7 +168,7 @@ class Store:
                 self.log(msg, callback)
         return num
 
-    def remove_store_files(self, store_id: str, callback: callable = None) -> int:
+    def remove_store_files(self, store_id: str, callback: Optional[callable] = None) -> int:
         """
         Remove all files from store
 
@@ -192,8 +193,8 @@ class Store:
             items: dict,
             order: str = "asc",
             limit: int = 100,
-            after: str = None,
-            callback: callable = None
+            after: Optional[str] = None,
+            callback: Optional[callable] = None
     ) -> dict:
         """
         Import vector stores from API
@@ -318,7 +319,7 @@ class Store:
             items: list,
             order: str = "asc",
             limit: int = 100,
-            after: str = None,
+            after: Optional[str] = None,
     ) -> list:
         """
         Get all vector stores IDs
@@ -358,7 +359,7 @@ class Store:
             items: list,
             order: str = "asc",
             limit: int = 100,
-            after: str = None,
+            after: Optional[str] = None,
     ) -> list:
         """
         Get all vector store files IDs
@@ -426,7 +427,7 @@ class Store:
             num += 1
         return num
 
-    def remove_all(self, callback: callable = None) -> int:
+    def remove_all(self, callback: Optional[callable] = None) -> int:
         """
         Remove all vector stores
 
@@ -495,7 +496,7 @@ class Store:
         if vector_store_file is not None:
             return vector_store_file
 
-    def import_stores_files(self, callback: callable = None) -> int:
+    def import_stores_files(self, callback: Optional[callable] = None) -> int:
         """
         Import all vector stores files
 
@@ -525,7 +526,7 @@ class Store:
             order: str = "asc",
             limit: int = 100,
             after: str = None,
-            callback: callable = None
+            callback: Optional[callable] = None
     ) -> list:
         """
         Import and get all vector store files IDs

pygpt_net/provider/gpt/vision.py

@@ -6,12 +6,13 @@
 # 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 00:00:00                  #
 # ================================================== #
 
 import base64
 import os
 import re
+from typing import Optional, Union
 
 from pygpt_net.core.types import (
     MODE_VISION,
@@ -77,8 +78,8 @@ class Vision:
             prompt: str,
             system_prompt: str,
             model: ModelItem,
-            history: list = None,
-            attachments: dict = None,
+            history: Optional[list] = None,
+            attachments: Optional[dict] = None,
     ) -> list:
         """
         Build chat messages list
@@ -160,8 +161,8 @@ class Vision:
 
     def build_content(
             self,
-            content: str or list,
-            attachments: dict = None,
+            content: Union[str, list],
+            attachments: Optional[dict] = None,
     ) -> list:
         """
         Build vision content

pygpt_net/tools/__init__.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.09 23:00:00                  #
 # ================================================== #
 
 from pygpt_net.ui.widget.dialog.base import BaseDialog
@@ -42,6 +42,14 @@ class Tools:
         if id in self.tools:
             return self.tools[id]
 
+    def get_all(self) -> dict:
+        """
+        Get all tools
+
+        :return: dict with tools
+        """
+        return self.tools
+
     def setup(self):
         """Setup tools"""
         self.setup_dialogs()

pygpt_net/tools/base.py

@@ -6,11 +6,13 @@
 # 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.12 04:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import QObject
+from PySide6.QtWidgets import QWidget
 
+from pygpt_net.core.tabs.tab import Tab
 from pygpt_net.ui.widget.dialog.base import BaseDialog
 
 class BaseTool(QObject):
@@ -25,6 +27,9 @@ class BaseTool(QObject):
         super(BaseTool, self).__init__()
         self.window = None
         self.id = ""
+        self.has_tab = False
+        self.tab_title = ""
+        self.tab_icon = ":/icons/build.svg"
 
     def setup(self):
         """Setup tool"""
@@ -83,6 +88,15 @@ class BaseTool(QObject):
         """
         return None
 
+    def as_tab(self, tab: Tab) -> QWidget or None:
+        """
+        Spawn and return tab instance
+
+        :param tab: Parent tab instance
+        :return: Tab widget instance
+        """
+        return None
+
     def get_lang_mappings(self) -> dict:
         """
         Get language mappings