turingpulse-sdk-cohere 1.0.0__tar.gz

turingpulse_sdk_cohere-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_cohere-1.0.0/PKG-INFO

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

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

turingpulse_sdk_cohere-1.0.0/turingpulse_sdk_cohere/__init__.py

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

turingpulse_sdk_cohere-1.0.0/turingpulse_sdk_cohere/_wrapper.py

@@ -0,0 +1,130 @@
+"""Cohere monkey-patch instrumentation for TuringPulse."""
+
+from __future__ import annotations
+
+import logging
+import os
+from contextvars import ContextVar
+from typing import Any, Optional
+
+from turingpulse_sdk import instrument, GovernanceDirective
+from turingpulse_sdk.config import MAX_FIELD_SIZE
+from turingpulse_sdk.context import current_context
+from turingpulse_sdk.exceptions import ConfigurationError
+
+logger = logging.getLogger("turingpulse.sdk.cohere")
+
+_INSTRUMENTING: ContextVar[bool] = ContextVar("_tp_cohere_instrumenting", default=False)
+
+_ORIGINAL_CHAT: Any = None
+
+
+def patch_cohere(
+    *,
+    name: str | None = None,
+    governance: Optional[GovernanceDirective] = None,
+) -> None:
+    """Monkey-patch ``cohere.ClientV2.chat``."""
+    global _ORIGINAL_CHAT
+
+    effective_name = name or os.getenv("TP_WORKFLOW_NAME", "")
+    if not effective_name:
+        raise ConfigurationError(
+            "patch_cohere() requires name= or TP_WORKFLOW_NAME to be set."
+        )
+
+    try:
+        from cohere import ClientV2
+    except ImportError as exc:
+        raise ImportError("cohere package is required: pip install cohere>=5.20.0") from exc
+
+    if _ORIGINAL_CHAT is not None:
+        logger.warning("Cohere is already patched — skipping")
+        return
+
+    _ORIGINAL_CHAT = ClientV2.chat
+
+    @instrument(name=effective_name, governance=governance)
+    def _patched_chat(self_client, *args: Any, **kwargs: Any) -> Any:
+        if _INSTRUMENTING.get(False):
+            return _ORIGINAL_CHAT(self_client, *args, **kwargs)
+        token = _INSTRUMENTING.set(True)
+        try:
+            response = _ORIGINAL_CHAT(self_client, *args, **kwargs)
+            _record_cohere_span(response, kwargs)
+            return response
+        finally:
+            _INSTRUMENTING.reset(token)
+
+    ClientV2.chat = _patched_chat
+    logger.info("Cohere patched for TuringPulse instrumentation")
+
+
+def unpatch_cohere() -> None:
+    """Restore original Cohere methods."""
+    global _ORIGINAL_CHAT
+    if _ORIGINAL_CHAT is None:
+        return
+    try:
+        from cohere import ClientV2
+        ClientV2.chat = _ORIGINAL_CHAT
+    except ImportError:
+        pass
+    _ORIGINAL_CHAT = None
+    logger.info("Cohere unpatched — original methods restored")
+
+
+def _record_cohere_span(response: Any, kwargs: dict) -> None:
+    ctx = current_context()
+    if not ctx:
+        return
+    ctx.framework = "cohere"
+    ctx.node_type = "llm"
+
+    usage = getattr(response, "usage", None)
+    if usage:
+        billed = getattr(usage, "billed_units", None)
+        if billed:
+            ctx.set_tokens(
+                getattr(billed, "input_tokens", 0) or 0,
+                getattr(billed, "output_tokens", 0) or 0,
+            )
+
+    ctx.set_model(kwargs.get("model", getattr(response, "model", "unknown")), "cohere")
+
+    messages = kwargs.get("messages", [])
+    if messages:
+        last_user = next(
+            (m for m in reversed(messages)
+             if (m.get("role") if isinstance(m, dict) else getattr(m, "role", "")) == "user"),
+            None,
+        )
+        if last_user:
+            content = last_user.get("content", "") if isinstance(last_user, dict) else getattr(last_user, "content", "")
+            ctx.set_prompt(str(content)[:MAX_FIELD_SIZE])
+
+    tokens = getattr(response, "meta", None)
+    if tokens:
+        tok = getattr(tokens, "tokens", None) if not isinstance(tokens, dict) else tokens.get("tokens")
+        if tok:
+            inp = (getattr(tok, "input_tokens", 0) if not isinstance(tok, dict) else tok.get("input_tokens", 0)) or 0
+            out = (getattr(tok, "output_tokens", 0) if not isinstance(tok, dict) else tok.get("output_tokens", 0)) or 0
+            if inp or out:
+                ctx.set_tokens(inp, out)
+
+    message = getattr(response, "message", None)
+    if message:
+        content = getattr(message, "content", [])
+        if content:
+            texts = [getattr(c, "text", "") for c in content if getattr(c, "type", "") == "text"]
+            if texts:
+                ctx.set_io(output_data="\n".join(texts)[:MAX_FIELD_SIZE])
+
+        tool_calls = getattr(message, "tool_calls", None)
+        if tool_calls:
+            for tc in tool_calls:
+                ctx.add_tool_call(
+                    tool_name=getattr(tc, "name", "unknown"),
+                    tool_args=getattr(tc, "parameters", {}),
+                    tool_id=getattr(tc, "id", None),
+                )