pygpt-net 2.7.4__py3-none-any.whl → 2.7.6__py3-none-any.whl

pygpt_net/provider/core/{assistant_store → remote_store}/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.12.14 22:00:00                  #
+# Updated Date: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 from .provider import *

pygpt_net/provider/core/{assistant_store → remote_store}/db_sqlite/patch.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.26 23:00:00                  #
+# Updated Date: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 from packaging.version import parse as parse_version, Version

pygpt_net/provider/core/{assistant_store → remote_store}/db_sqlite/provider.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 22:00:00                  #
+# Updated Date: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 import uuid
@@ -14,8 +14,8 @@ from typing import Dict
 
 from packaging.version import Version
 
-from pygpt_net.item.assistant import AssistantStoreItem
-from pygpt_net.provider.core.assistant_store.base import BaseProvider
+from pygpt_net.item.store import RemoteStoreItem
+from pygpt_net.provider.core.remote_store.base import BaseProvider
 
 from .patch import Patch
 from .storage import Storage
@@ -28,7 +28,7 @@ class DbSqliteProvider(BaseProvider):
         self.patcher = Patch(window, self)
         self.storage = Storage(window)
         self.id = "db_sqlite"
-        self.type = "assistant_store"
+        self.type = "remote_store"
 
     def attach(self, window):
         self.window = window
@@ -51,11 +51,11 @@ class DbSqliteProvider(BaseProvider):
         """
         return str(uuid.uuid4())
 
-    def create(self, store: AssistantStoreItem) -> int:
+    def create(self, store: RemoteStoreItem) -> int:
         """
         Create new and return its ID
 
-        :param store: AssistantStoreItem
+        :param store: RemoteStoreItem
         :return: store ID
         """
         if store.record_id is None or store.record_id == "":
@@ -63,15 +63,16 @@ class DbSqliteProvider(BaseProvider):
             store.record_id = self.storage.insert(store)
         return store.record_id
 
-    def load_all(self) -> Dict[str, AssistantStoreItem]:
+    def load_all(self, provider: str) -> Dict[str, RemoteStoreItem]:
         """
         Load stores from DB
 
+        :param provider: provider ID
         :return: stores dict
         """
-        return self.storage.get_all()
+        return self.storage.get_all(provider)
 
-    def load(self, id: int) -> AssistantStoreItem:
+    def load(self, id: int) -> RemoteStoreItem:
         """
         Load store from DB
 
@@ -80,11 +81,11 @@ class DbSqliteProvider(BaseProvider):
         """
         return self.storage.get_by_id(id)
 
-    def save(self, store: AssistantStoreItem):
+    def save(self, store: RemoteStoreItem):
         """
         Save store to DB
 
-        :param store: AssistantStoreItem
+        :param store: RemoteStoreItem
         """
         try:
             self.storage.save(store)
@@ -92,11 +93,11 @@ class DbSqliteProvider(BaseProvider):
             self.window.core.debug.log(e)
             print("Error while saving stored: {}".format(str(e)))
 
-    def save_all(self, items: Dict[str, AssistantStoreItem]):
+    def save_all(self, items: Dict[str, RemoteStoreItem]):
         """
         Save all stores to DB
 
-        :param items: dict of AssistantStoreItem objects
+        :param items: dict of RemoteStoreItem objects
         """
         try:
             for id in items:
@@ -124,12 +125,12 @@ class DbSqliteProvider(BaseProvider):
         """
         return self.storage.delete_by_store_id(id)
 
-    def truncate(self) -> bool:
+    def truncate(self, provider: str) -> bool:
         """
         Truncate all stores
 
         :return: True if truncated
         """
-        return self.storage.truncate_all()
+        return self.storage.truncate_all(provider)
 
 

pygpt_net/provider/core/{assistant_store → remote_store}/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.12.14 22:00:00                  #
+# Updated Date: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 import time
@@ -14,7 +14,7 @@ from typing import Dict
 
 from sqlalchemy import text
 
-from pygpt_net.item.assistant import AssistantStoreItem
+from pygpt_net.item.store import RemoteStoreItem
 
 from .utils import pack_item_value, unpack_store
 
@@ -35,36 +35,37 @@ class Storage:
         """
         self.window = window
 
-    def get_all(self) -> Dict[str, AssistantStoreItem]:
+    def get_all(self, provider: str) -> Dict[str, RemoteStoreItem]:
         """
-        Return dict with AssistantStoreItem objects, indexed by ID
+        Return dict with RemoteStoreItem objects, indexed by ID
 
-        :return: dict of AssistantStoreItem objects
+        :param provider: provider ID
+        :return: dict of RemoteStoreItem objects
         """
         stmt = text("""
-            SELECT * FROM remote_store
-        """)
+            SELECT * FROM remote_store WHERE provider = :provider
+        """).bindparams(provider=provider)
         items = {}
         db = self.window.core.db.get_db()
         with db.connect() as conn:
             result = conn.execute(stmt)
             for row in result:
-                store = AssistantStoreItem()
+                store = RemoteStoreItem()
                 unpack_store(store, row._asdict())
                 items[store.id] = store  # index by store ID
         return items
 
-    def get_by_id(self, id: int) -> AssistantStoreItem:
+    def get_by_id(self, id: int) -> RemoteStoreItem:
         """
-        Return AssistantStoreItem by ID
+        Return RemoteStoreItem by ID
 
         :param id: store item ID
-        :return: AssistantStoreItem
+        :return: RemoteStoreItem
         """
         stmt = text("""
             SELECT * FROM remote_store WHERE id = :id LIMIT 1
         """).bindparams(id=id)
-        store = AssistantStoreItem()
+        store = RemoteStoreItem()
         db = self.window.core.db.get_db()
         with db.connect() as conn:
             result = conn.execute(stmt)
@@ -72,17 +73,17 @@ class Storage:
                 unpack_store(store, row._asdict())
         return store
 
-    def get_by_store_id(self, id: str) -> AssistantStoreItem:
+    def get_by_store_id(self, id: str) -> RemoteStoreItem:
         """
-        Return AssistantStoreItem by ID
+        Return RemoteStoreItem by ID
 
         :param id: store_id
-        :return: AssistantStoreItem
+        :return: RemoteStoreItem
         """
         stmt = text("""
             SELECT * FROM remote_store WHERE store_id = :id LIMIT 1
         """).bindparams(id=id)
-        store = AssistantStoreItem()
+        store = RemoteStoreItem()
         db = self.window.core.db.get_db()
         with db.connect() as conn:
             result = conn.execute(stmt)
@@ -90,16 +91,17 @@ class Storage:
                 unpack_store(store, row._asdict())
         return store
 
-    def truncate_all(self) -> bool:
+    def truncate_all(self, provider: str) -> bool:
         """
         Truncate all stores items
 
+        :param provider: provider ID
         :return: True if truncated
         """
         db = self.window.core.db.get_db()
         with db.begin() as conn:
-            conn.execute(text("DELETE FROM remote_store"))
-            conn.execute(text("DELETE FROM sqlite_sequence WHERE name='remote_store'"))
+            conn.execute(text("DELETE FROM remote_store WHERE provider = :provider").bindparams(provider=provider))
+            # conn.execute(text("DELETE FROM sqlite_sequence WHERE name='remote_store'"))
         return True
 
     def delete_by_id(self, id: int) -> bool:
@@ -132,11 +134,11 @@ class Storage:
             conn.execute(stmt)
             return True
 
-    def save(self, store: AssistantStoreItem):
+    def save(self, store: RemoteStoreItem):
         """
         Insert or update store item
 
-        :param store: AssistantStoreItem object
+        :param store: RemoteStoreItem object
         """
         db = self.window.core.db.get_db()
         with db.begin() as conn:
@@ -145,6 +147,7 @@ class Storage:
                 UPDATE remote_store
                 SET
                     name = :name,
+                    provider = :provider,
                     description = :description,
                     expire_days = :expire_days,
                     usage_bytes = :usage_bytes,
@@ -160,6 +163,7 @@ class Storage:
                 """).bindparams(
                 id=store.record_id,
                 name=store.name,
+                provider=store.provider,
                 description=store.description,
                 expire_days=store.expire_days,
                 usage_bytes=store.usage_bytes,
@@ -174,11 +178,11 @@ class Storage:
             )
             conn.execute(stmt)
 
-    def insert(self, store: AssistantStoreItem) -> int:
+    def insert(self, store: RemoteStoreItem) -> int:
         """
         Insert store item
 
-        :param store: AssistantStoreItem object
+        :param store: RemoteStoreItem object
         :return: store item ID
         """
         db = self.window.core.db.get_db()
@@ -189,6 +193,7 @@ class Storage:
                 store_id,
                 uuid,
                 name,
+                provider,
                 description,
                 expire_days,
                 usage_bytes,
@@ -206,6 +211,7 @@ class Storage:
                 :store_id,
                 :uuid,
                 :name,
+                :provider,
                 :description,
                 :expire_days,
                 :usage_bytes,
@@ -222,6 +228,7 @@ class Storage:
             store_id=store.id,
             uuid=store.uuid,
             name=store.name,
+            provider=store.provider,
             description=store.description,
             expire_days=int(store.expire_days or 0),
             usage_bytes=int(store.usage_bytes or 0),

pygpt_net/provider/core/{assistant_store → remote_store}/db_sqlite/utils.py

@@ -6,20 +6,20 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.26 23:00:00                  #
+# Updated Date: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 import json
 
 from pygpt_net.utils import unpack_var
-from pygpt_net.item.assistant import AssistantStoreItem
+from pygpt_net.item.store import RemoteStoreItem
 
 
-def unpack_store(store: AssistantStoreItem, row: dict) -> AssistantStoreItem:
+def unpack_store(store: RemoteStoreItem, row: dict) -> RemoteStoreItem:
     """
     Unpack store item from DB row
 
-    :param store: store item (AssistantStoreItem)
+    :param store: store item (RemoteStoreItem)
     :param row: DB row
     :return: store item
     """
@@ -31,6 +31,7 @@ def unpack_store(store: AssistantStoreItem, row: dict) -> AssistantStoreItem:
     store.last_active = unpack_var(row['last_active_ts'], "int")
     store.last_sync = unpack_var(row['last_sync_ts'], "int")
     store.name = row['name']
+    store.provider = row['provider']
     store.description = row['description']
     store.expire_days = unpack_var(row['expire_days'], "int")
     store.usage_bytes = unpack_var(row['usage_bytes'], "int")

pygpt_net/provider/core/{assistant_store → remote_store}/json_file.py

@@ -6,14 +6,14 @@
 # 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: 2026.01.02 20:00:00                  #
 # ================================================== #
 
 import json
 import os
 import uuid
 
-from pygpt_net.item.assistant import AssistantStoreItem
+from pygpt_net.item.store import RemoteStoreItem
 from pygpt_net.provider.core.assistant.base import BaseProvider
 
 
@@ -22,7 +22,7 @@ class JsonFileProvider(BaseProvider):
         super(JsonFileProvider, self).__init__(window)
         self.window = window
         self.id = "json_file"
-        self.type = "assistant_store"
+        self.type = "remote_store"
         self.config_file = 'assistants_vector_store.json'
 
     def create_id(self) -> str:
@@ -33,11 +33,11 @@ class JsonFileProvider(BaseProvider):
         """
         return str(uuid.uuid4())
 
-    def create(self, store: AssistantStoreItem) -> str:
+    def create(self, store: RemoteStoreItem) -> str:
         """
         Create new and return its ID
 
-        :param store: AssistantStoreItem
+        :param store: RemoteStoreItem
         :return: vector store ID
         """
         if store.id is None or store.id == "":
@@ -61,7 +61,7 @@ class JsonFileProvider(BaseProvider):
                     # deserialize
                     for id in data['items']:
                         item = data['items'][id]
-                        store = AssistantStoreItem()
+                        store = RemoteStoreItem()
                         self.deserialize(item, store)
                         items[id] = store
         except Exception as e:
@@ -109,7 +109,7 @@ class JsonFileProvider(BaseProvider):
         pass
 
     @staticmethod
-    def serialize(item: AssistantStoreItem) -> dict:
+    def serialize(item: RemoteStoreItem) -> dict:
         """
         Serialize item to dict
 
@@ -124,7 +124,7 @@ class JsonFileProvider(BaseProvider):
         }
 
     @staticmethod
-    def deserialize(data: dict, item: AssistantStoreItem):
+    def deserialize(data: dict, item: RemoteStoreItem):
         """
         Deserialize item from dict
 
@@ -140,7 +140,7 @@ class JsonFileProvider(BaseProvider):
         if 'file_ids' in data:
             item.file_ids = data['file_ids']
 
-    def dump(self, item: AssistantStoreItem) -> str:
+    def dump(self, item: RemoteStoreItem) -> str:
         """
         Dump to string
 

pygpt_net/provider/llms/google.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.17 20:00:00                  #
+# Updated Date: 2026.01.02 19:00:00                  #
 # ================================================== #
 
 from typing import Optional, List, Dict
@@ -70,7 +70,7 @@ class GoogleLLM(BaseLLM):
         # -----------------------------------------------------------
         built_tools = []
         try:
-            built_tools = window.core.api.google.build_remote_tools(model=model) or []
+            built_tools = window.core.api.google.remote_tools.build_remote_tools(model=model) or []
         except Exception as e:
             window.core.debug.log(e)