trulens-apps-opencode 2.14.0__tar.gz

trulens_apps_opencode-2.14.0/PKG-INFO

@@ -0,0 +1,26 @@
+Metadata-Version: 2.4
+Name: trulens-apps-opencode
+Version: 2.14.0
+Summary: OpenCode hook instrumentation plugin for TruLens.
+License: MIT
+Author: Snowflake Inc.
+Author-email: ml-observability-wg-dl@snowflake.com
+Requires-Python: >=3.9,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: trulens-core (>=2.0.0,<3.0.0)
+Project-URL: Homepage, https://trulens.org/
+Project-URL: Repository, https://github.com/truera/trulens
+Description-Content-Type: text/markdown
+
+# trulens-apps-opencode
+
+OpenCode plugin configuration and payload mapping for TruLens client-hook
+instrumentation.
+

trulens_apps_opencode-2.14.0/README.md

@@ -0,0 +1,4 @@
+# trulens-apps-opencode
+
+OpenCode plugin configuration and payload mapping for TruLens client-hook
+instrumentation.

trulens_apps_opencode-2.14.0/pyproject.toml

@@ -0,0 +1,30 @@
+[build-system]
+build-backend = "poetry.core.masonry.api"
+requires = [
+  "poetry-core",
+]
+
+[tool.poetry]
+name = "trulens-apps-opencode"
+version = "2.14.0"
+description = "OpenCode hook instrumentation plugin for TruLens."
+authors = [
+  "Snowflake Inc. <ml-observability-wg-dl@snowflake.com>",
+]
+license = "MIT"
+readme = "README.md"
+packages = [
+  { include = "trulens" },
+]
+homepage = "https://trulens.org/"
+repository = "https://github.com/truera/trulens"
+
+[tool.poetry.dependencies]
+python = "^3.9"
+trulens-core = { version = "^2.0.0" }
+
+[tool.poetry.group.dev.dependencies]
+trulens-core = { path = "../../core" }
+
+[tool.poetry.plugins."trulens.client_hooks"]
+opencode = "trulens.apps.opencode:client_spec"

trulens_apps_opencode-2.14.0/trulens/apps/opencode/__init__.py

@@ -0,0 +1,5 @@
+"""OpenCode hook instrumentation plugin."""
+
+from trulens.apps.opencode.client import client_spec
+
+__all__ = ["client_spec"]

trulens_apps_opencode-2.14.0/trulens/apps/opencode/client.py

