pygent 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

pygent/__init__.py

@@ -8,5 +8,6 @@ except _metadata.PackageNotFoundError:  # pragma: no cover - fallback for tests
 
 from .agent import Agent, run_interactive  # noqa: E402,F401, must come after __version__
 from .models import Model, OpenAIModel  # noqa: E402,F401
+from .errors import PygentError, APIError  # noqa: E402,F401
 
-__all__ = ["Agent", "run_interactive", "Model", "OpenAIModel"]
+__all__ = ["Agent", "run_interactive", "Model", "OpenAIModel", "PygentError", "APIError"]

pygent/agent.py

@@ -10,6 +10,7 @@ from typing import Any, Dict, List
 
 from rich.console import Console
 from rich.panel import Panel
+from rich.markdown import Markdown
 
 from .runtime import Runtime
 from .tools import TOOL_SCHEMAS, execute_tool
@@ -19,6 +20,10 @@ 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, call the `stop` tool.\n"
+    "You can use the following tools:\n"
+    f"{json.dumps(TOOL_SCHEMAS, indent=2)}\n"
+    "You can also use the `continue` tool to continue the conversation.\n"
 )
 
 console = Console()
@@ -46,7 +51,8 @@ class Agent:
                 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)
+            markdown_response = Markdown(assistant_msg.content)
+            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:
@@ -56,7 +62,7 @@ class Agent:
         for _ in range(max_steps):
             assistant_msg = self.step(msg)
             calls = assistant_msg.tool_calls or []
-            if any(c.function.name == "stop" for c in calls):
+            if any(c.function.name in ("stop", "continue") for c in calls):
                 break
             msg = "continue"
 
@@ -66,9 +72,9 @@ def run_interactive(use_docker: bool | None = None) -> None:  # pragma: no cover
     console.print("[bold green]Pygent[/] iniciado. (digite /exit para sair)")
     try:
         while True:
-            user_msg = console.input("[cyan]vc> [/]" )
+            user_msg = console.input("[cyan]user> [/]" )
             if user_msg.strip() in {"/exit", "quit", "q"}:
                 break
-            agent.step(user_msg)
+            agent.run_until_stop(user_msg)
     finally:
         agent.runtime.cleanup()

pygent/errors.py

@@ -0,0 +1,6 @@
+class PygentError(Exception):
+    """Base error for the Pygent package."""
+
+
+class APIError(PygentError):
+    """Raised when the OpenAI API call fails."""

pygent/models.py

@@ -10,6 +10,7 @@ except ModuleNotFoundError:  # pragma: no cover - fallback to bundled client
     from . import openai_compat as openai
 
 from .openai_compat import Message
+from .errors import APIError
 
 
 class Model(Protocol):
@@ -24,10 +25,13 @@ class OpenAIModel:
     """Default model using the OpenAI-compatible API."""
 
     def chat(self, messages: List[Dict[str, Any]], model: str, tools: Any) -> Message:
-        resp = openai.chat.completions.create(
-            model=model,
-            messages=messages,
-            tools=tools,
-            tool_choice="auto",
-        )
-        return resp.choices[0].message
+        try:
+            resp = openai.chat.completions.create(
+                model=model,
+                messages=messages,
+                tools=tools,
+                tool_choice="auto",
+            )
+            return resp.choices[0].message
+        except Exception as exc:
+            raise APIError(str(exc)) from exc

pygent/openai_compat.py

@@ -2,7 +2,9 @@ import os
 import json
 from dataclasses import dataclass
 from typing import Any, Dict, List
-from urllib import request
+from urllib import request, error
+
+from .errors import APIError
 
 OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
@@ -39,8 +41,15 @@ def _post(path: str, payload: Dict[str, Any]) -> Dict[str, Any]:
     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())
+    try:
+        with request.urlopen(req) as resp:
+            return json.loads(resp.read().decode())
+    except error.HTTPError as exc:  # pragma: no cover - network dependent
+        raise APIError(f"HTTP error {exc.code}: {exc.reason}") from exc
+    except error.URLError as exc:  # pragma: no cover - network dependent
+        raise APIError(f"Connection error: {exc.reason}") from exc
+    except Exception as exc:  # pragma: no cover - fallback
+        raise APIError(str(exc)) from exc
 
 
 class _ChatCompletions:

pygent/runtime.py

