agentstack-cli 0.5.1rc2__tar.gz → 0.5.2__tar.gz

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-cli
-Version: 0.5.1rc2
+Version: 0.5.2
 Summary: Agent Stack CLI
 Author: IBM Corp.
 Requires-Dist: anyio~=4.10.0

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentstack-cli"
-version = "0.5.1-rc2"
+version = "0.5.2"
 description = "Agent Stack CLI"
 readme = "README.md"
 authors = [{ name = "IBM Corp." }]

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/__init__.py

@@ -10,7 +10,6 @@ import typer
 
 import agentstack_cli.commands.agent
 import agentstack_cli.commands.build
-import agentstack_cli.commands.mcp
 import agentstack_cli.commands.model
 import agentstack_cli.commands.platform
 import agentstack_cli.commands.self
@@ -29,14 +28,14 @@ Usage: agentstack [OPTIONS] COMMAND [ARGS]...
 ╭─ Getting Started ──────────────────────────────────────────────────────────╮
 │ ui       Launch the web interface                                          │
 │ list     View all available agents                                         │
+│ info     Show agent details                                                │
 │ run      Run an agent interactively                                        │
 ╰────────────────────────────────────────────────────────────────────────────╯
 
-╭─ Agent Management ─────────────────────────────────────────────────────────╮
+╭─ Agent Management [Admin only] ────────────────────────────────────────────╮
 │ add                               Install an agent (Docker, GitHub)        │
 │ remove                            Uninstall an agent                       │
 │ update                            Update an agent                          │
-│ info                              Show agent details                       │
 │ logs                              Stream agent execution logs              │
 │ env                               Manage agent environment variables       │
 │ build                             Build an agent remotely                  │
@@ -44,13 +43,13 @@ Usage: agentstack [OPTIONS] COMMAND [ARGS]...
 ╰────────────────────────────────────────────────────────────────────────────╯
 
 ╭─ Platform & Configuration ─────────────────────────────────────────────────╮
-│ model           Configure 15+ LLM providers                                │
-│ platform        Start, stop, or delete local platform                      │
+│ model           Configure 15+ LLM providers [Admin only]                   │
+│ platform        Start, stop, or delete local platform [Local only]         │
 │ server          Connect to remote Agent Stack servers                      │
-│ user            Manage users and roles                                     │
+│ user            Manage users and roles [Admin only]                        │
 │ self version    Show Agent Stack CLI and Platform version                  │
-│ self upgrade    Upgrade Agent Stack CLI and Platform                       │
-│ self uninstall  Uninstall Agent Stack CLI and Platform                     │
+│ self upgrade    Upgrade Agent Stack CLI and Platform [Local only]          │
+│ self uninstall  Uninstall Agent Stack CLI and Platform [Local only]        │
 ╰────────────────────────────────────────────────────────────────────────────╯
 
 ╭─ Options ──────────────────────────────────────────────────────────────────╮
@@ -74,13 +73,20 @@ def main(
         raise typer.Exit()
 
 
-app.add_typer(agentstack_cli.commands.model.app, name="model", no_args_is_help=True, help="Manage model providers.")
-app.add_typer(agentstack_cli.commands.agent.app, name="agent", no_args_is_help=True, help="Manage agents.")
 app.add_typer(
-    agentstack_cli.commands.platform.app, name="platform", no_args_is_help=True, help="Manage Agent Stack platform."
+    agentstack_cli.commands.model.app, name="model", no_args_is_help=True, help="Manage model providers. [Admin only]"
 )
 app.add_typer(
-    agentstack_cli.commands.mcp.app, name="mcp", no_args_is_help=True, help="Manage MCP servers and toolkits."
+    agentstack_cli.commands.agent.app,
+    name="agent",
+    no_args_is_help=True,
+    help="Manage agents. Some commands are [Admin only].",
+)
+app.add_typer(
+    agentstack_cli.commands.platform.app,
+    name="platform",
+    no_args_is_help=True,
+    help="Manage Agent Stack platform. [Local only]",
 )
 app.add_typer(agentstack_cli.commands.build.app, name="", no_args_is_help=True, help="Build agent images.")
 app.add_typer(
@@ -100,7 +106,7 @@ app.add_typer(
     agentstack_cli.commands.user.app,
     name="user",
     no_args_is_help=True,
-    help="Manage users.",
+    help="Manage users. [Admin only]",
 )
 
 

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/async_typer.py

@@ -82,15 +82,23 @@ class AsyncTyper(typer.Typer):
                     else:
                         return f(*args, **kwargs)
                 except* Exception as ex:
+                    is_permission_error = False
                     is_connect_error = False
                     for exc_type, message in extract_messages(ex):
                         err_console.print(format_error(exc_type, message))
                         is_connect_error = is_connect_error or exc_type in ["ConnectionError", "ConnectError"]
+                        is_permission_error = is_permission_error or (
+                            exc_type == "HTTPStatusError" and "403" in message
+                        )
                         err_console.print()
                     if is_connect_error:
                         err_console.hint(
                             "Start the Agent Stack platform using: [green]agentstack platform start[/green]. If that does not help, run [green]agentstack platform delete[/green] to clean up, then [green]agentstack platform start[/green] again."
                         )
+                    elif is_permission_error:
+                        err_console.hint(
+                            "This command requires higher permissions than your account currently has. Contact your administrator for assistance."
+                        )
                     else:
                         err_console.hint(
                             "Are you having consistent problems? If so, try these troubleshooting steps: [green]agentstack platform delete[/green] to remove the platform, and [green]agentstack platform start[/green] to recreate it."

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/agent.py

@@ -151,16 +151,32 @@ class ProviderUtils(BaseModel):
 app = AsyncTyper()
 
 processing_messages = [
-    "Buzzing with ideas...",
-    "Pollinating thoughts...",
-    "Honey of an answer coming up...",
-    "Swarming through data...",
-    "Bee-processing your request...",
-    "Hive mind activating...",
-    "Making cognitive honey...",
-    "Waggle dancing for answers...",
-    "Bee right back...",
-    "Extracting knowledge nectar...",
+    "Asking agents...",
+    "Booting up bots...",
+    "Calibrating cognition...",
+    "Directing drones...",
+    "Engaging engines...",
+    "Fetching functions...",
+    "Gathering goals...",
+    "Hardening hypotheses...",
+    "Interpreting intentions...",
+    "Juggling judgements...",
+    "Kernelizing knowledge...",
+    "Loading logic...",
+    "Mobilizing models...",
+    "Nudging networks...",
+    "Optimizing outputs...",
+    "Prompting pipelines...",
+    "Quantizing queries...",
+    "Refining responses...",
+    "Scaling stacks...",
+    "Tuning transformers...",
+    "Unifying understandings...",
+    "Vectorizing values...",
+    "Wiring workflows...",
+    "Xecuting xperiments...",
+    "Yanking YAMLs...",
+    "Zipping zettabytes...",
 ]
 
 configuration = Configuration()
@@ -175,7 +191,7 @@ async def add_agent(
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
 ) -> None:
-    """Add a docker image or GitHub repository.
+    """Add a docker image or GitHub repository. [Admin only]
 
     This command supports a variety of GitHub URL formats for deploying agents:
 
@@ -260,7 +276,7 @@ async def update_agent(
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
 ) -> None:
-    """Upgrade agent to a newer docker image or build from GitHub repository"""
+    """Upgrade agent to a newer docker image or build from GitHub repository. [Admin only]"""
     with verbosity(verbose):
         async with configuration.use_platform_client():
             providers = await Provider.list()
@@ -389,7 +405,7 @@ async def uninstall_agent(
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
     all: typing.Annotated[bool, typer.Option("--all", "-a", help="Remove all agents without selection.")] = False,
 ) -> None:
-    """Remove agent"""
+    """Remove agent. [Admin only]"""
     if search_path and all:
         console.error(
             "[bold]Cannot specify both --all and a search path."
@@ -440,7 +456,7 @@ async def stream_logs(
         str, typer.Argument(..., help="Short ID, agent name or part of the provider location")
     ],
 ):
-    """Stream agent provider logs"""
+    """Stream agent provider logs. [Admin only]"""
     announce_server_action(f"Streaming logs for '{search_path}' from")
     async with configuration.use_platform_client():
         provider = select_provider(search_path, await Provider.list()).id
@@ -1270,7 +1286,7 @@ async def add_env(
     env: typing.Annotated[list[str], typer.Argument(help="Environment variables to pass to agent")],
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
 ) -> None:
-    """Store environment variables"""
+    """Store environment variables. [Admin only]"""
     url = announce_server_action(f"Adding environment variables for '{search_path}' on")
     await confirm_server_action("Apply these environment variable changes on", url=url, yes=yes)
     env_vars = dict(parse_env_var(var) for var in env)
@@ -1286,7 +1302,7 @@ async def list_env(
         str, typer.Argument(..., help="Short ID, agent name or part of the provider location")
     ],
 ):
-    """List stored environment variables"""
+    """List stored environment variables. [Admin only]"""
     announce_server_action(f"Listing environment variables for '{search_path}' on")
     async with configuration.use_platform_client():
         provider = select_provider(search_path, await Provider.list())
@@ -1301,6 +1317,7 @@ async def remove_env(
     env: typing.Annotated[list[str], typer.Argument(help="Environment variable(s) to remove")],
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
 ):
+    """Remove environment variable(s). [Admin only]"""
     url = announce_server_action(f"Removing environment variables from '{search_path}' on")
     await confirm_server_action("Remove the selected environment variables on", url=url, yes=yes)
     async with configuration.use_platform_client():
@@ -1321,7 +1338,7 @@ async def list_feedback(
     limit: typing.Annotated[int, typer.Option("--limit", help="Number of results per page [default: 50]")] = 50,
     after_cursor: typing.Annotated[str | None, typer.Option("--after", help="Cursor for pagination")] = None,
 ):
-    """List your agent feedback"""
+    """List your agent feedback. [Admin only]"""
 
     announce_server_action("Listing feedback on")
 

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/build.py

@@ -61,7 +61,7 @@ async def client_side_build(
     vm_name: typing.Annotated[str, typer.Option(hidden=True)] = "agentstack",
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
 ):
-    """Build agent locally using Docker."""
+    """Build agent locally using Docker. [Local only]"""
     with verbosity(verbose):
         await run_command(["which", "docker"], "Checking docker")
         image_id = "agentstack-agent-build-tmp:latest"
@@ -210,7 +210,7 @@ async def server_side_build(
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
 ):
-    """Build agent from a GitHub repository in the platform."""
+    """Build agent from a GitHub repository in the platform. [Admin only]"""
 
     url = announce_server_action(f"Starting build for '{github_url}' on")
     await confirm_server_action("Proceed with building this agent on", url=url, yes=yes)

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/model.py

@@ -379,6 +379,7 @@ async def _select_default_model(capability: ModelCapability) -> str | None:
 
 @app.command("list")
 async def list_models():
+    """List all available models."""
     announce_server_action("Listing models on")
     async with configuration.use_platform_client():
         config = await SystemConfiguration.get()
@@ -411,7 +412,7 @@ async def _reset_configuration(existing_providers: list[ModelProvider] | None =
     await SystemConfiguration.update(default_embedding_model=None, default_llm_model=None)
 
 
-@app.command("setup")
+@app.command("setup", help="Interactive setup for LLM and embedding provider environment variables [Admin only]")
 async def setup(
     use_true_localhost: typing.Annotated[bool, typer.Option(hidden=True)] = False,
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
@@ -477,7 +478,7 @@ async def setup(
                 raise
 
 
-@app.command("change | select | default")
+@app.command("change | select | default", help="Change the default model [Admin only]")
 async def select_default_model(
     capability: typing.Annotated[
         ModelCapability | None, typer.Argument(help="Which default model to change (llm/embedding)")
@@ -524,19 +525,21 @@ def _list_providers(providers: list[ModelProvider]):
 
 @model_provider_app.command("list")
 async def list_model_providers():
+    """List all available model providers."""
     announce_server_action("Listing model providers on")
     async with configuration.use_platform_client():
         providers = await ModelProvider.list()
         _list_providers(providers)
 
 
-@model_provider_app.command("add")
+@model_provider_app.command("add", help="Add a new model provider [Admin only]")
 @app.command("add")
 async def add_provider(
     capability: typing.Annotated[
         ModelCapability | None, typer.Argument(help="Which default model to change (llm/embedding)")
     ] = None,
 ):
+    """Add a new model provider. [Admin only]"""
     announce_server_action("Adding provider for")
     if not capability:
         capability = await inquirer.select(  # type: ignore
@@ -577,7 +580,7 @@ def _select_provider(providers: list[ModelProvider], search_path: str) -> ModelP
     return selected_provider
 
 
-@model_provider_app.command("remove | rm | delete")
+@model_provider_app.command("remove | rm | delete", help="Remove a model provider [Admin only]")
 @app.command("remove | rm | delete")
 async def remove_provider(
     search_path: typing.Annotated[
@@ -639,7 +642,7 @@ async def ensure_llm_provider():
         if config.default_llm_model and not inconsistent:
             return
 
-    console.print("[bold]Welcome to 🐝 [red]Agent Stack[/red]![/bold]")
+    console.print("[bold]Welcome to [red]Agent Stack[/red]![/bold]")
     console.print("Let's start by configuring your LLM environment.\n")
     try:
         await setup()

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/platform/__init__.py

@@ -57,7 +57,7 @@ def get_driver(vm_name: str = "agentstack") -> BaseDriver:
         sys.exit(1)
 
 
-@app.command("start")
+@app.command("start", help="Start Agent Stack platform. [Local only]")
 async def start(
     set_values_list: typing.Annotated[
         list[str], typer.Option("--set", help="Set Helm chart values using <key>=<value> syntax", default_factory=list)
@@ -68,13 +68,19 @@ async def start(
             "--import", help="Import an image from a local Docker CLI into Agent Stack platform", default_factory=list
         ),
     ],
+    pull_on_host: typing.Annotated[
+        bool,
+        typer.Option(
+            "--pull-on-host",
+            help="Pull images on host Docker daemon and import them instead of pulling inside the VM. Acts as a pull cache layer.",
+        ),
+    ] = False,
     values_file: typing.Annotated[
         pathlib.Path | None, typer.Option("-f", help="Set Helm chart values using yaml values file")
     ] = None,
     vm_name: typing.Annotated[str, typer.Option(hidden=True)] = "agentstack",
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
 ):
-    """Start Agent Stack platform."""
     import agentstack_cli.commands.server
 
     values_file_path = None
@@ -87,7 +93,12 @@ async def start(
         driver = get_driver(vm_name=vm_name)
         await driver.create_vm()
         await driver.install_tools()
-        await driver.deploy(set_values_list=set_values_list, values_file=values_file_path, import_images=import_images)
+        await driver.deploy(
+            set_values_list=set_values_list,
+            values_file=values_file_path,
+            import_images=import_images,
+            pull_on_host=pull_on_host,
+        )
 
         with console.status("Waiting for Agent Stack platform to be ready...", spinner="dots"):
             timeout = datetime.timedelta(minutes=20)
@@ -124,12 +135,11 @@ async def start(
         await agentstack_cli.commands.server.server_login("http://localhost:8333")
 
 
-@app.command("stop")
+@app.command("stop", help="Stop Agent Stack platform. [Local only]")
 async def stop(
     vm_name: typing.Annotated[str, typer.Option(hidden=True)] = "agentstack",
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
 ):
-    """Stop Agent Stack platform."""
     with verbosity(verbose):
         driver = get_driver(vm_name=vm_name)
         if not await driver.status():
@@ -139,40 +149,37 @@ async def stop(
         console.success("Agent Stack platform stopped successfully.")
 
 
-@app.command("delete")
+@app.command("delete", help="Delete Agent Stack platform. [Local only]")
 async def delete(
     vm_name: typing.Annotated[str, typer.Option(hidden=True)] = "agentstack",
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
 ):
-    """Delete Agent Stack platform."""
     with verbosity(verbose):
         driver = get_driver(vm_name=vm_name)
         await driver.delete()
         console.success("Agent Stack platform deleted successfully.")
 
 
-@app.command("import")
+@app.command("import", help="Import a local docker image into the Agent Stack platform. [Local only]")
 async def import_image_cmd(
     tag: typing.Annotated[str, typer.Argument(help="Docker image tag to import")],
     vm_name: typing.Annotated[str, typer.Option(hidden=True)] = "agentstack",
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
 ):
-    """Import a local docker image into the Agent Stack platform."""
     with verbosity(verbose):
         driver = get_driver(vm_name=vm_name)
         if (await driver.status()) != "running":
             console.error("Agent Stack platform is not running.")
             sys.exit(1)
-        await driver.import_image(tag)
+        await driver.import_images(tag)
 
 
-@app.command("exec")
+@app.command("exec", help="For debugging -- execute a command inside the Agent Stack platform VM. [Local only]")
 async def exec_cmd(
     command: typing.Annotated[list[str] | None, typer.Argument()] = None,
     vm_name: typing.Annotated[str, typer.Option(hidden=True)] = "agentstack",
     verbose: typing.Annotated[bool, typer.Option("-v", "--verbose", help="Show verbose output")] = False,
 ):
-    """For debugging -- execute a command inside the Agent Stack platform VM."""
     with verbosity(verbose, show_success_status=False):
         driver = get_driver(vm_name=vm_name)
         if (await driver.status()) != "running":

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/platform/base_driver.py

@@ -15,6 +15,7 @@ from tenacity import AsyncRetrying, stop_after_attempt
 
 import agentstack_cli.commands.platform.istio
 from agentstack_cli.configuration import Configuration
+from agentstack_cli.utils import run_command
 
 
 class BaseDriver(abc.ABC):
@@ -46,7 +47,7 @@ class BaseDriver(abc.ABC):
     async def delete(self) -> None: ...
 
     @abc.abstractmethod
-    async def import_image(self, tag: str) -> None: ...
+    async def import_images(self, *tags: str) -> None: ...
 
     @abc.abstractmethod
     async def import_image_to_internal_registry(self, tag: str) -> None: ...
@@ -104,6 +105,7 @@ class BaseDriver(abc.ABC):
         set_values_list: list[str],
         values_file: pathlib.Path | None = None,
         import_images: list[str] | None = None,
+        pull_on_host: bool = False,
     ) -> None:
         await self.run_in_vm(
             ["sh", "-c", "mkdir -p /tmp/agentstack && cat >/tmp/agentstack/chart.tgz"],
@@ -140,22 +142,34 @@ class BaseDriver(abc.ABC):
                 "Listing necessary images",
             )
         ).stdout.decode()
-        for image in import_images or []:
-            await self.import_image(image)
-            self.loaded_images.add(image)
-        for image in {typing.cast(str, yaml.safe_load(line)) for line in images_str.splitlines()} - set(
-            import_images or []
-        ):
+
+        def canonify(tag: str) -> str:
+            return tag if "." in tag.split("/")[0] else f"docker.io/{tag}"
+
+        required_images = {canonify(typing.cast(str, yaml.safe_load(line))) for line in images_str.splitlines()}
+        images_to_import = {canonify(tag) for tag in import_images or []}
+        images_to_pull = required_images - images_to_import
+
+        if pull_on_host:
+            for image in images_to_pull:
+                await run_command(["docker", "pull", image], f"Pulling image {image} on host")
+            images_to_import = required_images
+            images_to_pull = set[str]()
+
+        if images_to_import:
+            await self.import_images(*images_to_import)
+
+        for image in images_to_pull:
             async for attempt in AsyncRetrying(stop=stop_after_attempt(5)):
                 with attempt:
                     attempt_num = attempt.retry_state.attempt_number
-                    image_id = image if "." in image.split("/")[0] else f"docker.io/{image}"
-                    self.loaded_images.add(image_id)
                     await self.run_in_vm(
-                        ["k3s", "ctr", "image", "pull", image_id],
+                        ["k3s", "ctr", "image", "pull", image],
                         f"Pulling image {image}" + (f" (attempt {attempt_num})" if attempt_num > 1 else ""),
                     )
 
+        self.loaded_images = required_images
+
         if any("auth.oidc.enabled=true" in value.lower() for value in set_values_list):
             await agentstack_cli.commands.platform.istio.install(driver=self)
 

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/platform/lima_driver.py

@@ -180,7 +180,7 @@ class LimaDriver(BaseDriver):
         )
 
     @typing.override
-    async def import_image(self, tag: str):
+    async def import_images(self, *tags: str):
         image_dir = anyio.Path("/tmp/agentstack")
         await image_dir.mkdir(exist_ok=True, parents=True)
         image_file = str(uuid.uuid4())
@@ -188,11 +188,12 @@ class LimaDriver(BaseDriver):
 
         try:
             await run_command(
-                ["docker", "image", "save", "-o", str(image_path), tag], f"Exporting image {tag} from Docker"
+                ["docker", "image", "save", "-o", str(image_path), *tags],
+                f"Exporting image{'' if len(tags) == 1 else 's'} {', '.join(tags)} from Docker",
             )
             await self.run_in_vm(
                 ["/bin/sh", "-c", f"k3s ctr images import /tmp/agentstack/{image_file}"],
-                f"Importing image {tag} into Agent Stack platform",
+                f"Importing image{'' if len(tags) == 1 else 's'} {', '.join(tags)} into Agent Stack platform",
             )
         finally:
             await image_path.unlink(missing_ok=True)

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/platform/wsl_driver.py

@@ -131,7 +131,11 @@ class WSLDriver(BaseDriver):
         set_values_list: list[str],
         values_file: pathlib.Path | None = None,
         import_images: list[str] | None = None,
+        pull_on_host: bool = False,
     ) -> None:
+        if pull_on_host:
+            raise NotImplementedError("Pulling on host is not supported on this platform.")
+
         host_ip = (
             (
                 await self.run_in_vm(
@@ -205,7 +209,7 @@ class WSLDriver(BaseDriver):
         await run_command(["wsl.exe", "--unregister", self.vm_name], "Deleting Agent Stack platform", check=False)
 
     @typing.override
-    async def import_image(self, tag: str) -> None:
+    async def import_images(self, *tags: str) -> None:
         raise NotImplementedError("Importing images is not supported on this platform.")
 
     @typing.override

{agentstack_cli-0.5.1rc2 → agentstack_cli-0.5.2}/src/agentstack_cli/commands/user.py

@@ -23,13 +23,12 @@ ROLE_DISPLAY = {
 }
 
 
-@app.command("list")
+@app.command("list", help="List platform users [Admin only]")
 async def list_users(
     email: typing.Annotated[str | None, typer.Option(help="Filter by email (case-insensitive partial match)")] = None,
     limit: typing.Annotated[int, typer.Option(help="Results per page (1-100)")] = 40,
     after: typing.Annotated[str | None, typer.Option(help="Pagination cursor (page_token)")] = None,
 ):
-    """List platform users (admin only)."""
     announce_server_action("Listing users on")
 
     async with configuration.use_platform_client():
@@ -68,13 +67,12 @@ async def list_users(
         console.print(f"\n[dim]Use --after {next_page_token} to see more[/dim]")
 
 
-@app.command("set-role")
+@app.command("set-role", help="Change user role [Admin only]")
 async def set_role(
     user_id: typing.Annotated[str, typer.Argument(help="User UUID")],
     role: typing.Annotated[UserRole, typer.Argument(help="Target role (admin, developer, user)")],
     yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
 ):
-    """Change user role (admin only)."""
     url = announce_server_action(f"Changing user {user_id} to role '{role}' on")
     await confirm_server_action("Proceed with role change on", url=url, yes=yes)
 

agentstack_cli-0.5.1rc2/src/agentstack_cli/commands/mcp.py

@@ -1,150 +0,0 @@
-# Copyright 2025 © BeeAI a Series of LF Projects, LLC
-# SPDX-License-Identifier: Apache-2.0
-
-import typing
-from enum import StrEnum
-
-import typer
-from rich.table import Column
-
-from agentstack_cli.api import api_request
-from agentstack_cli.async_typer import AsyncTyper, console, create_table
-from agentstack_cli.utils import announce_server_action, confirm_server_action, status
-
-app = AsyncTyper()
-
-
-class Transport(StrEnum):
-    SSE = "sse"
-    STREAMABLE_HTTP = "streamable_http"
-
-
-@app.command("add")
-async def add_provider(
-    name: typing.Annotated[str, typer.Argument(help="Name for the MCP server")],
-    location: typing.Annotated[str, typer.Argument(help="Location of the MCP server")],
-    transport: typing.Annotated[
-        Transport, typer.Argument(help="Transport the MCP server uses")
-    ] = Transport.STREAMABLE_HTTP,
-    yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
-) -> None:
-    """Install discovered MCP server."""
-
-    url = announce_server_action(f"Registering MCP server '{name}' on")
-    await confirm_server_action("Proceed with registering this MCP server on", url=url, yes=yes)
-    with status("Registering server to platform"):
-        await api_request(
-            "POST", "mcp/providers", json={"name": name, "location": location, "transport": transport.value}
-        )
-    console.print("Registering server to platform [[green]DONE[/green]]")
-    await list_providers()
-
-
-@app.command("list")
-async def list_providers():
-    """List MCP servers."""
-
-    announce_server_action("Listing MCP servers on")
-    providers = await api_request("GET", "mcp/providers")
-    assert providers
-    with create_table(
-        Column("Name"),
-        Column("Location"),
-        Column("Transport"),
-        Column("State"),
-        no_wrap=True,
-    ) as table:
-        for provider in providers:
-            table.add_row(provider["name"], provider["location"], provider["transport"], provider["state"])
-    console.print()
-    console.print(table)
-
-
-@app.command("remove | uninstall | rm | delete")
-async def uninstall_provider(
-    name: typing.Annotated[str, typer.Argument(help="Name of the MCP provider to remove")],
-    yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
-) -> None:
-    """Remove MCP server."""
-    url = announce_server_action(f"Removing MCP server '{name}' from")
-    await confirm_server_action("Proceed with removing this MCP server from", url=url, yes=yes)
-    provider = await _get_provider_by_name(name)
-    if provider:
-        await api_request("delete", f"mcp/providers/{provider['id']}")
-    else:
-        raise ValueError(f"Provider {name} not found")
-    await list_providers()
-
-
-tool_app = AsyncTyper()
-app.add_typer(tool_app, name="tool", no_args_is_help=True, help="Inspect tools.")
-
-
-@tool_app.command("list")
-async def list_tools() -> None:
-    """List tools."""
-
-    announce_server_action("Listing MCP tools on")
-    tools = await api_request("GET", "mcp/tools")
-    assert tools
-    with create_table(
-        Column("Name"),
-        Column("Description", max_width=30),
-        no_wrap=True,
-    ) as table:
-        for tool in tools:
-            table.add_row(tool["name"], tool["description"])
-    console.print()
-    console.print(table)
-
-
-toolkit_app = AsyncTyper()
-app.add_typer(toolkit_app, name="toolkit", no_args_is_help=True, help="Create toolkits.")
-
-
-@toolkit_app.command("create")
-async def toolkit(
-    tools: typing.Annotated[list[str], typer.Argument(help="Tools to put in the toolkit")],
-    yes: typing.Annotated[bool, typer.Option("--yes", "-y", help="Skip confirmation prompts.")] = False,
-) -> None:
-    """Create a toolkit."""
-
-    url = announce_server_action("Creating MCP toolkit on")
-    await confirm_server_action("Proceed with creating this MCP toolkit on", url=url, yes=yes)
-    api_tools = await _get_tools_by_names(tools)
-    assert api_tools
-    toolkit = await api_request("POST", "mcp/toolkits", json={"tools": [tool["id"] for tool in api_tools]})
-    assert toolkit
-    with create_table(Column("Location"), Column("Transport"), Column("Expiration")) as table:
-        table.add_row(toolkit["location"], toolkit["transport"], toolkit["expires_at"])
-    console.print()
-    console.print(table)
-
-
-async def _get_provider_by_name(name: str):
-    providers = await api_request("GET", "mcp/providers")
-    assert providers
-
-    for provider in providers:
-        if provider["name"] == name:
-            return provider
-
-    raise ValueError(f"Provider {name} not found")
-
-
-async def _get_tools_by_names(names: list[str]) -> list[dict[str, typing.Any]]:
-    all_tools = await api_request("GET", "mcp/tools")
-    assert all_tools
-
-    tools = []
-    for name in names:
-        found = False
-        for tool in all_tools:
-            if tool["name"] == name:
-                tools.append(tool)
-                found = True
-                break
-        if not found:
-            raise ValueError(f"Tool {name} not found")
-
-    return tools