agentic-threat-hunting-framework 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

{agentic_threat_hunting_framework-0.5.1.dist-info → agentic_threat_hunting_framework-0.5.2.dist-info}/METADATA

@@ -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.dist-info → agentic_threat_hunting_framework-0.5.2.dist-info}/RECORD

@@ -1,8 +1,8 @@
-agentic_threat_hunting_framework-0.5.1.dist-info/licenses/LICENSE,sha256=_KObErRfiKoolznt-DF0nJnr3U9Rdh7Z4Ba7G5qqckk,1071
+agentic_threat_hunting_framework-0.5.2.dist-info/licenses/LICENSE,sha256=_KObErRfiKoolznt-DF0nJnr3U9Rdh7Z4Ba7G5qqckk,1071
 athf/__init__.py,sha256=OrjZe8P97_BTEkscapnwSsqKSjwXNP9d8-HtGr19Ni0,241
 athf/__version__.py,sha256=wCIQoU9b7qKcSNQiIOgHaD2buzBC-dlQYtvg8X5WS4A,59
 athf/cli.py,sha256=108PnDRlaytJj9KzjzcTLljB3DeerMIXZOeAQJrmtPU,5052
-athf/plugin_system.py,sha256=c_9-oUiR6tuYpWpEmeVRayU8-TXlkjvZC3EUxuYWW4M,1515
+athf/plugin_system.py,sha256=eS_P56IAZQEtyefQkp-x2J-8pxKcZX_394-WoCY7M3U,2323
 athf/agents/__init__.py,sha256=iaSJpvnXm9rz4QS7gBrsaLEjm49uvsMs4BLPOJeyp78,346
 athf/agents/base.py,sha256=ZLqSW6I0pKqs1Z3OIoV8urkysMRzNDs52yNIxDgQjTU,3981
 athf/agents/llm/__init__.py,sha256=qSGA-NaInjsDkMpGQwnTz3S1OgCVlzetpMcDS_to1co,671
@@ -52,8 +52,8 @@ athf/data/prompts/ai-workflow.md,sha256=rZtOcGuAEi35qx7182TwHJEORdz1-RxkZMBVkg61
 athf/data/prompts/basic-prompts.md,sha256=2bunpO35RoBdJWYthXVi40RNl2UWrfwOaFthBLHF5sU,8463
 athf/data/templates/HUNT_LOCK.md,sha256=zXxHaKMWbRDLewLTegYJMbXRM72s9gFFvjdwFfGNeJE,7386
 athf/utils/__init__.py,sha256=aEAPI1xnAsowOtc036cCb9ZOek5nrrfevu8PElhbNgk,30
-agentic_threat_hunting_framework-0.5.1.dist-info/METADATA,sha256=A9gPoTr0bXjdPo_l76Fn1tIxJFAL3RFQRRUPSDT6N8I,15949
-agentic_threat_hunting_framework-0.5.1.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-agentic_threat_hunting_framework-0.5.1.dist-info/entry_points.txt,sha256=GopR2iTiBs-yNMWiUZ2DaFIFglXxWJx1XPjTa3ePtfE,39
-agentic_threat_hunting_framework-0.5.1.dist-info/top_level.txt,sha256=Cxxg6SMLfawDJWBITsciRzq27XV8fiaAor23o9Byoes,5
-agentic_threat_hunting_framework-0.5.1.dist-info/RECORD,,
+agentic_threat_hunting_framework-0.5.2.dist-info/METADATA,sha256=YfuSvYBFEkY6HIsnVOIuC7Yjx16mCRvycZHfXAwSZIA,15949
+agentic_threat_hunting_framework-0.5.2.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+agentic_threat_hunting_framework-0.5.2.dist-info/entry_points.txt,sha256=GopR2iTiBs-yNMWiUZ2DaFIFglXxWJx1XPjTa3ePtfE,39
+agentic_threat_hunting_framework-0.5.2.dist-info/top_level.txt,sha256=Cxxg6SMLfawDJWBITsciRzq27XV8fiaAor23o9Byoes,5
+agentic_threat_hunting_framework-0.5.2.dist-info/RECORD,,

athf/plugin_system.py

@@ -1,17 +1,27 @@
 """Plugin system for ATHF extensions."""
-from typing import Dict, Type, Callable
-import importlib.metadata
+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] = {}
+    _agents: Dict[str, Type[Any]] = {}
     _commands: Dict[str, Command] = {}
 
     @classmethod
-    def register_agent(cls, name: str, agent_class: Type) -> None:
+    def register_agent(cls, name: str, agent_class: Type[Any]) -> None:
         """Register an agent plugin."""
         cls._agents[name] = agent_class
 
@@ -21,12 +31,12 @@ class PluginRegistry:
         cls._commands[name] = command
 
     @classmethod
-    def get_agent(cls, name: str) -> Type:
+    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) -> Command:
+    def get_command(cls, name: str) -> Optional[Command]:
         """Get registered command by name."""
         return cls._commands.get(name)
 
@@ -34,14 +44,26 @@ class PluginRegistry:
     def load_plugins(cls) -> None:
         """Auto-discover and load all installed plugins."""
         try:
-            for ep in importlib.metadata.entry_points(group='athf.commands'):
+            # 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:
-            for ep in importlib.metadata.entry_points(group='athf.agents'):
+            # 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: