DATABRICKS

Command	Description
`export-evaluation`	Export an evaluation’s data items to a new Databricks Delta table.
`export-model`	Register an Oumi model in Unity Catalog as a servable MLflow chat model.
`import-dataset`	Import a Databricks table or volume file as an Oumi dataset.
`login`	Sign in to a Databricks workspace and save it and a SQL warehouse to the profile.

EXPORT-EVALUATION

Export an evaluation’s data items to a new Databricks Delta table.

oumi-cli databricks export-evaluation [OPTIONS] EVALUATION_ID

Arguments

Argument	Type	Required	Description
`EVALUATION_ID`	integer	✓	ID of the evaluation (see `oumi-cli evaluations list`).

Options

Option	Type	Required	Description
`--to`	text	✓	New managed table, catalog.schema.table. Refuses an existing table (exit 6).
`--project`	text		Project that owns the evaluation; falls back to OUMI_PROJECT_ID or the profile.

oumi-cli databricks export-model [OPTIONS] MODEL_ID

Arguments

Argument	Type	Required	Description
`MODEL_ID`	integer	✓	ID of the model (see `oumi-cli models list`).

Options

Option	Type	Required	Description
`--to`	text	✓	New Unity Catalog model, catalog.schema.model. Refuses an existing one (exit 6).
`--model-version`	integer		Version ID to export. Defaults to the latest version.
`--project`	text		Project that owns the model; falls back to OUMI_PROJECT_ID or the profile.

Import a Databricks table or volume file as an Oumi dataset.

oumi-cli databricks import-dataset [OPTIONS] SOURCE

Arguments

Argument	Type	Required	Description
`SOURCE`	text	✓	A table (catalog.schema.table) or a volume file (/Volumes/c/s/v/file.jsonl).

Options

Option	Type	Description
`--prompt-col`	text	Table column holding the user turn. Needs `--response-col`.
`--response-col`	text	Table column holding the assistant turn.
`--limit`	integer range	Import at most this many table rows.
`--display-name`	text	Display name for the dataset. Defaults to the source name.
`--dataset-id`	integer	Upload as a new version of this existing dataset. Omit to create a new one.
`--project`	text	Project to upload the dataset into; falls back to OUMI_PROJECT_ID or the profile.
`--wait` / `--no-wait`	flag	Wait for the dataset ingest to finish (default); `--no-wait` returns the operation ID immediately.
`--dry-run`	flag	Read from Databricks and preview the rows; uploads nothing to Oumi.

oumi-cli databricks login [OPTIONS]

Options

Option	Type	Required	Description
`--workspace-url`	text		Workspace URL, e.g. https://dbc-1.cloud.databricks.com. Defaults to the saved one.
`--warehouse`	text		SQL warehouse ID. Omit to use the only one visible, or choose on a TTY.