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

pygent/agent.py

@@ -35,7 +35,7 @@ class Agent:
         {"role": "system", "content": SYSTEM_MSG}
     ])
 
-    def step(self, user_msg: str) -> None:
+    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)
         self.history.append(assistant_msg)
@@ -47,6 +47,18 @@ class Agent:
                 console.print(Panel(output, title=f"tool:{call.function.name}"))
         else:
             console.print(assistant_msg.content)
+        return assistant_msg
+
+    def run_until_stop(self, user_msg: str, max_steps: int = 10) -> None:
+        """Run steps automatically until the model calls the ``stop`` tool or
+        the step limit is reached."""
+        msg = user_msg
+        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):
+                break
+            msg = "continue"
 
 
 def run_interactive(use_docker: bool | None = None) -> None:  # pragma: no cover

pygent/runtime.py

@@ -71,8 +71,8 @@ class Runtime:
         )
         return proc.stdout + proc.stderr
 
-    def write_file(self, rel_path: Union[str, Path], content: str) -> str:
-        p = self.base_dir / rel_path
+    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)
         return f"Wrote {p.relative_to(self.base_dir)}"

pygent/tools.py

@@ -35,6 +35,14 @@ TOOL_SCHEMAS = [
             },
         },
     },
+    {
+        "type": "function",
+        "function": {
+            "name": "stop",
+            "description": "Stop the autonomous loop.",
+            "parameters": {"type": "object", "properties": {}},
+        },
+    },
 ]
 
 # --------------- dispatcher ---------------
@@ -47,4 +55,6 @@ def execute_tool(call: Any, rt: Runtime) -> str:  # pragma: no cover, Any→open
         return rt.bash(**args)
     if name == "write_file":
         return rt.write_file(**args)
+    if name == "stop":
+        return "Stopping."
     return f"⚠️ unknown tool {name}"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygent
-Version: 0.1.5
+Version: 0.1.7
 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
@@ -8,8 +8,7 @@ Project-URL: Repository, https://github.com/marianochaves/pygent
 Requires-Python: >=3.9
 License-File: LICENSE
 Requires-Dist: rich>=13.7.0
-Provides-Extra: llm
-Requires-Dist: openai>=1.0.0; extra == "llm"
+Requires-Dist: openai>=1.0.0
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Provides-Extra: docs

{pygent-0.1.5.dist-info → pygent-0.1.7.dist-info}/RECORD

@@ -1,16 +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/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=33y4jieNeyZ-9nxtVlOmO236u2fDAAv2GaaEWTQDdm8,3173
-pygent/tools.py,sha256=Ru2_voFgPUVc6YgBTRVByn7vWTxXAXT-loAWFMkXHno,1460
+pygent/runtime.py,sha256=MxuYTIzVqXOLGuM-2J8TtJRjud1Z-LYHt5dtCVpW27E,3165
+pygent/tools.py,sha256=8-jvqYeiJOlZ2ku1MTnBnK2O1m90hkrboq9aZrMORr0,1732
 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.7.dist-info/licenses/LICENSE,sha256=rIktBU2VR4kHzsWul64cbom2zHIgGqYmABoZwSur6T8,1071
+pygent-0.1.7.dist-info/METADATA,sha256=E-A18f6OmzOQiRBF2tb0a0SZgj2q2Y4EY9UA4BRr3FI,877
+pygent-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygent-0.1.7.dist-info/entry_points.txt,sha256=b9j216E5UpuMrQWRZrwyEmacNEAYvw1tCKkZqdIVIOc,70
+pygent-0.1.7.dist-info/top_level.txt,sha256=P26IYsb-ThK5IkGP_bRuGJQ0Q_Y8JCcbYqVpvULdxDw,7
+pygent-0.1.7.dist-info/RECORD,,