pygpt-net 2.7.3__py3-none-any.whl → 2.7.5__py3-none-any.whl

pygpt_net/item/store.py

@@ -0,0 +1,238 @@
+#!/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: 2026.01.02 20:00:00                  #
+# ================================================== #
+
+import json
+import time
+from dataclasses import dataclass, field
+from typing import Optional
+
+from pygpt_net.item.attachment import AttachmentItem
+
+
+@dataclass(slots=True)
+class RemoteStoreItem:
+    id: Optional[object] = None
+    record_id: Optional[object] = None
+    uuid: Optional[object] = None
+    name: Optional[object] = None
+    provider: Optional[object] = None
+    description: Optional[object] = None
+    status: dict = field(default_factory=dict)
+    last_status: str = ""
+    expire_days: int = 0
+    usage_bytes: int = 0
+    bytes: int = 0
+    num_files: int = 0
+    is_thread: bool = False
+    created: int = 0
+    updated: int = 0
+    last_active: int = 0
+    last_sync: int = 0
+    file_ids: list = field(default_factory=list)
+
+    def __init__(self):
+        """Assistant vector store item"""
+        self.id = None
+        self.record_id = None
+        self.uuid = None
+        self.name = None
+        self.provider = None
+        self.description = None
+        self.status = {}
+        self.last_status = ""
+        self.expire_days = 0
+        self.usage_bytes = 0
+        self.bytes = 0
+        self.num_files = 0
+        self.is_thread = False
+        self.created = int(time.time())
+        self.updated = int(time.time())
+        self.last_active = int(time.time())
+        self.last_sync = int(time.time())
+
+    def reset(self):
+        """Reset store"""
+        self.id = None
+        self.record_id = None
+        self.uuid = None
+        self.name = None
+        self.provider = None
+        self.description = None
+        self.status = {}
+        self.last_status = ""
+        self.expire_days = 0
+        self.usage_bytes = 0
+        self.num_files = 0
+        self.bytes = 0
+        self.is_thread = False
+        self.last_active = int(time.time())
+        self.last_sync = int(time.time())
+
+    def to_dict(self) -> dict:
+        """
+        Return as dictionary
+
+        :return: dictionary
+        """
+        return {
+            "id": self.id,
+            "uuid": self.uuid,
+            "name": self.name,
+            "provider": self.provider,
+            "description": self.description,
+            "last_status": self.last_status,
+            "expire_days": self.expire_days,
+            "usage_bytes": self.usage_bytes,
+            "num_files": self.num_files,
+            "status": self.status,
+            "is_thread": self.is_thread,
+        }
+
+    def from_dict(self, data: dict):
+        """
+        Load from dictionary
+
+        :param data: dictionary
+        """
+        self.id = data.get('id', None)
+        self.name = data.get('name', None)
+        self.provider = data.get('provider', None)
+        self.uuid = data.get('uuid', None)
+        self.expire_days = data.get('expire_days', 0)
+        self.status = data.get('status', {})
+
+    def get_file_count(self):
+        """
+        Return number of files in store
+
+        :return: number of files
+        """
+        num = self.num_files
+        if self.status and isinstance(self.status, dict) and 'file_counts' in self.status:
+            if 'completed' in self.status['file_counts']:
+                num = int(self.status['file_counts']['completed'] or 0)
+        return num
+
+    def dump(self) -> str:
+        """
+        Dump item to string
+
+        :return: serialized item
+        """
+        try:
+            return json.dumps(self.to_dict())
+        except Exception as e:
+            pass
+        return ""
+
+    def __str__(self):
+        """To string"""
+        return self.dump()
+
+
+@dataclass(slots=True)
+class RemoteFileItem:
+    id: Optional[object] = None
+    record_id: Optional[object] = None
+    name: Optional[object] = None
+    provider: Optional[object] = None
+    path: Optional[object] = None
+    file_id: Optional[object] = None
+    store_id: Optional[object] = None
+    thread_id: Optional[object] = None
+    uuid: Optional[object] = None
+    size: int = 0
+    created: int = 0
+    updated: int = 0
+
+    def __init__(self):
+        """Assistant file item"""
+        self.id = None
+        self.record_id = None
+        self.name = None
+        self.provider = None
+        self.path = None
+        self.file_id = None
+        self.store_id = None
+        self.thread_id = None
+        self.uuid = None
+        self.size = 0
+        self.created = 0
+        self.updated = 0
+
+    def reset(self):
+        """Reset store"""
+        self.id = None
+        self.record_id = None
+        self.name = None
+        self.provider = None
+        self.path = None
+        self.file_id = None
+        self.store_id = None
+        self.thread_id = None
+        self.uuid = None
+        self.size = 0
+        self.created = 0
+        self.updated = 0
+
+    def to_dict(self) -> dict:
+        """
+        Return as dictionary
+
+        :return: dictionary
+        """
+        return {
+            "id": self.id,
+            "name": self.name,
+            "provider": self.provider,
+            "path": self.path,
+            "file_id": self.file_id,
+            "store_id": self.store_id,
+            "thread_id": self.thread_id,
+            "uuid": self.uuid,
+            "size": self.size,
+            "created": self.created,
+            "updated": self.updated,
+        }
+
+    def from_dict(self, data: dict):
+        """
+        Load from dictionary
+
+        :param data: dictionary
+        """
+        self.id = data.get('id', None)
+        self.name = data.get('name', None)
+        self.provider = data.get('provider', None)
+        self.path = data.get('path', None)
+        self.file_id = data.get('file_id', None)
+        self.store_id = data.get('store_id', None)
+        self.thread_id = data.get('thread_id', None)
+        self.uuid = data.get('uuid', None)
+        self.size = data.get('size', 0)
+        self.created = data.get('created', 0)
+        self.updated = data.get('updated', 0)
+
+    def dump(self) -> str:
+        """
+        Dump item to string
+
+        :return: serialized item
+        """
+        try:
+            return json.dumps(self.to_dict())
+        except Exception as e:
+            pass
+        return ""
+
+    def __str__(self):
+        """To string"""
+        return self.dump()

