agentstack-cli 0.4.3rc2__tar.gz → 0.5.0__tar.gz

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/PKG-INFO

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

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/pyproject.toml

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

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/__init__.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import logging
+import re
 import typing
 from copy import deepcopy
 
@@ -14,6 +15,7 @@ import agentstack_cli.commands.model
 import agentstack_cli.commands.platform
 import agentstack_cli.commands.self
 import agentstack_cli.commands.server
+import agentstack_cli.commands.user
 from agentstack_cli.async_typer import AliasGroup, AsyncTyper
 from agentstack_cli.configuration import Configuration
 
@@ -47,6 +49,7 @@ Usage: agentstack [OPTIONS] COMMAND [ARGS]...
 │ model           Configure 15+ LLM providers                                │
 │ platform        Start, stop, or delete local platform                      │
 │ server          Connect to remote Agent Stack servers                      │
+│ user            Manage users and roles                                     │
 │ 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                     │
@@ -83,6 +86,12 @@ app.add_typer(
     help="Manage Agent Stack installation.",
     hidden=True,
 )
+app.add_typer(
+    agentstack_cli.commands.user.app,
+    name="user",
+    no_args_is_help=True,
+    help="Manage users.",
+)
 
 
 agent_alias = deepcopy(agentstack_cli.commands.agent.app)
@@ -110,9 +119,19 @@ async def ui():
     import agentstack_cli.commands.model
 
     await agentstack_cli.commands.model.ensure_llm_provider()
-    webbrowser.open(
-        "http://localhost:8334"
-    )  # TODO: This always opens the local UI, how to open the UI of a logged in server instead?
+
+    config = Configuration()
+    active_server = config.auth_manager.active_server
+
+    if active_server:
+        if re.search(r"(localhost|127\.0\.0\.1):8333", active_server):
+            ui_url = re.sub(r":8333", ":8334", active_server)
+        else:
+            ui_url = active_server
+    else:
+        ui_url = "http://localhost:8334"
+
+    webbrowser.open(ui_url)
 
 
 if __name__ == "__main__":

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/api.py

@@ -1,6 +1,7 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
 import json
+import logging
 import re
 import urllib
 import urllib.parse
@@ -12,6 +13,7 @@ from typing import Any
 
 import httpx
 import openai
+import pydantic
 from a2a.client import A2AClientHTTPError, Client, ClientConfig, ClientFactory
 from a2a.types import AgentCard
 from httpx import HTTPStatusError
@@ -20,6 +22,8 @@ from httpx._types import RequestFiles
 from agentstack_cli import configuration
 from agentstack_cli.configuration import Configuration
 
+logger = logging.getLogger(__name__)
+
 config = Configuration()
 
 API_BASE_URL = "api/v1/"
@@ -102,6 +106,26 @@ async def api_stream(
                 yield jsonlib.loads(re.sub("^data:", "", line).strip())
 
 
+async def fetch_server_version() -> str | None:
+    """Fetch server version from OpenAPI schema."""
+
+    class OpenAPIInfo(pydantic.BaseModel):
+        version: str
+
+    class OpenAPISchema(pydantic.BaseModel):
+        info: OpenAPIInfo
+
+    try:
+        response = await api_request("GET", "openapi.json", use_auth=False)
+        if not response:
+            return None
+        schema = OpenAPISchema.model_validate(response)
+        return schema.info.version
+    except Exception as e:
+        logger.warning("Failed to fetch server version: %s", e)
+        return None
+
+
 @asynccontextmanager
 async def a2a_client(agent_card: AgentCard, use_auth: bool = True) -> AsyncIterator[Client]:
     try:

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/commands/agent.py

@@ -61,7 +61,7 @@ from agentstack_sdk.a2a.extensions.common.form import (
     TextField,
     TextFieldValue,
 )
-from agentstack_sdk.platform import BuildState, ModelProvider, Provider
+from agentstack_sdk.platform import BuildState, File, ModelProvider, Provider, UserFeedback
 from agentstack_sdk.platform.context import Context, ContextPermissions, ContextToken, Permissions
 from agentstack_sdk.platform.model_provider import ModelCapability
 from InquirerPy import inquirer
@@ -152,12 +152,50 @@ configuration = Configuration()
 
 @app.command("add")
 async def add_agent(
-    location: typing.Annotated[str, typer.Argument(help="Agent location (public docker image or github url)")],
+    location: typing.Annotated[
+        str | None, typer.Argument(help="Agent location (public docker image or github url)")
+    ] = None,
     dockerfile: typing.Annotated[str | None, typer.Option(help="Use custom dockerfile path")] = None,
     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 [aliases: install]"""
+    if location is None:
+        repo_input = (
+            await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+                message="Enter GitHub repository (owner/repo or full URL):",
+            ).execute_async()
+            or ""
+        )
+
+        match = re.search(r"^(?:(?:https?://)?(?:www\.)?github\.com/)?([^/]+)/([^/?&]+)", repo_input)
+        if not match:
+            raise ValueError(f"Invalid GitHub URL format: {repo_input}. Expected 'owner/repo' or a full GitHub URL.")
+
+        owner, repo = match.group(1), match.group(2).removesuffix(".git")
+
+        async with httpx.AsyncClient() as client:
+            response = await client.get(
+                f"https://api.github.com/repos/{owner}/{repo}/tags",
+                headers={"Accept": "application/vnd.github.v3+json"},
+            )
+            tags = [tag["name"] for tag in response.json()] if response.status_code == 200 else []
+
+        if tags:
+            selected_tag = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+                message="Select a tag to use:",
+                choices=tags,
+            ).execute_async()
+        else:
+            selected_tag = (
+                await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+                    message="Enter tag to use:",
+                ).execute_async()
+                or "main"
+            )
+
+        location = f"https://github.com/{owner}/{repo}@{selected_tag}"
+
     url = announce_server_action(f"Installing agent '{location}' for")
     await confirm_server_action("Proceed with installing this agent on", url=url, yes=yes)
     with verbosity(verbose):
@@ -182,9 +220,11 @@ async def add_agent(
 @app.command("update")
 async def update_agent(
     search_path: typing.Annotated[
-        str, typer.Argument(..., help="Short ID, agent name or part of the provider location of agent to replace")
-    ],
-    location: typing.Annotated[str, typer.Argument(help="Agent location (public docker image or github url)")],
+        str | None, typer.Argument(help="Short ID, agent name or part of the provider location of agent to replace")
+    ] = None,
+    location: typing.Annotated[
+        str | None, typer.Argument(help="Agent location (public docker image or github url)")
+    ] = None,
     dockerfile: typing.Annotated[str | None, typer.Option(help="Use custom dockerfile path")] = None,
     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,
@@ -192,7 +232,58 @@ async def update_agent(
     """Upgrade agent to a newer docker image or build from GitHub repository"""
     with verbosity(verbose):
         async with configuration.use_platform_client():
-            provider = select_provider(search_path, providers=await Provider.list())
+            providers = await Provider.list()
+
+        if search_path is None:
+            if not providers:
+                console.error("No agents found. Add an agent first using 'agentstack agent add'.")
+                sys.exit(1)
+
+            provider_choices = [
+                Choice(value=p, name=f"{p.agent_card.name} ({ProviderUtils.short_location(p)})") for p in providers
+            ]
+            provider = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+                message="Select an agent to update:",
+                choices=provider_choices,
+            ).execute_async()
+            if not provider:
+                console.error("No agent selected. Exiting.")
+                sys.exit(1)
+        else:
+            provider = select_provider(search_path, providers=providers)
+
+        if location is None and is_github_url(provider.source):
+            match = re.search(r"^(?:(?:https?://)?(?:www\.)?github\.com/)?([^/]+)/([^/@?&]+)", provider.source)
+            if match:
+                owner, repo = match.group(1), match.group(2).removesuffix(".git")
+
+                async with httpx.AsyncClient() as client:
+                    response = await client.get(
+                        f"https://api.github.com/repos/{owner}/{repo}/tags",
+                        headers={"Accept": "application/vnd.github.v3+json"},
+                    )
+                    tags = [tag["name"] for tag in response.json()] if response.status_code == 200 else []
+
+                if tags:
+                    selected_tag = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+                        message="Select a new tag to use:",
+                        choices=tags,
+                    ).execute_async()
+                    if selected_tag:
+                        location = f"https://github.com/{owner}/{repo}@{selected_tag}"
+
+        if location is None:
+            location = (
+                await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+                    message="Enter new agent location (public docker image or github url):",
+                    default=provider.source,
+                ).execute_async()
+                or ""
+            )
+
+        if not location:
+            console.error("No location provided. Exiting.")
+            sys.exit(1)
 
         url = announce_server_action(f"Upgrading agent from '{provider.source}' to {location}")
         await confirm_server_action("Proceed with upgrading agent on", url=url, yes=yes)
@@ -560,10 +651,13 @@ async def _run_agent(
                                     match part.root.file:
                                         case FileWithBytes(bytes=bytes_str):
                                             full_path.write_bytes(base64.b64decode(bytes_str))
-                                        case FileWithUri(uri=uri):  # TODO process platform file uri
-                                            async with httpx.AsyncClient() as httpx_client:
-                                                resp = await httpx_client.get(uri)
-                                                full_path.write_bytes(base64.b64decode(resp.content))
+                                        case FileWithUri(uri=uri):
+                                            if uri.startswith("agentstack://"):
+                                                async with File.load_content(uri.removeprefix("agentstack://")) as file:
+                                                    full_path.write_bytes(file.content)
+                                            else:
+                                                async with httpx.AsyncClient() as httpx_client:
+                                                    full_path.write_bytes((await httpx_client.get(uri)).content)
                                     console.print(f"📁 Saved {full_path}")
                                 case TextPart(text=text):
                                     full_path.write_text(text)
@@ -770,24 +864,41 @@ def _create_input_handler(
 @app.command("run")
 async def run_agent(
     search_path: typing.Annotated[
-        str, typer.Argument(..., help="Short ID, agent name or part of the provider location")
-    ],
+        str | None,
+        typer.Argument(
+            help="Short ID, agent name or part of the provider location",
+        ),
+    ] = None,
     input: typing.Annotated[
         str | None,
         typer.Argument(
-            default_factory=lambda: None if sys.stdin.isatty() else sys.stdin.read(),
             help="Agent input as text or JSON",
         ),
-    ],
+    ] = None,
     dump_files: typing.Annotated[
         Path | None, typer.Option(help="Folder path to save any files returned by the agent")
     ] = None,
 ) -> None:
     """Run an agent."""
-    announce_server_action(f"Running agent '{search_path}' on")
+    if search_path is not None and input is None and sys.stdin.isatty():
+        input = sys.stdin.read()
     async with configuration.use_platform_client():
         providers = await Provider.list()
         await ensure_llm_provider()
+
+        if search_path is None:
+            if not providers:
+                err_console.error("No agents found. Add an agent first using 'agentstack agent add'.")
+                sys.exit(1)
+            search_path = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+                message="Select an agent to run:",
+                choices=[provider.agent_card.name for provider in providers],
+            ).execute_async()
+            if search_path is None:
+                err_console.error("No agent selected. Exiting.")
+                sys.exit(1)
+
+        announce_server_action(f"Running agent '{search_path}' on")
         provider = select_provider(search_path, providers=providers)
 
         context = await Context.create(
@@ -1053,3 +1164,63 @@ async def remove_env(
         provider = select_provider(search_path, await Provider.list())
         await provider.update_variables(variables=dict.fromkeys(env))
     await _list_env(provider)
+
+
+feedback_app = AsyncTyper()
+app.add_typer(feedback_app, name="feedback", help="Manage user feedback for your agents", no_args_is_help=True)
+
+
+@feedback_app.command("list")
+async def list_feedback(
+    search_path: typing.Annotated[
+        str | None, typer.Argument(help="Short ID, agent name or part of the provider location")
+    ] = None,
+    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"""
+
+    announce_server_action("Listing feedback on")
+
+    provider_id = None
+
+    async with configuration.use_platform_client():
+        if search_path:
+            providers = await Provider.list()
+            provider = select_provider(search_path, providers)
+            provider_id = str(provider.id)
+
+        response = await UserFeedback.list(
+            provider_id=provider_id,
+            limit=limit,
+            after_cursor=after_cursor,
+        )
+
+    if not response.items:
+        console.print("No feedback found.")
+        return
+
+    with create_table(
+        Column("Rating", style="yellow", ratio=1),
+        Column("Agent", style="cyan", ratio=2),
+        Column("Task ID", style="dim", ratio=1),
+        Column("Comment", ratio=3),
+        Column("Tags", ratio=2),
+        Column("Date", style="dim", ratio=1),
+    ) as table:
+        for item in response.items:
+            rating_icon = "✓" if item.rating == 1 else "✗"
+            agent_name = item.agent_name or str(item.provider_id)[:8]
+            task_id_short = str(item.task_id)[:8]
+            comment = item.comment or ""
+            if len(comment) > 50:
+                comment = comment[:50] + "..."
+            tags = ", ".join(item.comment_tags or []) if item.comment_tags else "-"
+            created_at = item.created_at.strftime("%Y-%m-%d")
+
+            table.add_row(rating_icon, agent_name, task_id_short, comment, tags, created_at)
+
+    console.print(table)
+    console.print(f"Showing {len(response.items)} of {response.total_count} total feedback entries")
+    if response.has_more and response.next_page_token:
+        console.print(f"Use --after {response.next_page_token} to see more")

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/commands/build.py

@@ -152,10 +152,10 @@ async def _server_side_build(
     verbose: bool = False,
 ) -> ProviderBuild:
     build = None
-    try:
-        from agentstack_cli.commands.agent import select_provider
-        from agentstack_cli.configuration import Configuration
+    from agentstack_cli.commands.agent import select_provider
+    from agentstack_cli.configuration import Configuration
 
+    try:
         if replace and add:
             raise ValueError("Cannot specify both replace and add options.")
 
@@ -181,8 +181,9 @@ async def _server_side_build(
                     print_log(message, ansi_mode=True, out_console=err_console)
             return await build.get()
     except (KeyboardInterrupt, CancelledError):
-        if build:
-            await build.delete()
+        async with Configuration().use_platform_client():
+            if build:
+                await build.delete()
         console.error("Build aborted.")
         raise
 

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/commands/model.py

@@ -354,7 +354,8 @@ async def _select_default_model(capability: ModelCapability) -> str | None:
                 if capability == ModelCapability.LLM:
                     test_response = await client.chat.completions.create(
                         model=selected_model,
-                        max_completion_tokens=500,  # reasoning models need some tokens to think about this
+                        # reasoning models need some tokens to think about this
+                        max_completion_tokens=500 if not selected_model.startswith("mistral") else None,
                         messages=[
                             {
                                 "role": "system",

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/commands/platform/base_driver.py

@@ -10,7 +10,6 @@ from subprocess import CompletedProcess
 from textwrap import dedent
 
 import anyio
-import pydantic
 import yaml
 from tenacity import AsyncRetrying, stop_after_attempt
 
@@ -193,27 +192,3 @@ class BaseDriver(abc.ABC):
                 ["k3s", "kubectl", "rollout", "restart", "deployment"],
                 "Restarting deployments to load imported images",
             )
-
-    async def version(self) -> str | None:
-        if (await self.status()) != "running":
-            return None
-        HelmStatus = typing.TypedDict("HelmStatus", {"status": str, "app_version": str})
-        helm_status = pydantic.TypeAdapter(list[HelmStatus]).validate_json(
-            (
-                await self.run_in_vm(
-                    [
-                        "/usr/local/bin/helm",
-                        "--kubeconfig=/etc/rancher/k3s/k3s.yaml",
-                        "ls",
-                        "--namespace=default",
-                        "--filter=^agentstack$",
-                        "-o",
-                        "json",
-                    ],
-                    "Getting Agent Stack platform version",
-                )
-            ).stdout
-        )
-        if helm_status[0]["status"] != "deployed":
-            return None
-        return helm_status[0]["app_version"]

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/commands/self.py

@@ -15,6 +15,7 @@ import typer
 from InquirerPy import inquirer
 
 import agentstack_cli.commands.platform
+from agentstack_cli.api import fetch_server_version
 from agentstack_cli.async_typer import AsyncTyper
 from agentstack_cli.commands.model import setup as model_setup
 from agentstack_cli.configuration import Configuration
@@ -46,7 +47,8 @@ async def version(
     """Print version of the Agent Stack CLI."""
     with verbosity(verbose=verbose):
         cli_version = importlib.metadata.version("agentstack-cli")
-        platform_version = await agentstack_cli.commands.platform.get_driver().version()
+        platform_version = await fetch_server_version()
+        active_server = configuration.auth_manager.active_server
 
         latest_cli_version: str | None = None
         with console.status("Checking for newer version...", spinner="dots"):
@@ -64,6 +66,7 @@ async def version(
         console.print(
             f"agentstack-platform version: [bold]{platform_version.replace('-', '') if platform_version is not None else 'not running'}[/bold]"
         )
+        console.print(f"          agentstack server: [bold]{active_server if active_server else 'none'}[/bold]")
         console.print()
 
         if latest_cli_version and packaging.version.parse(latest_cli_version) > packaging.version.parse(cli_version):

agentstack_cli-0.5.0/src/agentstack_cli/commands/user.py

@@ -0,0 +1,92 @@
+# Copyright 2025 © BeeAI a Series of LF Projects, LLC
+# SPDX-License-Identifier: Apache-2.0
+
+import typing
+from datetime import datetime
+
+import typer
+from agentstack_sdk.platform import User
+from agentstack_sdk.platform.user import UserRole
+from rich.table import Column
+
+from agentstack_cli.async_typer import AsyncTyper, console, create_table
+from agentstack_cli.configuration import Configuration
+from agentstack_cli.utils import announce_server_action, confirm_server_action
+
+app = AsyncTyper()
+configuration = Configuration()
+
+ROLE_DISPLAY = {
+    "admin": "[red]admin[/red]",
+    "developer": "[cyan]developer[/cyan]",
+    "user": "user",
+}
+
+
+@app.command("list")
+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():
+        result = await User.list(email=email, limit=limit, page_token=after)
+
+        items = result.items
+        has_more = result.has_more
+        next_page_token = result.next_page_token
+
+    with create_table(
+        Column("ID", style="yellow"),
+        Column("Email"),
+        Column("Role"),
+        Column("Created"),
+        Column("Role Updated"),
+        no_wrap=True,
+    ) as table:
+        for user in items:
+            role_display = ROLE_DISPLAY.get(user.role, user.role)
+
+            created_at = _format_date(user.created_at)
+            role_updated_at = _format_date(user.role_updated_at) if user.role_updated_at else "-"
+
+            table.add_row(
+                user.id,
+                user.email,
+                role_display,
+                created_at,
+                role_updated_at,
+            )
+
+    console.print()
+    console.print(table)
+
+    if has_more and next_page_token:
+        console.print(f"\n[dim]Use --after {next_page_token} to see more[/dim]")
+
+
+@app.command("set-role")
+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)
+
+    async with configuration.use_platform_client():
+        result = await User.set_role(user_id, UserRole(role))
+
+        role_display = ROLE_DISPLAY.get(result.new_role, result.new_role)
+
+        console.success(f"User role updated to [cyan]{role_display}[/cyan]")
+
+
+def _format_date(dt: datetime | None) -> str:
+    if not dt:
+        return "-"
+    return dt.strftime("%Y-%m-%d %H:%M")

{agentstack_cli-0.4.3rc2 → agentstack_cli-0.5.0}/src/agentstack_cli/utils.py

@@ -142,7 +142,7 @@ async def confirm_server_action(message: str, url: str | None = None, *, yes: bo
         return
     url = url or require_active_server()
     confirmed = await inquirer.confirm(  # type: ignore
-        message=f"{message} [cyan]{url}[/cyan]?", default=False
+        message=f"{message} {url}?", default=False
     ).execute_async()
     if not confirmed:
         console.info("Action cancelled.")
@@ -309,10 +309,15 @@ def print_log(line, ansi_mode=False, out_console: Console | None = None):
     def decode(text: str):
         return Text.from_ansi(text) if ansi_mode else text
 
-    if line["stream"] == "stderr":
-        (out_console or err_console).print(decode(line["message"]))
-    elif line["stream"] == "stdout":
-        (out_console or console).print(decode(line["message"]))
+    match line:
+        case {"stream": "stderr"}:
+            (out_console or err_console).print(decode(line["message"]))
+        case {"stream": "stdout"}:
+            (out_console or console).print(decode(line["message"]))
+        case {"event": "[DONE]"}:
+            return
+        case _:
+            (out_console or console).print(line)
 
 
 def is_github_url(url: str) -> bool: