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

pygpt_net/plugin/mailer/worker.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.12.15 04:00:00                  #
+# ================================================== #
+
+from PySide6.QtCore import Slot, Signal
+
+from pygpt_net.plugin.base.worker import BaseWorker, BaseSignals
+
+
+class WorkerSignals(BaseSignals):
+    output = Signal(object, str)
+    html_output = Signal(object)
+    ipython_output = Signal(object)
+    build_finished = Signal()
+    clear = Signal()
+
+
+class Worker(BaseWorker):
+    def __init__(self, *args, **kwargs):
+        super(Worker, self).__init__()
+        self.signals = WorkerSignals()
+        self.args = args
+        self.kwargs = kwargs
+        self.plugin = None
+        self.cmds = None
+        self.ctx = None
+
+    @Slot()
+    def run(self):
+        responses = []
+        for item in self.cmds:
+            if self.is_stopped():
+                break
+            try:
+                response = None
+                if (item["cmd"] in self.plugin.allowed_cmds
+                        and (self.plugin.has_cmd(item["cmd"]) or 'force' in item)):
+
+                    if item["cmd"] == "send_email":
+                        response = self.cmd_send_mail(item)
+
+                    elif item["cmd"] == "get_emails":
+                        response = self.cmd_receive_emails(item)
+
+                    elif item["cmd"] == "get_email_body":
+                        response = self.cmd_get_email_body(item)
+
+                    if response:
+                        responses.append(response)
+
+            except Exception as e:
+                responses.append(
+                    self.make_response(
+                        item,
+                        self.throw_error(e)
+                    )
+                )
+
+        # send response
+        if len(responses) > 0:
+            self.reply_more(responses)
+
+    def cmd_send_mail(self, item: dict) -> dict:
+        """
+        Send email command
+
+        :param item: command item
+        :return: response item
+        """
+        request = self.from_request(item)
+        try:
+            result = self.plugin.runner.smtp_send_email(
+                ctx=self.ctx,
+                item=item,
+                request=request,
+            )
+        except Exception as e:
+            result = self.throw_error(e)
+        return self.make_response(item, result)
+
+    def cmd_receive_emails(self, item: dict) -> dict:
+        """
+        Receive emails command
+
+        :param item: command item
+        :return: response item
+        """
+        request = self.from_request(item)
+        try:
+            result = self.plugin.runner.smtp_receive_emails(
+                ctx=self.ctx,
+                item=item,
+                request=request,
+            )
+        except Exception as e:
+            result = self.throw_error(e)
+        return self.make_response(item, result)
+
+    def cmd_get_email_body(self, item: dict) -> dict:
+        """
+        Get email body command
+
+        :param item: command item
+        :return: response item
+        """
+        request = self.from_request(item)
+        try:
+            result = self.plugin.runner.smtp_get_email_body(
+                ctx=self.ctx,
+                item=item,
+                request=request,
+            )
+        except Exception as e:
+            result = self.throw_error(e)
+        return self.make_response(item, result)
+

pygpt_net/provider/agents/base.py

@@ -6,9 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.15 00:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
+
 class BaseAgent:
     def __init__(self, *args, **kwargs):
         self.id = ""
@@ -22,7 +25,7 @@ class BaseAgent:
         """
         return self.mode
 
-    def get_agent(self, window, kwargs: dict):
+    def get_agent(self, window, kwargs: Dict[str, Any]):
         """
         Return Agent provider instance
 

pygpt_net/provider/agents/openai.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.15 00:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
 from llama_index.agent.openai import OpenAIAgent as Agent
 
 from .base import BaseAgent
@@ -19,7 +21,7 @@ class OpenAIAgent(BaseAgent):
         self.id = "openai"
         self.mode = "step"  # step|plan
 
