steerable-agent-harness 0.2.0__py3-none-any.whl

steerable_agent_harness/__init__.py

@@ -0,0 +1,21 @@
+from .policy import ToolMode, PolicyDecision, decide_tool_mode
+from .budget import BudgetState, BudgetLimit, consume_budget
+from .retry import RetryPolicy, next_retry_delay_ms
+from .completion import is_terminal_result
+from .tracing import TraceSpan
+
+__version__ = "0.2.0"
+
+__all__ = [
+    "__version__",
+    "ToolMode",
+    "PolicyDecision",
+    "decide_tool_mode",
+    "BudgetState",
+    "BudgetLimit",
+    "consume_budget",
+    "RetryPolicy",
+    "next_retry_delay_ms",
+    "is_terminal_result",
+    "TraceSpan",
+]

steerable_agent_harness/budget.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+
+@dataclass(slots=True)
+class BudgetLimit:
+    max_tokens: int
+    max_steps: int
+    max_tool_calls: int
+
+
+@dataclass(slots=True)
+class BudgetState:
+    tokens_used: int = 0
+    steps_used: int = 0
+    tool_calls_used: int = 0
+
+
+def consume_budget(
+    state: BudgetState,
+    limits: BudgetLimit,
+    *,
+    tokens: int = 0,
+    step: bool = False,
+    tool_call: bool = False,
+) -> tuple[BudgetState, bool]:
+    next_state = BudgetState(
+        tokens_used=state.tokens_used + max(tokens, 0),
+        steps_used=state.steps_used + (1 if step else 0),
+        tool_calls_used=state.tool_calls_used + (1 if tool_call else 0),
+    )
+    exhausted = (
+        next_state.tokens_used > limits.max_tokens
+        or next_state.steps_used > limits.max_steps
+        or next_state.tool_calls_used > limits.max_tool_calls
+    )
+    return next_state, exhausted

steerable_agent_harness/completion.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Any
+
+
+def is_terminal_result(result: dict[str, Any] | None) -> bool:
+    if not result:
+        return False
+    if result.get("terminal") is True:
+        return True
+    if result.get("success") is False and result.get("needsFollowup") is not True:
+        return True
+    return False

steerable_agent_harness/generated.py

@@ -0,0 +1,159 @@
+from __future__ import annotations
+
+from typing import Any
+from pydantic import BaseModel
+
+class ChatAgent(BaseModel):
+    id: str
+    slug: str | None = None
+    name: str
+    icon: str | None = None
+    color: str | None = None
+    description: str | None = None
+    rolePrompt: str | None = None
+    forbiddenPrompt: str | None = None
+    skillIds: list[Any] | None = None
+    allowExternalSkills: bool | None = None
+    isBuiltin: bool | None = None
+    isArchived: bool | None = None
+    sortOrder: int | None = None
+    createdAt: str
+    updatedAt: str
+
+class ChatMessage(BaseModel):
+    id: str
+    chatId: str | None = None
+    role: str
+    content: str
+    agentId: str | None = None
+    toolCalls: list[Any] | None = None
+    toolResult: Any | None = None
+    createdAt: str
+    updatedAt: str | None = None
+
+class SSEEvent(BaseModel):
+    type: str
+    event: str | None = None
+    content: str | None = None
+    hint: str | None = None
+    message: str | None = None
+    code: str | None = None
+    orchestrationGroupId: str | None = None
+    taskId: str | None = None
+    messageId: str | None = None
+    payload: dict[str, Any] | None = None
+
+class AgentSession(BaseModel):
+    id: str | None = None
+    sessionId: str
+    userId: str
+    projectId: Any | None = None
+    chatId: str
+    currentStage: str
+    nextStage: Any | None = None
+    scenario: str | None = None
+    stageData: Any | None = None
+    isActive: bool
+    createdAt: str
+    updatedAt: str
+
+class HarnessTrace(BaseModel):
+    traceId: str
+    userId: Any | None = None
+    chatId: Any | None = None
+    sessionId: Any | None = None
+    assistantMessageId: Any | None = None
+    status: str
+    durationMs: Any | None = None
+    hadError: bool
+    errorMessage: Any | None = None
+    eventCount: int
+    spanCount: int
+    totalTokens: Any | None = None
+    modelId: Any | None = None
+    startedAtMs: Any | None = None
+    createdAt: str
+    updatedAt: str
+
+class TraceEvent(BaseModel):
+    id: str | None = None
+    traceId: str
+    kind: str
+    name: str
+    sequence: int
+    timestampMs: int
+    durationMs: Any | None = None
+    status: Any | None = None
+    payload: Any | None = None
+    createdAt: str | None = None
+
+class TraceSpan(BaseModel):
+    spanId: str
+    traceId: Any | None = None
+    parentSpanId: Any | None = None
+    name: str
+    kind: str | None = None
+    startMs: int
+    endMs: Any | None = None
+    durationMs: Any | None = None
+    status: str
+    attrs: dict[str, Any] | None = None
+
+class CommandSafetyPattern(BaseModel):
+    id: str
+    label: str
+    description: str
+    pattern: str
+    category: str
+    severity: str
+    platform: str
+
+class SidecarError(BaseModel):
+    code: int
+    message: str
+    data: Any | None = None
+    kind: str | None = None
+
+class SidecarHealth(BaseModel):
+    status: str
+    version: str
+    protocolVersion: str | None = None
+    uptimeMs: int
+    pid: int | None = None
+    pythonVersion: str | None = None
+    platform: str | None = None
+    loadedProviders: list[Any] | None = None
+    loadedTools: int | None = None
+    activeTraces: int | None = None
+    checks: dict[str, Any] | None = None
+
+class SidecarNotification(BaseModel):
+    jsonrpc: str
+    method: str
+    params: Any | None = None
+
+class SidecarRequest(BaseModel):
+    jsonrpc: str
+    id: Any
+    method: str
+    params: Any | None = None
+
+class SidecarResponse(BaseModel):
+    jsonrpc: str
+    id: Any
+    result: Any | None = None
+    error: Any | None = None
+
+class ToolCall(BaseModel):
+    id: str
+    name: str
+    arguments: dict[str, Any]
+
+class ToolResult(BaseModel):
+    success: bool
+    terminal: bool | None = None
+    needsFollowup: bool | None = None
+    nextAction: str | None = None
+    message: str | None = None
+    error: str | None = None
+    data: dict[str, Any] | None = None

steerable_agent_harness/policy.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Literal
+
+
+ToolMode = Literal["read", "safe_write", "destructive", "other"]
+
+
+@dataclass(slots=True)
+class PolicyDecision:
+    allowed: bool
+    tool_mode: ToolMode
+    reason: str
+
+
+def decide_tool_mode(tool_name: str) -> ToolMode:
+    normalized = tool_name.lower()
+    if normalized.startswith(("get_", "list_", "read_")):
+        return "read"
+    if normalized.startswith(("create_", "update_", "set_", "write_", "apply_")):
+        return "safe_write"
+    if normalized.startswith(("delete_", "drop_", "remove_", "destroy_")):
+        return "destructive"
+    return "other"

steerable_agent_harness/retry.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+import random
+
+
+@dataclass(slots=True)
+class RetryPolicy:
+    max_attempts: int = 3
+    base_delay_ms: int = 200
+    max_delay_ms: int = 5000
+    jitter: bool = True
+
+
+def next_retry_delay_ms(policy: RetryPolicy, attempt: int) -> int:
+    if attempt < 1:
+        attempt = 1
+    delay = min(policy.base_delay_ms * (2 ** (attempt - 1)), policy.max_delay_ms)
+    if policy.jitter:
+        delay = int(delay * random.uniform(0.8, 1.2))
+    return max(delay, 0)

steerable_agent_harness/tracing.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from datetime import datetime, timezone
+from typing import Any
+
+
+@dataclass(slots=True)
+class TraceSpan:
+    span_id: str
+    name: str
+    start_at: str = field(
+        default_factory=lambda: datetime.now(timezone.utc).isoformat()
+    )
+    end_at: str | None = None
+    attrs: dict[str, Any] = field(default_factory=dict)
+
+    def finish(self) -> None:
+        if self.end_at is None:
+            self.end_at = datetime.now(timezone.utc).isoformat()

steerable_agent_harness-0.2.0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: steerable-agent-harness
+Version: 0.2.0
+Summary: Steerable harness helpers for Python
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic>=2.10.0
+Requires-Dist: steerable-agent-protocol<1.0.0,>=0.1.0
+
+# steerable-agent-harness
+
+Python harness primitives and policy helpers for Steerable.

steerable_agent_harness-0.2.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+steerable_agent_harness/__init__.py,sha256=iexmxAnppYLxK7_KuC-O5wa3j0SIKW5D-R6d2Qz8tek,518
+steerable_agent_harness/budget.py,sha256=Cx38m3ifbbjVtG-FJPKpY4TrBsgS5AOTOaVGybkbUvk,927
+steerable_agent_harness/completion.py,sha256=FMqUyfyUssFmZXwA7ohwWMqj93gsXYsyd1VdfkZxCl0,343
+steerable_agent_harness/generated.py,sha256=PYEMt4BkfH0T8ret99arwkljF_F3J4rXaElKxEwqzGg,3826
+steerable_agent_harness/policy.py,sha256=L4oXfmMgKXgsbr_Q1-7uaeYf6AVzy5LTKWmVaNyH0IM,669
+steerable_agent_harness/retry.py,sha256=UKDAfpKCc92oWABQb2rPepeCg3RYt45X3lHL18OGMQo,528
+steerable_agent_harness/tracing.py,sha256=ru4Y7acHArbdC8aSaYJrzfXioCFcQQY2E-uaZ4ATTuM,531
+steerable_agent_harness-0.2.0.dist-info/METADATA,sha256=3htTtG8Kc-gx97SGUpHZOCoT8TM1VX9VY8if0vJ3tHI,351
+steerable_agent_harness-0.2.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+steerable_agent_harness-0.2.0.dist-info/top_level.txt,sha256=L99GRcOHaeL_gakhZZnmpEbK5yV4fOeIxXzWKW_2lDA,24
+steerable_agent_harness-0.2.0.dist-info/RECORD,,

steerable_agent_harness-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

steerable_agent_harness-0.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+steerable_agent_harness