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

pygpt_net/controller/chat/response.py

@@ -6,9 +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 08:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
 from pygpt_net.core.types import (
     MODE_AGENT_LLAMA,
     MODE_ASSISTANT,
@@ -29,7 +31,12 @@ class Response:
         super(Response, self).__init__()
         self.window = window
 
-    def handle(self, context: BridgeContext, extra: dict, status: bool):
+    def handle(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any],
+            status: bool
+    ):
         """
         Handle Bridge success
 
@@ -92,7 +99,11 @@ class Response:
         self.window.controller.chat.output.post_handle(ctx, mode, stream, reply, internal)
         self.window.controller.chat.output.handle_end(ctx, mode)  # handle end.
 
-    def begin(self, context: BridgeContext, extra: dict):
+    def begin(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any]
+    ):
         """
         Handle Bridge begin
 
@@ -104,7 +115,11 @@ class Response:
         if msg:
             self.window.update_status(msg)
 
-    def append(self, context: BridgeContext, extra: dict):
+    def append(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any]
+    ):
         """
         Handle Bridge append (agent mode)
 
@@ -204,7 +219,11 @@ class Response:
         if ctx.extra is not None and (type(ctx.extra) == dict and "agent_finish" in ctx.extra):
             self.window.controller.agent.llama.on_finish(ctx)  # evaluate response and continue if needed
 
-    def end(self, context: BridgeContext, extra: dict):
+    def end(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any]
+    ):
         """
         Handle Bridge end
 
@@ -223,7 +242,11 @@ class Response:
             "id": "chat",
         }))
 
-    def failed(self, context: BridgeContext, extra: dict):
+    def failed(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any]
+    ):
         """
         Handle Bridge failed
 
@@ -240,7 +263,11 @@ class Response:
             "id": "chat",
         }))
 
-    def update_status(self, context: BridgeContext, extra: dict):
+    def update_status(
+            self,
+            context: BridgeContext,
+            extra: Dict[str, Any]
+    ):
         """
         Handle Bridge evaluate
 

pygpt_net/controller/chat/stream.py

@@ -6,9 +6,11 @@
 # 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.14 00:00:00                  #
 # ================================================== #
 
+from typing import Any
+
 from PySide6.QtWidgets import QApplication
 
 from pygpt_net.core.events import RenderEvent
@@ -175,7 +177,7 @@ class Stream:
         # log
         self.log("[chat] Stream end.")
 
-    def log(self, data: any):
+    def log(self, data: Any):
         """
         Log data to debug
 

pygpt_net/controller/chat/text.py

@@ -6,9 +6,11 @@
 # 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.14 00:00:00                  #
 # ================================================== #
 
+from typing import Optional
+
 from pygpt_net.core.types import (
     MODE_AGENT,
     MODE_AGENT_LLAMA,
@@ -37,9 +39,9 @@ class Text:
             text: str,
             reply: bool = False,
             internal: bool = False,
-            prev_ctx: CtxItem = None,
-            parent_id: str = None,
-            multimodal_ctx: MultimodalContext = None,
+            prev_ctx: Optional[CtxItem] = None,
+            parent_id: Optional[str] = None,
+            multimodal_ctx: Optional[MultimodalContext] = None,
     ) -> CtxItem:
         """
         Send text message

pygpt_net/controller/command.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.14 08:00:00                  #
 # ================================================== #
 
 import json
@@ -27,7 +27,10 @@ class Command:
         """
         self.window = window
         self.stop = False
-        self.flush_events = [Event.CMD_EXECUTE, Event.CMD_INLINE]
+        self.flush_events = [
+            Event.CMD_EXECUTE,
+            Event.CMD_INLINE,
+        ]
 
     def dispatch(self, event: Event, all: bool = False):
         """
@@ -88,7 +91,12 @@ class Command:
         worker.kwargs['finished_signal'] = worker.signals.finished
         self.window.threadpool.start(worker)
 
