aigp-strands 1.0.0__tar.gz

aigp_strands-1.0.0/.gitignore

@@ -0,0 +1,40 @@
+# Python
+__pycache__/
+*.py[cod]
+*.egg-info/
+dist/
+build/
+.venv/
+.eggs/
+
+# Node
+node_modules/
+dist/
+
+# Go
+bin/
+
+# Rust
+target/
+
+# .NET
+bin/
+obj/
+
+# Java
+*.class
+out/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Secrets (never commit)
+.env
+*.pem
+*.key

aigp_strands-1.0.0/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: aigp-strands
+Version: 1.0.0
+Summary: AIGP governance adapter for Strands Agents SDK
+Requires-Python: >=3.11
+Requires-Dist: aigp-agent-core>=1.0.0
+Provides-Extra: otel
+Requires-Dist: opentelemetry-api>=1.20; extra == 'otel'

aigp_strands-1.0.0/aigp_strands/__init__.py

@@ -0,0 +1,8 @@
+"""AIGP-Strands — Governance adapter for Strands Agents SDK."""
+
+__version__ = "1.0.0"
+
+from .handler import AigpCallbackHandler
+from .governance import AigpGovernance
+
+__all__ = ["AigpGovernance", "AigpCallbackHandler"]

aigp_strands-1.0.0/aigp_strands/governance.py

@@ -0,0 +1,17 @@
+"""AigpGovernance — entry point for Strands agent governance."""
+
+from aigp_agent_core import AgentGovernance
+from .handler import AigpCallbackHandler
+
+
+class AigpGovernance(AgentGovernance):
+    """Strands-specific governance wrapper.
+
+    Usage:
+        governance = AigpGovernance(gov_url=..., app_id=..., hmac_secret=...)
+        agent = Agent(model=..., callback_handler=governance.callback_handler())
+    """
+
+    def callback_handler(self) -> AigpCallbackHandler:
+        """Create a Strands callback handler wired to this governance instance."""
+        return AigpCallbackHandler(self)

aigp_strands-1.0.0/aigp_strands/handler.py

@@ -0,0 +1,61 @@
+"""Strands Callback Handler — maps Strands lifecycle to AIGP governance."""
+
+import asyncio
+import logging
+import time
+
+logger = logging.getLogger(__name__)
+
+
+class AigpCallbackHandler:
+    """Strands SDK callback handler implementing AIGP governance.
+
+    Attach to any Strands Agent:
+        agent = Agent(model=..., callback_handler=handler)
+    """
+
+    def __init__(self, governance):
+        self._gov = governance
+        self._tool_start_times: dict[str, float] = {}
+
+    def on_agent_start(self, agent_name: str, model_id: str, prompt: str, **kwargs):
+        """Called when agent starts processing. Sends AIGP CHECK."""
+        user_id = kwargs.get("user_id", "")
+        self._gov.pre_invoke(agent_name, model_id, user_id)
+
+    def on_model_start(self, model_id: str, prompt: str, **kwargs):
+        """Called before each model invocation within a cycle."""
+        pass  # Trace stage already handled by pre_invoke
+
+    def on_model_end(self, response: str, usage: dict = None, **kwargs):
+        """Called after model returns. Accumulates tokens."""
+        self._gov.on_model_call(usage=usage or {})
+
+    def on_tool_start(self, tool_name: str, params: dict = None, **kwargs):
+        """Called before tool execution."""
+        self._tool_start_times[tool_name] = time.time()
+
+    def on_tool_end(self, tool_name: str, result: str = "", **kwargs):
+        """Called after tool execution. Records tool span."""
+        start = self._tool_start_times.pop(tool_name, time.time())
+        duration_ms = int((time.time() - start) * 1000)
+        try:
+            asyncio.get_event_loop().run_until_complete(
+                self._gov.on_tool_call(tool_name, duration_ms=duration_ms, result=result)
+            )
+        except RuntimeError:
+            asyncio.run(self._gov.on_tool_call(tool_name, duration_ms=duration_ms, result=result))
+
+    def on_agent_end(self, result: str = "", **kwargs):
+        """Called when agent completes. Sends RECORD + TRACE + evidence."""
+        try:
+            asyncio.get_event_loop().run_until_complete(self._gov.post_invoke())
+        except RuntimeError:
+            asyncio.run(self._gov.post_invoke())
+
+    def on_agent_error(self, error: Exception, **kwargs):
+        """Called on agent error."""
+        try:
+            asyncio.get_event_loop().run_until_complete(self._gov.on_error(error))
+        except RuntimeError:
+            asyncio.run(self._gov.on_error(error))

aigp_strands-1.0.0/pyproject.toml

@@ -0,0 +1,13 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "aigp-strands"
+version = "1.0.0"
+description = "AIGP governance adapter for Strands Agents SDK"
+requires-python = ">=3.11"
+dependencies = ["aigp-agent-core>=1.0.0"]
+
+[project.optional-dependencies]
+otel = ["opentelemetry-api>=1.20"]