pygent 0.1.0__tar.gz

pygent-0.1.0/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/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,40 @@
+# Pygent
+
+Pygent é um assistente de código minimalista que executa cada tarefa em um
+container Docker isolado. O objetivo é facilitar execução de comandos de forma
+segura e reprodutível, mantendo o histórico de mensagens da conversa.
+
+## Instalação
+
+Recomendação mais simples é usar `pip`:
+
+```bash
+pip install -e .
+```
+
+O projeto requer Python ≥ 3.9 e depende de `docker`, `openai` e `rich`.
+
+## Uso via CLI
+
+Após instalado, inicie a interface interativa com:
+
+```bash
+pygent
+```
+
+Digita mensagens normalmente e utilize `/exit` para sair.
+
+## Uso via API
+
+Também é possível integrar diretamente com o código Python:
+
+```python
+from pygent import Agent
+
+ag = Agent()
+ag.step("echo hello")  # roda dentro do container
+ag.runtime.cleanup()
+```
+
+Veja a pasta `examples/` para scripts mais completos.
+

pygent-0.1.0/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-0.1.0/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-0.1.0/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-0.1.0/pygent/py.typed

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

pygent-0.1.0/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-0.1.0/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/pygent.egg-info/PKG-INFO

@@ -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/pygent.egg-info/SOURCES.txt

@@ -0,0 +1,17 @@
+LICENSE
+README.md
+pyproject.toml
+pygent/__init__.py
+pygent/agent.py
+pygent/cli.py
+pygent/py.typed
+pygent/runtime.py
+pygent/tools.py
+pygent.egg-info/PKG-INFO
+pygent.egg-info/SOURCES.txt
+pygent.egg-info/dependency_links.txt
+pygent.egg-info/entry_points.txt
+pygent.egg-info/requires.txt
+pygent.egg-info/top_level.txt
+tests/test_tools.py
+tests/test_version.py

pygent-0.1.0/pygent.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

pygent-0.1.0/pygent.egg-info/entry_points.txt

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

pygent-0.1.0/pygent.egg-info/requires.txt

@@ -0,0 +1,9 @@
+openai>=1.0.0
+docker>=7.0.0
+rich>=13.7.0
+
+[docs]
+mkdocs
+
+[test]
+pytest

pygent-0.1.0/pygent.egg-info/top_level.txt

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

pygent-0.1.0/pyproject.toml

@@ -0,0 +1,25 @@
+[project]
+name = "pygent"
+version = "0.1.0"
+description = "A minimal agentic coding assistant that runs each task in a secure container."
+authors = [ { name = "Mariano Chaves", email = "mchaves.software@gmail.com" } ]
+requires-python = ">=3.9"
+dependencies = [
+    "openai>=1.0.0",   # interface com modelos da OpenAI
+    "docker>=7.0.0",    # manipular containers
+    "rich>=13.7.0"      # saídas coloridas (opcional)
+]
+
+[project.scripts]
+pygent = "pygent.cli:main"
+
+[project.optional-dependencies]
+test = ["pytest"]
+docs = ["mkdocs"]
+
+[tool.setuptools.package-data]
+"pygent" = ["py.typed"]
+
+[build-system]
+requires = ["setuptools>=64", "wheel"]
+build-backend = "setuptools.build_meta"

pygent-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

pygent-0.1.0/tests/test_tools.py

@@ -0,0 +1,43 @@
+import os
+import sys
+import types
+
+sys.modules.setdefault('openai', types.ModuleType('openai'))
+sys.modules.setdefault('docker', types.ModuleType('docker'))
+rich_mod = types.ModuleType('rich')
+console_mod = types.ModuleType('console')
+console_mod.Console = lambda *a, **k: None
+panel_mod = types.ModuleType('panel')
+panel_mod.Panel = lambda *a, **k: None
+sys.modules.setdefault('rich', rich_mod)
+sys.modules.setdefault('rich.console', console_mod)
+sys.modules.setdefault('rich.panel', panel_mod)
+
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+from pygent import tools
+
+class DummyRuntime:
+    def bash(self, cmd: str):
+        return f"ran {cmd}"
+    def write_file(self, path: str, content: str):
+        return f"wrote {path}"
+
+def test_execute_bash():
+    call = type('Call', (), {
+        'function': type('Func', (), {
+            'name': 'bash',
+            'arguments': '{"cmd": "ls"}'
+        })
+    })()
+    assert tools.execute_tool(call, DummyRuntime()) == 'ran ls'
+
+
+def test_execute_write_file():
+    call = type('Call', (), {
+        'function': type('Func', (), {
+            'name': 'write_file',
+            'arguments': '{"path": "foo.txt", "content": "bar"}'
+        })
+    })()
+    assert tools.execute_tool(call, DummyRuntime()) == 'wrote foo.txt'

pygent-0.1.0/tests/test_version.py

@@ -0,0 +1,19 @@
+import importlib
+import sys
+import types
+
+# Stub external dependencies so the package can be imported without network
+sys.modules.setdefault('openai', types.ModuleType('openai'))
+sys.modules.setdefault('docker', types.ModuleType('docker'))
+rich_mod = types.ModuleType('rich')
+console_mod = types.ModuleType('console')
+console_mod.Console = lambda *a, **k: None
+panel_mod = types.ModuleType('panel')
+panel_mod.Panel = lambda *a, **k: None
+sys.modules.setdefault('rich', rich_mod)
+sys.modules.setdefault('rich.console', console_mod)
+sys.modules.setdefault('rich.panel', panel_mod)
+
+def test_version_string():
+    pkg = importlib.import_module('pygent')
+    assert isinstance(pkg.__version__, str)