-    def worker(self, event: Event, window, finished_signal: Signal):
+    def worker(
+            self,
+            event: Event,
+            window,
+            finished_signal: Signal
+    ):
         """
         Command worker callback
 

pygpt_net/controller/config/__init__.py

@@ -6,9 +6,11 @@
 # 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: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Dict
+
 from .field.checkbox import Checkbox
 from .field.combo import Combo
 from .field.cmd import Cmd
@@ -36,7 +38,11 @@ class Config:
         self.slider = Slider(window)
         self.textarea = Textarea(window)
 
-    def load_options(self, parent_id: str, options: dict):
+    def load_options(
+            self,
+            parent_id: str,
+            options: Dict[str, Dict[str, Any]]
+    ):
         """
         Load options
 
@@ -47,7 +53,12 @@ class Config:
             option = options[key]
             self.apply(parent_id, key, option)
 
-    def apply(self, parent_id: str, key: str, option: dict):
+    def apply(
+            self,
+            parent_id: str,
+            key: str,
+            option: Dict[str, Any]
+    ):
         """
         Apply option to field handler based on type
 
@@ -73,7 +84,13 @@ class Config:
         elif option['type'] == 'cmd':
             self.cmd.apply(parent_id, key, option)
 
-    def apply_value(self, parent_id: str, key: str, option: dict, value):
+    def apply_value(
+            self,
+            parent_id: str,
+            key: str,
+            option: Dict[str, Any],
+            value: Any
+    ):
         """
         Apply value to option
 
@@ -85,7 +102,13 @@ class Config:
         option['value'] = value
         self.apply(parent_id, key, option)
 
-    def get_value(self, parent_id: str, key: str, option: dict, idx: bool = False) -> any:
+    def get_value(
+            self,
+            parent_id: str,
+            key: str,
+            option: Dict[str, Any],
+            idx: bool = False
+    ) -> Any:
         """
         Get value from field handler based on type
 
@@ -113,7 +136,12 @@ class Config:
         elif option['type'] == 'cmd':
             return self.cmd.get_value(parent_id, key, option)
 
-    def update_combo(self, parent_id: str, key: str, items: dict):
+    def update_combo(
+            self,
+            parent_id: str,
+            key: str,
+            items: Dict[str, str]
+    ):
         """
         Update combo items
 

pygpt_net/controller/config/field/checkbox.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.01.30 20:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Dict
+
+
 class Checkbox:
     def __init__(self, window=None):
         """
@@ -22,7 +25,7 @@ class Checkbox:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ):
         """
         Apply value to checkbox
@@ -42,7 +45,7 @@ class Checkbox:
             parent_id: str,
             key: str,
             option: dict,
-            value: any,
+            value: Any,
             hooks: bool = True
     ):
         """
@@ -68,7 +71,7 @@ class Checkbox:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ) -> bool:
         """
         Get checkbox value

pygpt_net/controller/config/field/cmd.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.12 06:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
 from pygpt_net.utils import trans
 
 
@@ -25,7 +27,7 @@ class Cmd:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ):
         """
         Apply values to command fields
@@ -57,7 +59,7 @@ class Cmd:
             self,
             parent_id: str,
             key: str,
-            values: dict,
+            values: Dict[str, Any],
             idx: int
     ):
         """
@@ -77,7 +79,7 @@ class Cmd:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ) -> dict:
         """
         Get command values
@@ -97,7 +99,7 @@ class Cmd:
     def to_options(
             self,
             parent_id: str,
-            option: dict
+            option: Dict[str, Any]
     ) -> dict:
         """
         Convert dictionary items option to options

pygpt_net/controller/config/field/combo.py

@@ -6,9 +6,12 @@
 # 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: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Dict
+
+
 class Combo:
     def __init__(self, window=None):
         """
@@ -22,7 +25,7 @@ class Combo:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ):
         """
         Apply value to combobox
@@ -44,8 +47,8 @@ class Combo:
             self,
             parent_id: str,
             key: str,
-            option: dict,
-            value: any,
+            option: Dict[str, Any],
+            value: Any,
             hooks: bool = True
     ):
         """
@@ -71,7 +74,7 @@ class Combo:
             self,
             parent_id: str,
             key: str,
-            option: dict,
+            option: Dict[str, Any],
             idx: bool = False
     ) -> str or int:
         """
@@ -88,7 +91,12 @@ class Combo:
         else:
             return self.window.ui.config[parent_id][key].combo.currentData()
 
