anyscale 0.26.64__py3-none-any.whl → 0.26.66__py3-none-any.whl

anyscale/_private/anyscale_client/common.py

@@ -57,7 +57,7 @@ from anyscale.utils.workspace_notification import WorkspaceNotification
 # Maybe just make it part of the release process to update it, or fetch the
 # default builds and get the latest one. The best thing to do is probably
 # to populate this in the backend.
-DEFAULT_RAY_VERSION = "2.49.2"  # RAY_RELEASE_UPDATE: update to latest version.
+DEFAULT_RAY_VERSION = "2.50.0"  # RAY_RELEASE_UPDATE: update to latest version.
 DEFAULT_PYTHON_VERSION = "py311"
 RUNTIME_ENV_PACKAGE_FORMAT = "pkg_{content_hash}.zip"
 

anyscale/_private/docgen/__main__.py

@@ -736,23 +736,42 @@ def generate(
         raise RuntimeError(f"output_dir '{output_dir}' does not exist.")
 
     gen = MarkdownGenerator(ALL_MODULES)
-    gen.generate()
 
     generated_files = set()
     os.makedirs(output_dir, exist_ok=True)
+
+    # Create legacy subdirectory
+    legacy_dir = os.path.join(output_dir, "legacy")
+    os.makedirs(legacy_dir, exist_ok=True)
+
     for filename, content in gen.generate().items():
         generated_files.add(filename)
         full_path = os.path.join(output_dir, filename)
+
+        # Create directory if it doesn't exist (for legacy/ files)
+        dir_path = os.path.dirname(full_path)
+        if dir_path:
+            os.makedirs(dir_path, exist_ok=True)
+
         print(f"Writing output file {full_path}")
         with open(full_path, "w") as f:
             f.write(content)
 
     if remove_existing:
-        to_remove = set(os.listdir(output_dir)) - generated_files
+        # Get all existing files (including in subdirectories)
+        existing_files = set()
+        for root, _dirs, files in os.walk(output_dir):
+            for file in files:
+                rel_path = os.path.relpath(os.path.join(root, file), output_dir)
+                existing_files.add(rel_path)
+
+        # Remove files that weren't generated
+        to_remove = existing_files - generated_files
         for path in to_remove:
             full_path = os.path.join(output_dir, path)
-            print(f"Removing existing file {full_path}")
-            os.unlink(full_path)
+            if os.path.exists(full_path):
+                print(f"Removing existing file {full_path}")
+                os.unlink(full_path)
 
 
 if __name__ == "__main__":

anyscale/_private/docgen/generator.py

@@ -106,11 +106,9 @@ class MarkdownGenerator:
 
         Returns a dictionary of filename to generated file contents.
 
-        Each module will contain three to four sections:
-            - Models
-            - CLI
-            - SDK
-            - CLI (legacy)
+        Each module will generate two files:
+        - Main file: contains current Models, CLI, and SDK
+        - Legacy file (in legacy/ subfolder): contains legacy CLI, SDK, and Models
         """
 
         output_files: Dict[str, str] = {}
@@ -119,6 +117,7 @@ class MarkdownGenerator:
             os.path.join(os.path.dirname(__file__), "models.md"),
         )
         for m in self._modules:
+            # Generate main (current) documentation
             output = "import Tabs from '@theme/Tabs';\n"
             output += "import TabItem from '@theme/TabItem';\n\n"
             output += f"# {m.title} API Reference\n\n"
@@ -127,12 +126,17 @@ class MarkdownGenerator:
             output += self._generate_clis(m)
             output += self._generate_sdks(m)
             output += self._generate_models(m)
-            output += self._generate_legacy_clis(m)
-            output += self._generate_legacy_sdks(m, legacy_sdks)
-            output += self._generate_legacy_models(m, legacy_models)
 
             output_files[m.filename] = output
 
+            # Generate legacy documentation if any legacy content exists
+            legacy_content = self._generate_legacy_content(
+                m, legacy_sdks, legacy_models
+            )
+            if legacy_content:
+                legacy_filename = f"legacy/{m.filename}"
+                output_files[legacy_filename] = legacy_content
+
         return output_files
 
     def _generate_sdks(self, m: Module) -> str:
@@ -146,15 +150,23 @@ class MarkdownGenerator:
 
         return output
 
-    def _generate_legacy_sdks(self, m: Module, legacy_sdks: List[LegacySDK]) -> str:
+    def _generate_legacy_sdks(
+        self, m: Module, legacy_sdks: List[LegacySDK], for_legacy_file: bool = False
+    ) -> str:
         if not m.legacy_sdk_commands:
             return ""
-        output = f"## {m.legacy_title or m.title} SDK <span class='label-h2 label-legacy'>Legacy</span>\n"
+        if for_legacy_file:
+            output = f"## {m.legacy_title or m.title} SDK\n"
+        else:
+            output = f"## {m.legacy_title or m.title} SDK <span class='label-h2 label-legacy'>Legacy</span>\n"
         output += ANYSCALE_SDK_INTRO + "\n"
         for legacy_sdk_str, new_sdk in m.legacy_sdk_commands.items():
             legacy_sdk = next(sdk for sdk in legacy_sdks if sdk.name == legacy_sdk_str)
             output += "\n" + self._gen_markdown_for_legacy_sdk_command(
-                legacy_sdk, new_sdk, sdk_prefix=m.sdk_prefix
+                legacy_sdk,
+                new_sdk,
+                sdk_prefix=m.sdk_prefix,
+                for_legacy_file=for_legacy_file,
             )
 
         return output
@@ -170,21 +182,28 @@ class MarkdownGenerator:
         return output
 
     def _generate_legacy_models(
-        self, m: Module, legacy_models: List[LegacyModel]
+        self, m: Module, legacy_models: List[LegacyModel], for_legacy_file: bool = False
     ) -> str:
         """Generate documentation for a list of legacy models."""
         if not m.legacy_sdk_models:
             return ""
-        output = f"## {m.legacy_title or m.title} Models <span class='label-h2 label-legacy'>Legacy</span>\n"
+        if for_legacy_file:
+            output = f"## {m.legacy_title or m.title} Models\n"
+        else:
+            output = f"## {m.legacy_title or m.title} Models <span class='label-h2 label-legacy'>Legacy</span>\n"
         for model_str in m.legacy_sdk_models:
             legacy_model = next(
                 model for model in legacy_models if model.name == model_str
             )
-            output += "\n" + self._gen_markdown_for_legacy_model(legacy_model)
+            output += "\n" + self._gen_markdown_for_legacy_model(
+                legacy_model, for_legacy_file
+            )
 
         return output
 
-    def _generate_clis(self, m: Module, is_legacy_cli: bool = False) -> str:
+    def _generate_clis(
+        self, m: Module, is_legacy_cli: bool = False, for_legacy_file: bool = False
+    ) -> str:
         """Generate CLI documentation for a module.
 
         Returns a tuple of CLI and legacy CLI documentation.
@@ -196,7 +215,10 @@ class MarkdownGenerator:
             return ""
 
         if is_legacy_cli:
-            output = f'## {m.legacy_title or m.title} CLI <span class="label-h2 label-legacy">Legacy</span>\n'
+            if for_legacy_file:
+                output = f"## {m.legacy_title or m.title} CLI\n"
+            else:
+                output = f'## {m.legacy_title or m.title} CLI <span class="label-h2 label-legacy">Legacy</span>\n'
         else:
             output = f"## {m.title} CLI\n"
 
@@ -210,8 +232,60 @@ class MarkdownGenerator:
 
         return output
 
-    def _generate_legacy_clis(self, m: Module) -> str:
-        return self._generate_clis(m, is_legacy_cli=True)
+    def _generate_legacy_clis(self, m: Module, for_legacy_file: bool = False) -> str:
+        return self._generate_clis(
+            m, is_legacy_cli=True, for_legacy_file=for_legacy_file
+        )
+
+    def _generate_legacy_content(
+        self, m: Module, legacy_sdks: List[LegacySDK], legacy_models: List[LegacyModel]
+    ) -> str:
+        """Generate legacy documentation content for a module.
+        
+        Returns empty string if no legacy content exists.
+        """
+        legacy_cli = self._generate_legacy_clis(m, for_legacy_file=True)
+        legacy_sdk = self._generate_legacy_sdks(m, legacy_sdks, for_legacy_file=True)
+        legacy_model = self._generate_legacy_models(
+            m, legacy_models, for_legacy_file=True
+        )
+
+        # If no legacy content exists, return empty string
+        if not (legacy_cli or legacy_sdk or legacy_model):
+            return ""
+
+        # Build legacy documentation
+        output = "import Tabs from '@theme/Tabs';\n"
+        output += "import TabItem from '@theme/TabItem';\n\n"
+        output += f"# {m.legacy_title or m.title} API Reference (Legacy)\n\n"
+        output += ":::warning\n"
+        output += (
+            "These APIs are legacy and deprecated. Please use the [current APIs](../"
+            + m.filename
+            + ") instead.\n"
+        )
+        output += ":::\n\n"
+
+        output += legacy_cli
+        output += legacy_sdk
+        output += legacy_model
+
+        return output
+
+    def _transform_links_for_legacy_file(self, text: str) -> str:
+        """Transform links in legacy content for use in legacy files.
+        
+        This removes -legacy suffixes and adjusts cross-module references.
+        """
+        import re
+
+        # Remove -legacy suffix from local anchors
+        text = re.sub(r"\(#([a-z]+)-legacy\)", r"(#\1)", text)
+
+        # Transform cross-module references from file.md#anchor-legacy to file.md#anchor
+        text = re.sub(r"\(([a-z-]+\.md)#([a-z]+)-legacy\)", r"(\1#\2)", text)
+
+        return text
 
     def _get_anchor(self, t: ModelType):
         """Get a markdown anchor (link) to the given type's docs."""
@@ -575,7 +649,11 @@ class MarkdownGenerator:
         return md
 
     def _gen_markdown_for_legacy_sdk_command(
-        self, legacy_sdk: LegacySDK, new_sdk: Optional[Callable], sdk_prefix: str
+        self,
+        legacy_sdk: LegacySDK,
+        new_sdk: Optional[Callable],
+        sdk_prefix: str,
+        for_legacy_file: bool = False,
     ) -> str:
         """Generate a markdown section for a legacy SDK command.
 
@@ -584,29 +662,44 @@ class MarkdownGenerator:
             - Arguments (docstrings pulled from __arg_docstrings__ magic attribute)
             - Returns (if the return type annotation is not None)
         """
-        md = f"### `{legacy_sdk.name}` <span class='label-h3 label-legacy'>Legacy</span>\n"
-        if new_sdk:
-            md += ":::warning\n"
-            md += f"This command is deprecated. Upgrade to [{sdk_prefix}.{new_sdk.__name__}]({self._get_sdk_anchor(new_sdk, sdk_prefix)}). \n"
-        elif sdk_prefix == CLUSTER_SDK_PREFIX:
-            # Cluster SDK commands have special handling.
-            md += ":::warning[Upgrade recommended]\n"
-            md += "Cluster commands are deprecated. Use [workspaces](./workspaces.md), [jobs](./job-api.md), or [services](./service-api.md) APIs based on your specific needs.\n"
+        if for_legacy_file:
+            md = f"### `{legacy_sdk.name}`\n"
+            docstring = self._transform_links_for_legacy_file(legacy_sdk.docstring)
+            md += _escape_mdx_content(docstring) + "\n"
         else:
-            md += ":::warning[Limited support]\n"
-            md += "This command is not actively maintained. Use with caution.\n"
-        md += ":::\n"
-        md += _escape_mdx_content(legacy_sdk.docstring) + "\n"
+            md = f"### `{legacy_sdk.name}` <span class='label-h3 label-legacy'>Legacy</span>\n"
+            if new_sdk:
+                md += ":::warning\n"
+                md += f"This command is deprecated. Upgrade to [{sdk_prefix}.{new_sdk.__name__}]({self._get_sdk_anchor(new_sdk, sdk_prefix)}). \n"
+            elif sdk_prefix == CLUSTER_SDK_PREFIX:
+                # Cluster SDK commands have special handling.
+                md += ":::warning[Upgrade recommended]\n"
+                md += "Cluster commands are deprecated. Use [workspaces](./workspaces.md), [jobs](./job-api.md), or [services](./service-api.md) APIs based on your specific needs.\n"
+            else:
+                md += ":::warning[Limited support]\n"
+                md += "This command is not actively maintained. Use with caution.\n"
+            md += ":::\n"
+            md += _escape_mdx_content(legacy_sdk.docstring) + "\n"
 
         return md
 
-    def _gen_markdown_for_legacy_model(self, legacy_model: LegacyModel) -> str:
+    def _gen_markdown_for_legacy_model(
+        self, legacy_model: LegacyModel, for_legacy_file: bool = False
+    ) -> str:
         """Generate a markdown section for a legacy model.
 
         The sections will be:
             - All fields and their types
         """
-        md = f"### `{legacy_model.name}` <span class='label-h3 label-legacy'>Legacy</span> {{#{legacy_model.name.lower()}-legacy}}\n"
-        md += _escape_mdx_content(legacy_model.docstring) + "\n"
+        if for_legacy_file:
+            # In legacy files, don't use -legacy suffix for anchors
+            md = f"### `{legacy_model.name}` {{#{legacy_model.name.lower()}}}\n"
+            # Transform the docstring to remove -legacy suffixes from links
+            docstring = self._transform_links_for_legacy_file(legacy_model.docstring)
+            md += _escape_mdx_content(docstring) + "\n"
+        else:
+            # In main files, use -legacy suffix
+            md = f"### `{legacy_model.name}` <span class='label-h3 label-legacy'>Legacy</span> {{#{legacy_model.name.lower()}-legacy}}\n"
+            md += _escape_mdx_content(legacy_model.docstring) + "\n"
 
         return md

anyscale/_private/docgen/generator_legacy.py

@@ -42,8 +42,14 @@ def _build_model_to_module_mapping() -> Dict[str, str]:
     return model_name_to_file
 
 
-def _transform_legacy_links(text: str) -> str:
-    """Transform legacy model links to include proper cross-module references."""
+def _transform_legacy_links(text: str, for_legacy_file: bool = True) -> str:
+    """Transform legacy model links to include proper cross-module references.
+    
+    Args:
+        text: The text containing links to transform
+        for_legacy_file: If True, generate links for legacy files (in legacy/ subfolder).
+                        If False, generate links for main files with -legacy anchors.
+    """
     # Build the mapping dynamically
     model_mapping = _build_model_to_module_mapping()
 
@@ -51,20 +57,37 @@ def _transform_legacy_links(text: str) -> str:
         model_name = match.group(1)
         if model_name in model_mapping:
             module_file = model_mapping[model_name]
-            return f"({module_file}#{model_name}-legacy)"
+            if for_legacy_file:
+                # In legacy files, link to other legacy files without -legacy suffix
+                return f"({module_file}#{model_name})"
+            else:
+                # In main files, link with -legacy suffix
+                return f"({module_file}#{model_name}-legacy)"
         else:
             # Fallback for unmapped models (stay in same file)
-            return f"(#{model_name}-legacy)"
+            if for_legacy_file:
+                return f"(#{model_name})"
+            else:
+                return f"(#{model_name}-legacy)"
 
     # Transform links from (#modelname) to proper cross-module references
     text = re.sub(r"\(#([a-z]+)\)", replace_link, text)
 
-    # Transform workspace command references to point to workspaces.md
-    text = re.sub(
-        r"\(#anyscale-workspace_v2-([a-z]+)\)",
-        r"(workspaces.md#anyscale-workspace_v2-\1)",
-        text,
-    )
+    # Transform workspace command references
+    if for_legacy_file:
+        # In legacy files, point to legacy workspace file
+        text = re.sub(
+            r"\(#anyscale-workspace_v2-([a-z]+)\)",
+            r"(workspaces.md#anyscale-workspace_v2-\1)",
+            text,
+        )
+    else:
+        # In main files, point to main workspace file
+        text = re.sub(
+            r"\(#anyscale-workspace_v2-([a-z]+)\)",
+            r"(workspaces.md#anyscale-workspace_v2-\1)",
+            text,
+        )
 
     return text
 
@@ -88,7 +111,7 @@ class LegacySDK:
             else:
                 # First transform ./models.md links, then transform local links
                 line = re.sub(r"\(./models\.md#([a-z]+)\)", r"(#\1-legacy)", line)
-                line = _transform_legacy_links(line)
+                line = _transform_legacy_links(line, for_legacy_file=False)
                 docstring += line + "\n"
 
         return cls(name=name, docstring=docstring.strip())
@@ -111,7 +134,7 @@ class LegacyModel:
             if line.startswith("## "):
                 name = line[3:]
             else:
-                docstring += _transform_legacy_links(line) + "\n"
+                docstring += _transform_legacy_links(line, for_legacy_file=False) + "\n"
 
         return cls(name=name, docstring=docstring.strip())
 

anyscale/client/README.md

@@ -135,6 +135,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_workspace_from_template_api_v2_experimental_workspaces_from_template_post**](docs/DefaultApi.md#create_workspace_from_template_api_v2_experimental_workspaces_from_template_post) | **POST** /api/v2/experimental_workspaces/from_template | Create Workspace From Template
 *DefaultApi* | [**delete_cloud_api_v2_clouds_cloud_id_delete**](docs/DefaultApi.md#delete_cloud_api_v2_clouds_cloud_id_delete) | **DELETE** /api/v2/clouds/{cloud_id} | Delete Cloud
 *DefaultApi* | [**delete_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_identity_id_delete**](docs/DefaultApi.md#delete_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_identity_id_delete) | **DELETE** /api/v2/clouds/{cloud_id}/collaborators/{identity_id} | Delete Cloud Collaborator
+*DefaultApi* | [**delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete**](docs/DefaultApi.md#delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete) | **DELETE** /api/v2/integrations/connections/databricks/{connection_id} | Delete Databricks Connection
 *DefaultApi* | [**delete_hosted_cloud_api_v2_clouds_hosted_cloud_id_delete**](docs/DefaultApi.md#delete_hosted_cloud_api_v2_clouds_hosted_cloud_id_delete) | **DELETE** /api/v2/clouds/hosted/{cloud_id} | Delete Hosted Cloud
 *DefaultApi* | [**delete_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_delete**](docs/DefaultApi.md#delete_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_delete) | **DELETE** /api/v2/instance_usage_budgets/{instance_usage_budget_id} | Delete Instance Usage Budget
 *DefaultApi* | [**delete_machine_api_v2_machines_delete_post**](docs/DefaultApi.md#delete_machine_api_v2_machines_delete_post) | **POST** /api/v2/machines/delete | Delete Machine
@@ -161,6 +162,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**execute_interactive_command_api_v2_sessions_session_id_execute_interactive_command_post**](docs/DefaultApi.md#execute_interactive_command_api_v2_sessions_session_id_execute_interactive_command_post) | **POST** /api/v2/sessions/{session_id}/execute_interactive_command | Execute Interactive Command
 *DefaultApi* | [**fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post**](docs/DefaultApi.md#fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post) | **POST** /api/v2/aggregated_instance_usage/cloud | Fetch Usage Group By Cloud
 *DefaultApi* | [**fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post**](docs/DefaultApi.md#fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post) | **POST** /api/v2/aggregated_instance_usage/cluster | Fetch Usage Group By Cluster
+*DefaultApi* | [**fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post**](docs/DefaultApi.md#fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post) | **POST** /api/v2/aggregated_instance_usage/cluster_type | Fetch Usage Group By Cluster Type
 *DefaultApi* | [**fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post**](docs/DefaultApi.md#fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post) | **POST** /api/v2/aggregated_instance_usage/instance_type | Fetch Usage Group By Instance Type
 *DefaultApi* | [**fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post**](docs/DefaultApi.md#fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post) | **POST** /api/v2/aggregated_instance_usage/project | Fetch Usage Group By Project
 *DefaultApi* | [**fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post**](docs/DefaultApi.md#fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post) | **POST** /api/v2/aggregated_instance_usage/user | Fetch Usage Group By User
@@ -193,6 +195,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get**](docs/DefaultApi.md#get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get) | **GET** /api/v2/experimental_cron_jobs/{cron_job_id} | Get Cron Job
 *DefaultApi* | [**get_dags_api_v2_dataset_runs_dags_get**](docs/DefaultApi.md#get_dags_api_v2_dataset_runs_dags_get) | **GET** /api/v2/dataset_runs/dags | Get Dags
 *DefaultApi* | [**get_dashboard_api_v2_dataset_runs_get**](docs/DefaultApi.md#get_dashboard_api_v2_dataset_runs_get) | **GET** /api/v2/dataset_runs/ | Get Dashboard
+*DefaultApi* | [**get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get**](docs/DefaultApi.md#get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get) | **GET** /api/v2/integrations/connections/databricks/{connection_id} | Get Databricks Connection
 *DefaultApi* | [**get_dataset_api_v2_datasets_dataset_id_get**](docs/DefaultApi.md#get_dataset_api_v2_datasets_dataset_id_get) | **GET** /api/v2/datasets/{dataset_id} | Get Dataset
 *DefaultApi* | [**get_dataset_download_info_api_v2_datasets_download_info_get**](docs/DefaultApi.md#get_dataset_download_info_api_v2_datasets_download_info_get) | **GET** /api/v2/datasets/download_info | Get Dataset Download Info
 *DefaultApi* | [**get_dataset_download_url_api_v2_datasets_download_get**](docs/DefaultApi.md#get_dataset_download_url_api_v2_datasets_download_get) | **GET** /api/v2/datasets/download | Get Dataset Download Url
@@ -226,6 +229,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_model_api_v2_llm_models_model_id_get**](docs/DefaultApi.md#get_model_api_v2_llm_models_model_id_get) | **GET** /api/v2/llm/models/{model_id} | Get Model
 *DefaultApi* | [**get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get**](docs/DefaultApi.md#get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get) | **GET** /api/v2/llm/models/get_by_job_id/{job_id} | Get Model By Job Id
 *DefaultApi* | [**get_nodes_api_v2_cluster_dashboard_get**](docs/DefaultApi.md#get_nodes_api_v2_cluster_dashboard_get) | **GET** /api/v2/cluster_dashboard/ | Get Nodes
+*DefaultApi* | [**get_oauth_login_url_api_v2_integrations_oauth_login_get**](docs/DefaultApi.md#get_oauth_login_url_api_v2_integrations_oauth_login_get) | **GET** /api/v2/integrations/oauth/login | Get Oauth Login Url
 *DefaultApi* | [**get_operator_events_api_v2_dataset_runs_operator_events_get**](docs/DefaultApi.md#get_operator_events_api_v2_dataset_runs_operator_events_get) | **GET** /api/v2/dataset_runs/operator_events | Get Operator Events
 *DefaultApi* | [**get_operator_metrics_api_v2_dataset_runs_operator_get**](docs/DefaultApi.md#get_operator_metrics_api_v2_dataset_runs_operator_get) | **GET** /api/v2/dataset_runs/operator | Get Operator Metrics
 *DefaultApi* | [**get_or_create_build_from_image_uri_api_v2_builds_get_or_create_build_from_image_uri_post**](docs/DefaultApi.md#get_or_create_build_from_image_uri_api_v2_builds_get_or_create_build_from_image_uri_post) | **POST** /api/v2/builds/get_or_create_build_from_image_uri | Get Or Create Build From Image Uri
@@ -283,9 +287,12 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_api_keys_for_admin_api_v2_users_admin_api_keys_get**](docs/DefaultApi.md#list_api_keys_for_admin_api_v2_users_admin_api_keys_get) | **GET** /api/v2/users/admin/api_keys | List Api Keys For Admin
 *DefaultApi* | [**list_application_templates_api_v2_application_templates_get**](docs/DefaultApi.md#list_application_templates_api_v2_application_templates_get) | **GET** /api/v2/application_templates/ | List Application Templates
 *DefaultApi* | [**list_builds_api_v2_builds_get**](docs/DefaultApi.md#list_builds_api_v2_builds_get) | **GET** /api/v2/builds/ | List Builds
+*DefaultApi* | [**list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get**](docs/DefaultApi.md#list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get) | **GET** /api/v2/data_catalogs/connections/{connection_id}/catalogs | List Catalogs In Connection
 *DefaultApi* | [**list_clouds_api_v2_clouds_get**](docs/DefaultApi.md#list_clouds_api_v2_clouds_get) | **GET** /api/v2/clouds/ | List Clouds
 *DefaultApi* | [**list_configurations_api_v2_organization_configurations_get**](docs/DefaultApi.md#list_configurations_api_v2_organization_configurations_get) | **GET** /api/v2/organization_configurations/ | List Configurations
 *DefaultApi* | [**list_cron_jobs_api_v2_experimental_cron_jobs_get**](docs/DefaultApi.md#list_cron_jobs_api_v2_experimental_cron_jobs_get) | **GET** /api/v2/experimental_cron_jobs/ | List Cron Jobs
+*DefaultApi* | [**list_data_catalog_connections_api_v2_data_catalogs_connections_get**](docs/DefaultApi.md#list_data_catalog_connections_api_v2_data_catalogs_connections_get) | **GET** /api/v2/data_catalogs/connections | List Data Catalog Connections
+*DefaultApi* | [**list_databricks_connections_api_v2_integrations_connections_databricks_get**](docs/DefaultApi.md#list_databricks_connections_api_v2_integrations_connections_databricks_get) | **GET** /api/v2/integrations/connections/databricks | List Databricks Connections
 *DefaultApi* | [**list_dataset_versions_api_v2_datasets_versions_get**](docs/DefaultApi.md#list_dataset_versions_api_v2_datasets_versions_get) | **GET** /api/v2/datasets/versions | List Dataset Versions
 *DefaultApi* | [**list_datasets_api_v2_datasets_get**](docs/DefaultApi.md#list_datasets_api_v2_datasets_get) | **GET** /api/v2/datasets/ | List Datasets
 *DefaultApi* | [**list_decorated_clusters_api_v2_decorated_sessions_get**](docs/DefaultApi.md#list_decorated_clusters_api_v2_decorated_sessions_get) | **GET** /api/v2/decorated_sessions/ | List Decorated Clusters
@@ -318,6 +325,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_workspaces_api_v2_experimental_workspaces_get**](docs/DefaultApi.md#list_workspaces_api_v2_experimental_workspaces_get) | **GET** /api/v2/experimental_workspaces/ | List Workspaces
 *DefaultApi* | [**login_user_api_v2_users_login_post**](docs/DefaultApi.md#login_user_api_v2_users_login_post) | **POST** /api/v2/users/login | Login User
 *DefaultApi* | [**logout_user_api_v2_users_logout_post**](docs/DefaultApi.md#logout_user_api_v2_users_logout_post) | **POST** /api/v2/users/logout | Logout User
+*DefaultApi* | [**oauth_callback_api_v2_integrations_oauth_callback_get**](docs/DefaultApi.md#oauth_callback_api_v2_integrations_oauth_callback_get) | **GET** /api/v2/integrations/oauth/callback | Oauth Callback
 *DefaultApi* | [**patch_workspace_api_v2_experimental_workspaces_workspace_id_patch**](docs/DefaultApi.md#patch_workspace_api_v2_experimental_workspaces_workspace_id_patch) | **PATCH** /api/v2/experimental_workspaces/{workspace_id} | Patch Workspace
 *DefaultApi* | [**pause_cron_job_api_v2_experimental_cron_jobs_cron_job_id_pause_post**](docs/DefaultApi.md#pause_cron_job_api_v2_experimental_cron_jobs_cron_job_id_pause_post) | **POST** /api/v2/experimental_cron_jobs/{cron_job_id}/pause | Pause Cron Job
 *DefaultApi* | [**produce_analytics_event_api_v2_analytics_post**](docs/DefaultApi.md#produce_analytics_event_api_v2_analytics_post) | **POST** /api/v2/analytics/ | Produce Analytics Event
@@ -337,6 +345,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**redirect_to_tools_api_v2_services_v2_service_id_tools_get**](docs/DefaultApi.md#redirect_to_tools_api_v2_services_v2_service_id_tools_get) | **GET** /api/v2/services-v2/{service_id}/tools | Redirect To Tools
 *DefaultApi* | [**register_api_v2_kubernetes_manager_register_post**](docs/DefaultApi.md#register_api_v2_kubernetes_manager_register_post) | **POST** /api/v2/kubernetes_manager/register | Register
 *DefaultApi* | [**register_build_api_v2_anyscaled_register_build_post**](docs/DefaultApi.md#register_build_api_v2_anyscaled_register_build_post) | **POST** /api/v2/anyscaled/register_build | Register Build
+*DefaultApi* | [**register_databricks_connection_api_v2_integrations_connections_databricks_post**](docs/DefaultApi.md#register_databricks_connection_api_v2_integrations_connections_databricks_post) | **POST** /api/v2/integrations/connections/databricks | Register Databricks Connection
 *DefaultApi* | [**register_machine_api_v2_anyscaled_register_machine_post**](docs/DefaultApi.md#register_machine_api_v2_anyscaled_register_machine_post) | **POST** /api/v2/anyscaled/register_machine | Register Machine
 *DefaultApi* | [**register_user_api_v2_users_post**](docs/DefaultApi.md#register_user_api_v2_users_post) | **POST** /api/v2/users/ | Register User
 *DefaultApi* | [**register_v2_api_v2_anyscaled_register_v2_post**](docs/DefaultApi.md#register_v2_api_v2_anyscaled_register_v2_post) | **POST** /api/v2/anyscaled/register_v2 | Register V2
@@ -364,6 +373,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**search_tags_api_v2_tags_get**](docs/DefaultApi.md#search_tags_api_v2_tags_get) | **GET** /api/v2/tags/ | Search Tags
 *DefaultApi* | [**set_resource_quota_status_api_v2_resource_quotas_resource_quota_id_status_patch**](docs/DefaultApi.md#set_resource_quota_status_api_v2_resource_quotas_resource_quota_id_status_patch) | **PATCH** /api/v2/resource_quotas/{resource_quota_id}/status | Set Resource Quota Status
 *DefaultApi* | [**show_one_time_password_source_api_v2_users_show_otp_source_post**](docs/DefaultApi.md#show_one_time_password_source_api_v2_users_show_otp_source_post) | **POST** /api/v2/users/show_otp_source | Show One Time Password Source
+*DefaultApi* | [**sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get**](docs/DefaultApi.md#sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get) | **GET** /api/v2/organizations/{organization_id}/sso_admin_link | Sso Admin Link
 *DefaultApi* | [**sso_login_info_api_v2_users_sso_login_info_get**](docs/DefaultApi.md#sso_login_info_api_v2_users_sso_login_info_get) | **GET** /api/v2/users/sso_login_info | Sso Login Info
 *DefaultApi* | [**sso_login_test_api_v2_users_sso_login_test_get**](docs/DefaultApi.md#sso_login_test_api_v2_users_sso_login_test_get) | **GET** /api/v2/users/sso_login_test | Sso Login Test
 *DefaultApi* | [**start_session_api_v2_sessions_session_id_start_post**](docs/DefaultApi.md#start_session_api_v2_sessions_session_id_start_post) | **POST** /api/v2/sessions/{session_id}/start | Start Session
@@ -381,7 +391,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post**](docs/DefaultApi.md#try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post) | **POST** /api/v2/aioa_cloud_waitlist/claim_cloud | Try Claim Cloud
 *DefaultApi* | [**try_login_api_v2_users_try_login_post**](docs/DefaultApi.md#try_login_api_v2_users_try_login_post) | **POST** /api/v2/users/try-login | Try Login
 *DefaultApi* | [**update_cloud_auto_add_user_api_v2_clouds_cloud_id_auto_add_user_put**](docs/DefaultApi.md#update_cloud_auto_add_user_api_v2_clouds_cloud_id_auto_add_user_put) | **PUT** /api/v2/clouds/{cloud_id}/auto_add_user | Update Cloud Auto Add User
-*DefaultApi* | [**update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put**](docs/DefaultApi.md#update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put) | **PUT** /api/v2/clouds/{cloud_id}/update_deployment | Update Cloud Deployment
 *DefaultApi* | [**update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put**](docs/DefaultApi.md#update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put) | **PUT** /api/v2/clouds/{cloud_id}/deployment/{cloud_deployment_id}/config | Update Cloud Deployment Config
 *DefaultApi* | [**update_cloud_resources_api_v2_clouds_cloud_id_resources_put**](docs/DefaultApi.md#update_cloud_resources_api_v2_clouds_cloud_id_resources_put) | **PUT** /api/v2/clouds/{cloud_id}/resources | Update Cloud Resources
 *DefaultApi* | [**update_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_put**](docs/DefaultApi.md#update_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_put) | **PUT** /api/v2/clouds_with_cloud_resource_gcp_router/{cloud_id} | Update Cloud With Cloud Resource
@@ -397,6 +406,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**update_resource_notification_api_v2_resource_notifications_resource_notification_id_put**](docs/DefaultApi.md#update_resource_notification_api_v2_resource_notifications_resource_notification_id_put) | **PUT** /api/v2/resource_notifications/{resource_notification_id} | Update Resource Notification
 *DefaultApi* | [**update_resource_quota_api_v2_resource_quotas_resource_quota_id_patch**](docs/DefaultApi.md#update_resource_quota_api_v2_resource_quotas_resource_quota_id_patch) | **PATCH** /api/v2/resource_quotas/{resource_quota_id} | Update Resource Quota
 *DefaultApi* | [**update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put**](docs/DefaultApi.md#update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put) | **PUT** /api/v2/clouds/{cloud_id}/update_system_cluster_config | Update System Cluster Config
+*DefaultApi* | [**update_template_api_v2_workspace_templates_template_id_put**](docs/DefaultApi.md#update_template_api_v2_workspace_templates_template_id_put) | **PUT** /api/v2/workspace_templates/{template_id} | Update Template
 *DefaultApi* | [**upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post**](docs/DefaultApi.md#upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post) | **POST** /api/v2/session_commands/{session_command_id}/upload_logs | Upload Session Command Logs
 *DefaultApi* | [**upsert_billing_version_api_v2_organization_billing_billing_versions_put**](docs/DefaultApi.md#upsert_billing_version_api_v2_organization_billing_billing_versions_put) | **PUT** /api/v2/organization_billing/billing_versions | Upsert Billing Version
 *DefaultApi* | [**upsert_resource_tags_api_v2_tags_resource_put**](docs/DefaultApi.md#upsert_resource_tags_api_v2_tags_resource_put) | **PUT** /api/v2/tags/resource | Upsert Resource Tags
@@ -550,6 +560,7 @@ Class | Method | HTTP request | Description
  - [ComputeTemplateQuery](docs/ComputeTemplateQuery.md)
  - [ComputetemplateResponse](docs/ComputetemplateResponse.md)
  - [ComputetemplateconfigResponse](docs/ComputetemplateconfigResponse.md)
+ - [ConnectionType](docs/ConnectionType.md)
  - [CreateAioaCloudWaitlist](docs/CreateAioaCloudWaitlist.md)
  - [CreateAnalyticsEvent](docs/CreateAnalyticsEvent.md)
  - [CreateAppConfig](docs/CreateAppConfig.md)
@@ -605,6 +616,18 @@ Class | Method | HTTP request | Description
  - [CreditsV2](docs/CreditsV2.md)
  - [CustomerAlertStatus](docs/CustomerAlertStatus.md)
  - [DashboardPanel](docs/DashboardPanel.md)
+ - [DataCatalog](docs/DataCatalog.md)
+ - [DataCatalogConnection](docs/DataCatalogConnection.md)
+ - [DataCatalogConnectionStatus](docs/DataCatalogConnectionStatus.md)
+ - [DataCatalogProvider](docs/DataCatalogProvider.md)
+ - [DatabricksConnectionConfig](docs/DatabricksConnectionConfig.md)
+ - [DatabricksConnectionInfo](docs/DatabricksConnectionInfo.md)
+ - [DatabricksConnectionResponse](docs/DatabricksConnectionResponse.md)
+ - [DatabricksRegisterRequest](docs/DatabricksRegisterRequest.md)
+ - [DatabricksconnectioninfoResponse](docs/DatabricksconnectioninfoResponse.md)
+ - [DatabricksconnectionresponseResponse](docs/DatabricksconnectionresponseResponse.md)
+ - [DatacatalogListResponse](docs/DatacatalogListResponse.md)
+ - [DatacatalogconnectionListResponse](docs/DatacatalogconnectionListResponse.md)
  - [DataplaneServices](docs/DataplaneServices.md)
  - [Dataset](docs/Dataset.md)
  - [DatasetDag](docs/DatasetDag.md)
@@ -684,6 +707,7 @@ Class | Method | HTTP request | Description
  - [DescribesystemworkloadresponseResponse](docs/DescribesystemworkloadresponseResponse.md)
  - [DetachMachinePoolFromCloudRequest](docs/DetachMachinePoolFromCloudRequest.md)
  - [DetachmachinepoolfromcloudresponseResponse](docs/DetachmachinepoolfromcloudresponseResponse.md)
+ - [DomainVerification](docs/DomainVerification.md)
  - [EditableCloudResource](docs/EditableCloudResource.md)
  - [EditableCloudResourceGCP](docs/EditableCloudResourceGCP.md)
  - [EmailVerificationRequest](docs/EmailVerificationRequest.md)
@@ -760,6 +784,7 @@ Class | Method | HTTP request | Description
  - [KubernetesmanagerregistrationresponseResponse](docs/KubernetesmanagerregistrationresponseResponse.md)
  - [LBResource](docs/LBResource.md)
  - [LbresourceResponse](docs/LbresourceResponse.md)
+ - [ListDatabricksConnections](docs/ListDatabricksConnections.md)
  - [ListMachinePoolsResponse](docs/ListMachinePoolsResponse.md)
  - [ListMachinesResponse](docs/ListMachinesResponse.md)
  - [ListRaySessionsResponse](docs/ListRaySessionsResponse.md)
@@ -830,6 +855,9 @@ Class | Method | HTTP request | Description
  - [NotificationChannelEmailConfig](docs/NotificationChannelEmailConfig.md)
  - [NotificationChannelSlackConfig](docs/NotificationChannelSlackConfig.md)
  - [NotificationChannelWebhookConfig](docs/NotificationChannelWebhookConfig.md)
+ - [OAuthConnectionResponse](docs/OAuthConnectionResponse.md)
+ - [OauthAuthUrlResponse](docs/OauthAuthUrlResponse.md)
+ - [OauthconnectionresponseResponse](docs/OauthconnectionresponseResponse.md)
  - [ObjectStorage](docs/ObjectStorage.md)
  - [OperationError](docs/OperationError.md)
  - [OperationProgress](docs/OperationProgress.md)
@@ -926,6 +954,9 @@ Class | Method | HTTP request | Description
  - [RunAttemptStatus](docs/RunAttemptStatus.md)
  - [RunStatus](docs/RunStatus.md)
  - [S3DownloadLocation](docs/S3DownloadLocation.md)
+ - [SSOConfig](docs/SSOConfig.md)
+ - [SSOConnection](docs/SSOConnection.md)
+ - [SSOConnectionState](docs/SSOConnectionState.md)
  - [SSOLoginInfo](docs/SSOLoginInfo.md)
  - [SSOMode](docs/SSOMode.md)
  - [SUPPORTEDBASEIMAGESENUM](docs/SUPPORTEDBASEIMAGESENUM.md)
@@ -969,6 +1000,7 @@ Class | Method | HTTP request | Description
  - [ShowOTPSourceReturnApiModel](docs/ShowOTPSourceReturnApiModel.md)
  - [ShowotpsourcereturnapimodelResponse](docs/ShowotpsourcereturnapimodelResponse.md)
  - [SortOrder](docs/SortOrder.md)
+ - [SsoconfigResponse](docs/SsoconfigResponse.md)
  - [SsologininfoResponse](docs/SsologininfoResponse.md)
  - [StartSessionOptions](docs/StartSessionOptions.md)
  - [StopSessionOptions](docs/StopSessionOptions.md)
@@ -1026,17 +1058,20 @@ Class | Method | HTTP request | Description
  - [UpdatePaymentInfo](docs/UpdatePaymentInfo.md)
  - [UpdateProjectCollaborator](docs/UpdateProjectCollaborator.md)
  - [UpdateResourceQuota](docs/UpdateResourceQuota.md)
+ - [UpdateWorkspaceTemplate](docs/UpdateWorkspaceTemplate.md)
  - [UpdatemachinepoolresponseResponse](docs/UpdatemachinepoolresponseResponse.md)
  - [UploadSessionCommandLogsLocations](docs/UploadSessionCommandLogsLocations.md)
  - [UploadsessioncommandlogslocationsResponse](docs/UploadsessioncommandlogslocationsResponse.md)
  - [UpsertResourceTagsRequest](docs/UpsertResourceTagsRequest.md)
  - [UsageByCloud](docs/UsageByCloud.md)
  - [UsageByCluster](docs/UsageByCluster.md)
+ - [UsageByClusterType](docs/UsageByClusterType.md)
  - [UsageByInstanceType](docs/UsageByInstanceType.md)
  - [UsageByProject](docs/UsageByProject.md)
  - [UsageByUser](docs/UsageByUser.md)
  - [UsagebycloudListResponse](docs/UsagebycloudListResponse.md)
  - [UsagebyclusterListResponse](docs/UsagebyclusterListResponse.md)
+ - [UsagebyclustertypeListResponse](docs/UsagebyclustertypeListResponse.md)
  - [UsagebyinstancetypeListResponse](docs/UsagebyinstancetypeListResponse.md)
  - [UsagebyprojectListResponse](docs/UsagebyprojectListResponse.md)
  - [UsagebyuserListResponse](docs/UsagebyuserListResponse.md)
@@ -1049,6 +1084,7 @@ Class | Method | HTTP request | Description
  - [UtmFields](docs/UtmFields.md)
  - [ValidateOTPParamsApiModel](docs/ValidateOTPParamsApiModel.md)
  - [ValidationError](docs/ValidationError.md)
+ - [ValidationStatus](docs/ValidationStatus.md)
  - [WaitlistStatusResponse](docs/WaitlistStatusResponse.md)
  - [WaitlistStatusType](docs/WaitlistStatusType.md)
  - [WaitliststatusresponseResponse](docs/WaitliststatusresponseResponse.md)