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

pygent/__init__.py

@@ -1,4 +1,4 @@
-"""Pacote Pygent."""
+"""Pygent package."""
 from importlib import metadata as _metadata
 
 try:

pygent/agent.py

@@ -1,4 +1,4 @@
-"""Camada de orquestração: recebe mensagens, chama OpenAI, delega ferramentas."""
+"""Orchestration layer: receives messages, calls the OpenAI-compatible backend and dispatches tools."""
 
 import json
 import os
@@ -8,14 +8,17 @@ import time
 from dataclasses import dataclass, field
 from typing import Any, Dict, List
 
-import openai
+try:
+    import openai  # type: ignore
+except ModuleNotFoundError:  # pragma: no cover - fallback to bundled client
+    from . import openai_compat as 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")
+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."

pygent/cli.py

@@ -1,12 +1,12 @@
-"""Ponto de entrada da CLI do Pygent."""
+"""Command-line entry point for Pygent."""
 import argparse
 
 from .agent import run_interactive
 
 def main() -> None:  # pragma: no cover
     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.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.set_defaults(use_docker=None)
     args = parser.parse_args()
     run_interactive(use_docker=args.use_docker)

pygent/openai_compat.py

@@ -0,0 +1,71 @@
+import os
+import json
+from dataclasses import dataclass
+from typing import Any, Dict, List
+from urllib import request
+
+OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
+
+@dataclass
+class ToolCallFunction:
+    name: str
+    arguments: str
+
+@dataclass
+class ToolCall:
+    id: str
+    type: str
+    function: ToolCallFunction
+
+@dataclass
+class Message:
+    role: str
+    content: str | None = None
+    tool_calls: List[ToolCall] | None = None
+
+@dataclass
+class Choice:
+    message: Message
+
+@dataclass
+class ChatCompletion:
+    choices: List[Choice]
+
+
+def _post(path: str, payload: Dict[str, Any]) -> Dict[str, Any]:
+    data = json.dumps(payload).encode()
+    headers = {"Content-Type": "application/json"}
+    if OPENAI_API_KEY:
+        headers["Authorization"] = f"Bearer {OPENAI_API_KEY}"
+    req = request.Request(f"{OPENAI_BASE_URL}{path}", data=data, headers=headers)
+    with request.urlopen(req) as resp:
+        return json.loads(resp.read().decode())
+
+
+class _ChatCompletions:
+    def create(self, model: str, messages: List[Dict[str, Any]], tools: Any = None, tool_choice: str | None = "auto") -> ChatCompletion:
+        payload: Dict[str, Any] = {"model": model, "messages": messages}
+        if tools is not None:
+            payload["tools"] = tools
+        if tool_choice is not None:
+            payload["tool_choice"] = tool_choice
+        raw = _post("/chat/completions", payload)
+        choices: List[Choice] = []
+        for ch in raw.get("choices", []):
+            msg_data = ch.get("message", {})
+            tool_calls = []
+            for tc in msg_data.get("tool_calls", []):
+                func = ToolCallFunction(**tc.get("function", {}))
+                tool_calls.append(ToolCall(id=tc.get("id", ""), type=tc.get("type", ""), function=func))
+            msg = Message(role=msg_data.get("role", ""), content=msg_data.get("content"), tool_calls=tool_calls or None)
+            choices.append(Choice(message=msg))
+        return ChatCompletion(choices=choices)
+
+
+class _Chat:
+    def __init__(self) -> None:
+        self.completions = _ChatCompletions()
+
+
+chat = _Chat()

pygent/runtime.py

@@ -1,4 +1,4 @@
-"""Executa comandos em um container Docker, caindo para execução local se necessário."""
+"""Run commands in a Docker container, falling back to local execution if needed."""
 from __future__ import annotations
 
 import os
@@ -16,7 +16,7 @@ except Exception:  # pragma: no cover - optional dependency
 
 
 class Runtime:
-    """Executa comandos em um container Docker ou localmente se Docker faltar."""
+    """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:
         self.base_dir = Path(tempfile.mkdtemp(prefix="pygent_"))
@@ -48,7 +48,7 @@ class Runtime:
 
     # ---------------- public API ----------------
     def bash(self, cmd: str, timeout: int = 30) -> str:
-        """Roda comando no container ou localmente e devolve a saída."""
+        """Run a command in the container or locally and return the output."""
         if self._use_docker and self.container is not None:
             res = self.container.exec_run(
                 cmd,

pygent/tools.py

@@ -1,4 +1,4 @@
-"""Mapa de ferramentas disponíveis para o agente."""
+"""Map of tools available to the agent."""
 from __future__ import annotations
 import json
 from typing import Any, Dict

pygent/ui.py

@@ -0,0 +1,36 @@
+from .agent import Agent, _chat
+from .runtime import Runtime
+from .tools import execute_tool
+
+
+def run_gui(use_docker: bool | None = None) -> None:
+    """Launch a simple Gradio chat interface."""
+    try:
+        import gradio as gr
+    except ModuleNotFoundError as exc:  # pragma: no cover - optional
+        raise SystemExit(
+            "Gradio is required for the GUI. Install with 'pip install pygent[ui]'"
+        ) from exc
+
+    agent = Agent(runtime=Runtime(use_docker=use_docker))
+
+    def _respond(message: str, history: list[tuple[str, str]] | None) -> str:
+        agent.history.append({"role": "user", "content": message})
+        assistant_msg = _chat(agent.history)
+        agent.history.append(assistant_msg)
+        reply = assistant_msg.content or ""
+        if assistant_msg.tool_calls:
+            for call in assistant_msg.tool_calls:
+                output = execute_tool(call, agent.runtime)
+                agent.history.append({"role": "tool", "content": output, "tool_call_id": call.id})
+                reply += f"\n\n[tool:{call.function.name}]\n{output}"
+        return reply
+
+    try:
+        gr.ChatInterface(_respond, title="Pygent").launch()
+    finally:
+        agent.runtime.cleanup()
+
+
+def main() -> None:  # pragma: no cover
+    run_gui()

{pygent-0.1.1.dist-info → pygent-0.1.3.dist-info}/METADATA

@@ -1,18 +1,21 @@
 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.
+Version: 0.1.3
+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
 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: llm
+Requires-Dist: openai>=1.0.0; extra == "llm"
 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"
+Provides-Extra: ui
+Requires-Dist: gradio; extra == "ui"
 Dynamic: license-file

pygent-0.1.3.dist-info/RECORD

@@ -0,0 +1,14 @@
+pygent/__init__.py,sha256=3YOE3tjTGEc987Vz-TqmYwQ4ogLwTmue642Enf4NVBg,360
+pygent/agent.py,sha256=1CGNB0rUmn-oiZ6HGfbr7fBwB_VtIVMR4xP3fISy_YM,2228
+pygent/cli.py,sha256=Hz2FZeNMVhxoT5DjCqphXla3TisGJtPEz921LEcpxrA,527
+pygent/openai_compat.py,sha256=mS6ntl70jpVH3JzfNYEDhg-z7QIQcMqQTuEV5ja7VOo,2173
+pygent/py.typed,sha256=0Wh72UpGSn4lSGW-u3xMV9kxcBHMdwE15IGUqiJTwqo,52
+pygent/runtime.py,sha256=33y4jieNeyZ-9nxtVlOmO236u2fDAAv2GaaEWTQDdm8,3173
+pygent/tools.py,sha256=Ru2_voFgPUVc6YgBTRVByn7vWTxXAXT-loAWFMkXHno,1460
+pygent/ui.py,sha256=DSW1o3gdhuEdJkyBkJmPE_NUHgvowzMjW2Hs2kGh_4A,1278
+pygent-0.1.3.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.3.dist-info/METADATA,sha256=tyOn6Jgwd70ALAlVa21m7XulzU4nnGCC_KKquKNKsWg,913
+pygent-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.3.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
+pygent-0.1.3.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.3.dist-info/RECORD,,

{pygent-0.1.1.dist-info → pygent-0.1.3.dist-info}/entry_points.txt

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

pygent-0.1.1.dist-info/RECORD

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