-    def update_list(self, parent_id: str, key: str, items: dict):
+    def update_list(
+            self,
+            parent_id: str,
+            key: str,
+            items: Dict[str, str]
+    ):
         """
         Update combobox items
 

pygpt_net/controller/config/field/dictionary.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.12 06:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any, List
+
+from pygpt_net.ui.widget.option.dictionary import OptionDict
 from pygpt_net.utils import trans
 
 
@@ -25,7 +28,7 @@ class Dictionary:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ):
         """
         Apply values to dictionary
@@ -44,7 +47,7 @@ class Dictionary:
             self,
             parent_id: str,
             key: str,
-            values: dict,
+            values: Dict[str, Any],
             idx: int
     ):
         """
@@ -64,8 +67,8 @@ class Dictionary:
             self,
             parent_id: str,
             key: str,
-            option: dict
-    ) -> list:
+            option: Dict[str, Any]
+    ) -> List[Dict[str, Any]]:
         """
         Get dictionary values
 
@@ -78,7 +81,7 @@ class Dictionary:
 
     def delete_item(
             self,
-            parent_object,
+            parent_object: OptionDict,
             id: str,
             force: bool = False,
             hooks: bool = True
@@ -117,8 +120,8 @@ class Dictionary:
     def to_options(
             self,
             parent_id: str,
-            option: dict
-    ) -> dict:
+            option: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """
         Convert dictionary items option to options
 
@@ -145,8 +148,8 @@ class Dictionary:
     def append_editor(
             self,
             id: str,
-            option: dict,
-            data: dict
+            option: Dict[str, Any],
+            data: Dict[str, Any]
     ):
         """
         Apply dict item option sub-values to dict editor
@@ -173,7 +176,7 @@ class Dictionary:
             self,
             option_key: str,
             parent: str,
-            fields: dict,
+            fields: Dict[str, Dict[str, Any]],
             hooks: bool = True
     ):
         """

pygpt_net/controller/config/field/input.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.15 12:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Dict, Optional, Union
+
+
 class Input:
     def __init__(self, window=None):
         """
@@ -22,7 +25,7 @@ class Input:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ):
         """
         Apply value to input
@@ -60,8 +63,8 @@ class Input:
             self,
             parent_id: str,
             key: str,
-            option: dict,
-            value: any,
+            option: Dict[str, Any],
+            value: Any,
             hooks: bool = True
     ):
         """
@@ -90,8 +93,8 @@ class Input:
             self,
             parent_id: str,
             key: str,
-            option: dict
-    ) -> any:
+            option: Dict[str, Any]
+    ) -> Optional[Union[int, float, str]]:
         """
         Get value from input
 

pygpt_net/controller/config/field/slider.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.20 06:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Optional, Dict, Union
+
+
 class Slider:
     def __init__(self, window=None):
         """
@@ -22,8 +25,8 @@ class Slider:
             self,
             parent_id: str,
             key: str,
-            option: dict,
-            type: str = None
+            option: Dict[str, Any],
+            type: Optional[str] = None
     ):
         """
         Apply value to slider
@@ -89,9 +92,9 @@ class Slider:
             self,
             parent_id: str,
             key: str,
-            option: dict,
-            value: any,
-            type: str = None,
+            option: Dict[str, Any],
+            value: Any,
+            type: Optional[str] = None,
             hooks: bool = True
     ):
         """
@@ -121,8 +124,8 @@ class Slider:
             self,
             parent_id: str,
             key: str,
-            option: dict
-    ) -> any:
+            option: Dict[str, Any]
+    ) -> Union[int, float]:
         """
         Get slider value
 

pygpt_net/controller/config/field/textarea.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.11 01:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Any, Dict
+
+
 class Textarea:
     def __init__(self, window=None):
         """
@@ -22,7 +25,7 @@ class Textarea:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ):
         """
         Apply value to textarea
@@ -38,8 +41,8 @@ class Textarea:
             self,
             parent_id: str,
             key: str,
-            option: dict,
-            value: any,
+            option: Dict[str, Any],
+            value: Any,
             hooks: bool = True
     ):
         """
@@ -68,7 +71,7 @@ class Textarea:
             self,
             parent_id: str,
             key: str,
-            option: dict
+            option: Dict[str, Any]
     ) -> str:
         """
         Get value of textarea