praisonai-code 0.0.39__tar.gz → 0.0.41__tar.gz

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonai-code
-Version: 0.0.39
+Version: 0.0.41
 Summary: Agentic terminal CLI for PraisonAI — the terminal-native agent (run, chat, code, runtime, cli backends) extracted from the praisonai wrapper.
 Author: Mervin Praison
 License: MIT
@@ -80,9 +80,10 @@ importing the wrapper. Only commands in `_WRAPPER_RESIDENT_COMMANDS` (see
 |---------|-------------------|-------|
 | `praisonai-code --version` | Yes | |
 | `run --help`, `config`, `doctor` | Yes | |
-| `run --output verbose "…"` | Yes | In-process `Agent` (verified). `run` modes: `silent` (default), `actions`, `verbose`, `json`, `stream` |
-| `run "…"` (default) | Yes | Resolves via `praisonai_code.cli.commands.run` |
-| `run --output actions "…"` | Yes (intended) | Currently affected by an `auto_save` shadow-import bug — tracked separately |
+| `run --output actions "…"` | Yes | In-process `Agent` (structured events). `run` modes: `silent` (default), `plain`, `actions`, `verbose`, `json`, `stream`, `stream-json` |
+| `run --output json/stream/stream-json "…"` | Yes | In-process `Agent` (structured output) |
+| `run "…"` (default) | Needs `praisonai` | Human-readable text run delegates to the wrapper's `handle_direct_prompt`; standalone gates with an install hint pointing to `--output actions` |
+| `run --output plain/verbose/silent "…"` | Needs `praisonai` | Same text path as default; standalone gates with an install hint |
 | `chat --output plain "…"` | Yes | One-shot; interactive REPL also in code package. `chat` modes: `actions` (default), `plain`, `verbose`, `json`, `silent` |
 | `code --help` | Yes | Full code assistant command registered |
 | `daemon start` (foreground) | Yes | |

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/README.md

@@ -58,9 +58,10 @@ importing the wrapper. Only commands in `_WRAPPER_RESIDENT_COMMANDS` (see
 |---------|-------------------|-------|
 | `praisonai-code --version` | Yes | |
 | `run --help`, `config`, `doctor` | Yes | |
-| `run --output verbose "…"` | Yes | In-process `Agent` (verified). `run` modes: `silent` (default), `actions`, `verbose`, `json`, `stream` |
-| `run "…"` (default) | Yes | Resolves via `praisonai_code.cli.commands.run` |
-| `run --output actions "…"` | Yes (intended) | Currently affected by an `auto_save` shadow-import bug — tracked separately |
+| `run --output actions "…"` | Yes | In-process `Agent` (structured events). `run` modes: `silent` (default), `plain`, `actions`, `verbose`, `json`, `stream`, `stream-json` |
+| `run --output json/stream/stream-json "…"` | Yes | In-process `Agent` (structured output) |
+| `run "…"` (default) | Needs `praisonai` | Human-readable text run delegates to the wrapper's `handle_direct_prompt`; standalone gates with an install hint pointing to `--output actions` |
+| `run --output plain/verbose/silent "…"` | Needs `praisonai` | Same text path as default; standalone gates with an install hint |
 | `chat --output plain "…"` | Yes | One-shot; interactive REPL also in code package. `chat` modes: `actions` (default), `plain`, `verbose`, `json`, `silent` |
 | `code --help` | Yes | Full code assistant command registered |
 | `daemon start` (foreground) | Yes | |

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/__init__.py

@@ -12,6 +12,6 @@ the remaining terminal-agent modules, with PEP 562 shims left behind at the
 old ``praisonai.*`` paths for backward compatibility.
 """
 
-__version__ = "0.0.39"
+__version__ = "0.0.41"
 
 __all__ = ["__version__"]

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/_wrapper_bridge.py

@@ -51,3 +51,48 @@ def optional_wrapper_attr(module_name: str, attr: str, default: T | None = None)
         return get_wrapper_attr(module_name, attr)
     except (ImportError, AttributeError):
         return default
+
+
+def run_wrapper_command(argv: list[str], *, feature: str) -> None:
+    """Re-enter the legacy ``PraisonAI().main()`` for a wrapper-resident command.
+
+    Typer stub commands (``agents``, ``workflow``, ``registry`` and peers) still
+    dispatch to the legacy CLI by mutating ``sys.argv`` and calling
+    ``PraisonAI().main()``. On a standalone ``pip install praisonai-code`` the
+    legacy path imports ``praisonai.cli.legacy.dispatch.argparse_builder`` which
+    is wrapper-only, so it raises a Rich ``ImportError`` traceback before any
+    feature handler runs (issue #2837).
+
+    This helper guards that re-entry: when the ``praisonai`` wrapper is not
+    installed it fails fast with a single-line install hint (exit code 1) instead
+    of a traceback. When the wrapper is present it performs the ``sys.argv``
+    mutation + ``main()`` call exactly as before, restoring ``sys.argv`` after.
+    A clean ``SystemExit`` (code ``0``/``None``) is swallowed so the calling
+    Typer command returns normally, while any non-zero exit code is re-raised so
+    the shell sees the wrapper's failure.
+
+    Args:
+        argv: Legacy argv tokens (e.g. ``["agents", "list"]``) without the
+            leading program name.
+        feature: Command group name used in the standalone hint message.
+    """
+    import sys
+
+    if not wrapper_available():
+        print(
+            f"{feature} requires the full wrapper. {_INSTALL_HINT}",
+            file=sys.stderr,
+        )
+        raise SystemExit(1)
+
+    from praisonai_code.cli.main import PraisonAI
+
+    original_argv = sys.argv
+    sys.argv = ["praisonai"] + argv
+    try:
+        PraisonAI().main()
+    except SystemExit as exc:
+        if exc.code:
+            raise
+    finally:
+        sys.argv = original_argv

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/app.py

@@ -15,6 +15,7 @@ from typer.main import get_command as typer_get_command
 
 from .output.console import OutputController, OutputMode, set_output_controller
 from .state.identifiers import create_context
+from .help_categories import category_for
 
 # Single source of truth for the terminal fallback model; entry points route
 # through resolve_default_model() and never re-declare this literal.
@@ -116,6 +117,20 @@ def _setup_langextract_observability(*, verbose: bool = False) -> None:
             typer.echo(f"Warning: failed to initialize langextract observability: {e}", err=True)
 
 
+def _help_panel_unset(command: click.Command) -> bool:
+    """Return True when ``command`` has no explicit ``rich_help_panel``.
+
+    Typer leaves the attribute as a ``DefaultPlaceholder`` (not ``None``) for
+    commands that don't set one, so both cases must be treated as "unset" before
+    we assign a category.
+    """
+    panel = getattr(command, "rich_help_panel", None)
+    if panel is None:
+        return True
+    # ``DefaultPlaceholder`` is Typer's sentinel for "value not provided".
+    return type(panel).__name__ == "DefaultPlaceholder"
+
+
 class OutputFormat(str, Enum):
     """Output format options."""
     text = "text"
@@ -320,7 +335,19 @@ class LazyCommandGroup(TyperGroup):
         return sorted(list(commands))
     
     def get_command(self, ctx: click.Context, name: str) -> Optional[click.Command]:
-        """Lazily import and return the command."""
+        """Lazily import and return the command, tagged with its help category."""
+        command = self._resolve_command(ctx, name)
+        if command is not None and _help_panel_unset(command):
+            # Group the command into a categorised ``--help`` panel. This is the
+            # single hook Typer's rich renderer reads; setting it lazily here keeps
+            # the registry the source of truth without importing every command up
+            # front just to categorise it. Commands that explicitly declare their
+            # own ``rich_help_panel`` keep it.
+            command.rich_help_panel = category_for(name)
+        return command
+
+    def _resolve_command(self, ctx: click.Context, name: str) -> Optional[click.Command]:
+        """Lazily import and return the command (without category tagging)."""
         # First check if command is already registered (e.g., retrieval commands)
         existing = super().get_command(ctx, name)
         if existing is not None:

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/agents.py

@@ -14,23 +14,13 @@ def agents_list(
     verbose: bool = typer.Option(False, "--verbose", "-v", help="Show detailed info"),
 ):
     """List available agents."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['agents', 'list']
     if verbose:
         argv.append('--verbose')
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="agents")
 
 
 @app.command("create")
@@ -39,23 +29,13 @@ def agents_create(
     template: str = typer.Option(None, "--template", "-t", help="Template to use"),
 ):
     """Create a new agent."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['agents', 'create', name]
     if template:
         argv.extend(['--template', template])
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="agents")
 
 
 @app.command("info")
@@ -63,18 +43,8 @@ def agents_info(
     name: str = typer.Argument(..., help="Agent name"),
 ):
     """Show agent information."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['agents', 'info', name]
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="agents")

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/auth.py

@@ -80,7 +80,14 @@ def _validate_with_live_call(provider: str, api_key: str, base_url: Optional[str
         return False, f"API key invalid: {str(e)}"
 
 
-def _run_oauth_login(provider: str, output, base_url, model, no_browser: bool) -> None:
+def _run_oauth_login(
+    provider: str,
+    output,
+    base_url,
+    model,
+    no_browser: bool,
+    overrides: Optional[dict] = None,
+) -> None:
     """Run the browser/device-code OAuth flow and store the resulting tokens."""
     from ..configuration.oauth import run_oauth_login
 
@@ -95,6 +102,7 @@ def _run_oauth_login(provider: str, output, base_url, model, no_browser: bool) -
     try:
         config, tokens = run_oauth_login(
             provider,
+            overrides=overrides,
             open_browser=not no_browser,
             on_prompt=_on_prompt,
         )
@@ -148,6 +156,11 @@ def auth_login(
     model: Optional[str] = typer.Option(None, "--model", help="Default model for this provider"),
     skip_validation: bool = typer.Option(False, "--skip-validation", help="Skip API key validation"),
     no_browser: bool = typer.Option(False, "--no-browser", help="Do not auto-open the browser for OAuth"),
+    client_id: Optional[str] = typer.Option(None, "--client-id", help="OAuth client id (overrides the built-in provider entry)"),
+    token_url: Optional[str] = typer.Option(None, "--token-url", help="OAuth token endpoint (for providers not in the built-in registry)"),
+    device_authorization_url: Optional[str] = typer.Option(None, "--device-authorization-url", help="OAuth device-code endpoint (RFC 8628)"),
+    authorization_url: Optional[str] = typer.Option(None, "--authorization-url", help="OAuth authorization endpoint (RFC 7636 PKCE)"),
+    scope: Optional[str] = typer.Option(None, "--scope", help="OAuth scope(s) to request"),
 ):
     """
     Store API credentials for a provider.
@@ -156,10 +169,22 @@ def auth_login(
         praisonai auth login openai
         praisonai auth login openai --key sk-...
         echo "sk-..." | praisonai auth login openai --key-stdin
-        praisonai auth login openai --method oauth
+        praisonai auth login github --method oauth --client-id <app-id>
     """
     output = get_output_controller()
 
+    # Assemble any OAuth endpoint overrides supplied on the command line. These
+    # let users sign in to a built-in provider that needs an app client id, or
+    # to a provider not in the registry (e.g. a self-hosted gateway).
+    oauth_overrides = {
+        "client_id": client_id,
+        "token_url": token_url,
+        "device_authorization_url": device_authorization_url,
+        "authorization_url": authorization_url,
+        "scope": scope,
+    }
+    oauth_overrides = {k: v for k, v in oauth_overrides.items() if v is not None}
+
     # Decide between OAuth and API-key login. ``auto`` selects OAuth only when
     # the provider has an OAuth config AND no key was supplied; otherwise it
     # falls back to the API-key path (unchanged behaviour).
@@ -168,14 +193,26 @@ def auth_login(
         output.print_error("Invalid auth method. Use: auto, apikey, or oauth")
         raise typer.Exit(1)
     if method == "oauth":
-        _run_oauth_login(provider, output, base_url, model, no_browser)
+        _run_oauth_login(provider, output, base_url, model, no_browser, oauth_overrides)
         return
     if method == "auto" and not key and not key_stdin:
         try:
-            from ..configuration.oauth import provider_supports_oauth
-            if provider_supports_oauth(provider):
-                _run_oauth_login(provider, output, base_url, model, no_browser)
-                return
+            from ..configuration.oauth import provider_supports_oauth, provider_requires_client_id
+            if provider_supports_oauth(provider, oauth_overrides):
+                # Don't silently start an OAuth flow that will immediately fail
+                # for lack of a client id; fall through to API-key login unless
+                # the user explicitly supplied one.
+                if not provider_requires_client_id(provider, oauth_overrides):
+                    _run_oauth_login(provider, output, base_url, model, no_browser, oauth_overrides)
+                    return
+                # Known OAuth provider but no client id yet: surface the OAuth
+                # path so the user isn't left wondering why they're being asked
+                # for an API key, then continue to the API-key fallback below.
+                output.print_info(
+                    f"{provider} supports OAuth sign-in. To use it, re-run with: "
+                    f"praisonai auth login {provider} --method oauth "
+                    f"--client-id <your-oauth-app-client-id>"
+                )
         except Exception:
             # Fall through to API-key login on any OAuth detection error.
             pass

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/call.py

@@ -24,8 +24,7 @@ def call_main(
         praisonai call
         praisonai call --model gpt-4o
     """
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['call']
     if model:
@@ -33,13 +32,4 @@ def call_main(
     if verbose:
         argv.append('--verbose')
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="call")

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/commit.py

@@ -24,8 +24,7 @@ def commit_main(
         praisonai commit -m "Fix bug"
         praisonai commit --all --push
     """
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['commit']
     if message:
@@ -35,13 +34,4 @@ def commit_main(
     if push:
         argv.append('--push')
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="commit")

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/config.py

@@ -632,6 +632,50 @@ def config_validate(
         raise typer.Exit(1) from e
 
 
+@app.command("provenance")
+def config_provenance(
+    key: Optional[str] = typer.Argument(
+        None,
+        help="Optional dotted key to inspect (e.g. agent.model). Omit for all keys.",
+    ),
+):
+    """Show the resolved value of each key and which file/layer supplied it."""
+    output = get_output_controller()
+
+    try:
+        resolver = get_resolver()
+        provenance = resolver.resolve_with_provenance()
+
+        if key:
+            entry = provenance.get(key)
+            if entry is None:
+                output.print_error(f"Key not found: {key}")
+                raise typer.Exit(1)
+            if output.is_json_mode:
+                output.print_json({"key": key, **entry})
+            else:
+                source = entry["source"] or "(runtime)"
+                output.print(f"{key} = {entry['value']}")
+                output.print(f"  [dim]layer:[/dim] {entry['layer']}  [dim]source:[/dim] {source}")
+            return
+
+        if output.is_json_mode:
+            output.print_json(provenance)
+            return
+
+        output.print("[bold]Resolved configuration (per-key provenance):[/bold]\n")
+        for dotted in sorted(provenance):
+            entry = provenance[dotted]
+            source = entry["source"] or f"({entry['layer']})"
+            output.print(f"  {dotted} = {entry['value']}")
+            output.print(f"      [dim]{entry['layer']} · {source}[/dim]")
+    except typer.Exit:
+        raise
+    except Exception as e:
+        output.print_error(f"Failed to resolve provenance: {e}")
+        raise typer.Exit(1) from e
+
+
 @app.command("sources")
 def config_sources():
     """List all configuration sources in precedence order."""

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/eval.py

@@ -17,21 +17,11 @@ def eval_accuracy(
     iterations: int = typer.Option(3, "--iterations", "-n", help="Number of iterations"),
 ):
     """Run accuracy evaluation."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['eval', 'accuracy', agent, '--input', input_text, '--expected', expected]
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="eval")
 
 
 @app.command("performance")
@@ -40,21 +30,11 @@ def eval_performance(
     iterations: int = typer.Option(50, "--iterations", "-n", help="Number of iterations"),
 ):
     """Run performance evaluation."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['eval', 'performance', agent, '--iterations', str(iterations)]
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="eval")
 
 
 @app.command("judge")
@@ -129,3 +109,9 @@ def list_judges_cmd():
     typer.echo("Registered Judge Types:")
     for name in judges:
         typer.echo(f"  - {name}")
+
+
+@app.command("list")
+def eval_list():
+    """List registered judge types (alias for list-judges)."""
+    list_judges_cmd()

praisonai_code-0.0.41/praisonai_code/cli/commands/hooks.py

@@ -0,0 +1,44 @@
+"""
+Hooks command group for PraisonAI CLI.
+
+Provides hook management commands.
+"""
+
+import typer
+
+app = typer.Typer(help="Hook management")
+
+
+@app.command("list")
+def hooks_list():
+    """List available hooks."""
+    from praisonai_code._wrapper_bridge import run_wrapper_command
+    
+    argv = ['hooks', 'list']
+    
+    run_wrapper_command(argv, feature="hooks")
+
+
+@app.command("add")
+def hooks_add(
+    name: str = typer.Argument(..., help="Hook name"),
+    event: str = typer.Option(..., "--event", "-e", help="Event to hook into"),
+):
+    """Add a hook."""
+    from praisonai_code._wrapper_bridge import run_wrapper_command
+    
+    argv = ['hooks', 'add', name, '--event', event]
+    
+    run_wrapper_command(argv, feature="hooks")
+
+
+@app.command("remove")
+def hooks_remove(
+    name: str = typer.Argument(..., help="Hook name to remove"),
+):
+    """Remove a hook."""
+    from praisonai_code._wrapper_bridge import run_wrapper_command
+    
+    argv = ['hooks', 'remove', name]
+    
+    run_wrapper_command(argv, feature="hooks")

{praisonai_code-0.0.39 → praisonai_code-0.0.41}/praisonai_code/cli/commands/memory.py

@@ -15,23 +15,24 @@ def memory_show(
     limit: int = typer.Option(10, "--limit", "-l", help="Number of memories to show"),
 ):
     """Show stored memories."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['memory', 'show']
     if user_id:
         argv.extend(['--user-id', user_id])
+    if limit is not None:
+        argv.extend(['--limit', str(limit)])
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="memory")
+
+
+@app.command("list")
+def memory_list(
+    user_id: str = typer.Option(None, "--user-id", help="User ID for memory isolation"),
+    limit: int = typer.Option(10, "--limit", "-l", help="Number of memories to show"),
+):
+    """List stored memories (alias for 'show')."""
+    memory_show(user_id=user_id, limit=limit)
 
 
 @app.command("add")
@@ -40,23 +41,13 @@ def memory_add(
     user_id: str = typer.Option(None, "--user-id", help="User ID for memory isolation"),
 ):
     """Add a memory."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['memory', 'add', content]
     if user_id:
         argv.extend(['--user-id', user_id])
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="memory")
 
 
 @app.command("search")
@@ -65,23 +56,13 @@ def memory_search(
     user_id: str = typer.Option(None, "--user-id", help="User ID for memory isolation"),
 ):
     """Search memories."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['memory', 'search', query]
     if user_id:
         argv.extend(['--user-id', user_id])
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="memory")
 
 
 @app.command("clear")
@@ -90,23 +71,13 @@ def memory_clear(
     force: bool = typer.Option(False, "--force", "-f", help="Force clear without confirmation"),
 ):
     """Clear all memories."""
-    from praisonai_code.cli.main import PraisonAI
-    import sys
+    from praisonai_code._wrapper_bridge import run_wrapper_command
     
     argv = ['memory', 'clear']
     if user_id:
         argv.extend(['--user-id', user_id])
     
-    original_argv = sys.argv
-    sys.argv = ['praisonai'] + argv
-    
-    try:
-        praison = PraisonAI()
-        praison.main()
-    except SystemExit:
-        pass
-    finally:
-        sys.argv = original_argv
+    run_wrapper_command(argv, feature="memory")
 
 
 # Learn subcommand group