pygpt-net 2.6.53__py3-none-any.whl → 2.6.54__py3-none-any.whl

pygpt_net/plugin/wolfram/__init__.py

@@ -0,0 +1,12 @@
+#!/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: 2025.06.30 02:00:00                  #
+# ================================================== #
+
+from .plugin import *

pygpt_net/plugin/wolfram/config.py

@@ -0,0 +1,214 @@
+#!/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: 2025.09.17 00:00:00                  #
+# ================================================== #
+
+from pygpt_net.plugin.base.config import BaseConfig, BasePlugin
+
+
+class Config(BaseConfig):
+    def __init__(self, plugin: BasePlugin = None, *args, **kwargs):
+        super(Config, self).__init__(plugin)
+        self.plugin = plugin
+
+    def from_defaults(self, plugin: BasePlugin = None):
+        # Endpoints / HTTP
+        plugin.add_option(
+            "api_base",
+            type="text",
+            value="https://api.wolframalpha.com",
+            label="API base",
+            description="Base API URL (default https://api.wolframalpha.com).",
+        )
+        plugin.add_option(
+            "http_timeout",
+            type="int",
+            value=30,
+            label="HTTP timeout (s)",
+            description="Requests timeout in seconds.",
+        )
+
+        # Auth / Keys
+        plugin.add_option(
+            "wa_appid",
+            type="text",
+            value="",
+            label="Wolfram Alpha AppID",
+            description="Your Wolfram Alpha AppID (https://developer.wolframalpha.com).",
+            secret=True,
+        )
+
+        # Defaults
+        plugin.add_option(
+            "units",
+            type="text",
+            value="metric",
+            label="Units",
+            description="metric | nonmetric (applied where supported).",
+        )
+        plugin.add_option(
+            "simple_background",
+            type="text",
+            value="white",
+            label="Simple API background",
+            description="white | transparent",
+        )
+        plugin.add_option(
+            "simple_layout",
+            type="text",
+            value="labelbar",
+            label="Simple API layout",
+            description="labelbar | inputonly | etc.",
+        )
+        plugin.add_option(
+            "simple_width",
+            type="int",
+            value=600,
+            label="Simple API width",
+            description="Image width in pixels (optional).",
+        )
+
+        # ---------------- Commands ----------------
+
+        # Generic endpoints
+        plugin.add_cmd(
+            "wa_short",
+            instruction="WolframAlpha: Short answer (concise text).",
+            params=[{"name": "query", "type": "str", "required": True, "description": "Natural language or math"}],
+            enabled=True,
+            description="Wolfram: short answer",
+            tab="query",
+        )
+        plugin.add_cmd(
+            "wa_spoken",
+            instruction="WolframAlpha: Spoken result (single sentence).",
+            params=[{"name": "query", "type": "str", "required": True, "description": "Natural language or math"}],
+            enabled=True,
+            description="Wolfram: spoken",
+            tab="query",
+        )
+        plugin.add_cmd(
+            "wa_simple",
+            instruction="WolframAlpha: Simple image result (PNG/GIF).",
+            params=[
+                {"name": "query", "type": "str", "required": True, "description": "Query"},
+                {"name": "out", "type": "str", "required": False, "description": "Output file path (relative or absolute)"},
+                {"name": "background", "type": "str", "required": False, "description": "white|transparent"},
+                {"name": "layout", "type": "str", "required": False, "description": "labelbar|inputonly|..."},
+                {"name": "width", "type": "int", "required": False, "description": "Image width"},
+            ],
+            enabled=True,
+            description="Wolfram: simple image",
+            tab="query",
+        )
+        plugin.add_cmd(
+            "wa_query",
+            instruction="WolframAlpha: Full JSON query (pods).",
+            params=[
+                {"name": "query", "type": "str", "required": True, "description": "Query"},
+                {"name": "format", "type": "str", "required": False, "description": "plaintext,image,..."},
+                {"name": "assumptions", "type": "list", "required": False, "description": "List of 'assumption' strings"},
+                {"name": "podstate", "type": "str", "required": False, "description": "Pod state id"},
+                {"name": "scantimeout", "type": "int", "required": False, "description": "Scan timeout (s)"},
+                {"name": "podtimeout", "type": "int", "required": False, "description": "Pod timeout (s)"},
+                {"name": "maxwidth", "type": "int", "required": False, "description": "Max image width"},
+                {"name": "download_images", "type": "bool", "required": False, "description": "Download pod images"},
+                {"name": "max_images", "type": "int", "required": False, "description": "Max images to download"},
+            ],
+            enabled=True,
+            description="Wolfram: full query (pods)",
+            tab="query",
+        )
+
+        # Math convenience
+        plugin.add_cmd(
+            "wa_calculate",
+            instruction="WolframAlpha: Evaluate/simplify numeric or symbolic expression.",
+            params=[{"name": "expr", "type": "str", "required": True, "description": "Expression"}],
+            enabled=True,
+            description="Math: calculate",
+            tab="math",
+        )
+        plugin.add_cmd(
+            "wa_solve",
+            instruction="WolframAlpha: Solve equation(s).",
+            params=[
+                {"name": "equation", "type": "str", "required": False, "description": "Single equation"},
+                {"name": "equations", "type": "list", "required": False, "description": "List of equations"},
+                {"name": "var", "type": "str", "required": False, "description": "Variable"},
+                {"name": "variables", "type": "list", "required": False, "description": "Variables list"},
+                {"name": "domain", "type": "str", "required": False, "description": "reals|integers|complexes"},
+            ],
+            enabled=True,
+            description="Math: solve",
+            tab="math",
+        )
+        plugin.add_cmd(
+            "wa_derivative",
+            instruction="WolframAlpha: Compute derivative.",
+            params=[
+                {"name": "expr", "type": "str", "required": True, "description": "Expression"},
+                {"name": "var", "type": "str", "required": False, "description": "Variable (default x)"},
+                {"name": "order", "type": "int", "required": False, "description": "Order (default 1)"},
+                {"name": "at", "type": "str", "required": False, "description": "Evaluation point, e.g. 'x=0'"},
+            ],
+            enabled=True,
+            description="Math: derivative",
+            tab="math",
+        )
+        plugin.add_cmd(
+            "wa_integral",
+            instruction="WolframAlpha: Compute integral (indef/def).",
+            params=[
+                {"name": "expr", "type": "str", "required": True, "description": "Expression"},
+                {"name": "var", "type": "str", "required": False, "description": "Variable (default x)"},
+                {"name": "a", "type": "str", "required": False, "description": "Lower limit"},
+                {"name": "b", "type": "str", "required": False, "description": "Upper limit"},
+            ],
+            enabled=True,
+            description="Math: integral",
+            tab="math",
+        )
+        plugin.add_cmd(
+            "wa_units_convert",
+            instruction="WolframAlpha: Convert units.",
+            params=[
+                {"name": "value", "type": "str", "required": True, "description": "Numeric value"},
+                {"name": "from", "type": "str", "required": True, "description": "From unit"},
+                {"name": "to", "type": "str", "required": True, "description": "To unit"},
+            ],
+            enabled=True,
+            description="Units: convert",
+            tab="units",
+        )
+        plugin.add_cmd(
+            "wa_matrix",
+            instruction="WolframAlpha: Matrix operations (determinant, inverse, eigenvalues, rank).",
+            params=[
+                {"name": "op", "type": "str", "required": False, "description": "determinant|inverse|eigenvalues|rank"},
+                {"name": "matrix", "type": "list", "required": True, "description": "List of lists, e.g. [[1,2],[3,4]]"},
+            ],
+            enabled=True,
+            description="Math: matrix ops",
+            tab="math",
+        )
+        plugin.add_cmd(
+            "wa_plot",
+            instruction="WolframAlpha: Plot function (image file via Simple API).",
+            params=[
+                {"name": "func", "type": "str", "required": True, "description": "Function f(x)"},
+                {"name": "var", "type": "str", "required": False, "description": "Variable (default x)"},
+                {"name": "a", "type": "str", "required": False, "description": "From"},
+                {"name": "b", "type": "str", "required": False, "description": "To"},
+                {"name": "out", "type": "str", "required": False, "description": "Output file path"},
+            ],
+            enabled=True,
+            description="Plots: function plot",
+            tab="plots",
+        )

