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

pygpt_net/plugin/audio_output/__init__.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.13 08:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Slot
@@ -127,12 +127,15 @@ class Plugin(BasePlugin):
         elif name == Event.CTX_AFTER:
             if mode == MODE_AUDIO:
                 return  # skip if audio mode
+            self.stop_audio()
             self.on_generate(ctx, event)
 
         elif name == Event.AUDIO_READ_TEXT:
+            self.stop_audio()
             self.on_generate(ctx, event)
 
         elif name == Event.AUDIO_PLAYBACK:
+            self.stop_audio()
             self.on_playback(ctx, event)
 
         elif name == Event.AUDIO_OUTPUT_STOP:

pygpt_net/plugin/base/config.py

@@ -6,16 +6,18 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.16 05:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
+from typing import Optional
+
 from .plugin import BasePlugin
 
 class BaseConfig:
     def __init__(self, plugin: BasePlugin = None, *args, **kwargs):
         self.plugin = plugin
 
-    def from_defaults(self, plugin: BasePlugin = None):
+    def from_defaults(self, plugin: Optional[BasePlugin] = None):
         """
         Set default options for plugin
 

pygpt_net/plugin/base/plugin.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.21 20:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
 import copy
+from typing import Optional
 
 from PySide6.QtCore import QObject, Slot
 
@@ -234,7 +235,7 @@ class BasePlugin(QObject):
         """
         return
 
-    def trans(self, text: str = None) -> str:
+    def trans(self, text: Optional[str] = None) -> str:
         """
         Translate text using plugin domain
 
@@ -264,7 +265,12 @@ class BasePlugin(QObject):
         """
         self.window.core.debug.info(data)
 
-    def reply(self, response: dict, ctx: CtxItem = None, extra_data: dict = None):
+    def reply(
+            self,
+            response: dict,
+            ctx: Optional[CtxItem] = None,
+            extra_data: Optional[dict] = None
+    ):
         """
         Send reply from plugin (command response)
 
@@ -329,7 +335,12 @@ class BasePlugin(QObject):
         }))
 
     @Slot(object, object, dict)
-    def handle_finished(self, response: dict, ctx: CtxItem = None, extra_data: dict = None):
+    def handle_finished(
+            self,
+            response: dict,
+            ctx: Optional[CtxItem] = None,
+            extra_data: Optional[dict] = None
+    ):
         """
         Handle finished response signal
 
@@ -354,7 +365,12 @@ class BasePlugin(QObject):
             self.window.dispatch(event)
 
     @Slot(object, object, dict)
-    def handle_finished_more(self, responses: list, ctx: CtxItem = None, extra_data: dict = None):
+    def handle_finished_more(
+            self,
+            responses: list,
+            ctx: Optional[CtxItem] = None,
+            extra_data: Optional[dict] = None
+    ):
         """
         Handle finished response signal
 
@@ -379,7 +395,11 @@ class BasePlugin(QObject):
         })
         self.window.dispatch(event)
 
-    def prepare_reply_ctx(self, response: dict, ctx: CtxItem = None) -> dict:
+    def prepare_reply_ctx(
+            self,
+            response: dict,
+            ctx: Optional[CtxItem] = None
+    ) -> dict:
         """
         Prepare reply context
 

pygpt_net/plugin/base/worker.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.24 04:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
+from typing import Optional, Any
+
 from PySide6.QtCore import QObject, QRunnable
 from typing_extensions import deprecated
 
@@ -17,7 +19,7 @@ from .signals import BaseSignals
 
 
 class BaseWorker(QObject, QRunnable):
-    def __init__(self, plugin: BasePlugin = None, *args, **kwargs):
+    def __init__(self, plugin: Optional[BasePlugin] = None, *args, **kwargs):
         QObject.__init__(self)
         QRunnable.__init__(self)
         self.plugin = plugin
@@ -63,7 +65,11 @@ class BaseWorker(QObject, QRunnable):
             self.signals.log.emit(msg)
 
     @deprecated("From 2.1.29: BaseWorker.response() is deprecated, use BaseWorker.reply() instead")
-    def response(self, response: dict, extra_data: dict = None):
+    def response(
+            self,
+            response: dict,
+            extra_data: Optional[dict] = None
+    ):
         """
         Emit finished signal (deprecated)
 
@@ -74,7 +80,11 @@ class BaseWorker(QObject, QRunnable):
         self.debug("BaseWorker.response is deprecated from 2.1.29, use BaseWorker.reply instead")
         self.reply(response, extra_data)
 
-    def reply(self, response: dict, extra_data: dict = None):
+    def reply(
+            self,
+            response: dict,
+            extra_data: Optional[dict] = None
+    ):
         """
         Emit finished signal (on reply from command output)
 
@@ -89,7 +99,11 @@ class BaseWorker(QObject, QRunnable):
         if self.signals is not None and hasattr(self.signals, "finished"):
             self.signals.finished.emit(response, self.ctx, extra_data)
 
-    def reply_more(self, responses: list, extra_data: dict = None):
+    def reply_more(
+            self,
+            responses: list,
+            extra_data: Optional[dict] = None
+    ):
         """
         Emit finished_more signal (on reply from command output, multiple responses)
 
@@ -161,7 +175,12 @@ class BaseWorker(QObject, QRunnable):
         """
         return {"cmd": item["cmd"]}
 
-    def make_response(self, item: dict, result: any, extra: dict = None) -> dict:
+    def make_response(
+            self,
+            item: dict,
+            result: any,
+            extra: Optional[dict] = None
+    ) -> dict:
         """
         Prepare response item
 
@@ -190,7 +209,11 @@ class BaseWorker(QObject, QRunnable):
         self.log(msg)
         return msg
 
-    def has_param(self, item: dict, param: str) -> bool:
+    def has_param(
+            self,
+            item: dict,
+            param: str
+    ) -> bool:
         """
         Check if item has parameter
 
@@ -202,7 +225,12 @@ class BaseWorker(QObject, QRunnable):
             return False
         return "params" in item and param in item["params"]
 
-    def get_param(self, item: dict, param: str, default: any = None) -> any:
+    def get_param(
+            self,
+            item: dict,
+            param: str,
+            default: Any = None
+    ) -> any:
         """
         Get parameter value from item
 
@@ -215,7 +243,7 @@ class BaseWorker(QObject, QRunnable):
             return item["params"][param]
         return default
 
-    def is_stopped(self):
+    def is_stopped(self) -> bool:
         """
         Check if worker is stopped
 

pygpt_net/plugin/cmd_code_interpreter/__init__.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.25 02:00:00                  #
+# Updated Date: 2024.12.12 01:00:00                  #
 # ================================================== #
 
 import os
@@ -118,42 +118,6 @@ class Plugin(BasePlugin):
                                            "directory is mapped as volume in host machine to: {}").format(legacy_data)
                 data['cmd'].append(cmd)  # append command
 
-    @Slot(object, str)
-    def handle_interpreter_output(self, data, type):
-        """
-        Handle interpreter output
-
-        :param data: output data
-        :param type: output type
-        """
-        if not self.get_option_value("attach_output"):
-            return
-        self.window.tools.get("interpreter").append_output(data, type)
-
-    @Slot()
-    def handle_interpreter_clear(self):
-        """Handle interpreter clear"""
-        self.window.tools.get("interpreter").clear_all()
-
-    @Slot(object)
-    def handle_html_output(self, data):
-        """
-        Handle HTML/JS canvas output
-
-        :param data: HTML/JS code
-        """
-        self.window.tools.get("html_canvas").set_output(data)
-        self.window.tools.get("html_canvas").open()
-    def get_interpreter(self):
-        """
-        Get interpreter
-
-        :return: interpreter
-        """
-        if self.get_option_value("sandbox_ipython"):
-            return self.ipython_docker
-        else:
-            return self.ipython_local
 
     def cmd(self, ctx: CtxItem, cmds: list, silent: bool = False):
         """
@@ -270,3 +234,41 @@ class Plugin(BasePlugin):
         self.window.tools.get("interpreter").append_output(cleaned_data)
         if self.window.tools.get("interpreter").opened:
             self.window.update_status("")
+
+    @Slot(object, str)
+    def handle_interpreter_output(self, data, type):
+        """
+        Handle interpreter output
+
+        :param data: output data
+        :param type: output type
+        """
+        if not self.get_option_value("attach_output"):
+            return
+        self.window.tools.get("interpreter").append_output(data, type)
+
+    @Slot()
+    def handle_interpreter_clear(self):
+        """Handle interpreter clear"""
+        self.window.tools.get("interpreter").clear_all()
+
+    @Slot(object)
+    def handle_html_output(self, data):
+        """
+        Handle HTML/JS canvas output
+
+        :param data: HTML/JS code
+        """
+        self.window.tools.get("html_canvas").set_output(data)
+        self.window.tools.get("html_canvas").auto_open()
+        
+    def get_interpreter(self):
+        """
+        Get interpreter
+
+        :return: interpreter
+        """
+        if self.get_option_value("sandbox_ipython"):
+            return self.ipython_docker
+        else:
+            return self.ipython_local

pygpt_net/plugin/cmd_code_interpreter/runner.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 04:00:00                  #
+# Updated Date: 2024.12.12 01:00:00                  #
 # ================================================== #
 
 import os.path
@@ -164,6 +164,14 @@ class Runner:
         """
         return self.plugin.get_option_value('sandbox_docker')
 
