pygent 0.1.13__tar.gz → 0.1.14__tar.gz

{pygent-0.1.13 → pygent-0.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.13
+Version: 0.1.14
 Summary: Pygent is a minimalist coding assistant that runs commands in a Docker container when available and falls back to local execution. See https://marianochaves.github.io/pygent for documentation and https://github.com/marianochaves/pygent for the source code.
 Author-email: Mariano Chaves <mchaves.software@gmail.com>
 Project-URL: Documentation, https://marianochaves.github.io/pygent

{pygent-0.1.13 → pygent-0.1.14}/pygent/agent.py

@@ -44,7 +44,10 @@ class Agent:
             self.history.append({"role": "system", "content": self.system_msg})
 
     def step(self, user_msg: str):
+        """Execute one round of interaction with the model."""
+
         self.history.append({"role": "user", "content": user_msg})
+
         assistant_msg = self.model.chat(
             self.history, self.model_name, tools.TOOL_SCHEMAS
         )
@@ -60,12 +63,36 @@ class Agent:
             console.print(Panel(markdown_response, title="Resposta do Agente", title_align="left", border_style="cyan"))
         return assistant_msg
 
-    def run_until_stop(self, user_msg: str, max_steps: int = 10) -> None:
-        """Run steps automatically until the model calls the ``stop`` tool or
-        the step limit is reached."""
+    def run_until_stop(
+        self,
+        user_msg: str,
+        max_steps: int = 20,
+        step_timeout: float | None = None,
+        max_time: float | None = None,
+    ) -> None:
+        """Run steps until ``stop`` is called or limits are reached."""
+
+        if step_timeout is None:
+            env = os.getenv("PYGENT_STEP_TIMEOUT")
+            step_timeout = float(env) if env else None
+        if max_time is None:
+            env = os.getenv("PYGENT_TASK_TIMEOUT")
+            max_time = float(env) if env else None
+
         msg = user_msg
+        start = time.monotonic()
+        self._timed_out = False
         for _ in range(max_steps):
+            if max_time is not None and time.monotonic() - start > max_time:
+                self.history.append({"role": "system", "content": f"[timeout after {max_time}s]"})
+                self._timed_out = True
+                break
+            step_start = time.monotonic()
             assistant_msg = self.step(msg)
+            if step_timeout is not None and time.monotonic() - step_start > step_timeout:
+                self.history.append({"role": "system", "content": f"[timeout after {step_timeout}s]"})
+                self._timed_out = True
+                break
             calls = assistant_msg.tool_calls or []
             if any(c.function.name in ("stop", "continue") for c in calls):
                 break

{pygent-0.1.13 → pygent-0.1.14}/pygent/task_manager.py

@@ -47,6 +47,8 @@ class TaskManager:
         parent_rt: Runtime,
         files: list[str] | None = None,
         parent_depth: int = 0,
+        step_timeout: float | None = None,
+        task_timeout: float | None = None,
     ) -> str:
         """Create a new agent and run ``prompt`` asynchronously."""
 
@@ -58,6 +60,13 @@ class TaskManager:
             if active >= self.max_tasks:
                 raise RuntimeError(f"max {self.max_tasks} tasks reached")
 
+        if step_timeout is None:
+            env = os.getenv("PYGENT_STEP_TIMEOUT")
+            step_timeout = float(env) if env else 60*5 # default 5 minutes
+        if task_timeout is None:
+            env = os.getenv("PYGENT_TASK_TIMEOUT")
+            task_timeout = float(env) if env else 60*20 # default 20 minutes
+
         agent = self.agent_factory()
         setattr(agent.runtime, "task_depth", parent_depth + 1)
         if files:
@@ -74,8 +83,15 @@ class TaskManager:
 
         def run() -> None:
             try:
-                agent.run_until_stop(prompt)
-                task.status = "finished"
+                agent.run_until_stop(
+                    prompt,
+                    step_timeout=step_timeout,
+                    max_time=task_timeout,
+                )
+                if getattr(agent, "_timed_out", False):
+                    task.status = f"timeout after {task_timeout}s"
+                else:
+                    task.status = "finished"
             except Exception as exc:  # pragma: no cover - error propagation
                 task.status = f"error: {exc}"
 

{pygent-0.1.13 → pygent-0.1.14}/pygent/tools.py

@@ -122,11 +122,19 @@ def _continue(rt: Runtime) -> str:  # pragma: no cover - side-effect free
                 "items": {"type": "string"},
                 "description": "Files to copy to the sub-agent before starting",
             },
+            "timeout": {"type": "number", "description": "Max seconds for the task"},
+            "step_timeout": {"type": "number", "description": "Time limit per step"},
         },
         "required": ["prompt"],
     },
 )
-def _delegate_task(rt: Runtime, prompt: str, files: list[str] | None = None) -> str:
+def _delegate_task(
+    rt: Runtime,
+    prompt: str,
+    files: list[str] | None = None,
+    timeout: float | None = None,
+    step_timeout: float | None = None,
+) -> str:
     if getattr(rt, "task_depth", 0) >= 1:
         return "error: delegation not allowed in sub-tasks"
     try:
@@ -135,6 +143,8 @@ def _delegate_task(rt: Runtime, prompt: str, files: list[str] | None = None) ->
             parent_rt=rt,
             files=files,
             parent_depth=getattr(rt, "task_depth", 0),
+            step_timeout=step_timeout,
+            task_timeout=timeout,
         )
     except RuntimeError as exc:
         return str(exc)

{pygent-0.1.13 → pygent-0.1.14}/pygent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.13
+Version: 0.1.14
 Summary: Pygent is a minimalist coding assistant that runs commands in a Docker container when available and falls back to local execution. See https://marianochaves.github.io/pygent for documentation and https://github.com/marianochaves/pygent for the source code.
 Author-email: Mariano Chaves <mchaves.software@gmail.com>
 Project-URL: Documentation, https://marianochaves.github.io/pygent

{pygent-0.1.13 → pygent-0.1.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pygent"
-version = "0.1.13"
+version = "0.1.14"
 description = "Pygent is a minimalist coding assistant that runs commands in a Docker container when available and falls back to local execution. See https://marianochaves.github.io/pygent for documentation and https://github.com/marianochaves/pygent for the source code."
 readme = "README.md"
 authors = [ { name = "Mariano Chaves", email = "mchaves.software@gmail.com" } ]

{pygent-0.1.13 → pygent-0.1.14}/tests/test_tasks.py

@@ -184,4 +184,20 @@ def test_task_limit():
     second = tools._delegate_task(Runtime(use_docker=False), prompt='run')
     assert 'max' in second
 
-    tm.tasks[tid].thread.join()
+    tm.tasks[tid].thread.join()
+
+
+def test_step_timeout():
+    ag = make_slow_agent()
+    ag.run_until_stop('run', step_timeout=0.05, max_steps=1)
+    assert 'timeout' in ag.history[-1]["content"]
+
+
+def test_task_timeout():
+    tm = TaskManager(agent_factory=make_slow_agent, max_tasks=1)
+    tools._task_manager = tm
+    rt = Runtime(use_docker=False)
+    tid = tm.start_task('run', rt, task_timeout=0.05, step_timeout=0.01)
+    tm.tasks[tid].thread.join()
+    assert 'timeout' in tm.status(tid)
+