pygpt_net/plugin/wolfram/plugin.py

@@ -0,0 +1,115 @@
+#!/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: 2025.08.15 23:00:00                  #
+# ================================================== #
+
+from pygpt_net.plugin.base.plugin import BasePlugin
+from pygpt_net.core.events import Event
+from pygpt_net.item.ctx import CtxItem
+
+from .config import Config
+
+
+class Plugin(BasePlugin):
+    def __init__(self, *args, **kwargs):
+        super(Plugin, self).__init__(*args, **kwargs)
+        self.id = "wolfram"
+        self.name = "Wolfram Alpha"
+        self.description = "Compute and solve with Wolfram Alpha: short answers, full JSON pods, math (solve, derivatives, integrals), unit conversions, matrix operations, and plots."
+        self.prefix = "API"
+        self.order = 100
+        self.allowed_cmds =  [
+            "wa_short",
+            "wa_spoken",
+            "wa_simple",
+            "wa_query",
+            "wa_calculate",
+            "wa_solve",
+            "wa_derivative",
+            "wa_integral",
+            "wa_units_convert",
+            "wa_matrix",
+            "wa_plot"
+        ]
+        self.use_locale = False
+        self.worker = None
+        self.config = Config(self)
+        self.init_options()
+
+    def init_options(self):
+        """Initialize options"""
+        self.config.from_defaults(self)
+
+    def handle(self, event: Event, *args, **kwargs):
+        """
+        Handle dispatched event
+
+        :param event: event object
+        :param args: event args
+        :param kwargs: event kwargs
+        """
+        name = event.name
+        data = event.data
+        ctx = event.ctx
+
+        if name == Event.CMD_SYNTAX:
+            self.cmd_syntax(data)
+
+        elif name == Event.CMD_EXECUTE:
+            self.cmd(
+                ctx,
+                data['commands'],
+            )
+
+    def cmd_syntax(self, data: dict):
+        """
+        Event: CMD_SYNTAX
+
+        :param data: event data dict
+        """
+        for option in self.allowed_cmds:
+            if self.has_cmd(option):
+                data['cmd'].append(self.get_cmd(option))  # append command
+
+    def cmd(self, ctx: CtxItem, cmds: list):
+        """
+        Event: CMD_EXECUTE
+
+        :param ctx: CtxItem
+        :param cmds: commands dict
+        """
+
+        from .worker import Worker
+
+        is_cmd = False
+        my_commands = []
+        for item in cmds:
+            if item["cmd"] in self.allowed_cmds:
+                my_commands.append(item)
+                is_cmd = True
+
+        if not is_cmd:
+            return
+
+        # set state: busy
+        self.cmd_prepare(ctx, my_commands)
+
+        try:
+            worker = Worker()
+            worker.from_defaults(self)
+            worker.cmds = my_commands
+            worker.ctx = ctx
+
+            if not self.is_async(ctx):
+                worker.run()
+                return
+            worker.run_async()
+
+        except Exception as e:
+            self.error(e)