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

pygent/__init__.py

@@ -9,5 +9,15 @@ 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
+from .tools import register_tool, tool  # noqa: E402,F401
 
-__all__ = ["Agent", "run_interactive", "Model", "OpenAIModel", "PygentError", "APIError"]
+__all__ = [
+    "Agent",
+    "run_interactive",
+    "Model",
+    "OpenAIModel",
+    "PygentError",
+    "APIError",
+    "register_tool",
+    "tool",
+]

pygent/agent.py

@@ -13,7 +13,7 @@ from rich.panel import Panel
 from rich.markdown import Markdown
 
 from .runtime import Runtime
-from .tools import TOOL_SCHEMAS, execute_tool
+from . import tools
 from .models import Model, OpenAIModel
 
 DEFAULT_MODEL = os.getenv("PYGENT_MODEL", "gpt-4.1-mini")
@@ -22,7 +22,7 @@ SYSTEM_MSG = (
     "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"
+    f"{json.dumps(tools.TOOL_SCHEMAS, indent=2)}\n"
     "You can also use the `continue` tool to continue the conversation.\n"
 )
 
@@ -36,18 +36,23 @@ class Agent:
     runtime: Runtime = field(default_factory=Runtime)
     model: Model = field(default_factory=OpenAIModel)
     model_name: str = DEFAULT_MODEL
-    history: List[Dict[str, Any]] = field(default_factory=lambda: [
-        {"role": "system", "content": SYSTEM_MSG}
-    ])
+    system_msg: str = SYSTEM_MSG
+    history: List[Dict[str, Any]] = field(default_factory=list)
+
+    def __post_init__(self) -> None:
+        if not self.history:
+            self.history.append({"role": "system", "content": self.system_msg})
 
     def step(self, user_msg: str):
         self.history.append({"role": "user", "content": user_msg})
-        assistant_msg = self.model.chat(self.history, self.model_name, TOOL_SCHEMAS)
+        assistant_msg = self.model.chat(
+            self.history, self.model_name, tools.TOOL_SCHEMAS
+        )
         self.history.append(assistant_msg)
 
         if assistant_msg.tool_calls:
             for call in assistant_msg.tool_calls:
-                output = execute_tool(call, self.runtime)
+                output = tools.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:

pygent/tools.py

@@ -1,70 +1,99 @@
-"""Map of tools available to the agent."""
+"""Tool registry and helper utilities."""
 from __future__ import annotations
+
 import json
-from typing import Any, Dict
+from typing import Any, Callable, Dict, List
 
 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"],
+
+# ---- registry ----
+TOOLS: Dict[str, Callable[..., str]] = {}
+TOOL_SCHEMAS: List[Dict[str, Any]] = []
+
+
+def register_tool(
+    name: str, description: str, parameters: Dict[str, Any], func: Callable[..., str]
+) -> None:
+    """Register a new callable tool."""
+    if name in TOOLS:
+        raise ValueError(f"tool {name} already registered")
+    TOOLS[name] = func
+    TOOL_SCHEMAS.append(
+        {
+            "type": "function",
+            "function": {
+                "name": name,
+                "description": description,
+                "parameters": parameters,
             },
-        },
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "stop",
-            "description": "Stop the autonomous loop.",
-            "parameters": {"type": "object", "properties": {}},
-        },
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "continue",
-            "description": "Continue the conversation.",
-            "parameters": {"type": "object", "properties": {}},
-        },
-    },
-]
+        }
+    )
+
+
+def tool(name: str, description: str, parameters: Dict[str, Any]):
+    """Decorator for registering a tool."""
+
+    def decorator(func: Callable[..., str]) -> Callable[..., str]:
+        register_tool(name, description, parameters, func)
+        return func
+
+    return decorator
 
-# --------------- dispatcher ---------------
 
-def execute_tool(call: Any, rt: Runtime) -> str:  # pragma: no cover, Any→openai.types.ToolCall
+def execute_tool(call: Any, rt: Runtime) -> str:  # pragma: no cover
+    """Dispatch a tool call."""
     name = call.function.name
     args: Dict[str, Any] = json.loads(call.function.arguments)
+    func = TOOLS.get(name)
+    if func is None:
+        return f"⚠️ unknown tool {name}"
+    return func(rt, **args)
+
+
+# ---- built-ins ----
+
+
+@tool(
+    name="bash",
+    description="Run a shell command inside the sandboxed container.",
+    parameters={
+        "type": "object",
+        "properties": {"cmd": {"type": "string", "description": "Command to execute"}},
+        "required": ["cmd"],
+    },
+)
+def _bash(rt: Runtime, cmd: str) -> str:
+    return rt.bash(cmd)
+
+
+@tool(
+    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"],
+    },
+)
+def _write_file(rt: Runtime, path: str, content: str) -> str:
+    return rt.write_file(path, content)
+
+
+@tool(
+    name="stop",
+    description="Stop the autonomous loop.",
+    parameters={"type": "object", "properties": {}},
+)
+def _stop(rt: Runtime) -> str:  # pragma: no cover - side-effect free
+    return "Stopping."
+
+
+@tool(
+    name="continue",
+    description="Continue the conversation.",
+    parameters={"type": "object", "properties": {}},
+)
+def _continue(rt: Runtime) -> str:  # pragma: no cover - side-effect free
+    return "Continuing the conversation."
 
-    if name == "bash":
-        return rt.bash(**args)
-    if name == "write_file":
-        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.11.dist-info → pygent-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.11
+Version: 0.1.12
 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 → pygent-0.1.12.dist-info}/RECORD

@@ -1,17 +1,17 @@
-pygent/__init__.py,sha256=mB_J_GPxLJEk_vHr9mj_4ttLIoSw1F-T1sKEitvRNNU,530
+pygent/__init__.py,sha256=L4IRxHYfzucHUjSMvYHG-1uoTe6XaYJkaHLXuGs4CI8,648
 pygent/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-pygent/agent.py,sha256=F8XzKzsPFrfgXpqizyOtpKERCy00FQYqI5SHFkOV8_Q,2897
+pygent/agent.py,sha256=ZyY9dEvB6bnM1ZFLh4XKQJ68v8xpRlP3HM2RRm6Y6Q0,3030
 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/tools.py,sha256=H2oice1Dyl-rw6Rq9hgWxpGSlxDD3ylCpxha8-4dkZQ,2568
 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.12.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.12.dist-info/METADATA,sha256=896zkz6xObtzwzVoM77ts8BYMgx-lNQzuUCe8YkBv74,878
+pygent-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.12.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
+pygent-0.1.12.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.12.dist-info/RECORD,,