agentic-threat-hunting-framework 0.5.1__tar.gz → 0.5.2__tar.gz

{agentic_threat_hunting_framework-0.5.1/agentic_threat_hunting_framework.egg-info → agentic_threat_hunting_framework-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-threat-hunting-framework
-Version: 0.5.1
+Version: 0.5.2
 Summary: Agentic Threat Hunting Framework - Memory and AI for threat hunters
 Author-email: Sydney Marrone <athf@nebulock.io>
 Maintainer-email: Sydney Marrone <athf@nebulock.io>

{agentic_threat_hunting_framework-0.5.1 → agentic_threat_hunting_framework-0.5.2/agentic_threat_hunting_framework.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-threat-hunting-framework
-Version: 0.5.1
+Version: 0.5.2
 Summary: Agentic Threat Hunting Framework - Memory and AI for threat hunters
 Author-email: Sydney Marrone <athf@nebulock.io>
 Maintainer-email: Sydney Marrone <athf@nebulock.io>

agentic_threat_hunting_framework-0.5.2/athf/plugin_system.py

@@ -0,0 +1,70 @@
+"""Plugin system for ATHF extensions."""
+from typing import Any, Dict, Optional, Type
+import sys
+from click import Command
+
+# Handle importlib.metadata API changes across Python versions
+if sys.version_info >= (3, 10):
+    from importlib.metadata import entry_points
+else:
+    # Python 3.8-3.9: use importlib_metadata backport API
+    try:
+        from importlib.metadata import entry_points
+    except ImportError:
+        from importlib_metadata import entry_points  # type: ignore
+
+
+class PluginRegistry:
+    """Central registry for ATHF plugins."""
+
+    _agents: Dict[str, Type[Any]] = {}
+    _commands: Dict[str, Command] = {}
+
+    @classmethod
+    def register_agent(cls, name: str, agent_class: Type[Any]) -> None:
+        """Register an agent plugin."""
+        cls._agents[name] = agent_class
+
+    @classmethod
+    def register_command(cls, name: str, command: Command) -> None:
+        """Register a CLI command plugin."""
+        cls._commands[name] = command
+
+    @classmethod
+    def get_agent(cls, name: str) -> Optional[Type[Any]]:
+        """Get registered agent by name."""
+        return cls._agents.get(name)
+
+    @classmethod
+    def get_command(cls, name: str) -> Optional[Command]:
+        """Get registered command by name."""
+        return cls._commands.get(name)
+
+    @classmethod
+    def load_plugins(cls) -> None:
+        """Auto-discover and load all installed plugins."""
+        try:
+            # Python 3.10+ uses group= parameter, 3.8-3.9 uses dict-like access
+            if sys.version_info >= (3, 10):
+                eps = entry_points(group='athf.commands')
+            else:
+                eps = entry_points().get('athf.commands', [])
+
+            for ep in eps:
+                command = ep.load()
+                cls.register_command(ep.name, command)
+        except Exception:
+            pass  # No plugins installed yet
+
+        try:
+            # Python 3.10+ uses group= parameter, 3.8-3.9 uses dict-like access
+            if sys.version_info >= (3, 10):
+                eps = entry_points(group='athf.agents')
+            else:
+                eps = entry_points().get('athf.agents', [])
+
+            for ep in eps:
+                agent = ep.load()
+                cls.register_agent(ep.name, agent)
+        except Exception:
+            pass  # No plugins installed yet

{agentic_threat_hunting_framework-0.5.1 → agentic_threat_hunting_framework-0.5.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentic-threat-hunting-framework"
-version = "0.5.1"
+version = "0.5.2"
 description = "Agentic Threat Hunting Framework - Memory and AI for threat hunters"
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.8"

agentic_threat_hunting_framework-0.5.1/athf/plugin_system.py

@@ -1,48 +0,0 @@
-"""Plugin system for ATHF extensions."""
-from typing import Dict, Type, Callable
-import importlib.metadata
-from click import Command
-
-
-class PluginRegistry:
-    """Central registry for ATHF plugins."""
-
-    _agents: Dict[str, Type] = {}
-    _commands: Dict[str, Command] = {}
-
-    @classmethod
-    def register_agent(cls, name: str, agent_class: Type) -> None:
-        """Register an agent plugin."""
-        cls._agents[name] = agent_class
-
-    @classmethod
-    def register_command(cls, name: str, command: Command) -> None:
-        """Register a CLI command plugin."""
-        cls._commands[name] = command
-
-    @classmethod
-    def get_agent(cls, name: str) -> Type:
-        """Get registered agent by name."""
-        return cls._agents.get(name)
-
-    @classmethod
-    def get_command(cls, name: str) -> Command:
-        """Get registered command by name."""
-        return cls._commands.get(name)
-
-    @classmethod
-    def load_plugins(cls) -> None:
-        """Auto-discover and load all installed plugins."""
-        try:
-            for ep in importlib.metadata.entry_points(group='athf.commands'):
-                command = ep.load()
-                cls.register_command(ep.name, command)
-        except Exception:
-            pass  # No plugins installed yet
-
-        try:
-            for ep in importlib.metadata.entry_points(group='athf.agents'):
-                agent = ep.load()
-                cls.register_agent(ep.name, agent)
-        except Exception:
-            pass  # No plugins installed yet