@@ -0,0 +1,238 @@
+"""Minimal OpenCode-native plugin specification."""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+import shutil
+import subprocess
+from typing import Any, Mapping
+
+from trulens.core.otel.client_hooks.clients import ClientSpec
+from trulens.core.otel.client_hooks.clients import FieldAliases
+
+_HOOK_EVENTS = (
+    "chat.message",
+    "tool.execute.before",
+    "tool.execute.after",
+    "experimental.text.complete",
+    "session.idle",
+    "session.error",
+    "file.edited",
+)
+
+
+def _first(payload: Mapping[str, Any], *keys: str) -> Any:
+    for key in keys:
+        value = payload.get(key)
+        if value is not None:
+            return value
+    return None
+
+
+def _nested(payload: Mapping[str, Any], *path: str) -> Any:
+    current: Any = payload
+    for key in path:
+        if not isinstance(current, Mapping):
+            return None
+        current = current.get(key)
+    return current
+
+
+def _prompt_from_parts(parts: Any) -> Any:
+    if not isinstance(parts, list):
+        return None
+    texts = [
+        part.get("text")
+        for part in parts
+        if isinstance(part, Mapping) and part.get("text")
+    ]
+    return "\n".join(texts) or None
+
+
+def extract_overrides(payload: Mapping[str, Any]) -> Mapping[str, Any]:
+    """Flatten OpenCode camelCase and nested event payloads."""
+
+    event = payload.get("event")
+    properties = event.get("properties") if isinstance(event, Mapping) else {}
+    if not isinstance(properties, Mapping):
+        properties = {}
+    event_name = payload.get("hook_event_name")
+    if event_name is None and isinstance(event, Mapping):
+        event_name = event.get("type")
+    model = _first(payload, "model")
+    if isinstance(model, Mapping):
+        model = model.get("modelID") or model.get("model_id")
+    prompt = _first(payload, "prompt") or _prompt_from_parts(
+        payload.get("parts")
+    )
+    if prompt is None:
+        message = payload.get("message")
+        if isinstance(message, Mapping):
+            prompt = message.get("content") or _prompt_from_parts(
+                message.get("parts")
+            )
+    overrides = {
+        "conversation_id": _first(
+            payload,
+            "session_id",
+            "sessionID",
+        )
+        or properties.get("sessionID")
+        or _nested(payload, "event", "properties", "sessionID"),
+        "turn_id": _first(payload, "message_id", "messageID"),
+        "operation_id": _first(payload, "call_id", "callID", "tool_call_id"),
+        "event_name": event_name,
+        "tool_name": _first(payload, "tool_name", "tool"),
+        "prompt": prompt,
+        "response": _first(payload, "response", "text")
+        or _nested(payload, "output", "text"),
+        "tool_input": _first(payload, "tool_input", "args")
+        or _nested(payload, "output", "args"),
+        "tool_output": _first(payload, "tool_output")
+        or _nested(payload, "output", "output"),
+    }
+    if model:
+        overrides["model"] = model
+    return {key: value for key, value in overrides.items() if value is not None}
+
+
+def _detect_version() -> str | None:
+    executable = shutil.which("opencode")
+    if executable is None:
+        return None
+    try:
+        result = subprocess.run(
+            [executable, "--version"],
+            capture_output=True,
+            check=True,
+            text=True,
+            timeout=5,
+        )
+    except (OSError, subprocess.SubprocessError):
+        return None
+    return result.stdout.strip() or None
+
+
+def _plugin(command: str, client_version: str | None = None) -> str:
+    encoded = json.dumps(command)
+    encoded_version = json.dumps(client_version)
+    return f"""// managed_by: trulens-client-hooks
+const COMMAND = {encoded}
+const VERSION = {encoded_version}
+async function ingest(payload) {{
+  try {{
+    const proc = Bun.spawn(["sh", "-c", COMMAND], {{
+      stdin: "pipe",
+      stdout: "ignore",
+      stderr: "pipe",
+    }})
+    proc.stdin.write(JSON.stringify(payload))
+    proc.stdin.end()
+    await proc.exited
+  }} catch (_error) {{
+    // Fail open: telemetry must never block OpenCode.
+  }}
+}}
+
+function textFromParts(parts) {{
+  if (!Array.isArray(parts)) {{
+    return undefined
+  }}
+  return parts
+    .map((part) => (part && part.text) || "")
+    .filter(Boolean)
+    .join("\\n") || undefined
+}}
+
+export const TruLensClientHooks = async ({{ directory }}) => {{
+  let lastSessionId
+  const send = async (payload) => {{
+    if (payload.session_id) {{
+      lastSessionId = payload.session_id
+    }}
+    await ingest({{ cwd: directory, client_version: VERSION, ...payload }})
+  }}
+  return {{
+    "chat.message": async (input, output) => {{
+      await send({{
+        session_id: input.sessionID,
+        message_id: input.messageID,
+        hook_event_name: "chat.message",
+        model: input.model && input.model.modelID,
+        prompt: textFromParts(output && output.parts),
+      }})
+    }},
+    "tool.execute.before": async (input, output) => {{
+      await send({{
+        session_id: input.sessionID,
+        hook_event_name: "tool.execute.before",
+        tool_name: input.tool,
+        tool_call_id: input.callID,
+        tool_input: output && output.args,
+      }})
+    }},
+    "tool.execute.after": async (input, output) => {{
+      await send({{
+        session_id: input.sessionID,
+        hook_event_name: "tool.execute.after",
+        tool_name: input.tool,
+        tool_call_id: input.callID,
+        tool_input: input.args,
+        tool_output: output && output.output,
+      }})
+    }},
+    "experimental.text.complete": async (input, output) => {{
+      await send({{
+        session_id: input.sessionID,
+        message_id: input.messageID,
+        response_message_id: input.messageID,
+        hook_event_name: "experimental.text.complete",
+        text: output && output.text,
+      }})
+    }},
+    event: async ({{ event }}) => {{
+      const type = event && event.type
+      const properties = (event && event.properties) || {{}}
+      const sessionId =
+        properties.sessionID || event.sessionID || lastSessionId
+      if (type === "session.idle" || type === "session.error") {{
+        await send({{
+          session_id: sessionId,
+          hook_event_name: type,
+          status: type === "session.error" ? "error" : "completed",
+          error:
+            type === "session.error"
+              ? properties.error || event.error
+              : undefined,
+        }})
+      }}
+      if (type === "file.edited") {{
+        await send({{
+          session_id: sessionId,
+          hook_event_name: "file.edited",
+          file_path: properties.file || event.path,
+        }})
+      }}
+    }},
+  }}
+}}
+"""
+
+
+client_spec = ClientSpec(
+    name="opencode",
+    aliases=("open-code",),
+    user_config_path=Path("~/.config/opencode/plugins/trulens-client-hooks.js"),
+    project_config_path=Path(".opencode/plugins/trulens-client-hooks.js"),
+    hook_events=_HOOK_EVENTS,
+    field_aliases=FieldAliases(
+        conversation=("session_id", "sessionID"),
+        turn=("message_id", "messageID"),
+        operation=("call_id", "callID", "tool_call_id"),
+        response=("text", "response", "output"),
+    ),
+    plugin_builder=_plugin,
+    version_detector=_detect_version,
+    extract_overrides=extract_overrides,
+)