pygent 0.1.2__tar.gz → 0.1.4__tar.gz

{pygent-0.1.2 → pygent-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.2
+Version: 0.1.4
 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
@@ -16,4 +16,6 @@ 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.2 → pygent-0.1.4}/README.md

@@ -8,6 +8,7 @@ Pygent is a coding assistant that executes each request inside an isolated Docke
 * Integrates with OpenAI-compatible models to orchestrate each step.
 * Persists the conversation history during the session.
 * Provides a small Python API for use in other projects.
+* Optional web interface via `pygent-ui`.
 
 ## Installation
 
@@ -26,7 +27,10 @@ To run commands in Docker containers also install `pygent[docker]`.
 Behaviour can be adjusted via environment variables:
 
 * `OPENAI_API_KEY` – key used to access the OpenAI API.
-* `PYGENT_MODEL` – model name used for requests (default `gpt-4o-mini-preview`).
+  Set this to your API key or a key from any compatible provider.
+* `OPENAI_BASE_URL` – base URL for OpenAI-compatible APIs
+  (defaults to ``https://api.openai.com/v1``).
+* `PYGENT_MODEL` – model name used for requests (default `gpt-4.1-mini`).
 * `PYGENT_IMAGE` – Docker image to create the container (default `python:3.12-slim`).
 * `PYGENT_USE_DOCKER` – set to `0` to disable Docker and run locally.
 
@@ -42,7 +46,10 @@ Use `--docker` to run commands inside a container (requires
 `pygent[docker]`). Use `--no-docker` or set `PYGENT_USE_DOCKER=0`
 to force local execution.
 
-Type messages normally; use `/exit` to end the session. Each command is executed in the container and the result shown in the terminal.
+Type messages normally; use `/exit` to end the session. Each command is executed
+in the container and the result shown in the terminal.
+For a minimal web interface run `pygent-ui` instead (requires `pygent[ui]`).
+
 
 ## API usage
 
@@ -57,7 +64,26 @@ ag.step("echo 'Hello World'")
 ag.runtime.cleanup()
 ```
 
-See the `examples/` folder for more complete scripts.
+See the `examples/` folder for more complete scripts. Models can be swapped by
+passing an object implementing the ``Model`` interface when creating the
+``Agent``. The default uses an OpenAI-compatible API, but custom models are
+easy to plug in.
+
+### Using OpenAI and other providers
+
+Set your OpenAI key:
+
+```bash
+export OPENAI_API_KEY="sk-..."
+```
+
+To use a different provider, set `OPENAI_BASE_URL` to the provider
+endpoint and keep `OPENAI_API_KEY` pointing to the correct key:
+
+```bash
+export OPENAI_BASE_URL="https://openrouter.ai/api/v1"
+export OPENAI_API_KEY="your-provider-key"
+```
 
 ## Development
 

{pygent-0.1.2 → pygent-0.1.4}/pygent/__init__.py

@@ -7,5 +7,6 @@ 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__
+from .models import Model, OpenAIModel  # noqa: E402,F401
 
-__all__ = ["Agent", "run_interactive"]
+__all__ = ["Agent", "run_interactive", "Model", "OpenAIModel"]

{pygent-0.1.2 → pygent-0.1.4}/pygent/agent.py

@@ -8,17 +8,14 @@ import time
 from dataclasses import dataclass, field
 from typing import Any, Dict, List
 
-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
+from .models import Model, OpenAIModel
 
-MODEL = os.getenv("PYGENT_MODEL", "gpt-4o-mini-preview")
+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."
@@ -27,26 +24,20 @@ SYSTEM_MSG = (
 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)
+    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}
     ])
 
     def step(self, user_msg: str) -> None:
         self.history.append({"role": "user", "content": user_msg})
-        assistant_msg = _chat(self.history)
+        assistant_msg = self.model.chat(self.history, self.model_name, TOOL_SCHEMAS)
         self.history.append(assistant_msg)
 
         if assistant_msg.tool_calls:

pygent-0.1.4/pygent/models.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+"""Model interface and default implementation for OpenAI-compatible APIs."""
+
+from typing import Any, Dict, List, Protocol
+
+try:
+    import openai  # type: ignore
+except ModuleNotFoundError:  # pragma: no cover - fallback to bundled client
+    from . import openai_compat as openai
+
+from .openai_compat import Message
+
+
+class Model(Protocol):
+    """Protocol for chat models used by :class:`~pygent.agent.Agent`."""
+
+    def chat(self, messages: List[Dict[str, Any]], model: str, tools: Any) -> Message:
+        """Return the assistant message for the given prompt."""
+        ...
+
+
+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

pygent-0.1.4/pygent/ui.py

@@ -0,0 +1,36 @@
+from .agent import Agent
+from .runtime import Runtime
+from .tools import execute_tool, TOOL_SCHEMAS
+
+
+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 = agent.model.chat(agent.history, agent.model_name, TOOL_SCHEMAS)
+        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.2 → pygent-0.1.4}/pygent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.2
+Version: 0.1.4
 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
@@ -16,4 +16,6 @@ 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.2 → pygent-0.1.4}/pygent.egg-info/SOURCES.txt

@@ -4,15 +4,18 @@ pyproject.toml
 pygent/__init__.py
 pygent/agent.py
 pygent/cli.py
+pygent/models.py
 pygent/openai_compat.py
 pygent/py.typed
 pygent/runtime.py
 pygent/tools.py
+pygent/ui.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_custom_model.py
 tests/test_tools.py
 tests/test_version.py

{pygent-0.1.2 → pygent-0.1.4}/pygent.egg-info/entry_points.txt

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

{pygent-0.1.2 → pygent-0.1.4}/pygent.egg-info/requires.txt

@@ -11,3 +11,6 @@ openai>=1.0.0
 
 [test]
 pytest
+
+[ui]
+gradio

{pygent-0.1.2 → pygent-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pygent"
-version = "0.1.2"
+version = "0.1.4"
 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"
@@ -13,6 +13,7 @@ llm = ["openai>=1.0.0"]  # OpenAI-compatible library (optional)
 test = ["pytest"]
 docs = ["mkdocs"]
 docker = ["docker>=7.0.0"]
+ui = ["gradio"]
 
 [project.urls]
 Documentation = "https://marianochaves.github.io/pygent"
@@ -20,6 +21,7 @@ Repository = "https://github.com/marianochaves/pygent"
 
 [project.scripts]
 pygent = "pygent.cli:main"
+pygent-ui = "pygent.ui:main"
 
 
 [tool.setuptools.package-data]

pygent-0.1.4/tests/test_custom_model.py

@@ -0,0 +1,28 @@
+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: type('C', (), {'print': 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 Agent, openai_compat
+
+class DummyModel:
+    def chat(self, messages, model, tools):
+        return openai_compat.Message(role='assistant', content='ok')
+
+
+def test_custom_model():
+    ag = Agent(model=DummyModel())
+    ag.step('hi')
+    assert ag.history[-1].content == 'ok'