agentstack-cli 0.6.1__tar.gz → 0.6.2rc2__tar.gz

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-cli
-Version: 0.6.1
+Version: 0.6.2rc2
 Summary: Agent Stack CLI
 Author: IBM Corp.
 Requires-Dist: anyio>=4.12.1

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentstack-cli"
-version = "0.6.1"
+version = "0.6.2-rc2"
 description = "Agent Stack CLI"
 readme = "README.md"
 authors = [{ name = "IBM Corp." }]
@@ -27,7 +27,7 @@ dependencies = [
 
 [dependency-groups]
 dev = [
-    "pyright>=1.1.407", # note: pyright 1.1.408 has a bug with multiple-inheritance
+    "pyrefly>=0.52.0",
     "pytest>=9.0.2",
     "ruff>=0.14.14",
     "wheel>=0.46.3",
@@ -74,8 +74,8 @@ lint.ignore = [
 ]
 force-exclude = true
 
-[tool.pyright]
-ignore = ["tests/**", "examples/cli.py"]
-venvPath = "."
-venv = ".venv"
-reportUnusedCallResult = false
+[tool.pyrefly]
+project-includes = [
+    "**/*.py*",
+    "**/*.ipynb",
+]

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/async_typer.py

@@ -22,6 +22,8 @@ from agentstack_cli.utils import extract_messages, format_error
 
 DEBUG = Configuration().debug
 
+sys.unraisablehook = lambda _: None  # Suppress benign cleanup errors
+
 
 class _LeftAlignedHeading(Heading):
     def __rich_console__(self, *args, **kwargs) -> RenderResult:

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/auth_manager.py

@@ -287,7 +287,10 @@ class AuthManager:
     def active_auth_server(self, auth_server: str | None) -> None:
         if auth_server is not None and (
             self._auth.active_server not in self._auth.servers
-            or auth_server not in self._auth.servers[self._auth.active_server].authorization_servers
+            or (
+                self._auth.active_server is not None
+                and auth_server not in self._auth.servers[self._auth.active_server].authorization_servers
+            )
         ):
             raise ValueError(f"Auth server {auth_server} not found in active server")
         self._auth.active_auth_server = auth_server

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/agent.py

@@ -93,12 +93,14 @@ from agentstack_cli.commands.build import _server_side_build
 from agentstack_cli.commands.model import ensure_llm_provider
 from agentstack_cli.configuration import Configuration
 
+# This is necessary for proper handling of arrow keys in interactive input
 if sys.platform != "win32":
+    import importlib
+
     try:
-        # This is necessary for proper handling of arrow keys in interactive input
-        import gnureadline as readline
+        readline = importlib.import_module("gnureadline")
     except ImportError:
-        import readline  # noqa: F401
+        readline = importlib.import_module("readline")
 
 from collections.abc import Callable
 from pathlib import Path
@@ -115,6 +117,8 @@ from agentstack_cli.async_typer import AsyncTyper, console, create_table, err_co
 from agentstack_cli.server_utils import announce_server_action, confirm_server_action
 from agentstack_cli.utils import (
     generate_schema_example,
+    get_github_repo_tags,
+    github_url_verbose_pattern,
     is_github_url,
     parse_env_var,
     print_log,
@@ -233,44 +237,50 @@ async def add_agent(
     - **Combined Formats**: `https://github.com/myorg/myrepo.git@v1.0.0#path=/path/to/agent`
     - **Enterprise GitHub**: `https://github.mycompany.com/myorg/myrepo`
     - **With a custom Dockerfile location**: `agentstack add --dockerfile /my-agent/path/to/Dockerfile "https://github.com/my-org/my-awesome-agents@main#path=/my-agent"`
-
-    [aliases: install]
     """
+    repo_input = location
     if location is None:
         repo_input = (
-            await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+            await inquirer.text(
                 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.")
+    if not repo_input:
+        console.error("No location provided. Exiting.")
+        sys.exit(1)
 
-        owner, repo = match.group(1), match.group(2).removesuffix(".git")
+    if match := re.match(github_url_verbose_pattern, repo_input, re.VERBOSE):
+        owner, repo, version, path = (
+            match.group("org"),
+            match.group("repo").removesuffix(".git"),
+            match.group("version"),
+            match.group("path"),
+        )
 
-        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 version is None and path is None:
+            host = match.group("host")
+            tags = await get_github_repo_tags(host, owner, repo)
 
-        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:",
+            if tags:
+                selected_tag = await inquirer.fuzzy(
+                    message="Select a tag to use:",
+                    choices=tags,
                 ).execute_async()
-                or "main"
-            )
+            else:
+                selected_tag = (
+                    await inquirer.text(
+                        message="Enter tag to use:",
+                    ).execute_async()
+                    or "main"
+                )
 
-        location = f"https://github.com/{owner}/{repo}@{selected_tag}"
+            location = f"https://github.com/{owner}/{repo}@{selected_tag}"
+        else:
+            location = repo_input
+    else:
+        location = repo_input
 
     url = announce_server_action(f"Installing agent '{location}' for")
     await confirm_server_action("Proceed with installing this agent on", url=url, yes=yes)
@@ -327,7 +337,7 @@ async def update_agent(
             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]
+            provider = await inquirer.fuzzy(
                 message="Select an agent to update:",
                 choices=provider_choices,
             ).execute_async()
@@ -337,20 +347,20 @@ async def update_agent(
         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 location is None and is_github_url(provider.origin):
+            match = re.match(github_url_verbose_pattern, provider.origin, re.VERBOSE)
+
             if match:
-                owner, repo = match.group(1), match.group(2).removesuffix(".git")
+                host, owner, repo = (
+                    match.group("host"),
+                    match.group("owner"),
+                    match.group("repo").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 []
+                tags = await get_github_repo_tags(host, owner, repo)
 
                 if tags:
-                    selected_tag = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+                    selected_tag = await inquirer.fuzzy(
                         message="Select a new tag to use:",
                         choices=tags,
                     ).execute_async()
@@ -359,9 +369,9 @@ async def update_agent(
 
         if location is None:
             location = (
-                await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+                await inquirer.text(
                     message="Enter new agent location (public docker image or github url):",
-                    default=provider.source,
+                    default=provider.origin.lstrip("git+"),
                 ).execute_async()
                 or ""
             )
@@ -370,7 +380,7 @@ async def update_agent(
             console.error("No location provided. Exiting.")
             sys.exit(1)
 
-        url = announce_server_action(f"Upgrading agent from '{provider.source}' to {location}")
+        url = announce_server_action(f"Upgrading agent from '{provider.origin}' to {location}")
         await confirm_server_action("Proceed with upgrading agent on", url=url, yes=yes)
 
         if is_github_url(location):
@@ -428,7 +438,7 @@ async def select_providers_multi(search_path: str, providers: list[Provider]) ->
     # Multiple matches - show selection menu
     choices = [Choice(value=p.id, name=f"{p.agent_card.name} - {p.id}") for p in provider_candidates.values()]
 
-    selected_ids = await inquirer.checkbox(  # pyright: ignore[reportPrivateImportUsage]
+    selected_ids = await inquirer.checkbox(
         message="Select agents to remove (use ↑/↓ to navigate, Space to select):", choices=choices
     ).execute_async()
 
@@ -512,7 +522,7 @@ async def _ask_form_questions(form_render: FormRender) -> FormResponse:
 
     for field in form_render.fields:
         if isinstance(field, TextField):
-            answer = await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+            answer = await inquirer.text(
                 message=field.label + ":",
                 default=field.default_value or "",
                 validate=EmptyInputValidator() if field.required else None,
@@ -520,7 +530,7 @@ async def _ask_form_questions(form_render: FormRender) -> FormResponse:
             form_values[field.id] = TextFieldValue(value=answer)
         elif isinstance(field, SingleSelectField):
             choices = [Choice(value=opt.id, name=opt.label) for opt in field.options]
-            answer = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+            answer = await inquirer.fuzzy(
                 message=field.label + ":",
                 choices=choices,
                 default=field.default_value,
@@ -529,7 +539,7 @@ async def _ask_form_questions(form_render: FormRender) -> FormResponse:
             form_values[field.id] = SingleSelectFieldValue(value=answer)
         elif isinstance(field, MultiSelectField):
             choices = [Choice(value=opt.id, name=opt.label) for opt in field.options]
-            answer = await inquirer.checkbox(  # pyright: ignore[reportPrivateImportUsage]
+            answer = await inquirer.checkbox(
                 message=field.label + ":",
                 choices=choices,
                 default=field.default_value,
@@ -538,14 +548,14 @@ async def _ask_form_questions(form_render: FormRender) -> FormResponse:
             form_values[field.id] = MultiSelectFieldValue(value=answer)
 
         elif isinstance(field, DateField):
-            year = await inquirer.text(  # pyright: ignore[reportPrivateImportUsage]
+            year = await inquirer.text(
                 message=f"{field.label} (year):",
                 validate=EmptyInputValidator() if field.required else None,
                 filter=lambda y: y.strip(),
             ).execute_async()
             if not year:
                 continue
-            month = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+            month = await inquirer.fuzzy(
                 message=f"{field.label} (month):",
                 validate=EmptyInputValidator() if field.required else None,
                 choices=[
@@ -558,7 +568,7 @@ async def _ask_form_questions(form_render: FormRender) -> FormResponse:
             ).execute_async()
             if not month:
                 continue
-            day = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+            day = await inquirer.fuzzy(
                 message=f"{field.label} (day):",
                 validate=EmptyInputValidator() if field.required else None,
                 choices=[
@@ -571,7 +581,7 @@ async def _ask_form_questions(form_render: FormRender) -> FormResponse:
             full_date = f"{year}-{month}-{day}"
             form_values[field.id] = DateFieldValue(value=full_date)
         elif isinstance(field, CheckboxField):
-            answer = await inquirer.confirm(  # pyright: ignore[reportPrivateImportUsage]
+            answer = await inquirer.confirm(
                 message=field.label + ":",
                 default=field.default_value,
                 long_instruction=field.content or "",
@@ -591,7 +601,7 @@ async def _ask_settings_questions(settings_render: SettingsRender) -> AgentRunSe
         if isinstance(field, CheckboxGroupField):
             checkbox_values: dict[str, SettingsCheckboxFieldValue] = {}
             for checkbox in field.fields:
-                answer = await inquirer.confirm(  # pyright: ignore[reportPrivateImportUsage]
+                answer = await inquirer.confirm(
                     message=checkbox.label + ":",
                     default=checkbox.default_value,
                 ).execute_async()
@@ -599,7 +609,7 @@ async def _ask_settings_questions(settings_render: SettingsRender) -> AgentRunSe
             settings_values[field.id] = CheckboxGroupFieldValue(values=checkbox_values)
         elif isinstance(field, SettingsSingleSelectField):
             choices = [Choice(value=opt.value, name=opt.label) for opt in field.options]
-            answer = await inquirer.fuzzy(  # pyright: ignore[reportPrivateImportUsage]
+            answer = await inquirer.fuzzy(
                 message=field.label + ":",
                 choices=choices,
                 default=field.default_value,
@@ -675,11 +685,7 @@ async def _run_agent(
                 else {}
             )
             | (
-                {
-                    FormServiceExtensionSpec.URI: {
-                        "form_fulfillments": {"initial_form": typing.cast(FormResponse, input).model_dump(mode="json")}
-                    }
-                }
+                {FormServiceExtensionSpec.URI: {"form_fulfillments": {"initial_form": input.model_dump(mode="json")}}}
                 if isinstance(input, FormResponse)
                 else {}
             )
@@ -898,7 +904,8 @@ class ShowConfig(InteractiveCommand):
                 schema_table.add_row(
                     prop,
                     json.dumps(required_schema),
-                    json.dumps(generate_schema_example(required_schema)),  # pyright: ignore [reportArgumentType]
+                    # pyrefly: ignore [bad-argument-type] -- probably a bug in Pyrefly
+                    json.dumps(generate_schema_example(required_schema)),
                 )
 
         renderables = [
@@ -1071,7 +1078,7 @@ async def run_agent(
             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]
+            search_path = await inquirer.fuzzy(
                 message="Select an agent to run:",
                 choices=[provider.agent_card.name for provider in providers],
             ).execute_async()

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/build.py

@@ -62,7 +62,6 @@ async def client_side_build(
 ):
     """Build agent locally using Docker. [Local only]"""
     with verbosity(verbose):
-        await run_command(["which", "docker"], "Checking docker")
         image_id = "agentstack-agent-build-tmp:latest"
         port = await find_free_port()
         dockerfile_args = ("-f", dockerfile) if dockerfile else ()

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/connector.py

@@ -75,7 +75,7 @@ async def select_connectors_multi(
     # Multiple matches - show selection menu
     choices = [Choice(value=c, name=f"{c.url} - {c.id} ({c.state})") for c in connector_candidates]
 
-    selected_connectors = await inquirer.checkbox(  # pyright: ignore[reportPrivateImportUsage]
+    selected_connectors = await inquirer.checkbox(
         message=f"Select connectors to {operation_name} (use ↑/↓ to navigate, Space to select):", choices=choices
     ).execute_async()
 

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/model.py

@@ -19,7 +19,6 @@ from agentstack_sdk.platform import (
 )
 from InquirerPy import inquirer
 from InquirerPy.base.control import Choice
-from InquirerPy.validator import EmptyInputValidator
 from rich.table import Column
 
 from agentstack_cli.api import openai_client
@@ -184,26 +183,26 @@ async def _add_provider(capability: ModelCapability, use_true_localhost: bool =
     base_url: str
     watsonx_project_id, watsonx_space_id = None, None
     choices = LLM_PROVIDERS if capability == ModelCapability.LLM else EMBEDDING_PROVIDERS
-    provider_type, provider_name, base_url = await inquirer.fuzzy(  # type: ignore
+    provider_type, provider_name, base_url = await inquirer.fuzzy(
         message=f"Select {capability} provider (type to search):", choices=choices
-    ).execute_async()
+    ).execute_async() or sys.exit(1)
 
     watsonx_project_or_space: str = ""
     watsonx_project_or_space_id: str = ""
 
     if provider_type == ModelProviderType.OTHER:
-        base_url: str = await inquirer.text(  # type: ignore
+        base_url: str = await inquirer.text(
             message="Enter the base URL of your API (OpenAI-compatible):",
-            validate=lambda url: (url.startswith(("http://", "https://")) or "URL must start with http:// or https://"),  # type: ignore
+            validate=lambda url: url.startswith(("http://", "https://")),
             transformer=lambda url: url.rstrip("/"),
-        ).execute_async()
+        ).execute_async() or sys.exit(1)
         if re.match(r"^https://[a-z0-9.-]+\.rits\.fmaas\.res\.ibm\.com/.*$", base_url):
             provider_type = ModelProviderType.RITS
             if not base_url.endswith("/v1"):
                 base_url = base_url.removesuffix("/") + "/v1"
 
     if provider_type == ModelProviderType.WATSONX:
-        region: str = await inquirer.select(  # type: ignore
+        region: str = await inquirer.select(
             message="Select IBM Cloud region:",
             choices=[
                 Choice(name="us-south", value="us-south"),
@@ -213,34 +212,36 @@ async def _add_provider(capability: ModelCapability, use_true_localhost: bool =
                 Choice(name="jp-tok", value="jp-tok"),
                 Choice(name="au-syd", value="au-syd"),
             ],
-        ).execute_async()
+        ).execute_async() or sys.exit(1)
         base_url: str = f"""https://{region}.ml.cloud.ibm.com"""
-        watsonx_project_or_space: str = await inquirer.select(  # type:ignore
+        watsonx_project_or_space: str = await inquirer.select(
             "Use a Project or a Space?", choices=["project", "space"]
-        ).execute_async()
+        ).execute_async() or sys.exit(1)
         if (
             not (watsonx_project_or_space_id := os.environ.get(f"WATSONX_{watsonx_project_or_space.upper()}_ID", ""))
-            or not await inquirer.confirm(  # type:ignore
+            or not await inquirer.confirm(
                 message=f"Use the {watsonx_project_or_space} id from environment variable 'WATSONX_{watsonx_project_or_space.upper()}_ID'?",
                 default=True,
             ).execute_async()
         ):
-            watsonx_project_or_space_id = await inquirer.text(  # type:ignore
+            watsonx_project_or_space_id = await inquirer.text(
                 message=f"Enter the {watsonx_project_or_space} id:"
-            ).execute_async()
+            ).execute_async() or sys.exit(1)
 
         watsonx_project_id = watsonx_project_or_space_id if watsonx_project_or_space == "project" else None
         watsonx_space_id = watsonx_project_or_space_id if watsonx_project_or_space == "space" else None
 
-    if (api_key := os.environ.get(f"{provider_type.upper()}_API_KEY")) is None or not await inquirer.confirm(  # type: ignore
-        message=f"Use the API key from environment variable '{provider_type.upper()}_API_KEY'?",
-        default=True,
-    ).execute_async():
-        api_key: str = (
-            "dummy"
-            if provider_type in {ModelProviderType.OLLAMA, ModelProviderType.JAN}
-            else await inquirer.secret(message="Enter API key:", validate=EmptyInputValidator()).execute_async()  # type: ignore
-        )
+    api_key: str = (
+        "dummy"
+        if provider_type in {ModelProviderType.OLLAMA, ModelProviderType.JAN}
+        else env_api_key
+        if (env_api_key := os.environ.get(f"{provider_type.upper()}_API_KEY"))
+        and await inquirer.confirm(
+            message=f"Use the API key from environment variable '{provider_type.upper()}_API_KEY'?",
+            default=True,
+        ).execute_async()
+        else await inquirer.secret(message="Enter API key:").execute_async() or ""
+    )
 
     try:
         if provider_type == ModelProviderType.OLLAMA:
@@ -264,13 +265,13 @@ async def _add_provider(capability: ModelCapability, use_true_localhost: bool =
                     message = f"Do you want to pull the recommended LLM model '{recommended_llm_model}'?"
                     if not available_models:
                         message = f"There are no locally available models in Ollama. {message}"
-                    if await inquirer.confirm(message, default=True).execute_async():  # type: ignore
+                    if await inquirer.confirm(message, default=True).execute_async():
                         await run_command(
                             [_ollama_exe(), "pull", recommended_llm_model], "Pulling the selected model", check=True
                         )
 
                 if recommended_embedding_model not in available_models and (
-                    await inquirer.confirm(  # type: ignore
+                    await inquirer.confirm(
                         message=f"Do you want to pull the recommended embedding model '{recommended_embedding_model}'?",
                         default=True,
                     ).execute_async()
@@ -293,8 +294,8 @@ async def _add_provider(capability: ModelCapability, use_true_localhost: bool =
             )
 
     except httpx.HTTPError as e:
-        if hasattr(e, "response") and hasattr(e.response, "json"):  # pyright: ignore [reportAttributeAccessIssue]
-            err = str(e.response.json().get("detail", str(e)))  # pyright: ignore [reportAttributeAccessIssue]
+        if hasattr(e, "response") and hasattr(e.response, "json"):
+            err = str(e.response.json().get("detail", str(e)))
         else:
             err = str(e)
         match provider_type:
@@ -336,12 +337,12 @@ async def _select_default_model(capability: ModelCapability) -> str | None:
     selected_model = (
         recommended_model
         if recommended_model
-        and await inquirer.confirm(  # type: ignore
+        and await inquirer.confirm(
             message=f"Do you want to use the recommended model as default: '{recommended_model}'?",
             default=True,
         ).execute_async()
         else (
-            await inquirer.fuzzy(  # type: ignore
+            await inquirer.fuzzy(
                 message="Select a model to be used as default (type to search):",
                 choices=sorted(available_models),
             ).execute_async()
@@ -429,7 +430,7 @@ async def setup(
                 console.warning("The following providers are already configured:\n")
                 _list_providers(existing_providers)
                 console.print()
-                if await inquirer.confirm(  # type: ignore
+                if await inquirer.confirm(
                     message="Do you want to reset the configuration?", default=True
                 ).execute_async():
                     with console.status("Resetting configuration...", spinner="dots"):
@@ -450,13 +451,13 @@ async def setup(
                     != ModelProviderType.RITS  # RITS does not support embeddings, but we treat it as OTHER
                     and (
                         llm_provider.type != ModelProviderType.OTHER  # OTHER may not support embeddings, so we ask
-                        or inquirer.confirm(  # type: ignore
+                        or inquirer.confirm(
                             "Do you want to also set up an embedding model from the same provider?", default=True
                         )
                     )
                 ):
                     default_embedding_model = await _select_default_model(ModelCapability.EMBEDDING)
-                elif await inquirer.confirm(  # type: ignore
+                elif await inquirer.confirm(
                     message="Do you want to configure an embedding provider? (recommended)", default=True
                 ).execute_async():
                     console.print("[bold]Setting up embedding provider...[/bold]")
@@ -490,7 +491,7 @@ async def select_default_model(
     url = announce_server_action("Updating default model for")
     await confirm_server_action("Proceed with updating default model on", url=url, yes=yes)
     if not capability:
-        capability = await inquirer.select(  # type: ignore
+        capability = await inquirer.select(
             message="Which default model would you like to change?",
             choices=[
                 Choice(name="llm", value=ModelCapability.LLM),
@@ -518,9 +519,17 @@ app.add_typer(model_provider_app, name="provider")
 
 
 def _list_providers(providers: list[ModelProvider]):
-    with create_table(Column("Type"), Column("Name"), Column("Base URL", ratio=1)) as provider_table:
+    with create_table(Column("Type"), Column("Name"), Column("State"), Column("Base URL", ratio=1)) as provider_table:
         for provider in providers:
-            provider_table.add_row(provider.type, provider.name, str(provider.base_url))
+            provider_table.add_row(
+                provider.type,
+                provider.name,
+                {
+                    "online": "[green]● connected[/green]",
+                    "offline": "[bright_black]○ disconnected[/bright_black]",
+                }.get(provider.state, provider.state or "<unknown>"),
+                str(provider.base_url),
+            )
     console.print(provider_table)
 
 
@@ -543,7 +552,7 @@ async def add_provider(
     """Add a new model provider. [Admin only]"""
     announce_server_action("Adding provider for")
     if not capability:
-        capability = await inquirer.select(  # type: ignore
+        capability = await inquirer.select(
             message="Which default provider would you like to add?",
             choices=[
                 Choice(name="llm", value=ModelCapability.LLM),
@@ -598,13 +607,15 @@ async def remove_provider(
         async with configuration.use_platform_client():
             providers = await ModelProvider.list()
 
-        if not search_path:
-            provider: ModelProvider = await inquirer.select(  # type: ignore
+        provider: ModelProvider = (
+            _select_provider(providers, search_path)
+            if search_path
+            else await inquirer.select(
                 message="Choose a provider to remove:",
                 choices=[Choice(name=f"{p.type} ({p.base_url})", value=p) for p in providers],
             ).execute_async()
-        else:
-            provider = _select_provider(providers, search_path)
+            or sys.exit(1)
+        )
 
         await provider.delete()
 

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/platform/base_driver.py

@@ -7,6 +7,7 @@ import json
 import pathlib
 import shlex
 import typing
+import uuid
 from enum import StrEnum
 from subprocess import CompletedProcess
 from textwrap import dedent
@@ -55,13 +56,87 @@ class BaseDriver(abc.ABC):
     async def delete(self) -> None: ...
 
     @abc.abstractmethod
-    async def import_images(self, *tags: str) -> None: ...
+    async def exec(self, command: list[str]) -> None: ...
 
     @abc.abstractmethod
-    async def import_image_to_internal_registry(self, tag: str) -> None: ...
+    def _get_export_import_paths(self) -> tuple[str, str]: ...
 
-    @abc.abstractmethod
-    async def exec(self, command: list[str]) -> None: ...
+    async def import_images(self, *tags: str) -> None:
+        if not tags:
+            return
+
+        host_path, guest_path = self._get_export_import_paths()
+
+        try:
+            await run_command(
+                ["docker", "image", "save", "-o", host_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 {guest_path}"],
+                f"Importing image{'' if len(tags) == 1 else 's'} {', '.join(tags)} into Agent Stack platform",
+            )
+        finally:
+            await anyio.Path(host_path).unlink(missing_ok=True)
+
+    async def import_image_to_internal_registry(self, tag: str) -> None:
+        host_path, guest_path = self._get_export_import_paths()
+
+        try:
+            await run_command(
+                ["docker", "image", "save", "-o", str(host_path), tag],
+                f"Exporting image {tag} from Docker",
+            )
+            job_name = f"push-{uuid.uuid4().hex[:6]}"
+            await self.run_in_vm(
+                ["k3s", "kubectl", "apply", "-f", "-"],
+                "Starting push job",
+                input=yaml.dump(
+                    {
+                        "apiVersion": "batch/v1",
+                        "kind": "Job",
+                        "metadata": {"name": job_name, "namespace": "default"},
+                        "spec": {
+                            "backoffLimit": 0,
+                            "ttlSecondsAfterFinished": 60,
+                            "template": {
+                                "spec": {
+                                    "restartPolicy": "Never",
+                                    "containers": [
+                                        {
+                                            "name": "crane",
+                                            "image": next(
+                                                (image for image in self.loaded_images if "alpine/crane" in image),
+                                                "ghcr.io/i-am-bee/alpine/crane:0.20.6",
+                                            ),
+                                            "command": [
+                                                "crane",
+                                                "push",
+                                                f"/workspace/{pathlib.Path(host_path).name}",
+                                                tag,
+                                                "--insecure",
+                                            ],
+                                            "volumeMounts": [{"name": "workspace", "mountPath": "/workspace"}],
+                                        }
+                                    ],
+                                    "volumes": [
+                                        {
+                                            "name": "workspace",
+                                            "hostPath": {"path": str(pathlib.PurePosixPath(guest_path).parent)},
+                                        }
+                                    ],
+                                }
+                            },
+                        },
+                    }
+                ).encode(),
+            )
+            await self.run_in_vm(
+                ["k3s", "kubectl", "wait", "--for=condition=complete", f"job/{job_name}", "--timeout=300s"],
+                "Waiting for push to complete",
+            )
+        finally:
+            await anyio.Path(host_path).unlink(missing_ok=True)
 
     def _canonify(self, tag: str) -> str:
         return tag if "." in tag.split("/")[0] else f"docker.io/{tag}"
@@ -92,7 +167,6 @@ class BaseDriver(abc.ABC):
         }
 
     async def install_tools(self) -> None:
-        # Configure k3s registry for local registry access
         registry_config = dedent(
             """\
             mirrors:
@@ -116,7 +190,7 @@ class BaseDriver(abc.ABC):
                     "sudo tee /etc/rancher/k3s/registries.yaml > /dev/null"
                 ),
             ],
-            "Configuring k3s registry",
+            "Configuring Kubernetes registry",
         )
 
         await self.run_in_vm(
@@ -125,7 +199,7 @@ class BaseDriver(abc.ABC):
                 "-c",
                 "which k3s || curl -sfL https://get.k3s.io | sh -s - --write-kubeconfig-mode 644 --https-listen-port=16443",
             ],
-            "Installing k3s",
+            "Installing Kubernetes",
         )
         await self.run_in_vm(
             [

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/platform/lima_driver.py

@@ -3,12 +3,14 @@
 
 import importlib.resources
 import os
+import pathlib
 import shutil
 import sys
 import tempfile
 import typing
 import uuid
 from subprocess import CompletedProcess
+from typing import TypedDict
 
 import anyio
 import psutil
@@ -62,9 +64,12 @@ class LimaDriver(BaseDriver):
             for line in result.stdout.decode().split("\n"):
                 if not line:
                     continue
-                status = pydantic.TypeAdapter(typing.TypedDict("Status", {"name": str, "status": str})).validate_json(
-                    line
-                )
+
+                class Status(TypedDict):
+                    name: str
+                    status: str
+
+                status = pydantic.TypeAdapter(Status).validate_json(line)
                 if status["name"] == self.vm_name:
                     return status["status"].lower()
             return None
@@ -102,7 +107,7 @@ class LimaDriver(BaseDriver):
             if total_memory_gib < 8:
                 console.warning("Less than 8 GB of RAM detected. Performance may be degraded.")
 
-            vm_memory_gib = round(min(8, max(3, total_memory_gib / 2)))
+            vm_memory_gib = round(min(8.0, max(3.0, total_memory_gib / 2)))
 
             with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete_on_close=False) as template_file:
                 template_file.write(
@@ -180,100 +185,20 @@ class LimaDriver(BaseDriver):
         )
 
     @typing.override
-    async def import_images(self, *tags: str):
-        if not tags:
-            return
-        image_dir = anyio.Path("/tmp/agentstack")
-        await image_dir.mkdir(exist_ok=True, parents=True)
-        image_file = str(uuid.uuid4())
-        image_path = image_dir / image_file
-
-        try:
-            await run_command(
-                ["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{'' if len(tags) == 1 else 's'} {', '.join(tags)} into Agent Stack platform",
-            )
-        finally:
-            await image_path.unlink(missing_ok=True)
-
-    @typing.override
-    async def import_image_to_internal_registry(self, tag: str) -> None:
-        # 1. Check if registry is running
-        try:
-            await self.run_in_vm(
-                ["k3s", "kubectl", "get", "svc", "agentstack-registry-svc"],
-                "Checking internal registry availability",
-            )
-        except Exception as e:
-            console.warning(f"Internal registry service not found. Push might fail: {e}")
-
-        # 2. Export image from Docker to shared temp dir
-        image_dir = anyio.Path("/tmp/agentstack")
-        await image_dir.mkdir(exist_ok=True, parents=True)
-        image_file = f"{uuid.uuid4()}.tar"
-        image_path = image_dir / image_file
-
-        try:
-            await run_command(
-                ["docker", "image", "save", "-o", str(image_path), tag],
-                f"Exporting image {tag} from Docker",
-            )
-
-            # 3 & 4. Run Crane Job
-            crane_image = "ghcr.io/i-am-bee/alpine/crane:0.20.6"
-            for image in self.loaded_images:
-                if "alpine/crane" in image:
-                    crane_image = image
-                    break
-
-            job_name = f"push-{uuid.uuid4().hex[:6]}"
-            job_def = {
-                "apiVersion": "batch/v1",
-                "kind": "Job",
-                "metadata": {"name": job_name, "namespace": "default"},
-                "spec": {
-                    "backoffLimit": 0,
-                    "ttlSecondsAfterFinished": 60,
-                    "template": {
-                        "spec": {
-                            "restartPolicy": "Never",
-                            "containers": [
-                                {
-                                    "name": "crane",
-                                    "image": crane_image,
-                                    "command": ["crane", "push", f"/workspace/{image_file}", tag, "--insecure"],
-                                    "volumeMounts": [{"name": "workspace", "mountPath": "/workspace"}],
-                                }
-                            ],
-                            "volumes": [{"name": "workspace", "hostPath": {"path": "/tmp/agentstack"}}],
-                        }
-                    },
-                },
-            }
-
-            await self.run_in_vm(
-                ["k3s", "kubectl", "apply", "-f", "-"], "Starting push job", input=yaml.dump(job_def).encode()
-            )
-            await self.run_in_vm(
-                ["k3s", "kubectl", "wait", "--for=condition=complete", f"job/{job_name}", "--timeout=300s"],
-                "Waiting for push to complete",
-            )
-            await self.run_in_vm(["k3s", "kubectl", "delete", "job", job_name], "Cleaning up push job")
-        finally:
-            await image_path.unlink(missing_ok=True)
+    def _get_export_import_paths(self) -> tuple[str, str]:
+        image_dir = pathlib.Path("/tmp/agentstack")
+        image_dir.mkdir(exist_ok=True, parents=True)
+        image_path = str(image_dir / f"{uuid.uuid4()}.tar")
+        return (image_path, image_path)
 
     @typing.override
     async def exec(self, command: list[str]):
         await anyio.run_process(
             [self.limactl_exe, "shell", f"--tty={sys.stdin.isatty()}", self.vm_name, "--", *command],
-            input=None if sys.stdin.isatty() else sys.stdin.read().encode(),
             check=False,
-            stdout=None,
-            stderr=None,
+            stdin=sys.stdin,
+            stdout=sys.stdout,
+            stderr=sys.stderr,
             env={**os.environ, "LIMA_HOME": str(Configuration().lima_home)},
             cwd="/",
         )

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/platform/wsl_driver.py

@@ -6,6 +6,7 @@ import os
 import pathlib
 import platform
 import sys
+import tempfile
 import textwrap
 import typing
 
@@ -132,9 +133,6 @@ class WSLDriver(BaseDriver):
         values_file: pathlib.Path | None = None,
         image_pull_mode: ImagePullMode = ImagePullMode.guest,
     ) -> None:
-        if image_pull_mode in {ImagePullMode.host, ImagePullMode.hybrid}:
-            raise NotImplementedError("Importing host images is not supported on Windows.")
-
         host_ip = (
             (
                 await self.run_in_vm(
@@ -170,7 +168,7 @@ class WSLDriver(BaseDriver):
 
             [Service]
             Type=simple
-            ExecStart=/bin/bash -c 'IFS=":" read svc port <<< "%i"; exec /usr/local/bin/kubectl port-forward --address=127.0.0.1 svc/$svc $port:$port'
+            ExecStart=/bin/bash -c 'IFS=":" read svc port <<< "%i"; exec /usr/local/bin/k3s kubectl port-forward --kubeconfig=/etc/rancher/k3s/k3s.yaml --address=127.0.0.1 svc/$svc $port:$port'
             Restart=on-failure
             User=root
 
@@ -207,20 +205,21 @@ class WSLDriver(BaseDriver):
     async def delete(self):
         await run_command(["wsl.exe", "--unregister", self.vm_name], "Deleting Agent Stack platform", check=False)
 
-    @typing.override
-    async def import_images(self, *tags: str) -> None:
-        raise NotImplementedError("Importing images is not supported on this platform.")
-
-    @typing.override
-    async def import_image_to_internal_registry(self, tag: str) -> None:
-        raise NotImplementedError("Importing images to internal registry is not supported on this platform.")
-
     @typing.override
     async def exec(self, command: list[str]):
         await anyio.run_process(
             ["wsl.exe", "--user", "root", "--distribution", self.vm_name, "--", *command],
-            input=None if sys.stdin.isatty() else sys.stdin.read().encode(),
             check=False,
-            stdout=None,
-            stderr=None,
+            stdin=sys.stdin,
+            stdout=sys.stdout,
+            stderr=sys.stderr,
+            cwd="/",
         )
+
+    @typing.override
+    def _get_export_import_paths(self) -> tuple[str, str]:
+        fd, tmp_path = tempfile.mkstemp(suffix=".tar")
+        os.close(fd)
+        windows_path = str(pathlib.Path(tmp_path).resolve().absolute())
+        wsl_path = f"/mnt/{windows_path[0].lower()}/{windows_path[2:].replace('\\', '/').removeprefix('/')}"
+        return (windows_path, wsl_path)

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/self.py

@@ -31,9 +31,9 @@ def _path() -> str:
     # These are PATHs where `uv` installs itself when installed through own install script
     # Package managers may install elsewhere, but that location should already be in PATH
     return os.pathsep.join(
-        [
-            *([xdg_bin_home] if (xdg_bin_home := os.getenv("XDG_BIN_HOME")) else []),
-            *([os.path.realpath(f"{xdg_data_home}/../bin")] if (xdg_data_home := os.getenv("XDG_DATA_HOME")) else []),
+        ([xdg_bin_home] if (xdg_bin_home := os.getenv("XDG_BIN_HOME")) else [])
+        + ([os.path.realpath(f"{xdg_data_home}/../bin")] if (xdg_data_home := os.getenv("XDG_DATA_HOME")) else [])
+        + [
             os.path.expanduser("~/.local/bin"),
             os.getenv("PATH", ""),
         ]
@@ -124,7 +124,7 @@ async def install(
         console.print()
         if (
             ready_to_start
-            and await inquirer.confirm(  # pyright: ignore[reportPrivateImportUsage]
+            and await inquirer.confirm(
                 message="Do you want to start the Agent Stack platform now? Will run: agentstack platform start",
                 default=True,
             ).execute_async()
@@ -139,7 +139,7 @@ async def install(
         already_configured = False
         if (
             already_started
-            and await inquirer.confirm(  # pyright: ignore[reportPrivateImportUsage]
+            and await inquirer.confirm(
                 message="Do you want to configure your LLM provider now? Will run: agentstack model setup", default=True
             ).execute_async()
         ):
@@ -151,7 +151,7 @@ async def install(
 
         if (
             already_configured
-            and await inquirer.confirm(  # pyright: ignore[reportPrivateImportUsage]
+            and await inquirer.confirm(
                 message="Do you want to open the web UI now? Will run: agentstack ui", default=True
             ).execute_async()
         ):

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/commands/server.py

@@ -78,7 +78,7 @@ def get_unique_app_name() -> str:
 async def server_login(server: typing.Annotated[str | None, typer.Argument()] = None):
     """Login to a server or switch between logged in servers."""
     server = server or (
-        await inquirer.select(  #  type: ignore
+        await inquirer.select(
             message="Select a server, or log in to a new one:",
             choices=[
                 *(
@@ -95,7 +95,7 @@ async def server_login(server: typing.Annotated[str | None, typer.Argument()] =
         if config.auth_manager.servers
         else None
     )
-    server = server or await inquirer.text(message="Enter server URL:").execute_async()  #  type: ignore
+    server = server or await inquirer.text(message="Enter server URL:").execute_async()
 
     if not server:
         raise RuntimeError("No server selected. Action cancelled.")
@@ -120,7 +120,7 @@ async def server_login(server: typing.Annotated[str | None, typer.Argument()] =
         elif len(auth_servers) == 1:
             auth_server = auth_servers[0]
         elif len(auth_servers) > 1:
-            auth_server = await inquirer.select(  #  type: ignore
+            auth_server = await inquirer.select(
                 message="Select an authorization server:",
                 choices=[
                     Choice(
@@ -199,13 +199,10 @@ async def server_login(server: typing.Annotated[str | None, typer.Argument()] =
         if len(auth_servers) == 1:
             auth_server = auth_servers[0]
         else:
-            auth_server = await inquirer.select(  # type: ignore
+            auth_server = await inquirer.select(
                 message="Select an authorization server:",
                 choices=auth_servers,
-            ).execute_async()
-
-        if not auth_server:
-            raise RuntimeError("No authorization server selected.")
+            ).execute_async() or sys.exit(1)
 
         async with httpx.AsyncClient() as client:
             try:
@@ -250,7 +247,7 @@ async def server_login(server: typing.Annotated[str | None, typer.Argument()] =
 
         if not client_id:
             client_id = (
-                await inquirer.text(  #  type: ignore
+                await inquirer.text(
                     message="Enter Client ID:",
                     instruction=f"(Redirect URI: {REDIRECT_URI})",
                 ).execute_async()
@@ -258,12 +255,7 @@ async def server_login(server: typing.Annotated[str | None, typer.Argument()] =
             )
             if not client_id:
                 raise RuntimeError("Client ID is mandatory. Action cancelled.")
-            client_secret = (
-                await inquirer.text(  #  type: ignore
-                    message="Enter Client Secret (optional):"
-                ).execute_async()
-                or None
-            )
+            client_secret = await inquirer.secret(message="Enter Client Secret (optional):").execute_async() or None
 
         code_verifier = generate_token(64)
 

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/configuration.py

@@ -87,7 +87,7 @@ class Configuration(pydantic_settings.BaseSettings):
         async with use_platform_client(
             auth=(self.username, self.password.get_secret_value()) if self.password else None,
             auth_token=auth_token,
-            base_url=self.auth_manager.active_server + "/",
+            base_url=(self.auth_manager.active_server or "") + "/",
         ) as client:
             yield client
 

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/server_utils.py

@@ -32,9 +32,7 @@ async def confirm_server_action(message: str, url: str | None = None, *, yes: bo
     if yes:
         return
     url = url or require_active_server()
-    confirmed = await inquirer.confirm(  # type: ignore
-        message=f"{message} {url}?", default=False
-    ).execute_async()
+    confirmed = await inquirer.confirm(message=f"{message} {url}?", default=False).execute_async()
     if not confirmed:
         console.info("Action cancelled.")
         sys.exit(1)

{agentstack_cli-0.6.1 → agentstack_cli-0.6.2rc2}/src/agentstack_cli/utils.py

@@ -288,10 +288,8 @@ def print_log(line, ansi_mode=False, out_console: Console | None = None):
             (out_console or console).print(line)
 
 
-def is_github_url(url: str) -> bool:
-    """This pattern is taken from agentstack_server.utils.github.GithubUrl, make sure to keep it in sync"""
-
-    pattern = r"""
+# ! This pattern is taken from agentstack_server.utils.github.GithubUrl, make sure to keep it in sync
+github_url_verbose_pattern = r"""
         ^
         (?:git\+)?                              # Optional git+ prefix
         https?://(?P<host>github(?:\.[^/]+)+)/  # GitHub host (github.com or github.enterprise.com)
@@ -307,7 +305,76 @@ def is_github_url(url: str) -> bool:
         (?:\#path=(?P<path>.+))?                # Optional path after #path=
         $
     """
-    return bool(re.match(pattern, url, re.VERBOSE))
+
+
+def is_github_url(url: str) -> bool:
+    return bool(re.match(github_url_verbose_pattern, url, re.VERBOSE))
+
+
+def get_local_github_token() -> str | None:
+    """Get GitHub token from standard local sources for authenticated API requests.
+
+    Checks multiple sources in order of preference:
+    1. GITHUB_TOKEN environment variable
+    2. GH_TOKEN environment variable (used by GitHub CLI)
+    3. GitHub CLI (gh auth token) if user is logged in
+
+    Authenticated requests have much higher rate limits (5000/hour vs 60/hour).
+    """
+    # Check environment variables first
+    if token := os.getenv("GITHUB_TOKEN"):
+        return token
+    if token := os.getenv("GH_TOKEN"):
+        return token
+
+    # Try GitHub CLI if available
+    try:
+        result = subprocess.run(
+            ["gh", "auth", "token"],
+            capture_output=True,
+            text=True,
+            timeout=5,
+        )
+        if result.returncode == 0 and (token := result.stdout.strip()):
+            return token
+    except (FileNotFoundError, subprocess.TimeoutExpired):
+        pass
+
+    return None
+
+
+async def get_github_repo_tags(host: str, owner: str, repo: str) -> list[str]:
+    headers = {"Accept": "application/vnd.github.v3+json"}
+
+    # Add authentication if GITHUB_TOKEN is available (avoids rate limiting)
+    if token := get_local_github_token():
+        headers["Authorization"] = f"token {token}"
+
+    # Determine API host
+    if host == "github.com":
+        api_host = "api.github.com"
+        if not get_local_github_token():
+            console.info(
+                "No GitHub token found. Using unauthenticated API (60 requests/hour limit).\n"
+                "For higher limits, run [green]gh auth login[/green] or set GITHUB_TOKEN or GH_TOKEN env variable."
+            )
+    else:
+        # GitHub Enterprise uses host/api/v3
+        api_host = f"{host}/api/v3"
+
+    async with httpx.AsyncClient() as client:
+        # Correct format: /repos/{owner}/{repo}/tags
+        url = f"https://{api_host}/repos/{owner}/{repo}/tags"
+        response = await client.get(url, headers=headers)
+
+        tags = [tag["name"] for tag in response.json()] if response.status_code == 200 else []
+
+        if not tags and response.status_code != 200:
+            console.warning(f"Failed to fetch tags from '{url}' (status code {response.status_code}). ")
+        elif not tags and response.status_code <= 200:
+            console.warning(f"No tags fetched from '{url}' (status code {response.status_code}). ")
+
+        return tags
 
 
 def get_httpx_response_error_details(response: httpx.Response | None) -> tuple[str, str] | None: