pygpt-net 2.6.37__py3-none-any.whl → 2.6.39__py3-none-any.whl

pygpt_net/CHANGELOG.txt

@@ -1,3 +1,15 @@
+2.6.39 (2025-09-06)
+
+- Added: Search input to the models editor.
+- Fixed: Brush color switching when changing modes in the Painter.
+
+2.6.38 (2025-09-05)
+
+- Fixed: Detection of chunk type in Ollama.
+- Fixed: Import of models with existing IDs.
+- Fixed: Updating Assistants UI list after create new Assistant.
+- Refactor and optimization.
+
 2.6.37 (2025-09-05)
 
 - Fixed: Function parameters sanitization in the Google Gen AI SDK.

pygpt_net/__init__.py

@@ -6,15 +6,15 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.05 00:00:00                  #
+# Updated Date: 2025.09.06 00:00:00                  #
 # ================================================== #
 
 __author__ = "Marcin Szczygliński"
 __copyright__ = "Copyright 2025, Marcin Szczygliński"
 __credits__ = ["Marcin Szczygliński"]
 __license__ = "MIT"
-__version__ = "2.6.37"
-__build__ = "2025-09-05"
+__version__ = "2.6.39"
+__build__ = "2025-09-06"
 __maintainer__ = "Marcin Szczygliński"
 __github__ = "https://github.com/szczyglis-dev/py-gpt"
 __report__ = "https://github.com/szczyglis-dev/py-gpt/issues"

pygpt_net/controller/chat/handler/anthropic_stream.py

@@ -154,8 +154,6 @@ def process_anthropic_chunk(ctx, core, state, chunk) -> Optional[str]:
                     state.usage_payload["out"] = out_tok
             delta = getattr(chunk, "delta", None)
             stop_reason = getattr(delta, "stop_reason", None) if delta else None
-            if stop_reason == "tool_use":
-                state.force_func_call = True
         except Exception:
             pass
         return None

pygpt_net/controller/chat/handler/worker.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.04 00:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 import io
@@ -16,6 +16,7 @@ from typing import Optional, Literal, Any
 from enum import Enum
 
 from PySide6.QtCore import QObject, Signal, Slot, QRunnable
+from openai.types.chat import ChatCompletionChunk
 
 from pygpt_net.core.events import RenderEvent
 from pygpt_net.core.text.utils import has_unclosed_code_tag
@@ -262,6 +263,10 @@ class StreamWorker(QRunnable):
             if not hasattr(chunk, "type") and not hasattr(chunk, "candidates"):
                 return ChunkType.LLAMA_CHAT
 
+        # fallback: OpenAI ChatCompletionChunk not caught above
+        if isinstance(chunk, ChatCompletionChunk):
+            return ChunkType.API_CHAT
+
         return ChunkType.RAW
 
     def _append_response(
@@ -344,7 +349,6 @@ class StreamWorker(QRunnable):
                 calls = xai_stream.xai_extract_tool_calls(state.xai_last_response)
                 if calls:
                     state.tool_calls = calls
-                    state.force_func_call = True
             except Exception:
                 pass
 

pygpt_net/controller/debug/debug.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.18 01:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 from datetime import datetime
@@ -31,6 +31,7 @@ class Debug(QObject):
         self.is_logger = False  # logger window opened
         self.is_app_log = False  # app log window opened
         self.allow_level_change = False  # allow changing log level
+        self._ids = None
 
     def update(self):
         """Update debug"""
@@ -100,12 +101,11 @@ class Debug(QObject):
 
         :param all: update all debug windows
         """
-        # not_realtime = ['context']
-        not_realtime = []
-        for id in self.window.controller.dialogs.debug.get_ids():
+        if self._ids is None:
+            self._ids = self.window.controller.dialogs.debug.get_ids()
+        for id in self._ids:
             if self.window.controller.dialogs.debug.is_active(id):
-                if all or id not in not_realtime:
-                    self.window.controller.dialogs.debug.update_worker(id)
+                self.window.controller.dialogs.debug.update_worker(id)
 
     def post_setup(self):
         """Post setup debug"""

pygpt_net/controller/model/editor.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.14 01:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 import copy
@@ -69,7 +69,6 @@ class Editor:
                 "type": "bool",
                 "label": "model.tool_calls",
                 "description": "model.tool_calls.desc",
-                "advanced": True,
             },
             "input": {
                 "type": "bool_list",  # list of comma separated values
@@ -424,15 +423,14 @@ class Editor:
 
         :param idx: tab index
         """
-        model_idx = 0
-        self.window.core.models.sort_items()
-        for id in self.window.core.models.get_ids():
-            if model_idx == idx:
-                self.current = id
-                break
-            model_idx += 1
-        current = self.window.ui.models['models.list'].index(idx, 0)
-        self.window.ui.nodes['models.list'].setCurrentIndex(current)
+        if idx is None:
+            return
+        # Resolve model id using the filtered view mapping
+        model_id = self.window.model_settings.get_model_id_by_row(idx)
+        if model_id is not None:
+            self.current = model_id
+            current = self.window.ui.models['models.list'].index(idx, 0)
+            self.window.ui.nodes['models.list'].setCurrentIndex(current)
 
     def set_tab_by_id(self, model_id: str):
         """
@@ -441,57 +439,37 @@ class Editor:
         :param model_id: model id
         """
         idx = self.get_tab_idx(model_id)
+        if idx is None:
+            return
         current = self.window.ui.models['models.list'].index(idx, 0)
         self.window.ui.nodes['models.list'].setCurrentIndex(current)
 
-    def get_tab_idx(self, model_id: str) -> int:
+    def get_tab_idx(self, model_id: str) -> Optional[int]:
         """
-        Get model list index
+        Get model list index (in the current filtered view)
 
         :param model_id: model id
         :return: list index
         """
-        model_idx = None
-        i = 0
-        self.window.core.models.sort_items()
-        for id in self.window.core.models.get_ids():
-            if id == model_id:
-                model_idx = i
-                break
-            i += 1
-        return model_idx
+        return self.window.model_settings.get_row_by_model_id(model_id)
 
-    def get_tab_by_id(self, model_id: str) -> int:
+    def get_tab_by_id(self, model_id: str) -> Optional[int]:
         """
-        Get model list index
+        Get model list index (alias to get_tab_idx for compatibility)
 
         :param model_id: model id
         :return: list index
         """
-        model_idx = None
-        i = 0
-        self.window.core.models.sort_items()
-        for id in self.window.core.models.get_ids():
-            if id == model_id:
-                model_idx = i
-                break
-            i += 1
-        return model_idx
+        return self.get_tab_idx(model_id)
 
     def get_model_by_tab_idx(self, idx: int) -> Optional[str]:
         """
-        Get model key by list index
+        Get model key by list index (in the current filtered view)
 
         :param idx: list index
         :return: model key
         """
-        model_idx = 0
-        self.window.core.models.sort_items()
-        for id in self.window.core.models.get_ids():
-            if model_idx == idx:
-                return id
-            model_idx += 1
-        return None
+        return self.window.model_settings.get_model_id_by_row(idx)
 
     def open_by_idx(self, idx: int):
         """
@@ -534,4 +512,4 @@ class Editor:
 
     def clear_selected(self):
         """Clear selected list"""
-        self.selected = []
+        self.selected = []

pygpt_net/controller/model/importer.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.26 23:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 import copy
@@ -503,7 +503,7 @@ class Importer:
                 return models
             else:
                 for model in ollama_models:
-                    name = model.get('name').replace(":latest", "")
+                    name = model.get('name')
                     m = self.window.core.models.create_empty(append=False)
                     m.id = name
                     m.name = name
@@ -541,6 +541,13 @@ class Importer:
                 base_models[key] = copy.deepcopy(self.pending[key])
                 base_models[key].imported = True
                 added = True
+            else:
+                # add provider suffix - to key
+                new_key = f"{key}-{self.provider}"
+                if new_key not in base_models:
+                    base_models[new_key] = copy.deepcopy(self.pending[key])
+                    base_models[new_key].imported = True
+                    added = True
         for key in list(self.removed.keys()):
             if key in base_models:
                 del base_models[key]

pygpt_net/controller/painter/common.py

@@ -56,10 +56,6 @@ class Common:
         :param enabled: bool
         """
         if enabled:
-            # keep UI color for compatibility
-            self.window.ui.nodes['painter.select.brush.color'].setCurrentText("Black")
-            self.window.ui.painter.set_brush_color(Qt.black)
-            # switch widget to brush mode (layered painting)
             self.window.ui.painter.set_mode("brush")
             self.window.core.config.set('painter.brush.mode', "brush")
             self.window.core.config.save()
@@ -71,10 +67,6 @@ class Common:
         :param enabled: bool
         """
         if enabled:
-            # keep UI color for compatibility
-            self.window.ui.nodes['painter.select.brush.color'].setCurrentText("White")
-            self.window.ui.painter.set_brush_color(Qt.white)
-            # switch widget to erase mode (layered erasing)
             self.window.ui.painter.set_mode("erase")
             self.window.core.config.set('painter.brush.mode', "erase")
             self.window.core.config.save()

pygpt_net/controller/plugins/plugins.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.15 23:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 from typing import List, Dict, Any, Optional
@@ -34,6 +34,8 @@ class Plugins:
         self.settings = Settings(window)
         self.presets = Presets(window)
         self.enabled = {}
+        self._ids = None
+        self._ids_with_update = None
         self._suspend_updates = 0
 
     def _begin_batch(self):
@@ -316,9 +318,15 @@ class Plugins:
     def on_post_update(self):
         """Called on post update"""
         pm = self.window.core.plugins
-        for pid in pm.get_ids():
+        if self._ids is None:
+            self._ids = pm.get_ids()
+        if self._ids_with_update is None:
+            self._ids_with_update = [pid for pid in self._ids if hasattr(self.window.core.plugins.get(pid), "on_post_update")]
+        if len(self._ids_with_update) == 0:
+            return
+        for pid in self._ids_with_update:
             if self.is_enabled(pid):
-                fn = getattr(pm.get(pid), "on_post_update", None)
+                fn = pm.get(pid).on_post_update
                 if callable(fn):
                     fn()
 

pygpt_net/controller/presets/presets.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.15 03:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 import re
@@ -440,7 +440,7 @@ class Presets:
         if preset_id and preset_id in w.core.presets.items:
             preset = w.core.presets.items[preset_id]
             preset.prompt = w.core.config.get('prompt')
-            w.core.presets.save(preset)
+            w.core.presets.save(preset_id)
 
     def select_model(self):
         """Select model by current preset"""

pygpt_net/core/ctx/bag.py

@@ -6,16 +6,21 @@
 # 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.09.05 18:00:00                  #
 # ================================================== #
 
 from typing import List
+from dataclasses import dataclass, field
 
 from pygpt_net.item.ctx import CtxItem
 
 
+@dataclass(slots=True)
 class Bag:
-    __slots__ = ('window', 'meta', 'tab_id', 'items')
+    window: object = None
+    meta: object = None
+    tab_id: int = 0
+    items: List[CtxItem] = field(default_factory=list)
 
     def __init__(self, window=None):
         """

pygpt_net/core/ctx/reply.py

@@ -6,12 +6,14 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.07.26 18:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
-from typing import Dict, Any
+from typing import Dict, Any, Optional
+from dataclasses import dataclass, field
 
 
+@dataclass(slots=True)
 class ReplyContext:
 
     AGENT_CONTINUE = "agent.continue"
@@ -21,6 +23,15 @@ class ReplyContext:
     EXPERT_CALL = "expert.call"
     EXPERT_RESPONSE = "expert.response"
 
+    type: Optional[object] = None
+    bridge_context: Optional[object] = None
+    ctx: Optional[object] = None
+    prev_ctx: Optional[object] = None
+    parent_id: Optional[object] = None
+    input: str = ""
+    internal: bool = False
+    cmds: list = field(default_factory=list)
+
     def __init__(self):
         """Reply context"""
         self.type = None
@@ -55,7 +66,11 @@ class ReplyContext:
             data["prev_ctx"] = self.prev_ctx.to_dict()
         return data
 
+
+@dataclass(slots=True)
 class Reply:
+    window: Optional[object] = None
+
     def __init__(self, window=None):
         """
         Reply core

pygpt_net/core/db/viewer.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: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 import json
@@ -65,16 +65,13 @@ class Viewer:
         limit_clause = f" LIMIT {limit} OFFSET {offset}"
 
         params = {}
-        if search_query:
-            search_clauses = [f"{column} LIKE :search_query" for column in search_fields]
-            where_clause = f" WHERE ({' OR '.join(search_clauses)})"
-            params['search_query'] = f"%{search_query}%"
-
-        # apply filters
-        # filters = {
-        #     "column1": "value1",
-        #     "column2": "value2"  # AND condition
-        # }
+        if search_query is not None:
+            sq = search_query.strip()
+            if sq:
+                search_clauses = [f"{column} LIKE :search_query" for column in search_fields]
+                where_clause = f" WHERE ({' OR '.join(search_clauses)})"
+                params['search_query'] = f"%{sq}%"
+
         if filters:
             filter_clauses = [f"{column} = :filter_{column}" for column in filters.keys()]
             if where_clause == "":
@@ -82,13 +79,13 @@ class Viewer:
             else:
                 where_clause += f" AND ({' AND '.join(filter_clauses)})"
             for column, value in filters.items():
-                params[f"filter_{column}"] = value  # filter placeholder prefixed with 'filter_'
+                params[f"filter_{column}"] = value
 
         query = f"{base_query}{where_clause}{order_clause}{limit_clause}"
         stmt = text(query).bindparams(**params)
         with self.database.get_db().connect() as conn:
             result = conn.execute(stmt).fetchall()
-            return result
+            return [tuple(r) for r in result]
 
     def count_rows(
             self,
@@ -116,15 +113,12 @@ class Viewer:
         else:
             search_fields = tables[table]['search_fields']
 
-        if search_query:
-            where_clause = f" WHERE {' OR '.join([f'{column} LIKE :search_query' for column in search_fields])}"
-            params['search_query'] = f"%{search_query}%"
+        if search_query is not None:
+            sq = search_query.strip()
+            if sq:
+                where_clause = f" WHERE {' OR '.join([f'{column} LIKE :search_query' for column in search_fields])}"
+                params['search_query'] = f"%{sq}%"
 
-        # apply filters
-        # filters = {
-        #     "column1": "value1",
-        #     "column2": "value2"  # AND condition
-        # }
         if filters:
             filter_clauses = [f"{column} = :filter_{column}" for column in filters.keys()]
             if where_clause == "":
@@ -132,13 +126,13 @@ class Viewer:
             else:
                 where_clause += f" AND ({' AND '.join(filter_clauses)})"
             for column, value in filters.items():
-                params[f"filter_{column}"] = value  # filter placeholder prefixed with 'filter_'
+                params[f"filter_{column}"] = value
 
         query = f"{base_query}{where_clause}"
         stmt = text(query).bindparams(**params)
         with self.database.get_db().connect() as conn:
             count = conn.execute(stmt).scalar()
-            return count
+            return int(count) if count is not None else 0
 
     def is_auto_backup(self) -> bool:
         """
@@ -154,14 +148,12 @@ class Viewer:
 
         :param data: Dictionary with table and row_id keys
         """
-        # create backup
         if self.is_auto_backup():
             backup_path = self.database.make_backup()
             if backup_path:
                 msg = f"[DB] Created DB backup: {backup_path}"
                 self.log(msg)
 
-        # delete row
         with self.database.get_db().begin() as conn:
             conn.execute(
                 text(f"DELETE FROM {data['table']} WHERE id = :row_id")
@@ -184,35 +176,30 @@ class Viewer:
         timestamp_columns = tables[data['table']]['timestamp_columns']
         primary_key = tables[data['table']]['primary_key']
 
-        # check JSON
         if field in json_columns or field.endswith("_json"):
             try:
-                value = json.dumps(json.loads(value))  # validate and pack JSON
+                value = json.dumps(json.loads(value))
             except:
                 raise ValueError(f"Invalid JSON value for column {field}")
 
-        # check timestamp
         if field in timestamp_columns or field.endswith("_ts"):
             try:
                 value = int(value)
             except:
                 raise ValueError(f"Invalid timestamp value for column {field}")
 
-        # check foreign id field
         if field.endswith("_id"):
             try:
                 value = int(value)
             except:
                 raise ValueError(f"Invalid _id value for column {field}")
 
-        # create backup
         if self.is_auto_backup():
             backup_path = self.database.make_backup()
             if backup_path:
                 msg = f"[DB] Created DB backup: {backup_path}"
                 self.log(msg)
 
-        # update row
         with self.database.get_db().begin() as conn:
             conn.execute(
                 text(f"UPDATE {data['table']} SET {data['field']} = :value WHERE {primary_key} = :id")
@@ -229,17 +216,15 @@ class Viewer:
         :param data: Dictionary with table key
         :param reset: Reset table sequence
         """
-        # create backup
         if self.is_auto_backup():
             backup_path = self.database.make_backup()
             if backup_path:
                 msg = f"[DB] Created DB backup: {backup_path}"
                 self.log(msg)
 
-        # truncate table
         with self.database.get_db().begin() as conn:
             conn.execute(text(f"DELETE FROM {data['table']}"))
-            if reset:  # reset table sequence (autoincrement)
+            if reset:
                 conn.execute(text(f"DELETE FROM sqlite_sequence WHERE name='{data['table']}'"))
                 msg = f"[DB] Truncated table {data['table']}"
             else:

pygpt_net/core/render/plain/pid.py

@@ -6,13 +6,24 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.12 19:00:00                  #
+# Updated Date: 2025.09.05 18:00:00                  #
 # ================================================== #
 
 import io
+from dataclasses import dataclass, field
 
+
+@dataclass(slots=True)
 class PidData:
 
+    pid: object = None
+    meta: object = None
+    images_appended: list = field(default_factory=list)
+    urls_appended: list = field(default_factory=list)
+    files_appended: list = field(default_factory=list)
+    _buffer: io.StringIO = field(default_factory=io.StringIO)
+    is_cmd: bool = False
+
     def __init__(self, pid, meta=None):
         """Pid Data"""
         self.pid = pid