Adding pythonic tool calling to agents from rigging.

Author: monoxgas

.hooks/generate_docs.py

@@ -42,7 +42,7 @@ def convert_details(self, el: t.Any, text: str, parent_tags: t.Any) -> t.Any:  #
         classes = el.get("class", [])
 
         # Handle source code details specially
-        if "quote" in classes:
+        if "mkdocstrings-source" in classes or "quote" in classes:
             summary = el.find("summary")
             if summary:
                 file_path = summary.get_text().replace("Source code in ", "").strip()
@@ -128,6 +128,10 @@ def generate_docs_for_module(
         module_data = self.handler.collect(module_path, options)
         html = self.handler.render(module_data, options)
 
+        if "Source code in " in html:
+            with open("debug.html", "w", encoding="utf-8") as f:
+                f.write(html)
+
         return str(
             CustomMarkdownConverter(
                 code_language="python",

dreadnode/agent/agent.py

@@ -272,6 +272,8 @@ def _get_transforms(self) -> list[rg.Transform]:
                     transforms.append(rg.transform.tools_to_json_with_tag_transform)
                 case "json":
                     transforms.append(rg.transform.tools_to_json_transform)
+                case "pythonic":
+                    transforms.append(rg.transform.tools_to_pythonic_transform)
 
         return transforms
 
@@ -931,7 +933,7 @@ def model_post_init(self, context: t.Any) -> None:
             self.hooks.append(
                 retry_with_feedback(
                     event_type=AgentStalled,
-                    feedback="Continue the task if possible, use the 'finish_task' tool to complete it, or 'give_up_on_task' if it cannot be completed.",
+                    feedback="No tool calls were observed. Continue the task if possible, use the 'finish_task' tool to complete it, or 'give_up_on_task' if it cannot be completed.",
                 )
             )
 

pyproject.toml

@@ -15,14 +15,14 @@ dependencies = [
     "coolname>=2.2.0,<3.0.0",
     "pandas>=2.2.3,<3.0.0",
     "fsspec[s3]>=2023.1.0,<=2025.3.0",
-    "rigging>=3.2.1,<4.0.0",
     "optuna>=4.5.0,<5.0.0",
     "numpy<=2.2.6", # 3.10 support was dropped in 2.3.0
     "universal-pathlib>=0.3.3,<0.4.0",
     "loguru>=0.7.3",
     "python-jsonpath>=2.0.1",
     "pyyaml>=6.0.2",
     "cyclopts>=4.2.0",
+    "rigging>=3.3.4",
 ]
 
 [project.optional-dependencies]