turingpulse-sdk-dspy 1.0.0__tar.gz

turingpulse_sdk_dspy-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_dspy-1.0.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: turingpulse-sdk-dspy
+Version: 1.0.0
+Summary: TuringPulse SDK integration for DSPy
+License-Expression: Apache-2.0
+Requires-Python: >=3.11
+Requires-Dist: dspy>=3.1.3
+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_dspy-1.0.0/pyproject.toml

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

turingpulse_sdk_dspy-1.0.0/turingpulse_sdk_dspy/__init__.py

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

turingpulse_sdk_dspy-1.0.0/turingpulse_sdk_dspy/_wrapper.py

@@ -0,0 +1,97 @@
+"""DSPy instrumentation for TuringPulse.
+
+Wraps DSPy module forward passes to capture LLM calls,
+optimization traces, and retrieval steps.
+"""
+
+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.dspy")
+
+_INSTRUMENTING: ContextVar[bool] = ContextVar("_tp_dspy_instrumenting", default=False)
+
+
+def instrument_dspy(
+    module,
+    *,
+    name: str,
+    governance: Optional[GovernanceDirective] = None,
+    model: Optional[str] = None,
+    provider: str = "openai",
+    kpis: Optional[Sequence["KPIConfig"]] = None,
+    metadata: Optional[Dict[str, str]] = None,
+):
+    """Wrap a DSPy module for TuringPulse observability.
+
+    Returns a callable wrapping ``module.forward()`` / ``module()``.
+
+    Args:
+        module: A ``dspy.Module`` instance (e.g., ``dspy.ChainOfThought``).
+        name: Workflow display name.
+        governance: Optional governance directive.
+        model: LLM model name override.
+        provider: LLM provider name.
+
+    Returns:
+        A callable wrapping the module's forward pass.
+    """
+
+    @instrument(name=name, governance=governance, kpis=kpis, metadata=metadata or {})
+    def _run(*args: Any, **kwargs: Any) -> Any:
+        token = _INSTRUMENTING.set(True)
+        try:
+            result = module(*args, **kwargs)
+
+            ctx = current_context()
+            if ctx:
+                ctx.framework = "dspy"
+                ctx.node_type = "workflow"
+
+                ctx.set_io(
+                    input_data=str(kwargs)[:MAX_FIELD_SIZE] if kwargs else str(args)[:MAX_FIELD_SIZE],
+                    output_data=str(result)[:MAX_FIELD_SIZE],
+                )
+
+                if model:
+                    ctx.set_model(model, provider)
+
+                try:
+                    import dspy
+                    lm = dspy.settings.lm
+                    if lm is not None:
+                        hist = getattr(lm, "history", [])
+                        if hist:
+                            total_prompt = 0
+                            total_completion = 0
+                            for entry in hist:
+                                usage = None
+                                if isinstance(entry, dict):
+                                    resp = entry.get("response")
+                                    if resp and hasattr(resp, "usage"):
+                                        usage = resp.usage
+                                    elif isinstance(resp, dict):
+                                        usage_d = resp.get("usage", {})
+                                        total_prompt += usage_d.get("prompt_tokens", 0) or 0
+                                        total_completion += usage_d.get("completion_tokens", 0) or 0
+                                        continue
+                                if usage:
+                                    total_prompt += getattr(usage, "prompt_tokens", 0) or 0
+                                    total_completion += getattr(usage, "completion_tokens", 0) or 0
+                            if total_prompt or total_completion:
+                                ctx.set_tokens(total_prompt, total_completion)
+                except Exception:
+                    pass
+
+            return result
+        finally:
+            _INSTRUMENTING.reset(token)
+
+    return _run