new: nerve namespaces command

Author: evilsocket

docs/assets/make_ns.py

@@ -19,7 +19,7 @@
             module = __import__(f"nerve.tools.namespaces.{modname}", fromlist=[""])
 
             doc = module.__doc__ or ""
-            print(f"## {modname}")
+            print(f"## {module.EMOJI} {modname}")
             print()
 
             if hasattr(module, "OPTIONAL_FEATURE"):

docs/namespaces.md

@@ -2,7 +2,7 @@
 
 Nerve offers a rich set of predefined tools, organized in namespaces, that the agent can import [via the `using` directive](index.md#usage). This page contains the list of namespaces available in Nerve, with the descriptive prompt that will be provided to the model.
 
-## anytool
+## 🔧 anytool
 
 Let the agent create its own tools in Python.
 
@@ -19,7 +19,7 @@ Let the agent create its own tools in Python.
 * `code` <i>(<class 'str'>)</i>: The Python code to create the tool.
 </details>
 
-## computer
+## 💻 computer
 
 > [!IMPORTANT]
 > This namespace is not available by default and requires the `computer_use` optional feature.
@@ -94,7 +94,7 @@ Computer use primitives for mouse, keyboard, and screen.
 
 </details>
 
-## filesystem
+## 📂 filesystem
 
 Read-only access primitives to the local filesystem.
 
@@ -117,7 +117,7 @@ Read-only access primitives to the local filesystem.
 * `path` <i>(<class 'str'>)</i>: The path to the file to read
 </details>
 
-## inquire
+## 💬 inquire
 
 Let the agent interactively ask questions to the user in a structured way.
 
@@ -157,7 +157,7 @@ Let the agent interactively ask questions to the user in a structured way.
 * `question` <i>(<class 'str'>)</i>: The question to ask the user.
 </details>
 
-## reasoning
+## 🧠 reasoning
 
 Simulates the reasoning process at runtime.
 
@@ -199,7 +199,7 @@ Simulates the reasoning process at runtime.
 * `thought` <i>(<class 'str'>)</i>: A thought to think about
 </details>
 
-## shell
+## 💻 shell
 
 Let the agent execute shell commands.
 
@@ -215,7 +215,7 @@ Let the agent execute shell commands.
 * `command` <i>(<class 'str'>)</i>: The shell command to execute
 </details>
 
-## task
+## ✅ task
 
 Let the agent autonomously set the task as complete or failed.
 
@@ -238,7 +238,7 @@ Let the agent autonomously set the task as complete or failed.
 * `reason` <i>(<class 'str'>)</i>: The reason why the task is impossible
 </details>
 
-## time
+## 🕒 time
 
 Provides tools for getting the current date and time and waiting for a given number of seconds.
 

nerve/cli/__init__.py

@@ -3,6 +3,7 @@
 import nerve
 from nerve.cli.agents import cli as agents_cli
 from nerve.cli.create import cli as create_cli
+from nerve.cli.namespaces import cli as namespaces_cli
 from nerve.cli.replay import cli as replay_cli
 from nerve.cli.run import cli as run_cli
 
@@ -16,8 +17,7 @@
 cli.add_typer(create_cli)
 cli.add_typer(run_cli)
 cli.add_typer(replay_cli)
-
-# TODO: namespaces command
+cli.add_typer(namespaces_cli)
 
 
 @cli.command(

nerve/cli/namespaces.py

@@ -0,0 +1,29 @@
+import pkgutil
+
+import typer
+from termcolor import colored
+
+import nerve
+import nerve.tools.namespaces as ns_pkg
+
+cli = typer.Typer(
+    no_args_is_help=True,
+    pretty_exceptions_enable=False,
+    context_settings={"help_option_names": ["-h", "--help"]},
+)
+
+
+@cli.command(
+    context_settings={"help_option_names": ["-h", "--help"]},
+    help="List all the available tools namespaces.",
+)
+def namespaces() -> None:
+    print(f"🧠 nerve v{nerve.__version__}")
+    print()
+
+    for _, modname, _ in pkgutil.iter_modules(ns_pkg.__path__):
+        if modname[0] != "_" and not modname.startswith("test_"):
+            module = __import__(f"nerve.tools.namespaces.{modname}", fromlist=[""])
+
+            doc = module.__doc__ or ""
+            print(f"{module.EMOJI} {colored(modname, 'magenta')} - {colored(doc.strip(), 'dark_grey')}")

nerve/tools/namespaces/anytool.py

@@ -11,6 +11,9 @@
 from nerve.runtime import state
 from nerve.tools.compiler import wrap_tool_function
 
+# for docs
+EMOJI = "🔧"
+
 
 def create_tool(
     code: Annotated[

nerve/tools/namespaces/computer.py

@@ -11,6 +11,9 @@
 
 # this is an extra feature, so we need to indicate it
 OPTIONAL_FEATURE = "computer_use"
+# for docs
+EMOJI = "💻"
+
 # max screenshot width
 MAX_WIDTH = 1280
 # typing delay in ms

nerve/tools/namespaces/filesystem.py

@@ -6,6 +6,9 @@
 from pathlib import Path
 from typing import Annotated
 
+# for docs
+EMOJI = "📂"
+
 # if set, the agent will only have access to these paths
 jail: list[str] = []
 

nerve/tools/namespaces/inquire.py

@@ -7,6 +7,9 @@
 import inquirer  # type: ignore
 from pydantic import Field
 
+# for docs
+EMOJI = "💬"
+
 
 def ask_question(
     question: Annotated[

nerve/tools/namespaces/reasoning.py

@@ -6,6 +6,9 @@
 
 import nerve.runtime.state as state
 
+# for docs
+EMOJI = "🧠"
+
 
 def think(thought: Annotated[str, "A thought to think about"]) -> None:
     """

nerve/tools/namespaces/shell.py

@@ -5,6 +5,9 @@
 import subprocess
 from typing import Annotated
 
+# for docs
+EMOJI = "💻"
+
 
 def _maybe_text(output: bytes) -> str | bytes:
     try: