Skip to content

Commit 83ccb30

Browse files
committed
fix: restore agent-helper list CLI commands
Re-register the list helper group in the CLI so agent-facing list subcommands are available again alongside inspect.
1 parent ad8da02 commit 83ccb30

1 file changed

Lines changed: 42 additions & 0 deletions

File tree

  • packages/data-designer/src/data_designer/cli

packages/data-designer/src/data_designer/cli/main.py

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,47 @@
9898
no_args_is_help=True,
9999
)
100100

101+
# Create list command group
102+
list_app = typer.Typer(
103+
name="list",
104+
help="List available types, model aliases, and persona datasets.",
105+
cls=create_lazy_typer_group(
106+
{
107+
"model-aliases": {
108+
"module": f"{_CMD}.agent_helpers.list",
109+
"attr": "model_aliases_command",
110+
"help": "List configured model aliases and backing models",
111+
},
112+
"persona-datasets": {
113+
"module": f"{_CMD}.agent_helpers.list",
114+
"attr": "persona_datasets_command",
115+
"help": "List Nemotron-Persona datasets and install status",
116+
},
117+
"columns": {
118+
"module": f"{_CMD}.agent_helpers.list",
119+
"attr": "column_types_command",
120+
"help": "List column type names and config classes",
121+
},
122+
"samplers": {
123+
"module": f"{_CMD}.agent_helpers.list",
124+
"attr": "sampler_types_command",
125+
"help": "List sampler type names and params classes",
126+
},
127+
"validators": {
128+
"module": f"{_CMD}.agent_helpers.list",
129+
"attr": "validator_types_command",
130+
"help": "List validator type names and params classes",
131+
},
132+
"processors": {
133+
"module": f"{_CMD}.agent_helpers.list",
134+
"attr": "processor_types_command",
135+
"help": "List processor type names and config classes",
136+
},
137+
}
138+
),
139+
no_args_is_help=True,
140+
)
141+
101142
# Create inspect command group
102143
inspect_app = typer.Typer(
103144
name="inspect",
@@ -145,6 +186,7 @@
145186

146187
# Add agent command groups
147188
title_agent_helpers = "Agent-Helper Commands"
189+
app.add_typer(list_app, name="list", rich_help_panel=title_agent_helpers)
148190
app.add_typer(inspect_app, name="inspect", rich_help_panel=title_agent_helpers)
149191

150192

0 commit comments

Comments
 (0)