pygpt-net 2.6.1__py3-none-any.whl → 2.6.6__py3-none-any.whl

pygpt_net/core/agents/runner.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 03:00:00                  #
+# Updated Date: 2025.08.14 13:00:00                  #
 # ================================================== #
 
 import asyncio
@@ -24,6 +24,8 @@ from pygpt_net.core.types import (
     AGENT_MODE_OPENAI,
 )
 
+from pygpt_net.item.ctx import CtxItem
+
 from .runners.llama_assistant import LlamaAssistant
 from .runners.llama_plan import LlamaPlan
 from .runners.llama_steps import LlamaSteps
@@ -31,10 +33,13 @@ from .runners.llama_workflow import LlamaWorkflow
 from .runners.openai_workflow import OpenAIWorkflow
 from .runners.helpers import Helpers
 from .runners.loop import Loop
-from ...item.ctx import CtxItem
-
 
 class Runner:
+
+    APPEND_SYSTEM_PROMPT_TO_MSG = [
+        "react",  # llama-index
+    ]
+
     def __init__(self, window=None):
         """
         Agent runner
@@ -52,9 +57,6 @@ class Runner:
         self.llama_steps = LlamaSteps(window)
         self.llama_workflow = LlamaWorkflow(window)
         self.openai_workflow = OpenAIWorkflow(window)
-        self.APPEND_SYSTEM_PROMPT_TO_MSG = [
-            "react",  # llama-index
-        ]
 
     def call(
             self,
@@ -85,8 +87,9 @@ class Runner:
 
             # prepare agent
             model = context.model
-            vector_store_idx = extra.get("agent_idx", None)            
+            vector_store_idx = extra.get("agent_idx", None)
             system_prompt = context.system_prompt
+            preset = context.preset
             max_steps = self.window.core.config.get("agent.llama.steps", 10)
             is_stream = self.window.core.config.get("stream", False)
             is_cmd = self.window.core.command.is_cmd(inline=False)
@@ -94,6 +97,10 @@ class Runner:
             llm = self.window.core.idx.llm.get(model, stream=False)
             workdir = self.window.core.config.get_workdir_prefix()
 
+            # vector store idx from preset
+            if preset:
+                vector_store_idx = preset.idx
+
             # tools
             self.window.core.agents.tools.context = context
             self.window.core.agents.tools.agent_idx = vector_store_idx
@@ -174,6 +181,7 @@ class Runner:
                 return asyncio.run(self.openai_workflow.run(**kwargs))
 
         except Exception as e:
+            print("Error in agent runner:", e)
             self.window.core.debug.error(e)
             self.last_error = e
             return False

pygpt_net/core/bridge/bridge.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.05 21:00:00                  #
+# Updated Date: 2025.08.15 23:00:00                  #
 # ================================================== #
 
 import time
@@ -39,10 +39,10 @@ class Bridge:
         self.last_call = None  # last API call time, for throttling
         self.last_context = None  # last context
         self.last_context_quick = None  # last context for quick call
-        self.sync_modes = [
+        self.sync_modes = (
             MODE_ASSISTANT,
             MODE_EXPERT,
-        ]
+        )
         self.worker = None
 
     def request(
@@ -59,7 +59,7 @@ class Bridge:
         if self.window.controller.kernel.stopped():
             return False
 
-        allowed_model_change = [MODE_VISION]
+        allowed_model_change = MODE_VISION
         is_virtual = False
         force_sync = False
 
@@ -81,7 +81,7 @@ class Bridge:
         context.parent_mode = base_mode  # store base mode
 
         # get agent or expert internal sub-mode
-        if base_mode in [MODE_AGENT, MODE_EXPERT]:
+        if base_mode in (MODE_AGENT, MODE_EXPERT):
             is_virtual = True
             sub_mode = None  # inline switch to sub-mode, because agent is a virtual mode only
             if base_mode == MODE_AGENT: