> ## Documentation Index
> Fetch the complete documentation index at: https://docs.oumi.ai/llms.txt
> Use this file to discover all available pages before exploring further.

# DATABRICKS

> Import Databricks tables as datasets, and export evaluations and models to Databricks

| Command | Description |
| - | - |
| [`export-evaluation`](#export-evaluation) | Export an evaluation's data items to a new Databricks Delta table. |
| [`export-model`](#export-model) | Register an Oumi model in Unity Catalog as a servable MLflow chat model. |
| [`import-dataset`](#import-dataset) | Import a Databricks table or volume file as an Oumi dataset. |
| [`login`](#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.

```bash theme={null}
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. |

***

## EXPORT-MODEL

Register an Oumi model in Unity Catalog as a servable MLflow chat model.

```bash theme={null}
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-DATASET

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

```bash theme={null}
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 | Required | 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. |

***

## LOGIN

Sign in to a Databricks workspace and save it and a SQL warehouse to the profile.

```bash theme={null}
oumi-cli databricks login [OPTIONS]
```

**Options**

| Option | Type | Required | Description |
| - | - | - | - |
| `--workspace-url` | text | | Workspace URL, e.g. [https://dbc-1.cloud.databricks.com](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. |

***


This documentation is built and hosted on [Mintlify](https://mintlify.com), a developer documentation platform.