> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/gym/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/gym/_mcp/server.

# nemo_gym.model_registry

Registry of model servers under `responses_api_models/&lt;name&gt;/`.

Maps each model dir to the config flavors it ships (`configs/&lt;flavor&gt;.yaml`), so they can be
enumerated by the token passed to `--model-type` (see :attr:`ModelEntry.model_types`). Reads the
directory tree only; never loads a config.

## Module Contents

### Classes

| Name                                                | Description                                                  |
| --------------------------------------------------- | ------------------------------------------------------------ |
| [`ModelEntry`](#nemo_gym-model_registry-ModelEntry) | A discovered model server: its name, group, and config file. |

### Functions

| Name                                                                          | Description                                                                                          |
| ----------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------- |
| [`_discover_models_in_dir`](#nemo_gym-model_registry-_discover_models_in_dir) | Map model name -> :class:`ModelEntry` for every config flavor under one `responses_api_models/` dir. |
| [`discover_models`](#nemo_gym-model_registry-discover_models)                 | Map model name -> :class:`ModelEntry` for every discoverable model server.                           |

### Data

[`MODELS_DIR`](#nemo_gym-model_registry-MODELS_DIR)

[`MODELS_SUBDIR`](#nemo_gym-model_registry-MODELS_SUBDIR)

[`MODEL_CONFIGS_SUBDIR`](#nemo_gym-model_registry-MODEL_CONFIGS_SUBDIR)

### API

```python
class nemo_gym.model_registry.ModelEntry(
    name: str,
    model_group: str,
    config_path: pathlib.Path
)
```

Dataclass

A discovered model server: its name, group, and config file.

```python
nemo_gym.model_registry._discover_models_in_dir(
    models_dir: pathlib.Path
) -> typing.Dict[str, nemo_gym.model_registry.ModelEntry]
```

Map model name -> :class:`ModelEntry` for every config flavor under one `responses_api_models/` dir.

One entry per `configs/&lt;flavor&gt;.yaml`: `&lt;dir&gt;` for the flavor named after the model, `&lt;dir&gt;/&lt;flavor&gt;`
for the rest. A config-less dir contributes nothing. Returns an empty dict if the directory is missing.

```python
nemo_gym.model_registry.discover_models() -> typing.Dict[str, nemo_gym.model_registry.ModelEntry]
```

Map model name -> :class:`ModelEntry` for every discoverable model server.

Scans the `responses_api_models/` subdir of every :func:`~nemo_gym.discovery.component_search_roots`
root (`NEMO_GYM_EXTRA_ROOTS` + cwd + built-ins), merged so user models shadow same-named built-ins.

```python
nemo_gym.model_registry.MODELS_DIR = PARENT_DIR / MODELS_SUBDIR
```

```python
nemo_gym.model_registry.MODELS_SUBDIR = 'responses_api_models'
```

```python
nemo_gym.model_registry.MODEL_CONFIGS_SUBDIR = 'configs'
```