agent-runloop 0.2.0__tar.gz → 0.2.1__tar.gz

{agent_runloop-0.2.0 → agent_runloop-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: agent-runloop
-Version: 0.2.0
+Version: 0.2.1
 Summary: Explicit, durable agent loop: validated tool calls, human approval, budgets, crash recovery, and composable router / waterfall / parallel / handoff / refine blocks.
 Project-URL: Homepage, https://github.com/kroq86/agent-loop
 Project-URL: Repository, https://github.com/kroq86/agent-loop

{agent_runloop-0.2.0 → agent_runloop-0.2.1}/agent_loop/__init__.py

@@ -13,8 +13,8 @@ from agent_loop.compose import (
     Sequence, Step, Timeout, Verdict, chain_task, join_answers, majority_vote, step,
 )
 from agent_loop.loop import (
-    AgentDecision, AgentResult, AgentStep, Budget, EventSink, FinalCheck, Status, ToolSpec, cancel_agent,
-    current_call_id, recover_agent, resume_agent, run_agent, tool_error,
+    AgentDecision, AgentResult, AgentStep, Budget, EventSink, FinalCheck, Status, ToolSpec, ToolTimeout,
+    cancel_agent, current_call_id, recover_agent, resume_agent, run_agent, tool_error,
 )
 from agent_loop.runs import SqliteAgentRunStore
 
@@ -26,7 +26,7 @@ except PackageNotFoundError:  # running from a source checkout
 __all__ = [
     "Agent", "AgentDecision", "AgentResult", "AgentStep", "Budget", "Context", "EventSink", "Fallback", "FinalCheck",
     "Handoff", "HandoffNote", "LLMBackend", "LLMClassifier", "LLMCritic", "Outcome", "Parallel", "Refine", "Router",
-    "Sequence", "SqliteAgentRunStore", "Status", "Step", "Timeout", "ToolSpec", "Verdict", "cancel_agent",
+    "Sequence", "SqliteAgentRunStore", "Status", "Step", "Timeout", "ToolSpec", "ToolTimeout", "Verdict", "cancel_agent",
     "chain_task", "current_call_id", "hooks", "join_answers", "majority_vote", "recover_agent", "resume_agent",
     "run_agent", "step", "tool_error", "usage",
 ]

{agent_runloop-0.2.0 → agent_runloop-0.2.1}/agent_loop/loop.py

@@ -244,12 +244,24 @@ def tool_error(code: str, message: str) -> str:
     return json.dumps({"error": {"code": code, "message": message}})
 
 
+class ToolTimeout(Exception):
+    """A tool exceeded its own `timeout_s`. Deliberately not a TimeoutError:
+    that one means the RUN's deadline, which ends the run. A slow tool is a
+    tool failure the model sees and can work around."""
+
+
 @trace
 async def execute_tool(spec: ToolSpec, args: BaseModel, trace_id: str) -> Any:
     """Separate traced node so trace.html shows each tool's validated
     arguments and full result, under the run_agent node."""
     trace.meta(tool=spec.name, arguments=args.model_dump(), tool_version=spec.version)
-    return await asyncio.wait_for(spec.handler(args, trace_id), timeout=spec.timeout_s)
+    try:
+        return await asyncio.wait_for(spec.handler(args, trace_id), timeout=spec.timeout_s)
+    except TimeoutError:
+        # Only the tool's own timeout lands here: when the run's deadline
+        # fires, the wait is cancelled instead, and that surfaces as the
+        # run's TimeoutError one level up.
+        raise ToolTimeout(f"{spec.name} did not finish within its timeout of {spec.timeout_s}s") from None
 
 
 async def _execute(state: "_RunState", spec: ToolSpec, args: BaseModel, canonical: dict,

{agent_runloop-0.2.0 → agent_runloop-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "agent-runloop"
-version = "0.2.0"
+version = "0.2.1"
 description = "Explicit, durable agent loop: validated tool calls, human approval, budgets, crash recovery, and composable router / waterfall / parallel / handoff / refine blocks."
 readme = "README.md"
 license = "MIT"