swarmauri_tool_psutil 0.6.0.dev154__tar.gz

swarmauri_tool_psutil-0.6.0.dev154/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.3
+Name: swarmauri_tool_psutil
+Version: 0.6.0.dev154
+Summary: Swarmauri Psutil Tool.
+License: Apache-2.0
+Author: Jacob Stewart
+Author-email: jacob@swarmauri.com
+Requires-Python: >=3.10,<3.13
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: psutil (>=6.1.0,<7.0.0)
+Requires-Dist: swarmauri_base (>=0.6.0.dev154,<0.7.0)
+Requires-Dist: swarmauri_core (>=0.6.0.dev154,<0.7.0)
+Project-URL: Repository, http://github.com/swarmauri/swarmauri-sdk
+Description-Content-Type: text/markdown
+
+# Swarmauri Example Community Package

swarmauri_tool_psutil-0.6.0.dev154/README.md

@@ -0,0 +1 @@
+# Swarmauri Example Community Package

swarmauri_tool_psutil-0.6.0.dev154/pyproject.toml

@@ -0,0 +1,56 @@
+[tool.poetry]
+name = "swarmauri_tool_psutil"
+version = "0.6.0.dev154"
+description = "Swarmauri Psutil Tool."
+authors = ["Jacob Stewart <jacob@swarmauri.com>"]
+license = "Apache-2.0"
+readme = "README.md"
+repository = "http://github.com/swarmauri/swarmauri-sdk"
+classifiers = [
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12"
+]
+
+[tool.poetry.dependencies]
+python = ">=3.10,<3.13"
+
+# Swarmauri
+swarmauri_core = {version = "^0.6.0.dev154"}
+swarmauri_base = {version = "^0.6.0.dev154"}
+
+# Dependencies
+psutil = "^6.1.0"
+
+[tool.poetry.group.dev.dependencies]
+flake8 = "^7.0"
+pytest = "^8.0"
+pytest-asyncio = ">=0.24.0"
+pytest-xdist = "^3.6.1"
+pytest-json-report = "^1.5.0"
+python-dotenv = "*"
+requests = "^2.32.3"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.pytest.ini_options]
+norecursedirs = ["combined", "scripts"]
+
+markers = [
+    "test: standard test",
+    "unit: Unit tests",
+    "integration: Integration tests",
+    "acceptance: Acceptance tests",
+    "experimental: Experimental tests"
+]
+log_cli = true
+log_cli_level = "INFO"
+log_cli_format = "%(asctime)s [%(levelname)s] %(message)s"
+log_cli_date_format = "%Y-%m-%d %H:%M:%S"
+asyncio_default_fixture_loop_scope = "function"
+
+[tool.poetry.plugins."swarmauri.tools"]
+PsutilTool = "swarmauri_tool_psutil.PsutilTool:PsutilTool"

swarmauri_tool_psutil-0.6.0.dev154/swarmauri_tool_psutil/PsutilTool.py

@@ -0,0 +1,130 @@
+# standard/tools/concrete/PsutilTool.py
+import psutil
+from typing import Dict, Any, Literal, List, Callable
+from swarmauri_core.ComponentBase import ComponentBase
+from pydantic import Field
+from swarmauri.tools.base.ToolBase import ToolBase
+from swarmauri.tools.concrete.Parameter import Parameter
+
+
+@ComponentBase.register_type(ToolBase, "PsutilTool")
+class PsutilTool(ToolBase):
+    type: Literal["PsutilTool"] = "PsutilTool"
+    name: str = Field(
+        "PsutilTool", description="Tool to gather system information using psutil."
+    )
+    description: str = Field(
+        "This tool gathers system information like CPU, memory, disk, network, and sensors using the psutil library.",
+        description="Description of the PsutilTool",
+    )
+
+    parameters: List[Parameter] = Field(
+        default_factory=lambda: [
+            Parameter(
+                name="info_type",
+                type="string",
+                description="Type of system information to retrieve (cpu, memory, disk, network, sensors).",
+                required=True,
+            )
+        ]
+    )
+
+    def get_all_cpu_values(self) -> Dict[str, Any]:
+        return {
+            "cpu_times": psutil.cpu_times()._asdict(),
+            "cpu_percent": psutil.cpu_percent(interval=1),
+            "cpu_times_per_cpu": [
+                cpu._asdict() for cpu in psutil.cpu_times(percpu=True)
+            ],
+            "cpu_count": psutil.cpu_count(logical=True),
+            "cpu_frequency": psutil.cpu_freq()._asdict(),
+            "cpu_stats": psutil.cpu_stats()._asdict(),
+        }
+
+    def get_all_memory_values(self) -> Dict[str, Any]:
+        return {
+            "virtual_memory": psutil.virtual_memory()._asdict(),
+            "swap_memory": psutil.swap_memory()._asdict(),
+        }
+
+    def get_all_disk_values(self) -> Dict[str, Any]:
+        partitions = psutil.disk_partitions()
+        disk_usage = {
+            partition.device: psutil.disk_usage(partition.mountpoint)._asdict()
+            for partition in partitions
+        }
+        return {
+            "disk_partitions": [partition._asdict() for partition in partitions],
+            "disk_usage": disk_usage,
+            "disk_io_counters": psutil.disk_io_counters()._asdict(),
+        }
+
+    def get_all_network_values(self) -> Dict[str, Any]:
+        return {
+            "network_io_counters": psutil.net_io_counters()._asdict(),
+            "network_connections": [
+                conn._asdict() for conn in psutil.net_connections()
+            ],
+            "network_interfaces": {
+                iface: [addr._asdict() for addr in addrs]
+                for iface, addrs in psutil.net_if_addrs().items()
+            },
+        }
+
+    def get_all_sensors_values(self) -> Dict[str, Any]:
+        battery = psutil.sensors_battery()
+        temperatures = psutil.sensors_temperatures()
+        fans = psutil.sensors_fans()
+
+        return {
+            "battery": battery._asdict() if battery else None,
+            "temperatures": {
+                name: [temp._asdict() for temp in temps]
+                for name, temps in (temperatures or {}).items()
+            },
+            "fan_speeds": {
+                name: [fan._asdict() for fan in fans]
+                for name, fans in (fans or {}).items()
+            },
+        }
+
+    def __call__(self, info_type: str) -> Dict[str, Any]:
+        """
+        Call the appropriate method based on the provided info_type and return the corresponding system information.
+
+        Parameters:
+        info_type (str): The type of system information requested. Valid options are 'cpu', 'memory', 'disk',
+                         'network', or 'sensors'.
+
+        Returns:
+        Dict[str, Any]: A dictionary where the key is the `info_type` and the value is the result of the corresponding
+                        system information retrieval method.
+
+        Raises:
+        ValueError: If an invalid `info_type` is provided.
+
+        Example:
+        >>> tool = YourToolClass()
+        >>> result = tool("cpu")
+        >>> print(result)
+        {'cpu': {...}}
+        """
+
+        # Mapping info_type values to the corresponding methods
+        info_methods: Dict[str, Callable[[], Any]] = {
+            "cpu": self.get_all_cpu_values,
+            "memory": self.get_all_memory_values,
+            "disk": self.get_all_disk_values,
+            "network": self.get_all_network_values,
+            "sensors": self.get_all_sensors_values,
+        }
+
+        # Retrieve the appropriate method or raise an error if invalid info_type is provided
+        if info_type not in info_methods:
+            raise ValueError(
+                f"Invalid info_type '{info_type}' specified. Valid options are: {list(info_methods.keys())}."
+            )
+
+        # Execute the corresponding method and return the result
+        result = info_methods[info_type]()
+        return result

swarmauri_tool_psutil-0.6.0.dev154/swarmauri_tool_psutil/__init__.py

@@ -0,0 +1,15 @@
+from .PsutilTool import PsutilTool
+__version__ = "0.6.0.dev26"
+__long_desc__ = """
+
+# Swarmauri Psutil Plugin
+
+This repository includes a Psutil of a Swarmauri Plugin.
+
+Visit us at: https://swarmauri.com
+Follow us at: https://github.com/swarmauri
+Star us at: https://github.com/swarmauri/swarmauri-sdk
+
+"""
+
+