pygent 0.1.0__py3-none-any.whl

pygent/__init__.py

@@ -0,0 +1,11 @@
+"""Pacote Pygent."""
+from importlib import metadata as _metadata
+
+try:
+    __version__: str = _metadata.version(__name__)
+except _metadata.PackageNotFoundError:  # pragma: no cover - fallback for tests
+    __version__ = "0.0.0"
+
+from .agent import Agent, run_interactive  # noqa: E402,F401, must come after __version__
+
+__all__ = ["Agent", "run_interactive"]

pygent/agent.py

@@ -0,0 +1,68 @@
+"""Camada de orquestração: recebe mensagens, chama OpenAI, delega ferramentas."""
+
+import json
+import os
+import pathlib
+import uuid
+import time
+from dataclasses import dataclass, field
+from typing import Any, Dict, List
+
+import openai
+from rich.console import Console
+from rich.panel import Panel
+
+from .runtime import Runtime
+from .tools import TOOL_SCHEMAS, execute_tool
+
+MODEL = os.getenv("PYGENT_MODEL", "gpt-4o-mini-preview")
+SYSTEM_MSG = (
+    "You are Pygent, a sandboxed coding assistant.\n"
+    "Respond with JSON when you need to use a tool."
+)
+
+console = Console()
+
+
+def _chat(messages: List[Dict[str, str]]) -> str:
+    resp = openai.chat.completions.create(
+        model=MODEL,
+        messages=messages,
+        tools=TOOL_SCHEMAS,
+        tool_choice="auto",
+    )
+    return resp.choices[0].message
+
+
+@dataclass
+class Agent:
+    runtime: Runtime = field(default_factory=Runtime)
+    history: List[Dict[str, Any]] = field(default_factory=lambda: [
+        {"role": "system", "content": SYSTEM_MSG}
+    ])
+
+    def step(self, user_msg: str) -> None:
+        self.history.append({"role": "user", "content": user_msg})
+        assistant_msg = _chat(self.history)
+        self.history.append(assistant_msg)
+
+        if assistant_msg.tool_calls:
+            for call in assistant_msg.tool_calls:
+                output = execute_tool(call, self.runtime)
+                self.history.append({"role": "tool", "content": output, "tool_call_id": call.id})
+                console.print(Panel(output, title=f"tool:{call.function.name}"))
+        else:
+            console.print(assistant_msg.content)
+
+
+def run_interactive() -> None:  # pragma: no cover
+    agent = Agent()
+    console.print("[bold green]Pygent[/] iniciado. (digite /exit para sair)")
+    try:
+        while True:
+            user_msg = console.input("[cyan]vc> [/]" )
+            if user_msg.strip() in {"/exit", "quit", "q"}:
+                break
+            agent.step(user_msg)
+    finally:
+        agent.runtime.cleanup()

pygent/cli.py

@@ -0,0 +1,5 @@
+"""Ponto de entrada da CLI do Pygent."""
+from .agent import run_interactive
+
+def main() -> None:  # pragma: no cover
+    run_interactive()

pygent/py.typed

@@ -0,0 +1 @@
+"""empty file to indicate PEP-561 typing support"""

pygent/runtime.py

@@ -0,0 +1,53 @@
+"""Isola as execuções num container Docker efêmero."""
+from __future__ import annotations
+
+import os
+import shutil
+import tempfile
+import subprocess
+import uuid
+from pathlib import Path
+from typing import Union
+
+import docker
+
+
+class Runtime:
+    """Cada instância corresponde a um diretório + container dedicados."""
+
+    def __init__(self, image: str | 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,
+        )
+
+    # ---------------- 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()
+
+    def write_file(self, rel_path: Union[str, Path], content: str) -> str:
+        p = self.base_dir / rel_path
+        p.parent.mkdir(parents=True, exist_ok=True)
+        p.write_text(content)
+        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)

pygent/tools.py

@@ -0,0 +1,50 @@
+"""Mapa de ferramentas disponíveis para o agente."""
+from __future__ import annotations
+import json
+from typing import Any, Dict
+
+from .runtime import Runtime
+
+TOOL_SCHEMAS = [
+    {
+        "type": "function",
+        "function": {
+            "name": "bash",
+            "description": "Run a shell command inside the sandboxed container.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "cmd": {"type": "string", "description": "Command to execute"}
+                },
+                "required": ["cmd"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "write_file",
+            "description": "Create or overwrite a file in the workspace.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string"},
+                    "content": {"type": "string"},
+                },
+                "required": ["path", "content"],
+            },
+        },
+    },
+]
+
+# --------------- dispatcher ---------------
+
+def execute_tool(call: Any, rt: Runtime) -> str:  # pragma: no cover, Any→openai.types.ToolCall
+    name = call.function.name
+    args: Dict[str, Any] = json.loads(call.function.arguments)
+
+    if name == "bash":
+        return rt.bash(**args)
+    if name == "write_file":
+        return rt.write_file(**args)
+    return f"⚠️ unknown tool {name}"

pygent-0.1.0.dist-info/METADATA

@@ -0,0 +1,15 @@
+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

@@ -0,0 +1,12 @@
+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,,

pygent-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

pygent-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+pygent = pygent.cli:main

pygent-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Mariano Chaves
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

pygent-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+pygent