@@ -54,34 +54,42 @@ class Runtime:
         caller can display what was run.
         """
         if self._use_docker and self.container is not None:
-            res = self.container.exec_run(
+            try:
+                res = self.container.exec_run(
+                    cmd,
+                    workdir="/workspace",
+                    demux=True,
+                    tty=False,
+                    stdin=False,
+                    timeout=timeout,
+                )
+                stdout, stderr = (
+                    res.output if isinstance(res.output, tuple) else (res.output, b"")
+                )
+                output = (stdout or b"").decode() + (stderr or b"").decode()
+                return f"$ {cmd}\n{output}"
+            except Exception as exc:
+                return f"$ {cmd}\n[error] {exc}"
+        try:
+            proc = subprocess.run(
                 cmd,
-                workdir="/workspace",
-                demux=True,
-                tty=False,
-                stdin=False,
+                shell=True,
+                cwd=self.base_dir,
+                capture_output=True,
+                text=True,
+                stdin=subprocess.DEVNULL,
                 timeout=timeout,
             )
-            stdout, stderr = (
-                res.output if isinstance(res.output, tuple) else (res.output, b"")
-            )
-            output = (stdout or b"").decode() + (stderr or b"").decode()
-            return f"$ {cmd}\n{output}"
-        proc = subprocess.run(
-            cmd,
-            shell=True,
-            cwd=self.base_dir,
-            capture_output=True,
-            text=True,
-            stdin=subprocess.DEVNULL,
-            timeout=timeout,
-        )
-        return f"$ {cmd}\n{proc.stdout + proc.stderr}"
+            return f"$ {cmd}\n{proc.stdout + proc.stderr}"
+        except subprocess.TimeoutExpired:
+            return f"$ {cmd}\n[timeout after {timeout}s]"
+        except Exception as exc:
+            return f"$ {cmd}\n[error] {exc}"
 
     def write_file(self, path: Union[str, Path], content: str) -> str:
         p = self.base_dir / path
         p.parent.mkdir(parents=True, exist_ok=True)
-        p.write_text(content)
+        p.write_text(content, encoding="utf-8")
         return f"Wrote {p.relative_to(self.base_dir)}"
 
     def cleanup(self) -> None:

pygent/tools.py

@@ -43,6 +43,14 @@ TOOL_SCHEMAS = [
             "parameters": {"type": "object", "properties": {}},
         },
     },
+    {
+        "type": "function",
+        "function": {
+            "name": "continue",
+            "description": "Continue the conversation.",
+            "parameters": {"type": "object", "properties": {}},
+        },
+    },
 ]
 
 # --------------- dispatcher ---------------
@@ -57,4 +65,6 @@ def execute_tool(call: Any, rt: Runtime) -> str:  # pragma: no cover, Any→open
         return rt.write_file(**args)
     if name == "stop":
         return "Stopping."
+    if name == "continue":
+        return "Continuing the conversation."
     return f"⚠️ unknown tool {name}"

{pygent-0.1.9.dist-info → pygent-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.9
+Version: 0.1.11
 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

pygent-0.1.11.dist-info/RECORD

@@ -0,0 +1,17 @@
+pygent/__init__.py,sha256=mB_J_GPxLJEk_vHr9mj_4ttLIoSw1F-T1sKEitvRNNU,530
+pygent/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
+pygent/agent.py,sha256=F8XzKzsPFrfgXpqizyOtpKERCy00FQYqI5SHFkOV8_Q,2897
+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=xD77DPFUfmbAwsIA5SjlEmj_TimFtN3j-6h53aUVdF0,3821
+pygent/tools.py,sha256=2_mKDVYbiPzHqUW18ROAsvMmiZXgXd5bjlXTYyPJaHM,2032
+pygent/ui.py,sha256=xqPAvweghPOBBvoD72HzhN6zlXew_3inb8AN7Ck2zpQ,1328
+pygent-0.1.11.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.11.dist-info/METADATA,sha256=9MEoVJl3vxrzYjZYl_r-uyBdULn3y7G6Pzkcvolbw0E,878
+pygent-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.11.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
+pygent-0.1.11.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.11.dist-info/RECORD,,

pygent-0.1.9.dist-info/RECORD

@@ -1,16 +0,0 @@
-pygent/__init__.py,sha256=_YO8FYMUMAWlRYCa6OBsfmJ9P4mCiYoOqFBLrQDP8qQ,442
-pygent/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-pygent/agent.py,sha256=vmTNM1SOCR8e3CV4zBT9Uyzj3LdDglZekaYupjlpl0I,2487
-pygent/cli.py,sha256=Hz2FZeNMVhxoT5DjCqphXla3TisGJtPEz921LEcpxrA,527
-pygent/models.py,sha256=_3Y1Z5wL6FUqzC-EOjZe3Vkcq4SzbPdGz6TbshcEB98,992
-pygent/openai_compat.py,sha256=mS6ntl70jpVH3JzfNYEDhg-z7QIQcMqQTuEV5ja7VOo,2173
-pygent/py.typed,sha256=0Wh72UpGSn4lSGW-u3xMV9kxcBHMdwE15IGUqiJTwqo,52
-pygent/runtime.py,sha256=6VEV4DwpdfrhpGpUiBJr_O1vlo407o0Mvd8Mdhq964A,3417
-pygent/tools.py,sha256=8-jvqYeiJOlZ2ku1MTnBnK2O1m90hkrboq9aZrMORr0,1732
-pygent/ui.py,sha256=xqPAvweghPOBBvoD72HzhN6zlXew_3inb8AN7Ck2zpQ,1328
-pygent-0.1.9.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
-pygent-0.1.9.dist-info/METADATA,sha256=YzPl-3O1RvCKbCeZQTcJUNt2m3sw2G6THKEXpbPtM8o,877
-pygent-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygent-0.1.9.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
-pygent-0.1.9.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
-pygent-0.1.9.dist-info/RECORD,,