+    def is_sandbox_ipython(self) -> bool:
+        """
+        Check if sandbox is enabled for IPython
+
+        :return: True if sandbox is enabled
+        """
+        return self.plugin.get_option_value('sandbox_ipython')
+
     def get_docker(self) -> docker.client.DockerClient:
         """
         Get docker client
@@ -392,13 +400,14 @@ class Runner:
         :param all: execute all
         :return: response dict
         """
+        sandbox = self.is_sandbox_ipython()
         data = item["params"]['code']
         if not all:
             path = self.plugin.window.tools.get("interpreter").file_current
             if "path" in item["params"]:
                 path = item["params"]['path']
             msg = "Saving Python file: {}".format(path)
-            self.log(msg, sandbox=True)
+            self.log(msg, sandbox=sandbox)
             with open(self.prepare_path(path, on_host=True), 'w', encoding="utf-8") as file:
                 file.write(data)
         else:
@@ -412,12 +421,13 @@ class Runner:
 
         # run code in IPython interpreter
         msg = "Executing Python code: {}".format(item["params"]['code'])
-        self.log(msg, sandbox=True)
-        self.log("Connecting to IPython interpreter...", sandbox=True)
+        self.log(msg, sandbox=sandbox)
+        self.log("Connecting to IPython interpreter...", sandbox=sandbox)
         try:
-            self.log("Please wait...", sandbox=True)
+            self.log("Please wait...", sandbox=sandbox)
             result = self.plugin.get_interpreter().execute(data, current=False)
             result = self.handle_result_ipython(ctx, result)
+            self.log("Python Code Executed.", sandbox=sandbox)
         except Exception as e:
             self.error(e)
             result = str(e)
@@ -437,13 +447,14 @@ class Runner:
         :param all: execute all
         :return: response dict
         """
+        sandbox = self.is_sandbox_ipython()
         data = item["params"]['code']
         if not all:
             path = self.plugin.window.tools.get("interpreter").file_current
             if "path" in item["params"]:
                 path = item["params"]['path']
             msg = "Saving Python file: {}".format(path)
-            self.log(msg, sandbox=True)
+            self.log(msg, sandbox=sandbox)
             with open(self.prepare_path(path, on_host=True), 'w', encoding="utf-8") as file:
                 file.write(data)
         else:
@@ -457,12 +468,13 @@ class Runner:
 
         # run code in IPython interpreter
         msg = "Executing Python code: {}".format(item["params"]['code'])
-        self.log(msg, sandbox=True)
-        self.log("Connecting to IPython interpreter...", sandbox=True)
+        self.log(msg, sandbox=sandbox)
+        self.log("Connecting to IPython interpreter...", sandbox=sandbox)
         try:
-            self.log("Please wait...", sandbox=True)
+            self.log("Please wait...", sandbox=sandbox)
             result = self.plugin.get_interpreter().execute(data, current=True)
             result = self.handle_result_ipython(ctx, result)
+            self.log("Python Code Executed.", sandbox=sandbox)
         except Exception as e:
             self.error(e)
             result = str(e)
@@ -482,13 +494,14 @@ class Runner:
         :param all: execute all
         :return: response dict
         """
+        sandbox = self.is_sandbox_ipython()
         self.append_input("")
         self.send_interpreter_input("")  # send input to interpreter tool
 
         # restart IPython interpreter
-        self.log("Connecting to IPython interpreter...", sandbox=True)
+        self.log("Connecting to IPython interpreter...", sandbox=sandbox)
         try:
-            self.log("Restarting IPython kernel...", sandbox=True)
+            self.log("Restarting IPython kernel...", sandbox=sandbox)
             response = self.plugin.get_interpreter().restart_kernel()
         except Exception as e:
             self.error(e)
@@ -497,7 +510,7 @@ class Runner:
             result = "Kernel restarted"
         else:
             result = "Kernel not restarted"
-        self.log(result)
+        self.log(result, sandbox=sandbox)
         return {
             "request": request,
             "result": str(result),

pygpt_net/plugin/cmd_web/__init__.py

@@ -6,7 +6,7 @@
 # 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.13 08:00:00                  #
 # ================================================== #
 
 import ssl
@@ -40,6 +40,8 @@ class Plugin(BasePlugin):
             "web_url_raw",
             "web_index",
             "web_index_query",
+            "web_extract_links",
+            "web_extract_images",
         ]
         self.order = 100
         self.use_locale = True
@@ -269,26 +271,66 @@ class Plugin(BasePlugin):
 
         :param data: event data dict
         """
+        max_urls = self.get_option_value("max_open_urls")
+        is_summary = True
+        if self.get_option_value("raw"):
+            is_summary = False
+
         for option in self.allowed_cmds:
             if not self.has_cmd(option):
                 continue
 
-            # special case for web_index
             if option == "web_index":
                 data['cmd'].append(self.prepare_idx_syntax())  # prepare params
                 continue
             elif option == "web_index_query":
                 data['cmd'].append(self.prepare_idx_query_syntax())  # prepare params
                 continue
+            elif option == "web_url_open":
+                cmd = self.get_cmd(option)
+                try:
+                    cmd["instruction"] = cmd["instruction"].format(
+                        max_urls=max_urls,
+                    )
+                    if not is_summary:
+                        for param in list(cmd["params"]):
+                            if param["name"] == "summarize_prompt":
+                                cmd["params"].remove(param)
+                    data['cmd'].append(cmd)
+                    continue
+                except Exception as e:
+                    pass
+            elif option == "web_url_raw":
+                cmd = self.get_cmd(option)
+                try:
+                    cmd["instruction"] = cmd["instruction"].format(
+                        max_urls=max_urls,
+                    )
+                    if not is_summary:
+                        for param in list(cmd["params"]):
+                            if param["name"] == "summarize_prompt":
+                                cmd["params"].remove(param)
+                    data['cmd'].append(cmd)
+                    continue
+                except Exception as e:
+                    pass
             elif option == "web_search":
                 max_pages = self.get_option_value("num_pages")
                 cmd = self.get_cmd(option)
                 try:
-                    cmd["instruction"] = cmd["instruction"].format(max_pages=max_pages)
+                    cmd["instruction"] = cmd["instruction"].format(
+                        max_pages=max_pages,
+                        max_urls=max_urls,
+                    )
+                    if not is_summary:
+                        for param in list(cmd["params"]):
+                            if param["name"] == "summarize_prompt":
+                                cmd["params"].remove(param)
                     data['cmd'].append(cmd)
                     continue
                 except Exception as e:
                     pass
+
             data['cmd'].append(self.get_cmd(option))  # append command
 
     def cmd(self, ctx: CtxItem, cmds: list):
@@ -389,8 +431,6 @@ class Plugin(BasePlugin):
                     req,
                     timeout=self.get_option_value('timeout'),
                 ).read()
-        except HTTPError as e:
-            data = str(e)
         except Exception as e:
-            raise e
+            data = str(e)
         return data