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

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/__main__.py

@@ -0,0 +1,4 @@
+from .cli import main
+
+if __name__ == "__main__":
+    main()

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-4.1-mini")
+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/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/ui.py

@@ -1,6 +1,6 @@
-from .agent import Agent, _chat
+from .agent import Agent
 from .runtime import Runtime
-from .tools import execute_tool
+from .tools import execute_tool, TOOL_SCHEMAS
 
 
 def run_gui(use_docker: bool | None = None) -> None:
@@ -16,7 +16,7 @@ def run_gui(use_docker: bool | None = None) -> None:
 
     def _respond(message: str, history: list[tuple[str, str]] | None) -> str:
         agent.history.append({"role": "user", "content": message})
-        assistant_msg = _chat(agent.history)
+        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:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.3
+Version: 0.1.5
 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.5.dist-info/RECORD

@@ -0,0 +1,16 @@
+pygent/__init__.py,sha256=_YO8FYMUMAWlRYCa6OBsfmJ9P4mCiYoOqFBLrQDP8qQ,442
+pygent/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
+pygent/agent.py,sha256=aR1lHtI3ouI9rEMZLE5lr8tIdC9-A_ZtvdwsvqwxRMg,2010
+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=33y4jieNeyZ-9nxtVlOmO236u2fDAAv2GaaEWTQDdm8,3173
+pygent/tools.py,sha256=Ru2_voFgPUVc6YgBTRVByn7vWTxXAXT-loAWFMkXHno,1460
+pygent/ui.py,sha256=xqPAvweghPOBBvoD72HzhN6zlXew_3inb8AN7Ck2zpQ,1328
+pygent-0.1.5.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.5.dist-info/METADATA,sha256=Ajg8ejlOp6EVdNGBs589xrAqcnzwwlH19ZE49KIeDOU,913
+pygent-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.5.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
+pygent-0.1.5.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.5.dist-info/RECORD,,

pygent-0.1.3.dist-info/RECORD

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