harness-sdk-python 0.16.0__tar.gz → 0.17.0__tar.gz

{harness_sdk_python-0.16.0 → harness_sdk_python-0.17.0}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.5
 Name: harness-sdk-python
-Version: 0.16.0
+Version: 0.17.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
 Requires-Python: <4.0,>=3.12
-Requires-Dist: pinned<0.11,>=0.10.0
+Requires-Dist: pinned<0.12,>=0.11.0
 Requires-Dist: statewire<0.10,>=0.9.0
 Provides-Extra: deepagents
 Requires-Dist: deepagents>=0.6.12; extra == 'deepagents'

{harness_sdk_python-0.16.0 → harness_sdk_python-0.17.0}/examples/runs_app.py

@@ -20,10 +20,11 @@ from typing import Any
 from fastapi import FastAPI
 from langchain_core.messages import AIMessage, HumanMessage, ToolMessage
 from pinned import PinnedHost, PinnedWorker
+from pinned.testing import MemoryStateStore
 from statewire import Statewire, command, plain
 from statewire.langgraph import append_langgraph_event
 
-from harness_sdk import RunManager, linear_thread
+from harness_sdk import RunManager, linear_thread, run_manager
 
 
 def select_model() -> str:
@@ -81,19 +82,15 @@ class RunsHost(Statewire):
             role=lambda m: "user" if m["type"] == "human" else "assistant",
         )
         self.turn = 0
-        self.runs = RunManager(
-            state=self.state,
-            run=self.run,
-            thread=self.thread,
-            create_task=self.create_task,
-            schedule=self.schedule,
-            capabilities=("rewind",),
-        )
         yield
 
+    async def get_message_meta(self, message_id: str | None) -> dict[str, Any] | None:
+        return await self.thread.get_message_meta(message_id)
+
     async def get_message_child_id(self, parent_id: str) -> str | None:
         return await self.thread.get_message_child_id(parent_id)
 
+    @run_manager(capabilities=("rewind",))
     async def run(self, ctx: RunManager.RunContext) -> Any:
         if ctx.has_rollback:
             self.truncate(ctx.rollback_to)
@@ -158,38 +155,39 @@ class RunsHost(Statewire):
 
     @command("run/enqueue")
     async def run_enqueue(self, params, *, ctx):
-        return await self.runs.enqueue(params, applied=ctx.applied)
+        return await (await self.run.get()).enqueue(params, applied=ctx.applied)
 
     @command("run/steer")
     async def run_steer(self, params, *, ctx):
-        return await self.runs.steer(params, applied=ctx.applied)
+        return await (await self.run.get()).steer(params, applied=ctx.applied)
 
     @command("run/dequeue")
     async def run_dequeue(self, params):
-        return await self.runs.dequeue(params)
+        return await (await self.run.get()).dequeue(params)
 
     @command("run/edit")
     async def run_edit(self, params, *, ctx):
-        return await self.runs.edit(params, applied=ctx.applied)
+        return await (await self.run.get()).edit(params, applied=ctx.applied)
 
     @command("run/reload")
     async def run_reload(self, params, *, ctx):
-        return await self.runs.reload(params, applied=ctx.applied)
+        return await (await self.run.get()).reload(params, applied=ctx.applied)
 
     @command("run/stop")
     async def run_stop(self, params=None, *, ctx):
-        return await self.runs.stop(params, applied=ctx.applied)
+        return await (await self.run.get()).stop(params, applied=ctx.applied)
 
     @command("run/continue")
     async def run_continue(self, params=None, *, ctx):
-        return await self.runs.continue_run(applied=ctx.applied)
+        return await (await self.run.get()).continue_run(applied=ctx.applied)
 
     @command("run/input")
     async def run_input(self, params):
-        return await self.runs.input(params)
+        return await (await self.run.get()).input(params)
 
 
 worker = PinnedWorker(
+    state_store=MemoryStateStore(),
     pinboard_url=os.environ.get("PINBOARD_URL"),
     token=os.environ.get("PINBOARD_TOKEN"),
     advertise_url=os.environ.get("ADVERTISE_URL"),

{harness_sdk_python-0.16.0 → harness_sdk_python-0.17.0}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "harness-sdk-python"
-version = "0.16.0"
+version = "0.17.0"
 description = "RunManager: the harness-sdk runs subsystem for Python Statewire hosts"
 readme = "README.md"
 license = "MIT"
 requires-python = ">=3.12,<4.0"
-dependencies = ["statewire>=0.9.0,<0.10", "pinned>=0.10.0,<0.11"]
+dependencies = ["statewire>=0.9.0,<0.10", "pinned>=0.11.0,<0.12"]
 
 [project.optional-dependencies]
 postgres = ["langgraph-checkpoint-postgres>=2.0.0"]

harness_sdk_python-0.17.0/src/harness_sdk/__init__.py

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