arbis 0.4.5__py3-none-any.whl

arbis/__init__.py

@@ -0,0 +1,80 @@
+"""
+arbis: runtime workflow observation for agent systems.
+
+Instruments known SDK boundaries, collects execution events,
+and infers workflow graphs from observed behavior.
+"""
+from ._tracker import Tracker, discover
+from ._diff import diff
+from ._policy import (
+    PolicyEvaluationError,
+    PolicyEvalResult,
+    SemanticEvalResult,
+    SemanticExplainability,
+    SemanticRuleResult,
+)
+from ._streaming_errors import StreamingRequiredError
+from ._api import (
+    agent,
+    begin_agent,
+    begin_subagent,
+    begin_workflow,
+    branch,
+    end_agent,
+    end_subagent,
+    end_workflow,
+    evaluate_policy,
+    evaluate_rules,
+    event,
+    human_input,
+    route,
+    route_decision,
+    session,
+    step,
+    subagent,
+    tool,
+    wait,
+    workflow,
+)
+
+def __getattr__(name: str):
+    # Lazy so that ``import arbis`` never imports opentelemetry.
+    if name == "enable_otel":
+        from .otel import enable_otel
+
+        return enable_otel
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+__all__ = [
+    "discover",
+    "Tracker",
+    "diff",
+    "enable_otel",
+    "session",
+    "workflow",
+    "tool",
+    "wait",
+    "StreamingRequiredError",
+    "agent",
+    "subagent",
+    "branch",
+    "begin_workflow",
+    "end_workflow",
+    "begin_agent",
+    "end_agent",
+    "begin_subagent",
+    "end_subagent",
+    "human_input",
+    "event",
+    "step",
+    "evaluate_policy",
+    "evaluate_rules",
+    "PolicyEvalResult",
+    "PolicyEvaluationError",
+    "SemanticEvalResult",
+    "SemanticExplainability",
+    "SemanticRuleResult",
+    "route",
+    "route_decision",
+]

arbis/_agent_identity.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+import hashlib
+import json
+from typing import Any, Dict, Optional
+
+from ._turn_messages import normalize_turn_messages
+
+
+def stable_hash(value: Any) -> str:
+    payload = json.dumps(value, sort_keys=True, default=str)
+    return hashlib.sha256(payload.encode()).hexdigest()[:16]
+
+
+def tools_hash(input_data: Optional[Dict[str, Any]]) -> str:
+    if not input_data or not input_data.get("tools"):
+        return ""
+    names = []
+    for tool in input_data.get("tools", []):
+        if not isinstance(tool, dict):
+            continue
+        fn = tool.get("function") or {}
+        if isinstance(fn, dict) and fn.get("name"):
+            names.append(str(fn["name"]))
+    return stable_hash(sorted(names)) if names else stable_hash(input_data.get("tools"))
+
+
+def system_prompt_hash(input_data: Optional[Dict[str, Any]]) -> str:
+    if not input_data:
+        return ""
+    for msg in normalize_turn_messages(input_data):
+        if msg.get("role") == "system":
+            return stable_hash(msg.get("content"))
+    if input_data.get("system") is not None:
+        return stable_hash(input_data.get("system"))
+    if input_data.get("instructions") is not None:
+        return stable_hash(input_data.get("instructions"))
+    return ""
+
+
+def developer_prompt_hash(input_data: Optional[Dict[str, Any]]) -> str:
+    if not input_data:
+        return ""
+    for msg in normalize_turn_messages(input_data):
+        if msg.get("role") == "developer":
+            return stable_hash(msg.get("content"))
+    return ""
+
+
+def agent_type_hash(input_data: Optional[Dict[str, Any]]) -> Optional[str]:
+    if not input_data:
+        return None
+    return stable_hash(
+        (
+            system_prompt_hash(input_data),
+            developer_prompt_hash(input_data),
+            tools_hash(input_data),
+        )
+    )