diff --git a/README.md b/README.md
index 1b83d051..033943a9 100644
--- a/README.md
+++ b/README.md
@@ -845,10 +845,33 @@ Namespace | Resource | Operation | HTTP request |
 **MediaSets** | MediaSet | [**transform**](docs/v2/MediaSets/MediaSet.md#transform) | **POST** /v2/mediasets/{mediaSetRid}/items/{mediaItemRid}/transform |
 **MediaSets** | MediaSet | [**upload**](docs/v2/MediaSets/MediaSet.md#upload) | **POST** /v2/mediasets/{mediaSetRid}/items |
 **MediaSets** | MediaSet | [**upload_media**](docs/v2/MediaSets/MediaSet.md#upload_media) | **PUT** /v2/mediasets/media/upload |
+**Models** | Experiment | [**get**](docs/v2/Models/Experiment.md#get) | **GET** /v2/models/{modelRid}/experiments/{experimentRid} |
+**Models** | Experiment | [**search**](docs/v2/Models/Experiment.md#search) | **POST** /v2/models/{modelRid}/experiments/search |
+**Models** | ExperimentArtifactTable | [**json**](docs/v2/Models/ExperimentArtifactTable.md#json) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/artifactTables/{experimentArtifactTableName}/json |
+**Models** | ExperimentArtifactTable | [**parquet**](docs/v2/Models/ExperimentArtifactTable.md#parquet) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/artifactTables/{experimentArtifactTableName}/parquet |
+**Models** | ExperimentSeries | [**json**](docs/v2/Models/ExperimentSeries.md#json) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/series/{experimentSeriesName}/json |
+**Models** | ExperimentSeries | [**parquet**](docs/v2/Models/ExperimentSeries.md#parquet) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/series/{experimentSeriesName}/parquet |
+**Models** | LiveDeployment | [**create**](docs/v2/Models/LiveDeployment.md#create) | **POST** /v2/models/liveDeployments |
+**Models** | LiveDeployment | [**get**](docs/v2/Models/LiveDeployment.md#get) | **GET** /v2/models/liveDeployments/{liveDeploymentRid} |
+**Models** | LiveDeployment | [**list**](docs/v2/Models/LiveDeployment.md#list) | **GET** /v2/models/liveDeployments |
+**Models** | LiveDeployment | [**replace**](docs/v2/Models/LiveDeployment.md#replace) | **PUT** /v2/models/liveDeployments/{liveDeploymentRid} |
 **Models** | LiveDeployment | [**transform_json**](docs/v2/Models/LiveDeployment.md#transform_json) | **POST** /v2/models/liveDeployments/{liveDeploymentRid}/transformJson |
 **Models** | Model | [**create**](docs/v2/Models/Model.md#create) | **POST** /v2/models |
 **Models** | Model | [**get**](docs/v2/Models/Model.md#get) | **GET** /v2/models/{modelRid} |
 **Models** | Model | [**promote_version**](docs/v2/Models/Model.md#promote_version) | **POST** /v2/models/{modelRid}/promoteVersion |
+**Models** | ModelFunction | [**create**](docs/v2/Models/ModelFunction.md#create) | **POST** /v2/models/{modelRid}/function |
+**Models** | ModelFunction | [**get**](docs/v2/Models/ModelFunction.md#get) | **GET** /v2/models/{modelRid}/function |
+**Models** | ModelFunction | [**replace**](docs/v2/Models/ModelFunction.md#replace) | **PUT** /v2/models/{modelRid}/function |
+**Models** | ModelStudio | [**create**](docs/v2/Models/ModelStudio.md#create) | **POST** /v2/models/modelStudios |
+**Models** | ModelStudio | [**get**](docs/v2/Models/ModelStudio.md#get) | **GET** /v2/models/modelStudios/{modelStudioRid} |
+**Models** | ModelStudio | [**launch**](docs/v2/Models/ModelStudio.md#launch) | **POST** /v2/models/modelStudios/{modelStudioRid}/launch |
+**Models** | ModelStudioConfigVersion | [**create**](docs/v2/Models/ModelStudioConfigVersion.md#create) | **POST** /v2/models/modelStudios/{modelStudioRid}/configVersions |
+**Models** | ModelStudioConfigVersion | [**get**](docs/v2/Models/ModelStudioConfigVersion.md#get) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions/{modelStudioConfigVersionVersion} |
+**Models** | ModelStudioConfigVersion | [**latest**](docs/v2/Models/ModelStudioConfigVersion.md#latest) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions/latest |
+**Models** | ModelStudioConfigVersion | [**list**](docs/v2/Models/ModelStudioConfigVersion.md#list) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions |
+**Models** | ModelStudioRun | [**list**](docs/v2/Models/ModelStudioRun.md#list) | **GET** /v2/models/modelStudios/{modelStudioRid}/runs |
+**Models** | ModelStudioTrainer | [**get**](docs/v2/Models/ModelStudioTrainer.md#get) | **GET** /v2/models/modelStudioTrainers/{modelStudioTrainerTrainerId} |
+**Models** | ModelStudioTrainer | [**list**](docs/v2/Models/ModelStudioTrainer.md#list) | **GET** /v2/models/modelStudioTrainers |
 **Models** | ModelVersion | [**get**](docs/v2/Models/ModelVersion.md#get) | **GET** /v2/models/{modelRid}/versions/{modelVersionRid} |
 **Models** | ModelVersion | [**list**](docs/v2/Models/ModelVersion.md#list) | **GET** /v2/models/{modelRid}/versions |
 **Ontologies** | Action | [**apply**](docs/v2/Ontologies/Action.md#apply) | **POST** /v2/ontologies/{ontology}/actions/{action}/apply |
@@ -2154,7 +2177,6 @@ Namespace | Name | Import |
 **Models** | [ExperimentArtifactMetadata](docs/v2/Models/models/ExperimentArtifactMetadata.md) | `from foundry_sdk.v2.models.models import ExperimentArtifactMetadata` |
 **Models** | [ExperimentArtifactName](docs/v2/Models/models/ExperimentArtifactName.md) | `from foundry_sdk.v2.models.models import ExperimentArtifactName` |
 **Models** | [ExperimentAuthoringSource](docs/v2/Models/models/ExperimentAuthoringSource.md) | `from foundry_sdk.v2.models.models import ExperimentAuthoringSource` |
-**Models** | [ExperimentBranch](docs/v2/Models/models/ExperimentBranch.md) | `from foundry_sdk.v2.models.models import ExperimentBranch` |
 **Models** | [ExperimentCodeWorkspaceSource](docs/v2/Models/models/ExperimentCodeWorkspaceSource.md) | `from foundry_sdk.v2.models.models import ExperimentCodeWorkspaceSource` |
 **Models** | [ExperimentRid](docs/v2/Models/models/ExperimentRid.md) | `from foundry_sdk.v2.models.models import ExperimentRid` |
 **Models** | [ExperimentSdkSource](docs/v2/Models/models/ExperimentSdkSource.md) | `from foundry_sdk.v2.models.models import ExperimentSdkSource` |
diff --git a/docs-snippets-npm/package.json b/docs-snippets-npm/package.json
index 7355b4c9..3db29e72 100644
--- a/docs-snippets-npm/package.json
+++ b/docs-snippets-npm/package.json
@@ -24,7 +24,7 @@
   "sls": {
     "dependencies": {
       "com.palantir.foundry.api:api-gateway": {
-        "minVersion": "1.1656.0",
+        "minVersion": "1.1665.0",
         "maxVersion": "1.x.x",
         "optional": false
       }
diff --git a/docs-snippets-npm/src/index.ts b/docs-snippets-npm/src/index.ts
index 50ff4fc9..c03466f6 100644
--- a/docs-snippets-npm/src/index.ts
+++ b/docs-snippets-npm/src/index.ts
@@ -1390,7 +1390,7 @@ export const PYTHON_PLATFORM_SNIPPETS: SdkSnippets<typeof PLATFORM_API_DOCS_SPEC
         ],
         "v2.createModelStudioConfigVersion": [
           {
-            "template": "from foundry_sdk import FoundryClient\nimport foundry_sdk\nfrom pprint import pprint\n\nclient = FoundryClient(auth=foundry_sdk.UserTokenAuth(...), hostname=\"example.palantirfoundry.com\")\n\n# ModelStudioRid\nmodel_studio_rid = None\n# ModelStudioConfigVersionName | Human readable name of the configuration version and experiment.\nname = None\n# ResourceConfiguration | The compute resources allocated for training runs.\nresources = {\"gpu\": \"A100\"}\n# TrainerId | The identifier of the trainer to use for this configuration.\ntrainer_id = \"autogluon\"\n# ModelStudioWorkerConfig | The worker configuration including inputs, outputs, and custom settings.\nworker_config = None\n# Optional[str] | Changelog describing changes in this version.\nchangelog = None\n# Optional[PreviewMode] | Enables the use of preview functionality.\npreview = None\n\n\ntry:\n    api_response = client.models.ModelStudio.ConfigVersion.create(\n        model_studio_rid,\n        name=name,\n        resources=resources,\n        trainer_id=trainer_id,\n        worker_config=worker_config,\n        changelog=changelog,\n        preview=preview,\n    )\n    print(\"The create response:\\n\")\n    pprint(api_response)\nexcept foundry_sdk.PalantirRPCException as e:\n    print(\"HTTP error when calling ConfigVersion.create: %s\\n\" % e)"
+            "template": "from foundry_sdk import FoundryClient\nimport foundry_sdk\nfrom pprint import pprint\n\nclient = FoundryClient(auth=foundry_sdk.UserTokenAuth(...), hostname=\"example.palantirfoundry.com\")\n\n# ModelStudioRid\nmodel_studio_rid = None\n# ModelStudioConfigVersionName | Human readable name of the configuration version and experiment.\nname = \"Initial configuration\"\n# ResourceConfiguration | The compute resources allocated for training runs.\nresources = {\"memory\": \"4G\", \"cpu\": \"2\"}\n# TrainerId | The identifier of the trainer to use for this configuration.\ntrainer_id = \"ri.models..trainer.autogluon_tabular_regression\"\n# ModelStudioWorkerConfig | The worker configuration including inputs, outputs, and custom settings.\nworker_config = {\n    \"outputs\": {\n        \"model\": {\n            \"type\": \"model\",\n            \"modelRid\": \"ri.models.main.model.a1b2c3d4-e5f6-7890-abcd-ef1234567890\",\n        }\n    },\n    \"inputs\": {\n        \"input_df\": {\n            \"type\": \"dataset\",\n            \"rid\": \"ri.foundry.main.dataset.a1b2c3d4-e5f6-7890-abcd-ef1234567890\",\n            \"columnMapping\": {\"target_column\": [\"target\"]},\n            \"ignoreColumns\": [],\n            \"selectColumns\": [],\n        }\n    },\n}\n# Optional[str] | Changelog describing changes in this version.\nchangelog = None\n# Optional[PreviewMode] | Enables the use of preview functionality.\npreview = None\n\n\ntry:\n    api_response = client.models.ModelStudio.ConfigVersion.create(\n        model_studio_rid,\n        name=name,\n        resources=resources,\n        trainer_id=trainer_id,\n        worker_config=worker_config,\n        changelog=changelog,\n        preview=preview,\n    )\n    print(\"The create response:\\n\")\n    pprint(api_response)\nexcept foundry_sdk.PalantirRPCException as e:\n    print(\"HTTP error when calling ConfigVersion.create: %s\\n\" % e)"
           }
         ],
         "v2.getModelStudioConfigVersion": [
diff --git a/docs/v2/Admin/models/ListAuthenticationProvidersResponse.md b/docs/v2/Admin/models/ListAuthenticationProvidersResponse.md
index f62dbd5f..567bc7a1 100644
--- a/docs/v2/Admin/models/ListAuthenticationProvidersResponse.md
+++ b/docs/v2/Admin/models/ListAuthenticationProvidersResponse.md
@@ -6,7 +6,6 @@ ListAuthenticationProvidersResponse
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
 **data** | List[AuthenticationProvider] | Yes |  |
-**next_page_token** | Optional[PageToken] | No |  |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Admin/models/ListEnrollmentRoleAssignmentsResponse.md b/docs/v2/Admin/models/ListEnrollmentRoleAssignmentsResponse.md
index c12a1c42..df5fe6aa 100644
--- a/docs/v2/Admin/models/ListEnrollmentRoleAssignmentsResponse.md
+++ b/docs/v2/Admin/models/ListEnrollmentRoleAssignmentsResponse.md
@@ -6,7 +6,6 @@ ListEnrollmentRoleAssignmentsResponse
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
 **data** | List[EnrollmentRoleAssignment] | Yes |  |
-**next_page_token** | Optional[PageToken] | No |  |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Admin/models/ListOrganizationGuestMembersResponse.md b/docs/v2/Admin/models/ListOrganizationGuestMembersResponse.md
index f55c7557..21e57bd4 100644
--- a/docs/v2/Admin/models/ListOrganizationGuestMembersResponse.md
+++ b/docs/v2/Admin/models/ListOrganizationGuestMembersResponse.md
@@ -6,7 +6,6 @@ ListOrganizationGuestMembersResponse
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
 **data** | List[OrganizationGuestMember] | Yes |  |
-**next_page_token** | Optional[PageToken] | No |  |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Admin/models/ListOrganizationRoleAssignmentsResponse.md b/docs/v2/Admin/models/ListOrganizationRoleAssignmentsResponse.md
index 45df3068..00f1f668 100644
--- a/docs/v2/Admin/models/ListOrganizationRoleAssignmentsResponse.md
+++ b/docs/v2/Admin/models/ListOrganizationRoleAssignmentsResponse.md
@@ -6,7 +6,6 @@ ListOrganizationRoleAssignmentsResponse
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
 **data** | List[OrganizationRoleAssignment] | Yes |  |
-**next_page_token** | Optional[PageToken] | No |  |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Models/Experiment.md b/docs/v2/Models/Experiment.md
index 657bc241..362c792a 100644
--- a/docs/v2/Models/Experiment.md
+++ b/docs/v2/Models/Experiment.md
@@ -2,8 +2,8 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**get**](#get) | **GET** /v2/models/{modelRid}/experiments/{experimentRid} | Private Beta |
-[**search**](#search) | **POST** /v2/models/{modelRid}/experiments/search | Private Beta |
+[**get**](#get) | **GET** /v2/models/{modelRid}/experiments/{experimentRid} | Public Beta |
+[**search**](#search) | **POST** /v2/models/{modelRid}/experiments/search | Public Beta |
 
 # **get**
 Retrieve a single experiment with all metadata, parameters, series metadata, and summary metrics.
diff --git a/docs/v2/Models/ExperimentArtifactTable.md b/docs/v2/Models/ExperimentArtifactTable.md
index b8465877..aa27c143 100644
--- a/docs/v2/Models/ExperimentArtifactTable.md
+++ b/docs/v2/Models/ExperimentArtifactTable.md
@@ -2,8 +2,8 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**json**](#json) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/artifactTables/{experimentArtifactTableName}/json | Private Beta |
-[**parquet**](#parquet) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/artifactTables/{experimentArtifactTableName}/parquet | Private Beta |
+[**json**](#json) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/artifactTables/{experimentArtifactTableName}/json | Public Beta |
+[**parquet**](#parquet) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/artifactTables/{experimentArtifactTableName}/parquet | Public Beta |
 
 # **json**
 Read table data from an experiment artifact as a streamed binary response containing JSON.
diff --git a/docs/v2/Models/ExperimentSeries.md b/docs/v2/Models/ExperimentSeries.md
index b0b444da..c760d740 100644
--- a/docs/v2/Models/ExperimentSeries.md
+++ b/docs/v2/Models/ExperimentSeries.md
@@ -2,8 +2,8 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**json**](#json) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/series/{experimentSeriesName}/json | Private Beta |
-[**parquet**](#parquet) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/series/{experimentSeriesName}/parquet | Private Beta |
+[**json**](#json) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/series/{experimentSeriesName}/json | Public Beta |
+[**parquet**](#parquet) | **GET** /v2/models/{modelRid}/experiments/{experimentRid}/series/{experimentSeriesName}/parquet | Public Beta |
 
 # **json**
 Retrieve raw time-series data for a single series in JSON format.
diff --git a/docs/v2/Models/LiveDeployment.md b/docs/v2/Models/LiveDeployment.md
index d3f0d9a4..d739a0c2 100644
--- a/docs/v2/Models/LiveDeployment.md
+++ b/docs/v2/Models/LiveDeployment.md
@@ -2,10 +2,10 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**create**](#create) | **POST** /v2/models/liveDeployments | Private Beta |
-[**get**](#get) | **GET** /v2/models/liveDeployments/{liveDeploymentRid} | Private Beta |
-[**list**](#list) | **GET** /v2/models/liveDeployments | Private Beta |
-[**replace**](#replace) | **PUT** /v2/models/liveDeployments/{liveDeploymentRid} | Private Beta |
+[**create**](#create) | **POST** /v2/models/liveDeployments | Public Beta |
+[**get**](#get) | **GET** /v2/models/liveDeployments/{liveDeploymentRid} | Public Beta |
+[**list**](#list) | **GET** /v2/models/liveDeployments | Public Beta |
+[**replace**](#replace) | **PUT** /v2/models/liveDeployments/{liveDeploymentRid} | Public Beta |
 [**transform_json**](#transform_json) | **POST** /v2/models/liveDeployments/{liveDeploymentRid}/transformJson | Public Beta |
 
 # **create**
diff --git a/docs/v2/Models/ModelFunction.md b/docs/v2/Models/ModelFunction.md
index 450692f9..fda97c23 100644
--- a/docs/v2/Models/ModelFunction.md
+++ b/docs/v2/Models/ModelFunction.md
@@ -2,9 +2,9 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**create**](#create) | **POST** /v2/models/{modelRid}/function | Private Beta |
-[**get**](#get) | **GET** /v2/models/{modelRid}/function | Private Beta |
-[**replace**](#replace) | **PUT** /v2/models/{modelRid}/function | Private Beta |
+[**create**](#create) | **POST** /v2/models/{modelRid}/function | Public Beta |
+[**get**](#get) | **GET** /v2/models/{modelRid}/function | Public Beta |
+[**replace**](#replace) | **PUT** /v2/models/{modelRid}/function | Public Beta |
 
 # **create**
 Creates a function for the model.
diff --git a/docs/v2/Models/ModelStudio.md b/docs/v2/Models/ModelStudio.md
index 30363845..7432d59e 100644
--- a/docs/v2/Models/ModelStudio.md
+++ b/docs/v2/Models/ModelStudio.md
@@ -2,9 +2,9 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**create**](#create) | **POST** /v2/models/modelStudios | Private Beta |
-[**get**](#get) | **GET** /v2/models/modelStudios/{modelStudioRid} | Private Beta |
-[**launch**](#launch) | **POST** /v2/models/modelStudios/{modelStudioRid}/launch | Private Beta |
+[**create**](#create) | **POST** /v2/models/modelStudios | Public Beta |
+[**get**](#get) | **GET** /v2/models/modelStudios/{modelStudioRid} | Public Beta |
+[**launch**](#launch) | **POST** /v2/models/modelStudios/{modelStudioRid}/launch | Public Beta |
 
 # **create**
 Creates a new Model Studio.
diff --git a/docs/v2/Models/ModelStudioConfigVersion.md b/docs/v2/Models/ModelStudioConfigVersion.md
index 5ce0fd3f..870b1cdb 100644
--- a/docs/v2/Models/ModelStudioConfigVersion.md
+++ b/docs/v2/Models/ModelStudioConfigVersion.md
@@ -2,10 +2,10 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**create**](#create) | **POST** /v2/models/modelStudios/{modelStudioRid}/configVersions | Private Beta |
-[**get**](#get) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions/{modelStudioConfigVersionVersion} | Private Beta |
-[**latest**](#latest) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions/latest | Private Beta |
-[**list**](#list) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions | Private Beta |
+[**create**](#create) | **POST** /v2/models/modelStudios/{modelStudioRid}/configVersions | Public Beta |
+[**get**](#get) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions/{modelStudioConfigVersionVersion} | Public Beta |
+[**latest**](#latest) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions/latest | Public Beta |
+[**list**](#list) | **GET** /v2/models/modelStudios/{modelStudioRid}/configVersions | Public Beta |
 
 # **create**
 Creates a new Model Studio configuration version.
@@ -37,13 +37,29 @@ client = FoundryClient(auth=foundry_sdk.UserTokenAuth(...), hostname="example.pa
 # ModelStudioRid
 model_studio_rid = None
 # ModelStudioConfigVersionName | Human readable name of the configuration version and experiment.
-name = None
+name = "Initial configuration"
 # ResourceConfiguration | The compute resources allocated for training runs.
-resources = {"gpu": "A100"}
+resources = {"memory": "4G", "cpu": "2"}
 # TrainerId | The identifier of the trainer to use for this configuration.
-trainer_id = "autogluon"
+trainer_id = "ri.models..trainer.autogluon_tabular_regression"
 # ModelStudioWorkerConfig | The worker configuration including inputs, outputs, and custom settings.
-worker_config = None
+worker_config = {
+    "outputs": {
+        "model": {
+            "type": "model",
+            "modelRid": "ri.models.main.model.a1b2c3d4-e5f6-7890-abcd-ef1234567890",
+        }
+    },
+    "inputs": {
+        "input_df": {
+            "type": "dataset",
+            "rid": "ri.foundry.main.dataset.a1b2c3d4-e5f6-7890-abcd-ef1234567890",
+            "columnMapping": {"target_column": ["target"]},
+            "ignoreColumns": [],
+            "selectColumns": [],
+        }
+    },
+}
 # Optional[str] | Changelog describing changes in this version.
 changelog = None
 # Optional[PreviewMode] | Enables the use of preview functionality.
diff --git a/docs/v2/Models/ModelStudioRun.md b/docs/v2/Models/ModelStudioRun.md
index 598acb49..a20ae6b8 100644
--- a/docs/v2/Models/ModelStudioRun.md
+++ b/docs/v2/Models/ModelStudioRun.md
@@ -2,7 +2,7 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**list**](#list) | **GET** /v2/models/modelStudios/{modelStudioRid}/runs | Private Beta |
+[**list**](#list) | **GET** /v2/models/modelStudios/{modelStudioRid}/runs | Public Beta |
 
 # **list**
 Lists all runs for a Model Studio.
diff --git a/docs/v2/Models/ModelStudioTrainer.md b/docs/v2/Models/ModelStudioTrainer.md
index 75d79c23..7fbb56ba 100644
--- a/docs/v2/Models/ModelStudioTrainer.md
+++ b/docs/v2/Models/ModelStudioTrainer.md
@@ -2,8 +2,8 @@
 
 Method | HTTP request | Release Stage |
 ------------- | ------------- | ----- |
-[**get**](#get) | **GET** /v2/models/modelStudioTrainers/{modelStudioTrainerTrainerId} | Private Beta |
-[**list**](#list) | **GET** /v2/models/modelStudioTrainers | Private Beta |
+[**get**](#get) | **GET** /v2/models/modelStudioTrainers/{modelStudioTrainerTrainerId} | Public Beta |
+[**list**](#list) | **GET** /v2/models/modelStudioTrainers | Public Beta |
 
 # **get**
 Gets details about a specific trainer by its ID and optional version.
diff --git a/docs/v2/Models/models/DatasetInput.md b/docs/v2/Models/models/DatasetInput.md
index 60e0cf7b..27d768d1 100644
--- a/docs/v2/Models/models/DatasetInput.md
+++ b/docs/v2/Models/models/DatasetInput.md
@@ -5,7 +5,7 @@ Dataset input configuration.
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**rid** | RID | Yes | The RID of the input dataset. |
+**rid** | DatasetRid | Yes | The RID of the input dataset. |
 **column_mapping** | Dict[ColumnTypeSpecId, List[ColumnName]] | Yes | Mapping of column type spec IDs to column names. |
 **ignore_columns** | List[ColumnName] | Yes | Columns to ignore from the dataset. |
 **select_columns** | List[ColumnName] | Yes | Columns to select from the dataset. If empty, all columns not in ignoreColumns will be used. |
diff --git a/docs/v2/Models/models/DatasetSchemaNotFoundError.md b/docs/v2/Models/models/DatasetSchemaNotFoundError.md
index ad9f98cd..eed3940f 100644
--- a/docs/v2/Models/models/DatasetSchemaNotFoundError.md
+++ b/docs/v2/Models/models/DatasetSchemaNotFoundError.md
@@ -5,7 +5,7 @@ A schema could not be found for the specified dataset.
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**dataset_rid** | RID | Yes | The RID of the dataset whose schema was not found. |
+**dataset_rid** | DatasetRid | Yes | The RID of the dataset whose schema was not found. |
 **type** | Literal["datasetSchemaNotFound"] | Yes | None |
 
 
diff --git a/docs/v2/Models/models/Experiment.md b/docs/v2/Models/models/Experiment.md
index c8bfd829..e2659982 100644
--- a/docs/v2/Models/models/Experiment.md
+++ b/docs/v2/Models/models/Experiment.md
@@ -12,7 +12,7 @@ Experiment
 **source** | ExperimentSource | Yes |  |
 **status** | ExperimentStatus | Yes |  |
 **status_message** | Optional[str] | No |  |
-**branch** | ExperimentBranch | Yes |  |
+**branch** | BranchName | Yes |  |
 **parameters** | List[Parameter] | Yes |  |
 **series** | List[SeriesAggregations] | Yes |  |
 **summary_metrics** | List[SummaryMetric] | Yes |  |
diff --git a/docs/v2/Models/models/ExperimentBranch.md b/docs/v2/Models/models/ExperimentBranch.md
deleted file mode 100644
index 606f44d8..00000000
--- a/docs/v2/Models/models/ExperimentBranch.md
+++ /dev/null
@@ -1,11 +0,0 @@
-# ExperimentBranch
-
-ExperimentBranch
-
-## Type
-```python
-str
-```
-
-
-[[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Models/models/FieldValidationError.md b/docs/v2/Models/models/FieldValidationError.md
index e4e8599f..a4dadb44 100644
--- a/docs/v2/Models/models/FieldValidationError.md
+++ b/docs/v2/Models/models/FieldValidationError.md
@@ -5,7 +5,7 @@ A dataset column type is not compatible with the trainer's supported column type
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**dataset_rid** | RID | Yes | The RID of the dataset containing the invalid field. |
+**dataset_rid** | DatasetRid | Yes | The RID of the dataset containing the invalid field. |
 **field_name** | Optional[str] | No | The name of the dataset column or field that failed validation. |
 **field_type** | str | Yes | The type of the dataset field. |
 **type** | Literal["fieldValidationFailure"] | Yes | None |
diff --git a/docs/v2/Models/models/ListLiveDeploymentsResponse.md b/docs/v2/Models/models/ListLiveDeploymentsResponse.md
index def94713..9fde4f8f 100644
--- a/docs/v2/Models/models/ListLiveDeploymentsResponse.md
+++ b/docs/v2/Models/models/ListLiveDeploymentsResponse.md
@@ -6,7 +6,6 @@ ListLiveDeploymentsResponse
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
 **data** | List[LiveDeployment] | Yes |  |
-**next_page_token** | Optional[PageToken] | No |  |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Models/models/ListModelStudioTrainersResponse.md b/docs/v2/Models/models/ListModelStudioTrainersResponse.md
index 8ccdd707..3b1ceb69 100644
--- a/docs/v2/Models/models/ListModelStudioTrainersResponse.md
+++ b/docs/v2/Models/models/ListModelStudioTrainersResponse.md
@@ -6,7 +6,6 @@ ListModelStudioTrainersResponse
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
 **data** | List[ModelStudioTrainer] | Yes |  |
-**next_page_token** | Optional[PageToken] | No |  |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Models/models/MissingRequiredDatasetColumnError.md b/docs/v2/Models/models/MissingRequiredDatasetColumnError.md
index f37b9463..e335ab3e 100644
--- a/docs/v2/Models/models/MissingRequiredDatasetColumnError.md
+++ b/docs/v2/Models/models/MissingRequiredDatasetColumnError.md
@@ -5,7 +5,7 @@ The user-provided dataset is missing a column required by the trainer.
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**dataset_rid** | RID | Yes | The RID of the dataset missing the required column. |
+**dataset_rid** | DatasetRid | Yes | The RID of the dataset missing the required column. |
 **column_type_spec_id** | ColumnTypeSpecId | Yes | The trainer column type spec ID for the required column. |
 **column_names** | List[ColumnName] | Yes | The valid dataset column names that could map to this trainer column. |
 **type** | Literal["missingRequiredDatasetColumn"] | Yes | None |
diff --git a/docs/v2/Models/models/MissingWorkerConfigInputDatasetColumnMappingError.md b/docs/v2/Models/models/MissingWorkerConfigInputDatasetColumnMappingError.md
index efd93dc8..5f04b5a9 100644
--- a/docs/v2/Models/models/MissingWorkerConfigInputDatasetColumnMappingError.md
+++ b/docs/v2/Models/models/MissingWorkerConfigInputDatasetColumnMappingError.md
@@ -5,7 +5,7 @@ The provided worker config input dataset is missing a column mapping required by
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**dataset_rid** | RID | Yes | The RID of the dataset with the missing column mapping. |
+**dataset_rid** | DatasetRid | Yes | The RID of the dataset with the missing column mapping. |
 **column_type_spec_id** | ColumnTypeSpecId | Yes | The column type spec ID for the missing column. |
 **type** | Literal["missingWorkerConfigInputDatasetColumnMapping"] | Yes | None |
 
diff --git a/docs/v2/Models/models/ModelOutput.md b/docs/v2/Models/models/ModelOutput.md
index bededd72..e4987745 100644
--- a/docs/v2/Models/models/ModelOutput.md
+++ b/docs/v2/Models/models/ModelOutput.md
@@ -5,7 +5,7 @@ Model output configuration.
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**model_rid** | RID | Yes | The RID of the output model. |
+**model_rid** | ModelRid | Yes | The RID of the output model. |
 **type** | Literal["model"] | Yes | None |
 
 
diff --git a/docs/v2/Models/models/ModelStudioRunModelOutput.md b/docs/v2/Models/models/ModelStudioRunModelOutput.md
index 6b6bcc54..2b0ce06f 100644
--- a/docs/v2/Models/models/ModelStudioRunModelOutput.md
+++ b/docs/v2/Models/models/ModelStudioRunModelOutput.md
@@ -5,9 +5,9 @@ Resolved model output details for a Model Studio run.
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**model_rid** | RID | Yes | The RID of the model. |
-**model_version_rid** | RID | Yes | The RID of the model version created by this run. |
-**experiment_rid** | Optional[RID] | No | The RID of the experiment associated with this run, if any. |
+**model_rid** | ModelRid | Yes | The RID of the model. |
+**model_version_rid** | ModelVersionRid | Yes | The RID of the model version created by this run. |
+**experiment_rid** | Optional[ExperimentRid] | No | The RID of the experiment associated with this run, if any. |
 **type** | Literal["model"] | Yes | None |
 
 
diff --git a/docs/v2/Models/models/ModelStudioTrainer.md b/docs/v2/Models/models/ModelStudioTrainer.md
index e6dac591..0baf54da 100644
--- a/docs/v2/Models/models/ModelStudioTrainer.md
+++ b/docs/v2/Models/models/ModelStudioTrainer.md
@@ -8,7 +8,7 @@ ModelStudioTrainer
 **trainer_id** | TrainerId | Yes |  |
 **version** | TrainerVersion | Yes | The version of this trainer. |
 **name** | TrainerName | Yes | Human-readable name of the trainer. |
-**type** | TrainerType | Yes | The type/category of this trainer (e.g., TABULAR_CLASSIFICATION, TIME_SERIES). |
+**type** | TrainerType | Yes | The category of machine learning task this trainer is designed to solve. |
 **description** | TrainerDescription | Yes | Description of what this trainer does and its capabilities. |
 **custom_config_schema** | TrainerSchemaDefinition | Yes | JSON schema defining the custom configuration parameters for this trainer. |
 **inputs** | TrainerInputsSpecification | Yes | Input specifications for this trainer. |
diff --git a/docs/v2/Models/models/MultipleColumnsNotAllowedForTrainerError.md b/docs/v2/Models/models/MultipleColumnsNotAllowedForTrainerError.md
index e64b3ee9..09284360 100644
--- a/docs/v2/Models/models/MultipleColumnsNotAllowedForTrainerError.md
+++ b/docs/v2/Models/models/MultipleColumnsNotAllowedForTrainerError.md
@@ -5,7 +5,7 @@ Multiple columns were mapped but the trainer only allows a single column for thi
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**dataset_rid** | RID | Yes | The RID of the dataset with multiple columns mapped. |
+**dataset_rid** | DatasetRid | Yes | The RID of the dataset with multiple columns mapped. |
 **column_type_spec_id** | ColumnTypeSpecId | Yes | The column type spec ID that does not allow multiple columns. |
 **type** | Literal["multipleColumnsNotAllowedForTrainer"] | Yes | None |
 
diff --git a/docs/v2/Models/models/TrainerId.md b/docs/v2/Models/models/TrainerId.md
index 89df8181..736f182a 100644
--- a/docs/v2/Models/models/TrainerId.md
+++ b/docs/v2/Models/models/TrainerId.md
@@ -1,10 +1,10 @@
 # TrainerId
 
-The identifier for a trainer.
+The Resource Identifier (RID) of a trainer.
 
 ## Type
 ```python
-str
+RID
 ```
 
 
diff --git a/docs/v2/Models/models/TrainerType.md b/docs/v2/Models/models/TrainerType.md
index f8060c6a..69a5f33a 100644
--- a/docs/v2/Models/models/TrainerType.md
+++ b/docs/v2/Models/models/TrainerType.md
@@ -1,11 +1,16 @@
 # TrainerType
 
-The type/category of a trainer.
+The category of machine learning task a trainer is designed to solve. This determines the kind of modeling problem the trainer addresses and the shape of the inputs and outputs it expects.
 
-## Type
-```python
-str
-```
+
+| **Value** |
+| --------- |
+| `"GENERIC"` |
+| `"TIME_SERIES"` |
+| `"TABULAR_REGRESSION"` |
+| `"TABULAR_CLASSIFICATION"` |
+| `"LLM_FINETUNING"` |
+| `"VLM_FINETUNING"` |
 
 
 [[Back to Model list]](../../../../README.md#models-v2-link) [[Back to API list]](../../../../README.md#apis-v2-link) [[Back to README]](../../../../README.md)
diff --git a/docs/v2/Models/models/UnknownColumnSpecIdInConfigColumnMappingError.md b/docs/v2/Models/models/UnknownColumnSpecIdInConfigColumnMappingError.md
index dde4cda7..ef693715 100644
--- a/docs/v2/Models/models/UnknownColumnSpecIdInConfigColumnMappingError.md
+++ b/docs/v2/Models/models/UnknownColumnSpecIdInConfigColumnMappingError.md
@@ -5,7 +5,7 @@ The worker config column mapping contains an unknown column spec ID not found in
 ## Properties
 | Name | Type | Required | Description |
 | ------------ | ------------- | ------------- | ------------- |
-**dataset_rid** | RID | Yes | The RID of the dataset containing the unknown column mapping. |
+**dataset_rid** | DatasetRid | Yes | The RID of the dataset containing the unknown column mapping. |
 **column_type_spec_id** | ColumnTypeSpecId | Yes | The unknown column type spec ID. |
 **type** | Literal["unknownColumnSpecIdInConfigColumnMapping"] | Yes | None |
 
diff --git a/foundry_sdk/_version.py b/foundry_sdk/_version.py
index 96ad2237..d9901f11 100644
--- a/foundry_sdk/_version.py
+++ b/foundry_sdk/_version.py
@@ -17,4 +17,4 @@
 # using the autorelease bot
 __version__ = "0.0.0"
 
-__openapi_document_version__ = "1.1656.0"
+__openapi_document_version__ = "1.1665.0"
diff --git a/foundry_sdk/v2/admin/models.py b/foundry_sdk/v2/admin/models.py
index 134745ae..939be803 100644
--- a/foundry_sdk/v2/admin/models.py
+++ b/foundry_sdk/v2/admin/models.py
@@ -408,7 +408,6 @@ class ListAuthenticationProvidersResponse(core.ModelBase):
     """ListAuthenticationProvidersResponse"""
 
     data: typing.List[AuthenticationProvider]
-    next_page_token: typing.Optional[core_models.PageToken] = pydantic.Field(alias=str("nextPageToken"), default=None)  # type: ignore[literal-required]
 
 
 class ListAvailableOrganizationRolesResponse(core.ModelBase):
@@ -427,7 +426,6 @@ class ListEnrollmentRoleAssignmentsResponse(core.ModelBase):
     """ListEnrollmentRoleAssignmentsResponse"""
 
     data: typing.List[EnrollmentRoleAssignment]
-    next_page_token: typing.Optional[core_models.PageToken] = pydantic.Field(alias=str("nextPageToken"), default=None)  # type: ignore[literal-required]
 
 
 class ListGroupMembersResponse(core.ModelBase):
@@ -490,14 +488,12 @@ class ListOrganizationGuestMembersResponse(core.ModelBase):
     """ListOrganizationGuestMembersResponse"""
 
     data: typing.List[OrganizationGuestMember]
-    next_page_token: typing.Optional[core_models.PageToken] = pydantic.Field(alias=str("nextPageToken"), default=None)  # type: ignore[literal-required]
 
 
 class ListOrganizationRoleAssignmentsResponse(core.ModelBase):
     """ListOrganizationRoleAssignmentsResponse"""
 
     data: typing.List[OrganizationRoleAssignment]
-    next_page_token: typing.Optional[core_models.PageToken] = pydantic.Field(alias=str("nextPageToken"), default=None)  # type: ignore[literal-required]
 
 
 class ListUsersResponse(core.ModelBase):
diff --git a/foundry_sdk/v2/models/errors.py b/foundry_sdk/v2/models/errors.py
index 0e30659e..7e05ddc3 100644
--- a/foundry_sdk/v2/models/errors.py
+++ b/foundry_sdk/v2/models/errors.py
@@ -717,7 +717,7 @@ class TrainerNotFoundParameters(typing_extensions.TypedDict):
 
     __pydantic_config__ = {"extra": "allow"}  # type: ignore
 
-    trainerId: str
+    trainerId: models_models.TrainerId
 
 
 @dataclass
diff --git a/foundry_sdk/v2/models/models.py b/foundry_sdk/v2/models/models.py
index f22e06b7..d0fbda1d 100644
--- a/foundry_sdk/v2/models/models.py
+++ b/foundry_sdk/v2/models/models.py
@@ -141,7 +141,7 @@ class CreateModelVersionRequest(core.ModelBase):
 class DatasetInput(core.ModelBase):
     """Dataset input configuration."""
 
-    rid: core.RID
+    rid: core_models.DatasetRid
     """The RID of the input dataset."""
 
     column_mapping: typing.Dict[ColumnTypeSpecId, typing.List[core_models.ColumnName]] = pydantic.Field(alias=str("columnMapping"))  # type: ignore[literal-required]
@@ -159,7 +159,7 @@ class DatasetInput(core.ModelBase):
 class DatasetSchemaNotFoundError(core.ModelBase):
     """A schema could not be found for the specified dataset."""
 
-    dataset_rid: core.RID = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
+    dataset_rid: core_models.DatasetRid = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
     """The RID of the dataset whose schema was not found."""
 
     type: typing.Literal["datasetSchemaNotFound"] = "datasetSchemaNotFound"
@@ -245,7 +245,7 @@ class Experiment(core.ModelBase):
     source: ExperimentSource
     status: ExperimentStatus
     status_message: typing.Optional[str] = pydantic.Field(alias=str("statusMessage"), default=None)  # type: ignore[literal-required]
-    branch: ExperimentBranch
+    branch: core_models.BranchName
     parameters: typing.List[Parameter]
     series: typing.List[SeriesAggregations]
     summary_metrics: typing.List[SummaryMetric] = pydantic.Field(alias=str("summaryMetrics"))  # type: ignore[literal-required]
@@ -275,10 +275,6 @@ class ExperimentAuthoringSource(core.ModelBase):
     type: typing.Literal["authoring"] = "authoring"
 
 
-ExperimentBranch: typing_extensions.TypeAlias = str
-"""ExperimentBranch"""
-
-
 class ExperimentCodeWorkspaceSource(core.ModelBase):
     """Experiment created from a code workspace."""
 
@@ -317,7 +313,7 @@ class ExperimentSdkSource(core.ModelBase):
 class FieldValidationError(core.ModelBase):
     """A dataset column type is not compatible with the trainer's supported column types."""
 
-    dataset_rid: core.RID = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
+    dataset_rid: core_models.DatasetRid = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
     """The RID of the dataset containing the invalid field."""
 
     field_name: typing.Optional[str] = pydantic.Field(alias=str("fieldName"), default=None)  # type: ignore[literal-required]
@@ -432,7 +428,6 @@ class ListLiveDeploymentsResponse(core.ModelBase):
     """ListLiveDeploymentsResponse"""
 
     data: typing.List[LiveDeployment]
-    next_page_token: typing.Optional[core_models.PageToken] = pydantic.Field(alias=str("nextPageToken"), default=None)  # type: ignore[literal-required]
 
 
 class ListModelStudioConfigVersionsResponse(core.ModelBase):
@@ -453,7 +448,6 @@ class ListModelStudioTrainersResponse(core.ModelBase):
     """ListModelStudioTrainersResponse"""
 
     data: typing.List[ModelStudioTrainer]
-    next_page_token: typing.Optional[core_models.PageToken] = pydantic.Field(alias=str("nextPageToken"), default=None)  # type: ignore[literal-required]
 
 
 class ListModelVersionsResponse(core.ModelBase):
@@ -568,7 +562,7 @@ class LiveDeploymentStatus(core.ModelBase):
 class MissingRequiredDatasetColumnError(core.ModelBase):
     """The user-provided dataset is missing a column required by the trainer."""
 
-    dataset_rid: core.RID = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
+    dataset_rid: core_models.DatasetRid = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
     """The RID of the dataset missing the required column."""
 
     column_type_spec_id: ColumnTypeSpecId = pydantic.Field(alias=str("columnTypeSpecId"))  # type: ignore[literal-required]
@@ -583,7 +577,7 @@ class MissingRequiredDatasetColumnError(core.ModelBase):
 class MissingWorkerConfigInputDatasetColumnMappingError(core.ModelBase):
     """The provided worker config input dataset is missing a column mapping required by the trainer."""
 
-    dataset_rid: core.RID = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
+    dataset_rid: core_models.DatasetRid = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
     """The RID of the dataset with the missing column mapping."""
 
     column_type_spec_id: ColumnTypeSpecId = pydantic.Field(alias=str("columnTypeSpecId"))  # type: ignore[literal-required]
@@ -771,7 +765,7 @@ class ModelFunction(core.ModelBase):
 class ModelOutput(core.ModelBase):
     """Model output configuration."""
 
-    model_rid: core.RID = pydantic.Field(alias=str("modelRid"))  # type: ignore[literal-required]
+    model_rid: ModelRid = pydantic.Field(alias=str("modelRid"))  # type: ignore[literal-required]
     """The RID of the output model."""
 
     type: typing.Literal["model"] = "model"
@@ -870,13 +864,13 @@ class ModelStudioRun(core.ModelBase):
 class ModelStudioRunModelOutput(core.ModelBase):
     """Resolved model output details for a Model Studio run."""
 
-    model_rid: core.RID = pydantic.Field(alias=str("modelRid"))  # type: ignore[literal-required]
+    model_rid: ModelRid = pydantic.Field(alias=str("modelRid"))  # type: ignore[literal-required]
     """The RID of the model."""
 
-    model_version_rid: core.RID = pydantic.Field(alias=str("modelVersionRid"))  # type: ignore[literal-required]
+    model_version_rid: ModelVersionRid = pydantic.Field(alias=str("modelVersionRid"))  # type: ignore[literal-required]
     """The RID of the model version created by this run."""
 
-    experiment_rid: typing.Optional[core.RID] = pydantic.Field(alias=str("experimentRid"), default=None)  # type: ignore[literal-required]
+    experiment_rid: typing.Optional[ExperimentRid] = pydantic.Field(alias=str("experimentRid"), default=None)  # type: ignore[literal-required]
     """The RID of the experiment associated with this run, if any."""
 
     type: typing.Literal["model"] = "model"
@@ -893,7 +887,7 @@ class ModelStudioTrainer(core.ModelBase):
     """Human-readable name of the trainer."""
 
     type: TrainerType
-    """The type/category of this trainer (e.g., TABULAR_CLASSIFICATION, TIME_SERIES)."""
+    """The category of machine learning task this trainer is designed to solve."""
 
     description: TrainerDescription
     """Description of what this trainer does and its capabilities."""
@@ -1012,7 +1006,7 @@ class ModelVersionSdkSource(core.ModelBase):
 class MultipleColumnsNotAllowedForTrainerError(core.ModelBase):
     """Multiple columns were mapped but the trainer only allows a single column for this spec."""
 
-    dataset_rid: core.RID = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
+    dataset_rid: core_models.DatasetRid = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
     """The RID of the dataset with multiple columns mapped."""
 
     column_type_spec_id: ColumnTypeSpecId = pydantic.Field(alias=str("columnTypeSpecId"))  # type: ignore[literal-required]
@@ -1423,8 +1417,8 @@ class TableArtifactDetails(core.ModelBase):
 """Description of what a trainer does and its capabilities."""
 
 
-TrainerId: typing_extensions.TypeAlias = str
-"""The identifier for a trainer."""
+TrainerId: typing_extensions.TypeAlias = core.RID
+"""The Resource Identifier (RID) of a trainer."""
 
 
 TrainerInputsSpecification: typing_extensions.TypeAlias = typing.Any
@@ -1443,8 +1437,15 @@ class TableArtifactDetails(core.ModelBase):
 """JSON schema defining the custom configuration parameters for a trainer."""
 
 
-TrainerType: typing_extensions.TypeAlias = str
-"""The type/category of a trainer."""
+TrainerType: typing_extensions.TypeAlias = typing.Literal[
+    "GENERIC",
+    "TIME_SERIES",
+    "TABULAR_REGRESSION",
+    "TABULAR_CLASSIFICATION",
+    "LLM_FINETUNING",
+    "VLM_FINETUNING",
+]
+"""The category of machine learning task a trainer is designed to solve. This determines the kind of modeling problem the trainer addresses and the shape of the inputs and outputs it expects."""
 
 
 TrainerVersion: typing_extensions.TypeAlias = str
@@ -1487,7 +1488,7 @@ class TypeMismatchError(core.ModelBase):
 class UnknownColumnSpecIdInConfigColumnMappingError(core.ModelBase):
     """The worker config column mapping contains an unknown column spec ID not found in the trainer's column specification."""
 
-    dataset_rid: core.RID = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
+    dataset_rid: core_models.DatasetRid = pydantic.Field(alias=str("datasetRid"))  # type: ignore[literal-required]
     """The RID of the dataset containing the unknown column mapping."""
 
     column_type_spec_id: ColumnTypeSpecId = pydantic.Field(alias=str("columnTypeSpecId"))  # type: ignore[literal-required]
@@ -1580,7 +1581,6 @@ class UnsupportedTypeError(core.ModelBase):
     "ExperimentArtifactMetadata",
     "ExperimentArtifactName",
     "ExperimentAuthoringSource",
-    "ExperimentBranch",
     "ExperimentCodeWorkspaceSource",
     "ExperimentRid",
     "ExperimentSdkSource",