yycode 0.3.2__py3-none-any.whl

agent/tool_retry.py

@@ -0,0 +1,106 @@
+"""Retry utilities for agent tool execution."""
+
+import asyncio
+import inspect
+import time
+from typing import Callable, TypeVar
+
+from agent.llm_retry import LLMCallError
+
+T = TypeVar("T")
+
+
+def run_with_retry(
+    func: Callable[..., T],
+    max_retries: int = 2,
+    retry_delay: float = 0.5,
+    backoff_factor: float = 2.0,
+) -> T:
+    """Run a function with automatic retry on failure."""
+    retry_count = 0
+
+    while True:
+        try:
+            return func()
+        except Exception:
+            retry_count += 1
+            if retry_count > max_retries:
+                raise
+            delay = retry_delay * (backoff_factor ** (retry_count - 1))
+            time.sleep(delay)
+
+
+def run_tool_with_retry(
+    handler: Callable[..., str],
+    tool_name: str,
+    max_retries: int = 2,
+    **kwargs
+) -> str:
+    """Run a synchronous tool handler with retry and error handling."""
+    retry_count = 0
+
+    while retry_count <= max_retries:
+        try:
+            if not handler:
+                return f"Unknown tool: {tool_name}"
+            return handler(**kwargs)
+        except Exception as e:
+            retry_count += 1
+            if retry_count > max_retries:
+                return f"Error executing tool {tool_name}: {str(e)}"
+            time.sleep(0.5 * retry_count)
+
+
+async def async_run_tool_with_retry(
+    handler: Callable[..., str],
+    tool_name: str,
+    max_retries: int = 5,
+    timeout_seconds: float | None = None,
+    **kwargs
+) -> str:
+    """Run a sync or async tool handler with retry and error handling."""
+    retry_count = 0
+
+    # Filter out unexpected keyword arguments that the handler doesn't accept
+    if handler:
+        sig = inspect.signature(handler)
+        accepts_var_kwargs = any(
+            param.kind == inspect.Parameter.VAR_KEYWORD
+            for param in sig.parameters.values()
+        )
+        if not accepts_var_kwargs:
+            param_names = list(sig.parameters.keys())
+            kwargs = {k: v for k, v in kwargs.items() if k in param_names}
+
+    while retry_count <= max_retries:
+        try:
+            if not handler:
+                return f"Unknown tool: {tool_name}"
+            return await _run_once(handler, timeout_seconds, **kwargs)
+        except asyncio.TimeoutError:
+            retry_count += 1
+            if retry_count > max_retries:
+                return f"Error executing tool {tool_name}: Timeout after {timeout_seconds}s"
+            await asyncio.sleep(0.5 * retry_count)
+        except LLMCallError:
+            raise
+        except Exception as e:
+            retry_count += 1
+            if retry_count > max_retries:
+                return f"Error executing tool {tool_name}: {str(e)}"
+            await asyncio.sleep(0.5 * retry_count)
+
+
+async def _run_once(handler: Callable[..., str], timeout_seconds: float | None, **kwargs) -> str:
+    """Run a single tool attempt with an optional timeout."""
+    async def call_handler():
+        if inspect.iscoroutinefunction(handler):
+            return await handler(**kwargs)
+        result = await asyncio.to_thread(lambda: handler(**kwargs))
+        if inspect.isawaitable(result):
+            return await result
+        return result
+
+    if timeout_seconds is None:
+        return await call_handler()
+    return await asyncio.wait_for(call_handler(), timeout=timeout_seconds)

agent/tui/__init__.py

@@ -0,0 +1,14 @@
+"""Terminal UI support for yoyoagent."""
+
+from .approval import TuiApprovalAdapter
+from .runner import AgentTuiRunner
+from .state import PendingApproval, SubagentStatus, TimelineItem, TuiState
+
+__all__ = [
+    "AgentTuiRunner",
+    "PendingApproval",
+    "SubagentStatus",
+    "TimelineItem",
+    "TuiApprovalAdapter",
+    "TuiState",
+]