pygent 0.1.11__tar.gz → 0.1.12__tar.gz

{pygent-0.1.11 → pygent-0.1.12}/PKG-INFO

@@ -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 → pygent-0.1.12}/README.md

@@ -9,6 +9,7 @@ Pygent is a coding assistant that executes each request inside an isolated Docke
 * Persists the conversation history during the session.
 * Provides a small Python API for use in other projects.
 * Optional web interface via `pygent-ui`.
+* Register your own tools and customise the system prompt.
 
 ## Installation
 

{pygent-0.1.11 → pygent-0.1.12}/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-0.1.11 → pygent-0.1.12}/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-0.1.12/pygent/tools.py

@@ -0,0 +1,99 @@
+"""Tool registry and helper utilities."""
+from __future__ import annotations
+
+import json
+from typing import Any, Callable, Dict, List
+
+from .runtime import Runtime
+
+
+# ---- 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,
+            },
+        }
+    )
+
+
+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
+
+
+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."
+

{pygent-0.1.11 → pygent-0.1.12}/pygent.egg-info/PKG-INFO

@@ -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 → pygent-0.1.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pygent"
-version = "0.1.11"
+version = "0.1.12"
 description = "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."
 authors = [ { name = "Mariano Chaves", email = "mchaves.software@gmail.com" } ]
 requires-python = ">=3.9"

{pygent-0.1.11 → pygent-0.1.12}/tests/test_tools.py

@@ -26,7 +26,7 @@ sys.modules.setdefault('rich.syntax', syntax_mod)     # Adicionado
 
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
-from pygent import tools
+from pygent import tools, register_tool
 
 class DummyRuntime:
     def bash(self, cmd: str):
@@ -53,3 +53,24 @@ def test_execute_write_file():
     })()
     assert tools.execute_tool(call, DummyRuntime()) == 'wrote foo.txt'
 
+
+def test_register_and_execute_custom_tool():
+    def hello(rt, name: str):
+        return f"hi {name}"
+
+    register_tool(
+        "hello",
+        "greet",
+        {"type": "object", "properties": {"name": {"type": "string"}}, "required": ["name"]},
+        hello,
+    )
+
+    call = type('Call', (), {
+        'function': type('Func', (), {
+            'name': 'hello',
+            'arguments': '{"name": "bob"}'
+        })
+    })()
+    assert tools.execute_tool(call, DummyRuntime()) == 'hi bob'
+
+

pygent-0.1.11/pygent/tools.py

@@ -1,70 +0,0 @@
-"""Map of tools available to the agent."""
-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"],
-            },
-        },
-    },
-    {
-        "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": {}},
-        },
-    },
-]
-
-# --------------- 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)
-    if name == "stop":
-        return "Stopping."
-    if name == "continue":
-        return "Continuing the conversation."
-    return f"⚠️ unknown tool {name}"