harness-sdk-python 0.4.5__tar.gz → 0.6.0__tar.gz

{harness_sdk_python-0.4.5 → harness_sdk_python-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: harness-sdk-python
-Version: 0.4.5
+Version: 0.6.0
 Summary: RunManager: the harness-sdk runs subsystem for Python Statewire hosts
 Project-URL: Repository, https://github.com/assistant-ui/harness-sdk
 License-Expression: MIT
@@ -23,7 +23,7 @@ class MyHost(Statewire):
         self.runs = RunManager(
             state=self.state,
             start=self._start,
-            get_message_meta=self._get_message_meta,
+            thread=self._thread,
             create_task=self.create_task,
             capabilities=("rewind",),
         )

{harness_sdk_python-0.4.5 → harness_sdk_python-0.6.0}/README.md

@@ -11,7 +11,7 @@ class MyHost(Statewire):
         self.runs = RunManager(
             state=self.state,
             start=self._start,
-            get_message_meta=self._get_message_meta,
+            thread=self._thread,
             create_task=self.create_task,
             capabilities=("rewind",),
         )

{harness_sdk_python-0.4.5 → harness_sdk_python-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "harness-sdk-python"
-version = "0.4.5"
+version = "0.6.0"
 description = "RunManager: the harness-sdk runs subsystem for Python Statewire hosts"
 readme = "README.md"
 license = "MIT"

harness_sdk_python-0.6.0/src/harness_sdk/__init__.py

@@ -0,0 +1,4 @@
+from .linear_thread import linear_thread
+from .run_manager import RunManager
+
+__all__ = ["RunManager", "linear_thread"]

harness_sdk_python-0.6.0/src/harness_sdk/linear_thread.py

@@ -0,0 +1,52 @@
+from dataclasses import dataclass
+from typing import Any, Callable
+
+
+@dataclass(frozen=True)
+class LinearThread:
+    _messages: Callable[[], list[dict[str, Any]]]
+    _role: Callable[[dict[str, Any]], str]
+
+    async def get_message_meta(self, message_id: str | None) -> dict[str, Any] | None:
+        items = self._messages()
+        if message_id is None:
+            return {"isLeaf": not items}
+        for index, message in enumerate(items):
+            if message.get("id") == message_id:
+                return {
+                    "parentId": items[index - 1].get("id") if index > 0 else None,
+                    "role": self._role(message),
+                    "isLeaf": index == len(items) - 1,
+                    "onActiveBranch": True,
+                }
+        return None
+
+    async def get_leaf_message_id(self) -> str | None:
+        items = self._messages()
+        return items[-1].get("id") if items else None
+
+    async def get_message_child_id(self, parent_id: str) -> str | None:
+        """The id of the first non-tool message after parent_id; None when parent_id is unknown or only tools follow."""
+        items = self._messages()
+        index = next(
+            (i for i, m in enumerate(items) if m.get("id") == parent_id), None
+        )
+        if index is None:
+            return None
+        for message in items[index + 1 :]:
+            if message.get("type") != "tool":
+                return message.get("id")
+        return None
+
+
+def linear_thread(
+    *,
+    messages: Callable[[], list[dict[str, Any]]],
+    role: Callable[[dict[str, Any]], str],
+) -> LinearThread:
+    """RunManager thread projection over a linear message list: ``messages`` returns the current list, ``role`` maps a message to its role."""
+    if not callable(messages):
+        raise TypeError("messages must be callable")
+    if not callable(role):
+        raise TypeError("role must be callable")
+    return LinearThread(messages, role)