pygent 0.1.13__py3-none-any.whl → 0.1.15__py3-none-any.whl

pygent/__init__.py

@@ -1,6 +1,8 @@
 """Pygent package."""
 from importlib import metadata as _metadata
 
+from .config import load_config
+
 try:
     __version__: str = _metadata.version(__name__)
 except _metadata.PackageNotFoundError:  # pragma: no cover - fallback for tests
@@ -15,6 +17,7 @@ from .task_manager import TaskManager  # noqa: E402,F401
 __all__ = [
     "Agent",
     "run_interactive",
+    "load_config",
     "Model",
     "OpenAIModel",
     "PygentError",

pygent/agent.py

@@ -16,15 +16,20 @@ from .runtime import Runtime
 from . import tools
 from .models import Model, OpenAIModel
 
+DEFAULT_PERSONA = os.getenv("PYGENT_PERSONA", "You are Pygent, a sandboxed coding assistant.")
+
+def build_system_msg(persona: str) -> str:
+    return (
+        f"{persona}\n"
+        "Respond with JSON when you need to use a tool."
+        "If you need to stop or finished you task, call the `stop` tool.\n"
+        "You can use the following tools:\n"
+        f"{json.dumps(tools.TOOL_SCHEMAS, indent=2)}\n"
+        "You can also use the `continue` tool to request user input or continue the conversation.\n"
+    )
+
 DEFAULT_MODEL = os.getenv("PYGENT_MODEL", "gpt-4.1-mini")
-SYSTEM_MSG = (
-    "You are Pygent, a sandboxed coding assistant.\n"
-    "Respond with JSON when you need to use a tool."
-    "If you need to stop or finished you task, call the `stop` tool.\n"
-    "You can use the following tools:\n"
-    f"{json.dumps(tools.TOOL_SCHEMAS, indent=2)}\n"
-    "You can also use the `continue` tool to request user input or continue the conversation.\n"
-)
+SYSTEM_MSG = build_system_msg(DEFAULT_PERSONA)
 
 console = Console()
 
@@ -36,15 +41,21 @@ class Agent:
     runtime: Runtime = field(default_factory=Runtime)
     model: Model = field(default_factory=OpenAIModel)
     model_name: str = DEFAULT_MODEL
-    system_msg: str = SYSTEM_MSG
+    persona: str = DEFAULT_PERSONA
+    system_msg: str = field(default_factory=lambda: build_system_msg(DEFAULT_PERSONA))
     history: List[Dict[str, Any]] = field(default_factory=list)
 
     def __post_init__(self) -> None:
+        if not self.system_msg:
+            self.system_msg = build_system_msg(self.persona)
         if not self.history:
             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 +71,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/cli.py

@@ -1,12 +1,16 @@
 """Command-line entry point for Pygent."""
 import argparse
 
-from .agent import run_interactive
+from .config import load_config
 
 def main() -> None:  # pragma: no cover
     parser = argparse.ArgumentParser(prog="pygent")
     parser.add_argument("--docker", dest="use_docker", action="store_true", help="run commands in a Docker container")
     parser.add_argument("--no-docker", dest="use_docker", action="store_false", help="run locally")
+    parser.add_argument("-c", "--config", help="path to configuration file")
     parser.set_defaults(use_docker=None)
     args = parser.parse_args()
+    load_config(args.config)
+    from .agent import run_interactive
+
     run_interactive(use_docker=args.use_docker)

pygent/config.py

@@ -0,0 +1,40 @@
+import os
+import tomllib
+from pathlib import Path
+from typing import Any, Dict
+
+DEFAULT_CONFIG_FILES = [
+    Path("pygent.toml"),
+    Path.home() / ".pygent.toml",
+]
+
+def load_config(path: str | os.PathLike[str] | None = None) -> Dict[str, Any]:
+    """Load configuration from a TOML file and set environment variables.
+
+    Environment variables already set take precedence over file values.
+    Returns the configuration dictionary.
+    """
+    config: Dict[str, Any] = {}
+    paths = [Path(path)] if path else DEFAULT_CONFIG_FILES
+    for p in paths:
+        if p.is_file():
+            with p.open("rb") as fh:
+                try:
+                    data = tomllib.load(fh)
+                except Exception:
+                    continue
+            config.update(data)
+    # update environment without overwriting existing values
+    if "persona" in config and "PYGENT_PERSONA" not in os.environ:
+        os.environ["PYGENT_PERSONA"] = str(config["persona"])
+    if "task_personas" in config and "PYGENT_TASK_PERSONAS" not in os.environ:
+        if isinstance(config["task_personas"], list):
+            os.environ["PYGENT_TASK_PERSONAS"] = os.pathsep.join(str(p) for p in config["task_personas"])
+        else:
+            os.environ["PYGENT_TASK_PERSONAS"] = str(config["task_personas"])
+    if "initial_files" in config and "PYGENT_INIT_FILES" not in os.environ:
+        if isinstance(config["initial_files"], list):
+            os.environ["PYGENT_INIT_FILES"] = os.pathsep.join(str(p) for p in config["initial_files"])
+        else:
+            os.environ["PYGENT_INIT_FILES"] = str(config["initial_files"])
+    return config

pygent/runtime.py

@@ -18,8 +18,18 @@ except Exception:  # pragma: no cover - optional dependency
 class Runtime:
     """Executes commands in a Docker container or locally if Docker is unavailable."""
 
-    def __init__(self, image: str | None = None, use_docker: bool | None = None) -> None:
+    def __init__(
+        self,
+        image: str | None = None,
+        use_docker: bool | None = None,
+        initial_files: list[str] | None = None,
+    ) -> None:
         self.base_dir = Path(tempfile.mkdtemp(prefix="pygent_"))
+        if initial_files is None:
+            env_files = os.getenv("PYGENT_INIT_FILES")
+            if env_files:
+                initial_files = [f.strip() for f in env_files.split(os.pathsep) if f.strip()]
+        self._initial_files = initial_files or []
         self.image = image or os.getenv("PYGENT_IMAGE", "python:3.12-slim")
         env_opt = os.getenv("PYGENT_USE_DOCKER")
         if use_docker is None:
@@ -46,6 +56,16 @@ class Runtime:
             self.client = None
             self.container = None
 
+        # populate workspace with initial files
+        for fp in self._initial_files:
+            src = Path(fp).expanduser()
+            dest = self.base_dir / src.name
+            if src.is_dir():
+                shutil.copytree(src, dest, dirs_exist_ok=True)
+            elif src.exists():
+                dest.parent.mkdir(parents=True, exist_ok=True)
+                shutil.copy(src, dest)
+
     # ---------------- public API ----------------
     def bash(self, cmd: str, timeout: int = 30) -> str:
         """Run a command in the container or locally and return the output.

pygent/task_manager.py

@@ -30,14 +30,24 @@ class TaskManager:
 
     def __init__(
         self,
-        agent_factory: Callable[[], "Agent"] | None = None,
+        agent_factory: Callable[..., "Agent"] | None = None,
         max_tasks: int | None = None,
+        personas: list[str] | None = None,
     ) -> None:
         from .agent import Agent  # local import to avoid circular dependency
 
         env_max = os.getenv("PYGENT_MAX_TASKS")
         self.max_tasks = max_tasks if max_tasks is not None else int(env_max or "3")
-        self.agent_factory = agent_factory or Agent
+        if agent_factory is None:
+            self.agent_factory = lambda p=None: Agent(persona=p)
+        else:
+            self.agent_factory = agent_factory
+        env_personas = os.getenv("PYGENT_TASK_PERSONAS")
+        if personas is None and env_personas:
+            personas = [p.strip() for p in env_personas.split(os.pathsep) if p.strip()]
+        default_persona = os.getenv("PYGENT_PERSONA", "You are Pygent, a sandboxed coding assistant.")
+        self.personas = personas or [default_persona]
+        self._persona_idx = 0
         self.tasks: Dict[str, Task] = {}
         self._lock = threading.Lock()
 
@@ -47,6 +57,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,7 +70,24 @@ class TaskManager:
             if active >= self.max_tasks:
                 raise RuntimeError(f"max {self.max_tasks} tasks reached")
 
-        agent = self.agent_factory()
+        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
+
+        persona = self.personas[self._persona_idx % len(self.personas)]
+        self._persona_idx += 1
+        try:
+            agent = self.agent_factory(persona)
+        except TypeError:
+            agent = self.agent_factory()
+        setattr(agent, "persona", persona)
+        if not getattr(agent, "system_msg", None):
+            from .agent import build_system_msg  # lazy import
+
+            agent.system_msg = build_system_msg(persona)
         setattr(agent.runtime, "task_depth", parent_depth + 1)
         if files:
             for fp in files:
@@ -74,8 +103,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/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.dist-info → pygent-0.1.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.13
+Version: 0.1.15
 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
@@ -68,6 +68,7 @@ pygent
 Use `--docker` to run commands inside a container (requires
 `pygent[docker]`). Use `--no-docker` or set `PYGENT_USE_DOCKER=0`
 to force local execution.
+Pass `--config path/to/pygent.toml` to load settings from a file.
 
 Type messages normally; use `/exit` to end the session. Each command is executed
 in the container and the result shown in the terminal.

pygent-0.1.15.dist-info/RECORD

@@ -0,0 +1,19 @@
+pygent/__init__.py,sha256=PnTyTP8ObRQfKN0_8-BendG2gGiZhPz5iP9URrlPVwU,776
+pygent/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
+pygent/agent.py,sha256=0yo2K47PgvQJgH_y2xCCkjINks1pYoTQpztPn2J0srk,4437
+pygent/cli.py,sha256=MixuFYGWdZXka6p6ccl4uEoGKFAS5l3xDdUPzcq7y3g,670
+pygent/config.py,sha256=4Yn7kV62RaXa-DI5IqurgWvFZNNykjS0PS4MsUrFEzo,1636
+pygent/errors.py,sha256=s5FBg_v94coSgMh7cfkP4hVXafViGYgCY8QiT698-c4,155
+pygent/models.py,sha256=j3670gjUtvQRGZ5wqGDcQ7ZJVTdT5WiwL7nWTokeYzg,1141
+pygent/openai_compat.py,sha256=cyWFtXt6sDfOlsZd3FuRxbcZMm3WU-DLPBQpbmcuiW8,2617
+pygent/py.typed,sha256=0Wh72UpGSn4lSGW-u3xMV9kxcBHMdwE15IGUqiJTwqo,52
+pygent/runtime.py,sha256=dXk4mcYWdc3UzWN4WgyH-fsAUOlqR9L7cYAAf55Gu50,5121
+pygent/task_manager.py,sha256=pO4A9nAfeUylHc2h94K5JUPXTy8MXEtBbw_oYb8oOQk,5106
+pygent/tools.py,sha256=xHpUpG2QcBEvwTilpw8BDlYWY3KG9mItSt0FoKNkohA,5585
+pygent/ui.py,sha256=xqPAvweghPOBBvoD72HzhN6zlXew_3inb8AN7Ck2zpQ,1328
+pygent-0.1.15.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.15.dist-info/METADATA,sha256=Im7H9V0OHX487HxfIrYvN4m8HNElXzdT6kViNizeAw8,4344
+pygent-0.1.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.15.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
+pygent-0.1.15.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.15.dist-info/RECORD,,

pygent-0.1.13.dist-info/RECORD

@@ -1,18 +0,0 @@
-pygent/__init__.py,sha256=8TB-wzDeQV0YANON-q7sUkzlIyAmTN6gyz4R-axmoYU,724
-pygent/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-pygent/agent.py,sha256=6CfSg943GoT6roj5ZAH8nC5UrfxqmwEhIXHMYad2SFw,3073
-pygent/cli.py,sha256=Hz2FZeNMVhxoT5DjCqphXla3TisGJtPEz921LEcpxrA,527
-pygent/errors.py,sha256=s5FBg_v94coSgMh7cfkP4hVXafViGYgCY8QiT698-c4,155
-pygent/models.py,sha256=j3670gjUtvQRGZ5wqGDcQ7ZJVTdT5WiwL7nWTokeYzg,1141
-pygent/openai_compat.py,sha256=cyWFtXt6sDfOlsZd3FuRxbcZMm3WU-DLPBQpbmcuiW8,2617
-pygent/py.typed,sha256=0Wh72UpGSn4lSGW-u3xMV9kxcBHMdwE15IGUqiJTwqo,52
-pygent/runtime.py,sha256=KckSdTSBIooz181ef8krXbSgqTxIaYDMdCy0ryOi7KQ,4386
-pygent/task_manager.py,sha256=Dq_rezUYC4bzXc3kQ9zLIcjUv3kNc9xzB9xTSDJAzFA,3482
-pygent/tools.py,sha256=HbV01OlRRfzRUf4p6lujtKcHEc43vduC4vv2hq9N1bM,5252
-pygent/ui.py,sha256=xqPAvweghPOBBvoD72HzhN6zlXew_3inb8AN7Ck2zpQ,1328
-pygent-0.1.13.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
-pygent-0.1.13.dist-info/METADATA,sha256=Mop-IQFJad5onR92vwdBexRjVk7FWIHE5_k3y9hS-wI,4278
-pygent-0.1.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygent-0.1.13.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
-pygent-0.1.13.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
-pygent-0.1.13.dist-info/RECORD,,