pygent 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

pygent/agent.py

@@ -55,8 +55,8 @@ class Agent:
             console.print(assistant_msg.content)
 
 
-def run_interactive() -> None:  # pragma: no cover
-    agent = Agent()
+def run_interactive(use_docker: bool | None = None) -> None:  # pragma: no cover
+    agent = Agent(runtime=Runtime(use_docker=use_docker))
     console.print("[bold green]Pygent[/] iniciado. (digite /exit para sair)")
     try:
         while True:

pygent/cli.py

@@ -1,5 +1,12 @@
 """Ponto de entrada da CLI do Pygent."""
+import argparse
+
 from .agent import run_interactive
 
 def main() -> None:  # pragma: no cover
-    run_interactive()
+    parser = argparse.ArgumentParser(prog="pygent")
+    parser.add_argument("--docker", dest="use_docker", action="store_true", help="executar em container Docker")
+    parser.add_argument("--no-docker", dest="use_docker", action="store_false", help="executar localmente")
+    parser.set_defaults(use_docker=None)
+    args = parser.parse_args()
+    run_interactive(use_docker=args.use_docker)

pygent/runtime.py

@@ -1,43 +1,75 @@
-"""Isola as execuções num container Docker efêmero."""
+"""Executa comandos em um container Docker, caindo para execução local se necessário."""
 from __future__ import annotations
 
 import os
 import shutil
-import tempfile
 import subprocess
+import tempfile
 import uuid
 from pathlib import Path
 from typing import Union
 
-import docker
+try:  # Docker may not be available (e.g. Windows without Docker)
+    import docker  # type: ignore
+except Exception:  # pragma: no cover - optional dependency
+    docker = None
 
 
 class Runtime:
-    """Cada instância corresponde a um diretório + container dedicados."""
+    """Executa comandos em um container Docker ou localmente se Docker faltar."""
 
-    def __init__(self, image: str | None = None) -> None:
+    def __init__(self, image: str | None = None, use_docker: bool | None = None) -> None:
         self.base_dir = Path(tempfile.mkdtemp(prefix="pygent_"))
         self.image = image or os.getenv("PYGENT_IMAGE", "python:3.12-slim")
-        self.client = docker.from_env()
-        self.container = self.client.containers.run(
-            self.image,
-            name=f"pygent-{uuid.uuid4().hex[:8]}",
-            command="sleep infinity",
-            volumes={str(self.base_dir): {"bind": "/workspace", "mode": "rw"}},
-            working_dir="/workspace",
-            detach=True,
-            tty=True,
-            network_disabled=True,
-            mem_limit="512m",
-            pids_limit=256,
-        )
+        env_opt = os.getenv("PYGENT_USE_DOCKER")
+        if use_docker is None:
+            use_docker = (env_opt != "0") if env_opt is not None else True
+        self._use_docker = bool(docker) and use_docker
+        if self._use_docker:
+            try:
+                self.client = docker.from_env()
+                self.container = self.client.containers.run(
+                    self.image,
+                    name=f"pygent-{uuid.uuid4().hex[:8]}",
+                    command="sleep infinity",
+                    volumes={str(self.base_dir): {"bind": "/workspace", "mode": "rw"}},
+                    working_dir="/workspace",
+                    detach=True,
+                    tty=True,
+                    network_disabled=True,
+                    mem_limit="512m",
+                    pids_limit=256,
+                )
+            except Exception:
+                self._use_docker = False
+        if not self._use_docker:
+            self.client = None
+            self.container = None
 
     # ---------------- public API ----------------
     def bash(self, cmd: str, timeout: int = 30) -> str:
-        """Roda comando dentro do container e devolve saída combinada."""
-        res = self.container.exec_run(cmd, workdir="/workspace", demux=True, tty=False, timeout=timeout)
-        stdout, stderr = res.output if isinstance(res.output, tuple) else (res.output, b"")
-        return (stdout or b"").decode() + (stderr or b"").decode()
+        """Roda comando no container ou localmente e devolve a saída."""
+        if self._use_docker and self.container is not None:
+            res = self.container.exec_run(
+                cmd,
+                workdir="/workspace",
+                demux=True,
+                tty=False,
+                timeout=timeout,
+            )
+            stdout, stderr = (
+                res.output if isinstance(res.output, tuple) else (res.output, b"")
+            )
+            return (stdout or b"").decode() + (stderr or b"").decode()
+        proc = subprocess.run(
+            cmd,
+            shell=True,
+            cwd=self.base_dir,
+            capture_output=True,
+            text=True,
+            timeout=timeout,
+        )
+        return proc.stdout + proc.stderr
 
     def write_file(self, rel_path: Union[str, Path], content: str) -> str:
         p = self.base_dir / rel_path
@@ -46,8 +78,9 @@ class Runtime:
         return f"Wrote {p.relative_to(self.base_dir)}"
 
     def cleanup(self) -> None:
-        try:
-            self.container.kill()
-        finally:
-            self.container.remove(force=True)
-            shutil.rmtree(self.base_dir, ignore_errors=True)
+        if self._use_docker and self.container is not None:
+            try:
+                self.container.kill()
+            finally:
+                self.container.remove(force=True)
+        shutil.rmtree(self.base_dir, ignore_errors=True)

pygent-0.1.1.dist-info/METADATA

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: pygent
+Version: 0.1.1
+Summary: Pygent é um assistente de código minimalista que executa comandos em contêiner Docker quando disponível, com fallback para execução local. Veja https://marianochaves.github.io/pygent para a documentação e https://github.com/marianochaves/pygent para o código-fonte.
+Author-email: Mariano Chaves <mchaves.software@gmail.com>
+Project-URL: Documentation, https://marianochaves.github.io/pygent
+Project-URL: Repository, https://github.com/marianochaves/pygent
+Requires-Python: >=3.9
+License-File: LICENSE
+Requires-Dist: openai>=1.0.0
+Requires-Dist: rich>=13.7.0
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Provides-Extra: docs
+Requires-Dist: mkdocs; extra == "docs"
+Provides-Extra: docker
+Requires-Dist: docker>=7.0.0; extra == "docker"
+Dynamic: license-file

pygent-0.1.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+pygent/__init__.py,sha256=25N0WIIaSg0e3XXoDGl_OSBN2ADl-8cHm0C0EOR_opw,359
+pygent/agent.py,sha256=-cQnN5X_srm_rCo33GTpHjaieJ_ogsCbRyYVuy3HGlA,2071
+pygent/cli.py,sha256=P7N9nRRzJ64oMlWY_6sGytjVA1rBA37SYBiQzq4ey4I,527
+pygent/py.typed,sha256=0Wh72UpGSn4lSGW-u3xMV9kxcBHMdwE15IGUqiJTwqo,52
+pygent/runtime.py,sha256=1VaYyRXaRGKc6UkMpQTZJp-XIHItwp1HLZRU79-h-l8,3170
+pygent/tools.py,sha256=1mXaPHFtZwT9w8thDeneH-Ryd9CjViiWOeDE1BtRF6I,1471
+pygent-0.1.1.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.1.dist-info/METADATA,sha256=cQlQp5YpfbF3td0GAxZxyDcNNwmPZhpt-mxA6Cwn79w,839
+pygent-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.1.dist-info/entry_points.txt,sha256=ivw-s2f1abmFsbL4173DP1IuMS7sNxQ6gZuDLdu_jKQ,43
+pygent-0.1.1.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.1.dist-info/RECORD,,

pygent-0.1.0.dist-info/METADATA

@@ -1,15 +0,0 @@
-Metadata-Version: 2.4
-Name: pygent
-Version: 0.1.0
-Summary: A minimal agentic coding assistant that runs each task in a secure container.
-Author-email: Mariano Chaves <mchaves.software@gmail.com>
-Requires-Python: >=3.9
-License-File: LICENSE
-Requires-Dist: openai>=1.0.0
-Requires-Dist: docker>=7.0.0
-Requires-Dist: rich>=13.7.0
-Provides-Extra: test
-Requires-Dist: pytest; extra == "test"
-Provides-Extra: docs
-Requires-Dist: mkdocs; extra == "docs"
-Dynamic: license-file

pygent-0.1.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-pygent/__init__.py,sha256=25N0WIIaSg0e3XXoDGl_OSBN2ADl-8cHm0C0EOR_opw,359
-pygent/agent.py,sha256=vOV0Zx3JzeRtJb-QRdBr-DzFIPVIt2bcZb8xeHlOxI0,2003
-pygent/cli.py,sha256=qOuIHAtt--NFbPRl-RuGSK7mfFj-4CtU4rRr6KYPgKc,139
-pygent/py.typed,sha256=0Wh72UpGSn4lSGW-u3xMV9kxcBHMdwE15IGUqiJTwqo,52
-pygent/runtime.py,sha256=jaLhAT-2Wo6DxmPksFyHvVaCNOOu_c7BgrENsIuwTlI,1897
-pygent/tools.py,sha256=1mXaPHFtZwT9w8thDeneH-Ryd9CjViiWOeDE1BtRF6I,1471
-pygent-0.1.0.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
-pygent-0.1.0.dist-info/METADATA,sha256=Rjva3PKq9hN83Piv8w7ppyNZpgzOnWOjGOYXh7BI7v4,468
-pygent-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygent-0.1.0.dist-info/entry_points.txt,sha256=ivw-s2f1abmFsbL4173DP1IuMS7sNxQ6gZuDLdu_jKQ,43
-pygent-0.1.0.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
-pygent-0.1.0.dist-info/RECORD,,