-    def get_agent(self, window, kwargs: dict):
+    def get_agent(self, window, kwargs: Dict[str, Any]):
         """
         Return Agent provider instance
 

pygpt_net/provider/agents/openai_assistant.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.15 00:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
 from llama_index.agent.openai import OpenAIAssistantAgent as Agent
 
 from pygpt_net.core.bridge.context import BridgeContext
@@ -21,7 +23,7 @@ class OpenAIAssistantAgent(BaseAgent):
         self.id = "openai_assistant"
         self.mode = "assistant"  # step|plan|...
 
-    def get_agent(self, window, kwargs: dict):
+    def get_agent(self, window, kwargs: Dict[str, Any]):
         """
         Return Agent provider instance
 

pygpt_net/provider/agents/planner.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.15 00:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
 from llama_index.core.agent import (
     StructuredPlannerAgent,
     FunctionCallingAgentWorker,
@@ -23,7 +25,7 @@ class PlannerAgent(BaseAgent):
         self.id = "planner"
         self.mode = "plan"  # step|plan
 
-    def get_agent(self, window, kwargs: dict):
+    def get_agent(self, window, kwargs: Dict[str, Any]):
         """
         Return Agent provider instance
 

pygpt_net/provider/agents/react.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 22:00:00                  #
 # ================================================== #
 
+from typing import Dict, Any
+
 from llama_index.core.agent import ReActAgent as Agent
 from llama_index.core.agent.react_multimodal.step import (
     MultimodalReActAgentWorker,
@@ -26,7 +28,7 @@ class ReactAgent(BaseAgent):
         self.id = "react"
         self.mode = "step"  # step|plan
 
-    def get_agent(self, window, kwargs: dict):
+    def get_agent(self, window, kwargs: Dict[str, Any]):
         """
         Return Agent provider instance
 

pygpt_net/provider/audio_output/openai_tts.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.02.23 19:00:00                  #
+# Updated Date: 2024.12.14 18:00:00                  #
 # ================================================== #
 
 import os
@@ -26,14 +26,6 @@ class OpenAITextToSpeech(BaseProvider):
         self.plugin = kwargs.get("plugin")
         self.id = "openai_tts"
         self.name = "OpenAI TTS"
-        self.allowed_voices = [
-            'alloy',
-            'echo',
-            'fable',
-            'onyx',
-            'nova',
-            'shimmer',
-        ]
         self.allowed_models = [
             'tts-1',
             'tts-1-hd',
@@ -52,10 +44,11 @@ class OpenAITextToSpeech(BaseProvider):
         )
         self.plugin.add_option(
             "openai_voice",
-            type="text",
+            type="combo",
             value="alloy",
             label="Voice",
             tab="openai_tts",
+            use="audio_tts_whisper_voices",
             description="Specify voice, available voices: "
                         "alloy, echo, fable, onyx, nova, shimmer",
         )
@@ -71,9 +64,10 @@ class OpenAITextToSpeech(BaseProvider):
         output_file = self.plugin.output_file
         voice = self.plugin.get_option_value('openai_voice')
         model = self.plugin.get_option_value('openai_model')
+        allowed_voices = self.plugin.window.core.audio.whisper.get_voices()
         if model not in self.allowed_models:
             model = 'tts-1'
-        if voice not in self.allowed_voices:
+        if voice not in allowed_voices:
             voice = 'alloy'
         path = os.path.join(
             self.plugin.window.core.config.path,

pygpt_net/provider/core/assistant/base.py

@@ -6,9 +6,11 @@
 # 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: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict
+
 from packaging.version import Version
 
 from pygpt_net.item.assistant import AssistantItem
@@ -32,10 +34,10 @@ class BaseProvider:
     def create(self, assistant: AssistantItem) -> str:
         pass
 
-    def load(self) -> dict:
+    def load(self) -> Dict[str, AssistantItem]:
         pass
 
-    def save(self, items: dict):
+    def save(self, items: Dict[str, AssistantItem]):
         pass
 
     def remove(self, id: str):

pygpt_net/provider/core/assistant/json_file.py

@@ -6,12 +6,13 @@
 # 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: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import json
 import os
 import uuid
+from typing import Dict, Any
 
 from packaging.version import Version
 
@@ -48,7 +49,7 @@ class JsonFileProvider(BaseProvider):
             assistant.id = self.create_id()
         return assistant.id
 
-    def load(self) -> dict:
+    def load(self) -> Dict[str, AssistantItem]:
         """
         Load assistants from file
 
@@ -75,9 +76,11 @@ class JsonFileProvider(BaseProvider):
 
         return items
 
-    def save(self, items: dict):
+    def save(self, items: Dict[str, AssistantItem]):
         """
         Save assistants to file
+
+        :param items: dict of assistants
         """
         try:
             # update assistants
@@ -122,7 +125,7 @@ class JsonFileProvider(BaseProvider):
         return False
 
     @staticmethod
-    def serialize(item: AssistantItem) -> dict:
+    def serialize(item: AssistantItem) -> Dict[str, Any]:
         """
         Serialize item to dict
 
@@ -149,7 +152,7 @@ class JsonFileProvider(BaseProvider):
         }
 
     @staticmethod
-    def deserialize(data: dict, item: AssistantItem):
+    def deserialize(data: Dict[str, Any], item: AssistantItem):
         """
         Deserialize item from dict
 

pygpt_net/provider/core/assistant_file/base.py

@@ -6,9 +6,10 @@
 # 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 22:00:00                  #
 # ================================================== #
 
+from typing import Dict
 from packaging.version import Version
 
 from pygpt_net.item.assistant import AssistantFileItem
@@ -35,13 +36,13 @@ class BaseProvider:
     def load(self, id) ->  AssistantFileItem:
         pass
 
-    def load_all(self) -> dict:
+    def load_all(self) -> Dict[str, AssistantFileItem]:
         pass
 
     def save(self, file:  AssistantFileItem):
         pass
 
-    def save_all(self, items: dict):
+    def save_all(self, items: Dict[str, AssistantFileItem]):
         pass
 
     def remove(self, id):

pygpt_net/provider/core/assistant_file/db_sqlite/__init__.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.29 16:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import uuid
+from typing import Dict
 
 from packaging.version import Version
 
@@ -67,7 +68,7 @@ class DbSqliteProvider(BaseProvider):
             file.record_id = self.storage.insert(file)
         return file.record_id
 
-    def load_all(self) -> dict:
+    def load_all(self) -> Dict[str, AssistantFileItem]:
         """
         Load files from DB
 
@@ -96,7 +97,7 @@ class DbSqliteProvider(BaseProvider):
             self.window.core.debug.log(e)
             print("Error while saving filed: {}".format(str(e)))
 
-    def save_all(self, items: dict):
+    def save_all(self, items: Dict[str, AssistantFileItem]):
         """
         Save all files to DB
 

pygpt_net/provider/core/assistant_file/db_sqlite/storage.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.29 16:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import time
+from typing import Dict
 
 from sqlalchemy import text
 
@@ -34,7 +35,7 @@ class Storage:
         """
         self.window = window
 
-    def get_all(self) -> dict:
+    def get_all(self) -> Dict[str, AssistantFileItem]:
         """
         Return dict with AssistantFileItem objects, indexed by ID
 

pygpt_net/provider/core/assistant_store/base.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 22:00:00                  #
 # ================================================== #
 
+from typing import Dict
+
 from packaging.version import Version
 
 from pygpt_net.item.assistant import AssistantStoreItem
@@ -32,16 +34,16 @@ class BaseProvider:
     def create(self, store: AssistantStoreItem) -> str:
         pass
 
-    def load(self, id) -> dict:
+    def load(self, id) -> AssistantStoreItem:
         pass
 
-    def load_all(self) -> dict:
+    def load_all(self) -> Dict[str, AssistantStoreItem]:
         pass
 
     def save(self, file:  AssistantStoreItem):
         pass
 
-    def save_all(self, items: dict):
+    def save_all(self, items: Dict[str, AssistantStoreItem]):
         pass
 
     def remove(self, id: str):

pygpt_net/provider/core/assistant_store/db_sqlite/__init__.py

@@ -1,15 +1,16 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 # ================================================== #
-# This store is a part of PYGPT package               #
+# This store is a part of PYGPT package              #
 # Website: https://pygpt.net                         #
 # 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 22:00:00                  #
 # ================================================== #
 
 import uuid
+from typing import Dict
 
 from packaging.version import Version
 
@@ -62,7 +63,7 @@ class DbSqliteProvider(BaseProvider):
             store.record_id = self.storage.insert(store)
         return store.record_id
 
-    def load_all(self) -> dict:
+    def load_all(self) -> Dict[str, AssistantStoreItem]:
         """
         Load stores from DB
 
@@ -91,7 +92,7 @@ class DbSqliteProvider(BaseProvider):
             self.window.core.debug.log(e)
             print("Error while saving stored: {}".format(str(e)))
 
-    def save_all(self, items: dict):
+    def save_all(self, items: Dict[str, AssistantStoreItem]):
         """
         Save all stores to DB
 

pygpt_net/provider/core/assistant_store/db_sqlite/storage.py

@@ -1,19 +1,21 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 # ================================================== #
-# This store is a part of PYGPT package               #
+# This store is a part of PYGPT package              #
 # Website: https://pygpt.net                         #
 # 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 22:00:00                  #
 # ================================================== #
 
 import time
+from typing import Dict
 
 from sqlalchemy import text
 
 from pygpt_net.item.assistant import AssistantStoreItem
+
 from .utils import pack_item_value, unpack_store
 
 class Storage:
@@ -33,7 +35,7 @@ class Storage:
         """
         self.window = window
 
-    def get_all(self) -> dict:
+    def get_all(self) -> Dict[str, AssistantStoreItem]:
         """
         Return dict with AssistantStoreItem objects, indexed by ID
 

pygpt_net/provider/core/attachment/base.py

@@ -6,9 +6,11 @@
 # 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: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict
+
 from packaging.version import Version
 
 from pygpt_net.item.attachment import AttachmentItem
@@ -32,10 +34,10 @@ class BaseProvider:
     def create(self, attachment: AttachmentItem) -> str:
         pass
 
-    def load(self) -> dict:
+    def load(self) -> Dict[str, Dict[str, AttachmentItem]]:
         pass
 
-    def save(self, items: dict):
+    def save(self, items: Dict[str, Dict[str, AttachmentItem]]):
         pass
 
     def remove(self, id: str):

pygpt_net/provider/core/attachment/json_file.py

@@ -6,12 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.26 02:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
 import json
 import os
 import uuid
+from typing import Dict
 
 from packaging.version import Version
 
@@ -46,7 +47,7 @@ class JsonFileProvider(BaseProvider):
             attachment.id = self.create_id()
         return attachment.id
 
-    def load(self) -> dict:
+    def load(self) -> Dict[str, Dict[str, AttachmentItem]]:
         """
         Load attachments from file
 
@@ -74,7 +75,7 @@ class JsonFileProvider(BaseProvider):
 
         return items
 
-    def save(self, items: dict):
+    def save(self, items: Dict[str, Dict[str, AttachmentItem]]):
         """
         Save attachments to file
         """

pygpt_net/provider/core/calendar/base.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.01.06 04:00:00                  #
+# Updated Date: 2024.12.14 22:00:00                  #
 # ================================================== #
 
+from typing import Dict
+
 from packaging.version import Version
 
 from pygpt_net.item.calendar_note import CalendarNoteItem
@@ -35,13 +37,13 @@ class BaseProvider:
     def load(self, year, month, day) -> CalendarNoteItem:
         pass
 
-    def load_all(self) -> dict:
+    def load_all(self) -> Dict[str, CalendarNoteItem]:
         pass
 
     def save(self, notepad: CalendarNoteItem):
         pass
 
-    def save_all(self, items: dict):
+    def save_all(self, items: Dict[str, CalendarNoteItem]):
         pass
 
     def remove(self, year, month, day):