pygpt_net/launcher.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.19 07:00:00                  #
+# Updated Date: 2025.12.31 17:00:00                  #
 # ================================================== #
 
 import os
@@ -50,6 +50,31 @@ class Launcher:
         self.force_disable_gpu = False
         self.shortcut_filter = None
         self.workdir = None
+        self._preloader = None
+
+    @staticmethod
+    def _clean_multiprocessing_argv(argv):
+        """
+        Clear multiprocessing flags from argv
+
+        :param argv: list of command line arguments
+        """
+        skip_flags = {
+            "--multiprocessing-fork",
+            "--multiprocessing-spawn",
+            "--billiard-fork",
+            "--billiard-spawn",
+        }
+        skip_prefixes = (
+            "parent_pid=",
+            "pipe_handle=",
+            "forkserver_port=",
+            "forkserver_authkey=",
+        )
+        return [
+            a for a in argv
+            if a not in skip_flags and not any(a.startswith(p) for p in skip_prefixes)
+        ]
 
     def setup(self) -> dict:
         """
@@ -57,59 +82,65 @@ class Launcher:
 
         :return: dict with launcher arguments
         """
-        parser = argparse.ArgumentParser()
-        parser.add_argument(
-            "-d",
-            "--debug",
-            required=False,
-            help="debug mode (0=disabled, 1=info, 2=debug)",
-        )
-        parser.add_argument(
-            "-l",
-            "--legacy",
-            required=False,
-            help="force enable legacy mode (0=disabled, 1=enable)",
-        )
-        parser.add_argument(
-            "-n",
-            "--disable-gpu",
-            required=False,
-            help="force disable OpenGL (1=disabled, 0=enabled)",
-        )
-        parser.add_argument(
-            "-w",
-            "--workdir",
-            required=False,
-            help="force set workdir",
-        )
-        args = vars(parser.parse_args())
-
-        # set log level [ERROR|WARNING|INFO|DEBUG]
-        if "debug" in args and args["debug"] == "1":
-            print("** Debug mode enabled (1=INFO)")
-            Debug.init(INFO)
-            self.debug = True
-        elif "debug" in args and args["debug"] == "2":
-            print("** Debug mode enabled (2=DEBUG)")
-            Debug.init(DEBUG)
-            self.debug = True
-        else:
-            Debug.init(ERROR)  # default log level
-
-        # force legacy mode
-        if "legacy" in args and args["legacy"] == "1":
-            print("** Force legacy mode enabled")
-            self.force_legacy = True
-
-        # force disable GPU
-        if "disable-gpu" in args and args["disable-gpu"] == "1":
-            print("** Force disable GPU enabled")
-            self.force_disable_gpu = True
-
-        # force set workdir
-        if "workdir" in args and args["workdir"] is not None:
-            # set as environment variable
-            os.environ["PYGPT_WORKDIR"] = args["workdir"]
+        args = {}
+        try:
+            parser = argparse.ArgumentParser()
+            parser.add_argument(
+                "-d",
+                "--debug",
+                required=False,
+                help="debug mode (0=disabled, 1=info, 2=debug)",
+            )
+            parser.add_argument(
+                "-l",
+                "--legacy",
+                required=False,
+                help="force enable legacy mode (0=disabled, 1=enable)",
+            )
+            parser.add_argument(
+                "-n",
+                "--disable-gpu",
+                required=False,
+                help="force disable OpenGL (1=disabled, 0=enabled)",
+            )
+            parser.add_argument(
+                "-w",
+                "--workdir",
+                required=False,
+                help="force set workdir",
+            )
+            safe_argv = self._clean_multiprocessing_argv(sys.argv[1:])
+            known, unknown = parser.parse_known_args(safe_argv)
+            args = vars(known)
+
+            # set log level [ERROR|WARNING|INFO|DEBUG]
+            if "debug" in args and args["debug"] == "1":
+                print("** Debug mode enabled (1=INFO)")
+                Debug.init(INFO)
+                self.debug = True
+            elif "debug" in args and args["debug"] == "2":
+                print("** Debug mode enabled (2=DEBUG)")
+                Debug.init(DEBUG)
+                self.debug = True
+            else:
+                Debug.init(ERROR)  # default log level
+
+            # force legacy mode
+            if "legacy" in args and args["legacy"] == "1":
+                print("** Force legacy mode enabled")
+                self.force_legacy = True
+
+            # force disable GPU
+            if "disable_gpu" in args and args["disable_gpu"] == "1":
+                print("** Force disable GPU enabled")
+                self.force_disable_gpu = True
+
+            # force set workdir
+            if "workdir" in args and args["workdir"] is not None:
+                # set as environment variable
+                os.environ["PYGPT_WORKDIR"] = args["workdir"]
+        except Exception as e:
+            print(f"** Launcher setup error: {e}")
 
         return args
 
@@ -123,6 +154,13 @@ class Launcher:
         self.window = MainWindow(self.app, args=args)
         self.shortcut_filter = GlobalShortcutFilter(self.window)
 
+        # Connect the window "ready" signal to close the splash
+        if self._preloader is not None:
+            try:
+                self.window.appReady.connect(self._on_window_ready, Qt.QueuedConnection)
+            except Exception:
+                pass
+
     def handle_signal(self, signal_number, frame):
         """
         Handle termination signal (SIGTERM, SIGINT)
@@ -136,6 +174,28 @@ class Launcher:
             print("Shutting down...")
             self.window.close()
 
+    def attach_preloader(self, preloader):
+        """
+        Attach external splash controller. The splash will be closed
+        when the main window announces readiness.
+        """
+        self._preloader = preloader
+        if self.window is not None:
+            try:
+                self.window.appReady.connect(self._on_window_ready, Qt.QueuedConnection)
+            except Exception:
+                pass
+
+    def _on_window_ready(self):
+        """
+        Close preloader when the first frame of the main window is ready.
+        """
+        if self._preloader:
+            try:
+                self._preloader.close()
+            except Exception:
+                pass
+            self._preloader = None
 
     def add_plugin(self, plugin: BasePlugin):
         """
@@ -290,4 +350,4 @@ class Launcher:
         # self.window.core.debug.mem("INIT")  # debug memory usage
         signal.signal(signal.SIGTERM, self.handle_signal)
         signal.signal(signal.SIGINT, self.handle_signal)
-        sys.exit(self.app.exec())
+        sys.exit(self.app.exec())

pygpt_net/migrations/Version20260102190000.py

@@ -0,0 +1,35 @@
+#!/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: 2026.01.02 20:00:00                  #
+# ================================================== #
+
+from sqlalchemy import text
+
+from .base import BaseMigration
+
+
+class Version20260102190000(BaseMigration):
+    def __init__(self, window=None):
+        super(Version20260102190000, self).__init__(window)
+        self.window = window
+
+    def up(self, conn):
+        conn.execute(text("""
+        ALTER TABLE remote_store ADD COLUMN provider TEXT;
+        """))
+        conn.execute(text("""
+        ALTER TABLE remote_file ADD COLUMN provider TEXT;
+        """))
+        # set all to OpenAI by default
+        conn.execute(text("""
+        UPDATE remote_store SET provider = 'openai' WHERE provider IS NULL;
+        """))
+        conn.execute(text("""
+        UPDATE remote_file SET provider = 'openai' WHERE provider IS NULL;
+        """))

pygpt_net/migrations/__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.15 11:00:00                  #
+# Updated Date: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 from .Version20231227152900 import Version20231227152900  # 2.0.59
@@ -23,6 +23,7 @@ from .Version20240501030000 import Version20240501030000  # 2.2.7
 from .Version20241122130000 import Version20241122130000  # 2.4.21
 from .Version20241126170000 import Version20241126170000  # 2.4.34
 from .Version20241215110000 import Version20241215110000  # 2.4.43
+from .Version20260102190000 import Version20260102190000  # 2.7.5
 
 class Migrations:
     def __init__(self):
@@ -50,4 +51,5 @@ class Migrations:
             Version20241122130000(),  # 2.4.21
             Version20241126170000(),  # 2.4.34
             Version20241215110000(),  # 2.4.43
+            Version20260102190000(),  # 2.7.5
         ]