turingpulse-sdk-strands 1.0.0__tar.gz

turingpulse_sdk_strands-1.0.0/.gitignore

@@ -0,0 +1,42 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Virtual environments
+.venv/
+venv/
+ENV/
+
+# Distribution / packaging
+dist/
+build/
+*.egg-info/
+
+# Database files
+*.db
+*.sqlite3
+
+# Environment variables
+.env
+.env.local
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+
+# Logs
+*.log
+logs/
+
+# OS files
+.DS_Store
+Thumbs.db

turingpulse_sdk_strands-1.0.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: turingpulse-sdk-strands
+Version: 1.0.0
+Summary: TuringPulse SDK integration for Strands Agents
+License-Expression: Apache-2.0
+Requires-Python: >=3.11
+Requires-Dist: strands-agents>=1.27.0
+Requires-Dist: turingpulse-sdk>=1.0.0
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'

turingpulse_sdk_strands-1.0.0/pyproject.toml

@@ -0,0 +1,17 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "turingpulse-sdk-strands"
+version = "1.0.0"
+description = "TuringPulse SDK integration for Strands Agents"
+requires-python = ">=3.11"
+license = "Apache-2.0"
+dependencies = [
+    "turingpulse-sdk>=1.0.0",
+    "strands-agents>=1.27.0",
+]
+
+[project.optional-dependencies]
+dev = ["pytest>=8.0", "pytest-asyncio>=0.23"]

turingpulse_sdk_strands-1.0.0/turingpulse_sdk_strands/__init__.py

@@ -0,0 +1,6 @@
+"""TuringPulse SDK integration for Strands Agents."""
+
+from ._wrapper import instrument_strands
+
+__version__ = "0.1.0"
+__all__ = ["instrument_strands"]

turingpulse_sdk_strands-1.0.0/turingpulse_sdk_strands/_wrapper.py

@@ -0,0 +1,77 @@
+"""Strands Agents instrumentation for TuringPulse.
+
+Wraps Strands agent invocations to capture tool calls,
+model interactions, and reasoning chains.
+"""
+
+from __future__ import annotations
+
+import logging
+from contextvars import ContextVar
+from typing import Any, Dict, Optional, Sequence
+
+from turingpulse_sdk import instrument, GovernanceDirective
+from turingpulse_sdk.config import MAX_FIELD_SIZE
+from turingpulse_sdk.context import current_context
+
+logger = logging.getLogger("turingpulse.sdk.strands")
+
+_INSTRUMENTING: ContextVar[bool] = ContextVar("_tp_strands_instrumenting", default=False)
+
+
+def instrument_strands(
+    agent,
+    *,
+    name: str,
+    governance: Optional[GovernanceDirective] = None,
+    model: Optional[str] = None,
+    provider: str = "bedrock",
+    kpis: Optional[Sequence["KPIConfig"]] = None,
+    metadata: Optional[Dict[str, str]] = None,
+):
+    """Wrap a Strands Agent for TuringPulse observability.
+
+    Returns a callable that instruments the agent invocation.
+
+    Args:
+        agent: A Strands ``Agent`` instance.
+        name: Workflow display name.
+        governance: Optional governance directive.
+        model: LLM model name override.
+        provider: LLM provider name.
+
+    Returns:
+        A callable wrapping the agent's __call__ method.
+    """
+
+    @instrument(name=name, governance=governance, kpis=kpis, metadata=metadata or {})
+    def _run(*args: Any, **kwargs: Any) -> Any:
+        token = _INSTRUMENTING.set(True)
+        try:
+            result = agent(*args, **kwargs)
+
+            ctx = current_context()
+            if ctx:
+                ctx.framework = "strands"
+                ctx.node_type = "workflow"
+
+                ctx.set_io(
+                    input_data=str(args[0]) if args else str(kwargs),
+                    output_data=str(result)[:MAX_FIELD_SIZE],
+                )
+
+                if model:
+                    ctx.set_model(model, provider)
+
+                if hasattr(result, "metrics") and result.metrics:
+                    metrics = result.metrics
+                    ctx.set_tokens(
+                        getattr(metrics, "input_tokens", 0) or 0,
+                        getattr(metrics, "output_tokens", 0) or 0,
+                    )
+
+            return result
+        finally:
+            _INSTRUMENTING.reset(token)
+
+    return _run