@databricks/sdk-modelregistry 0.0.0-dev → 0.1.0-dev.2

package/dist/v1/model.js

@@ -0,0 +1,950 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+import { z } from 'zod';
+/**
+ * An action that a user (with sufficient permissions) could take on an activity or comment.
+ *
+ * For activities, valid values are:
+ * * `APPROVE_TRANSITION_REQUEST`: Approve a transition request
+ *
+ * * `REJECT_TRANSITION_REQUEST`: Reject a transition request
+ *
+ * * `CANCEL_TRANSITION_REQUEST`: Cancel (delete) a transition request
+ *
+ * For comments, valid values are:
+ * * `EDIT_COMMENT`: Edit the comment
+ *
+ * * `DELETE_COMMENT`: Delete the comment
+ */
+export var ActivityAction;
+(function (ActivityAction) {
+    /** Approve a transition request. Available to users with sufficient permissions. */
+    ActivityAction["APPROVE_TRANSITION_REQUEST"] = "APPROVE_TRANSITION_REQUEST";
+    /** Reject a transition request. Available to users with sufficient permissions. */
+    ActivityAction["REJECT_TRANSITION_REQUEST"] = "REJECT_TRANSITION_REQUEST";
+    /** Cancel a transition request. Available to the user who created the request. */
+    ActivityAction["CANCEL_TRANSITION_REQUEST"] = "CANCEL_TRANSITION_REQUEST";
+    /** Edit the comment */
+    ActivityAction["EDIT_COMMENT"] = "EDIT_COMMENT";
+    /** Delete the comment */
+    ActivityAction["DELETE_COMMENT"] = "DELETE_COMMENT";
+})(ActivityAction || (ActivityAction = {}));
+/**
+ * Type of activity. Valid values are:
+ * * `APPLIED_TRANSITION`: User applied the corresponding stage transition.
+ *
+ * * `REQUESTED_TRANSITION`: User requested the corresponding stage transition.
+ *
+ * * `CANCELLED_REQUEST`: User cancelled an existing transition request.
+ *
+ * * `APPROVED_REQUEST`: User approved the corresponding stage transition.
+ *
+ * * `REJECTED_REQUEST`: User rejected the coressponding stage transition.
+ *
+ * * `SYSTEM_TRANSITION`: For events performed as a side effect, such as archiving existing model versions in a stage.
+ */
+export var ActivityType;
+(function (ActivityType) {
+    /** Indicates that the corresponding stage transition was applied by user. */
+    ActivityType["APPLIED_TRANSITION"] = "APPLIED_TRANSITION";
+    /** Corresponding stage transition was requested by user. */
+    ActivityType["REQUESTED_TRANSITION"] = "REQUESTED_TRANSITION";
+    /** User cancelled an existing request. */
+    ActivityType["CANCELLED_REQUEST"] = "CANCELLED_REQUEST";
+    /** Corresponding transition request was approved by user. */
+    ActivityType["APPROVED_REQUEST"] = "APPROVED_REQUEST";
+    /** Corresponding transition request was rejected by user. */
+    ActivityType["REJECTED_REQUEST"] = "REJECTED_REQUEST";
+    /** User posted a new comment */
+    ActivityType["NEW_COMMENT"] = "NEW_COMMENT";
+    /** Corresponding transition for events such as archiving existing model versions */
+    ActivityType["SYSTEM_TRANSITION"] = "SYSTEM_TRANSITION";
+})(ActivityType || (ActivityType = {}));
+/**
+ * The status of the model version. Valid values are:
+ * * `PENDING_REGISTRATION`: Request to register a new model version is pending as server performs background tasks.
+ *
+ * * `FAILED_REGISTRATION`: Request to register a new model version has failed.
+ *
+ * * `READY`: Model version is ready for use.
+ */
+export var ModelVersionStatus;
+(function (ModelVersionStatus) {
+    /** Request to register a new model version is pending as server performs background tasks. */
+    ModelVersionStatus["PENDING_REGISTRATION"] = "PENDING_REGISTRATION";
+    /** Request to register a new model version has failed. */
+    ModelVersionStatus["FAILED_REGISTRATION"] = "FAILED_REGISTRATION";
+    /** Model version is ready for use. */
+    ModelVersionStatus["READY"] = "READY";
+})(ModelVersionStatus || (ModelVersionStatus = {}));
+/** Permission level of the requesting user on the object. For what is allowed at each level, see [MLflow Model permissions](..). */
+export var PermissionLevel;
+(function (PermissionLevel) {
+    PermissionLevel["CAN_MANAGE"] = "CAN_MANAGE";
+    /** reserved 1;  // IS_OWNER = 1; was DEPRECATED */
+    PermissionLevel["CAN_EDIT"] = "CAN_EDIT";
+    PermissionLevel["CAN_READ"] = "CAN_READ";
+    PermissionLevel["CAN_MANAGE_STAGING_VERSIONS"] = "CAN_MANAGE_STAGING_VERSIONS";
+    PermissionLevel["CAN_MANAGE_PRODUCTION_VERSIONS"] = "CAN_MANAGE_PRODUCTION_VERSIONS";
+    /**
+     * Only applicable to the root ACL path, for which it is the default value if no permissions are
+     * set explicitly for the user. It is the default set by the MLflow service and The ACL database
+     * does not understand this value.
+     */
+    PermissionLevel["CAN_CREATE_REGISTERED_MODEL"] = "CAN_CREATE_REGISTERED_MODEL";
+})(PermissionLevel || (PermissionLevel = {}));
+/**
+ * .. note::
+ * Experimental: This entity may change or be removed in a future release without warning.
+ * Email subscription types for registry notifications:
+ * - `ALL_EVENTS`: Subscribed to all events.
+ * - `DEFAULT`: Default subscription type.
+ * - `SUBSCRIBED`: Subscribed to notifications.
+ * - `UNSUBSCRIBED`: Not subscribed to notifications.
+ */
+export var RegistryEmailSubscriptionType;
+(function (RegistryEmailSubscriptionType) {
+    RegistryEmailSubscriptionType["ALL_EVENTS"] = "ALL_EVENTS";
+    RegistryEmailSubscriptionType["DEFAULT"] = "DEFAULT";
+    RegistryEmailSubscriptionType["SUBSCRIBED"] = "SUBSCRIBED";
+    RegistryEmailSubscriptionType["UNSUBSCRIBED"] = "UNSUBSCRIBED";
+})(RegistryEmailSubscriptionType || (RegistryEmailSubscriptionType = {}));
+export var RegistryWebhookEvent;
+(function (RegistryWebhookEvent) {
+    RegistryWebhookEvent["MODEL_VERSION_CREATED"] = "MODEL_VERSION_CREATED";
+    RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_STAGE"] = "MODEL_VERSION_TRANSITIONED_STAGE";
+    RegistryWebhookEvent["TRANSITION_REQUEST_CREATED"] = "TRANSITION_REQUEST_CREATED";
+    RegistryWebhookEvent["COMMENT_CREATED"] = "COMMENT_CREATED";
+    RegistryWebhookEvent["REGISTERED_MODEL_CREATED"] = "REGISTERED_MODEL_CREATED";
+    RegistryWebhookEvent["MODEL_VERSION_TAG_SET"] = "MODEL_VERSION_TAG_SET";
+    RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_TO_STAGING"] = "MODEL_VERSION_TRANSITIONED_TO_STAGING";
+    RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_TO_PRODUCTION"] = "MODEL_VERSION_TRANSITIONED_TO_PRODUCTION";
+    RegistryWebhookEvent["MODEL_VERSION_TRANSITIONED_TO_ARCHIVED"] = "MODEL_VERSION_TRANSITIONED_TO_ARCHIVED";
+    RegistryWebhookEvent["TRANSITION_REQUEST_TO_STAGING_CREATED"] = "TRANSITION_REQUEST_TO_STAGING_CREATED";
+    RegistryWebhookEvent["TRANSITION_REQUEST_TO_PRODUCTION_CREATED"] = "TRANSITION_REQUEST_TO_PRODUCTION_CREATED";
+    RegistryWebhookEvent["TRANSITION_REQUEST_TO_ARCHIVED_CREATED"] = "TRANSITION_REQUEST_TO_ARCHIVED_CREATED";
+})(RegistryWebhookEvent || (RegistryWebhookEvent = {}));
+/**
+ * Enable or disable triggering the webhook, or put the webhook into test mode. The default is `ACTIVE`:
+ * * `ACTIVE`: Webhook is triggered when an associated event happens.
+ *
+ * * `DISABLED`: Webhook is not triggered.
+ *
+ * * `TEST_MODE`: Webhook can be triggered through the test endpoint, but is not triggered on a real event.
+ */
+export var RegistryWebhookStatus;
+(function (RegistryWebhookStatus) {
+    /** Event and test triggers will be sent. */
+    RegistryWebhookStatus["ACTIVE"] = "ACTIVE";
+    /** No triggers will be sent. */
+    RegistryWebhookStatus["DISABLED"] = "DISABLED";
+    /** Test triggers will be sent, but not actual events. */
+    RegistryWebhookStatus["TEST_MODE"] = "TEST_MODE";
+})(RegistryWebhookStatus || (RegistryWebhookStatus = {}));
+export const unmarshalActivitySchema = z
+    .object({
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    activity_type: z.enum(ActivityType).optional(),
+    comment: z.string().optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    from_stage: z.string().optional(),
+    to_stage: z.string().optional(),
+    system_comment: z.string().optional(),
+    available_actions: z.array(z.enum(ActivityAction)).optional(),
+    id: z.string().optional(),
+})
+    .transform(d => ({
+    creationTimestamp: d.creation_timestamp,
+    userId: d.user_id,
+    activityType: d.activity_type,
+    comment: d.comment,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    fromStage: d.from_stage,
+    toStage: d.to_stage,
+    systemComment: d.system_comment,
+    availableActions: d.available_actions,
+    id: d.id,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalApproveTransitionRequest_ResponseSchema = z
+    .object({
+    activity: z.lazy(() => unmarshalActivitySchema).optional(),
+})
+    .transform(d => ({
+    activity: d.activity,
+}));
+export const unmarshalCommentObjectSchema = z
+    .object({
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    activity_type: z.enum(ActivityType).optional(),
+    comment: z.string().optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    from_stage: z.string().optional(),
+    to_stage: z.string().optional(),
+    system_comment: z.string().optional(),
+    available_actions: z.array(z.enum(ActivityAction)).optional(),
+    id: z.string().optional(),
+})
+    .transform(d => ({
+    creationTimestamp: d.creation_timestamp,
+    userId: d.user_id,
+    activityType: d.activity_type,
+    comment: d.comment,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    fromStage: d.from_stage,
+    toStage: d.to_stage,
+    systemComment: d.system_comment,
+    availableActions: d.available_actions,
+    id: d.id,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalCreateCommentRequest_ResponseSchema = z
+    .object({
+    comment: z.lazy(() => unmarshalCommentObjectSchema).optional(),
+})
+    .transform(d => ({
+    comment: d.comment,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalCreateModelVersionRequest_ResponseSchema = z
+    .object({
+    model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
+})
+    .transform(d => ({
+    modelVersion: d.model_version,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalCreateRegisteredModelRequest_ResponseSchema = z
+    .object({
+    registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
+})
+    .transform(d => ({
+    registeredModel: d.registered_model,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalCreateRegistryWebhookRequest_ResponseSchema = z
+    .object({
+    webhook: z.lazy(() => unmarshalRegistryWebhookSchema).optional(),
+})
+    .transform(d => ({
+    webhook: d.webhook,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalCreateTransitionRequest_ResponseSchema = z
+    .object({
+    request: z.lazy(() => unmarshalTransitionRequestSchema).optional(),
+})
+    .transform(d => ({
+    request: d.request,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteCommentRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteModelVersionRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteModelVersionTagRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteRegisteredModelRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteRegisteredModelTagRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteRegistryWebhookRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalDeleteTransitionRequest_ResponseSchema = z
+    .object({
+    activity: z.lazy(() => unmarshalActivitySchema).optional(),
+})
+    .transform(d => ({
+    activity: d.activity,
+}));
+export const unmarshalFeatureListSchema = z
+    .object({
+    features: z.array(z.lazy(() => unmarshalLinkedFeatureSchema)).optional(),
+})
+    .transform(d => ({
+    features: d.features,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalGetModelVersionDownloadUriRequest_ResponseSchema = z
+    .object({
+    artifact_uri: z.string().optional(),
+})
+    .transform(d => ({
+    artifactUri: d.artifact_uri,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalGetModelVersionRequest_ResponseSchema = z
+    .object({
+    model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
+})
+    .transform(d => ({
+    modelVersion: d.model_version,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalGetRegisteredModelDatabricksRequest_ResponseSchema = z
+    .object({
+    registered_model_databricks: z
+        .lazy(() => unmarshalRegisteredModelDatabricksSchema)
+        .optional(),
+})
+    .transform(d => ({
+    registeredModelDatabricks: d.registered_model_databricks,
+}));
+export const unmarshalHttpUrlSpecSchema = z
+    .object({
+    url: z.string().optional(),
+    enable_ssl_verification: z.boolean().optional(),
+    secret: z.string().optional(),
+    authorization: z.string().optional(),
+})
+    .transform(d => ({
+    url: d.url,
+    enableSslVerification: d.enable_ssl_verification,
+    secret: d.secret,
+    authorization: d.authorization,
+}));
+export const unmarshalJobSpecSchema = z
+    .object({
+    job_id: z.string().optional(),
+    workspace_url: z.string().optional(),
+    access_token: z.string().optional(),
+})
+    .transform(d => ({
+    jobId: d.job_id,
+    workspaceUrl: d.workspace_url,
+    accessToken: d.access_token,
+}));
+export const unmarshalLinkedFeatureSchema = z
+    .object({
+    feature_table_name: z.string().optional(),
+    feature_name: z.string().optional(),
+    feature_table_id: z.string().optional(),
+})
+    .transform(d => ({
+    featureTableName: d.feature_table_name,
+    featureName: d.feature_name,
+    featureTableId: d.feature_table_id,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalListLatestVersionsRequest_ResponseSchema = z
+    .object({
+    model_versions: z
+        .array(z.lazy(() => unmarshalModelVersionSchema))
+        .optional(),
+})
+    .transform(d => ({
+    modelVersions: d.model_versions,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalListRegisteredModelsRequest_ResponseSchema = z
+    .object({
+    registered_models: z
+        .array(z.lazy(() => unmarshalRegisteredModelSchema))
+        .optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    registeredModels: d.registered_models,
+    nextPageToken: d.next_page_token,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalListRegistryWebhooksRequest_ResponseSchema = z
+    .object({
+    webhooks: z
+        .array(z.lazy(() => unmarshalRegistryWebhookSchema))
+        .optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    webhooks: d.webhooks,
+    nextPageToken: d.next_page_token,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalListTransitionRequest_ResponseSchema = z
+    .object({
+    requests: z.array(z.lazy(() => unmarshalActivitySchema)).optional(),
+})
+    .transform(d => ({
+    requests: d.requests,
+}));
+export const unmarshalModelVersionSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    current_stage: z.string().optional(),
+    description: z.string().optional(),
+    source: z.string().optional(),
+    run_id: z.string().optional(),
+    status: z.enum(ModelVersionStatus).optional(),
+    status_message: z.string().optional(),
+    tags: z.array(z.lazy(() => unmarshalModelVersionTagSchema)).optional(),
+    run_link: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    creationTimestamp: d.creation_timestamp,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    userId: d.user_id,
+    currentStage: d.current_stage,
+    description: d.description,
+    source: d.source,
+    runId: d.run_id,
+    status: d.status,
+    statusMessage: d.status_message,
+    tags: d.tags,
+    runLink: d.run_link,
+}));
+export const unmarshalModelVersionDatabricksSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    current_stage: z.string().optional(),
+    description: z.string().optional(),
+    source: z.string().optional(),
+    run_id: z.string().optional(),
+    status: z.enum(ModelVersionStatus).optional(),
+    status_message: z.string().optional(),
+    open_requests: z.array(z.lazy(() => unmarshalActivitySchema)).optional(),
+    permission_level: z.enum(PermissionLevel).optional(),
+    tags: z.array(z.lazy(() => unmarshalModelVersionTagSchema)).optional(),
+    run_link: z.string().optional(),
+    email_subscription_status: z
+        .enum(RegistryEmailSubscriptionType)
+        .optional(),
+    feature_list: z.lazy(() => unmarshalFeatureListSchema).optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    creationTimestamp: d.creation_timestamp,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    userId: d.user_id,
+    currentStage: d.current_stage,
+    description: d.description,
+    source: d.source,
+    runId: d.run_id,
+    status: d.status,
+    statusMessage: d.status_message,
+    openRequests: d.open_requests,
+    permissionLevel: d.permission_level,
+    tags: d.tags,
+    runLink: d.run_link,
+    emailSubscriptionStatus: d.email_subscription_status,
+    featureList: d.feature_list,
+}));
+export const unmarshalModelVersionTagSchema = z
+    .object({
+    key: z.string().optional(),
+    value: z.string().optional(),
+})
+    .transform(d => ({
+    key: d.key,
+    value: d.value,
+}));
+export const unmarshalRegisteredModelSchema = z
+    .object({
+    name: z.string().optional(),
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    description: z.string().optional(),
+    latest_versions: z
+        .array(z.lazy(() => unmarshalModelVersionSchema))
+        .optional(),
+    tags: z.array(z.lazy(() => unmarshalRegisteredModelTagSchema)).optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    creationTimestamp: d.creation_timestamp,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    userId: d.user_id,
+    description: d.description,
+    latestVersions: d.latest_versions,
+    tags: d.tags,
+}));
+export const unmarshalRegisteredModelDatabricksSchema = z
+    .object({
+    name: z.string().optional(),
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    description: z.string().optional(),
+    latest_versions: z
+        .array(z.lazy(() => unmarshalModelVersionSchema))
+        .optional(),
+    id: z.string().optional(),
+    permission_level: z.enum(PermissionLevel).optional(),
+    tags: z.array(z.lazy(() => unmarshalRegisteredModelTagSchema)).optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    creationTimestamp: d.creation_timestamp,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    userId: d.user_id,
+    description: d.description,
+    latestVersions: d.latest_versions,
+    id: d.id,
+    permissionLevel: d.permission_level,
+    tags: d.tags,
+}));
+export const unmarshalRegisteredModelTagSchema = z
+    .object({
+    key: z.string().optional(),
+    value: z.string().optional(),
+})
+    .transform(d => ({
+    key: d.key,
+    value: d.value,
+}));
+export const unmarshalRegistryWebhookSchema = z
+    .object({
+    id: z.string().optional(),
+    events: z.array(z.enum(RegistryWebhookEvent)).optional(),
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    description: z.string().optional(),
+    status: z.enum(RegistryWebhookStatus).optional(),
+    http_url_spec: z.lazy(() => unmarshalHttpUrlSpecSchema).optional(),
+    job_spec: z.lazy(() => unmarshalJobSpecSchema).optional(),
+    model_name: z.string().optional(),
+})
+    .transform(d => ({
+    id: d.id,
+    events: d.events,
+    creationTimestamp: d.creation_timestamp,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    description: d.description,
+    status: d.status,
+    httpUrlSpec: d.http_url_spec,
+    jobSpec: d.job_spec,
+    modelName: d.model_name,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalRejectTransitionRequest_ResponseSchema = z
+    .object({
+    activity: z.lazy(() => unmarshalActivitySchema).optional(),
+})
+    .transform(d => ({
+    activity: d.activity,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalRenameRegisteredModelRequest_ResponseSchema = z
+    .object({
+    registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
+})
+    .transform(d => ({
+    registeredModel: d.registered_model,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalSearchModelVersionsRequest_ResponseSchema = z
+    .object({
+    model_versions: z
+        .array(z.lazy(() => unmarshalModelVersionSchema))
+        .optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    modelVersions: d.model_versions,
+    nextPageToken: d.next_page_token,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalSearchRegisteredModelsRequest_ResponseSchema = z
+    .object({
+    registered_models: z
+        .array(z.lazy(() => unmarshalRegisteredModelSchema))
+        .optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    registeredModels: d.registered_models,
+    nextPageToken: d.next_page_token,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalSetModelVersionTagRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalSetRegisteredModelTagRequest_ResponseSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalTestRegistryWebhookRequest_ResponseSchema = z
+    .object({
+    status_code: z.number().optional(),
+    body: z.string().optional(),
+})
+    .transform(d => ({
+    statusCode: d.status_code,
+    body: d.body,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalTransitionModelVersionStageDatabricksRequest_ResponseSchema = z
+    .object({
+    model_version_databricks: z
+        .lazy(() => unmarshalModelVersionDatabricksSchema)
+        .optional(),
+})
+    .transform(d => ({
+    modelVersionDatabricks: d.model_version_databricks,
+}));
+export const unmarshalTransitionRequestSchema = z
+    .object({
+    creation_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    user_id: z.string().optional(),
+    activity_type: z.enum(ActivityType).optional(),
+    comment: z.string().optional(),
+    last_updated_timestamp: z
+        .union([z.number(), z.bigint()])
+        .transform(v => BigInt(v))
+        .optional(),
+    from_stage: z.string().optional(),
+    to_stage: z.string().optional(),
+    system_comment: z.string().optional(),
+    available_actions: z.array(z.enum(ActivityAction)).optional(),
+    id: z.string().optional(),
+})
+    .transform(d => ({
+    creationTimestamp: d.creation_timestamp,
+    userId: d.user_id,
+    activityType: d.activity_type,
+    comment: d.comment,
+    lastUpdatedTimestamp: d.last_updated_timestamp,
+    fromStage: d.from_stage,
+    toStage: d.to_stage,
+    systemComment: d.system_comment,
+    availableActions: d.available_actions,
+    id: d.id,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalUpdateCommentRequest_ResponseSchema = z
+    .object({
+    comment: z.lazy(() => unmarshalCommentObjectSchema).optional(),
+})
+    .transform(d => ({
+    comment: d.comment,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalUpdateModelVersionRequest_ResponseSchema = z
+    .object({
+    model_version: z.lazy(() => unmarshalModelVersionSchema).optional(),
+})
+    .transform(d => ({
+    modelVersion: d.model_version,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalUpdateRegisteredModelRequest_ResponseSchema = z
+    .object({
+    registered_model: z.lazy(() => unmarshalRegisteredModelSchema).optional(),
+})
+    .transform(d => ({
+    registeredModel: d.registered_model,
+}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalUpdateRegistryWebhookRequest_ResponseSchema = z
+    .object({
+    webhook: z.lazy(() => unmarshalRegistryWebhookSchema).optional(),
+})
+    .transform(d => ({
+    webhook: d.webhook,
+}));
+export const marshalApproveTransitionRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    stage: z.string().optional(),
+    archiveExistingVersions: z.boolean().optional(),
+    comment: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    stage: d.stage,
+    archive_existing_versions: d.archiveExistingVersions,
+    comment: d.comment,
+}));
+export const marshalCreateCommentRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    comment: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    comment: d.comment,
+}));
+export const marshalCreateModelVersionRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    source: z.string().optional(),
+    runId: z.string().optional(),
+    tags: z.array(z.lazy(() => marshalModelVersionTagSchema)).optional(),
+    runLink: z.string().optional(),
+    description: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    source: d.source,
+    run_id: d.runId,
+    tags: d.tags,
+    run_link: d.runLink,
+    description: d.description,
+}));
+export const marshalCreateRegisteredModelRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    tags: z.array(z.lazy(() => marshalRegisteredModelTagSchema)).optional(),
+    description: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    tags: d.tags,
+    description: d.description,
+}));
+export const marshalCreateRegistryWebhookRequestSchema = z
+    .object({
+    modelName: z.string().optional(),
+    events: z.array(z.enum(RegistryWebhookEvent)).optional(),
+    description: z.string().optional(),
+    status: z.enum(RegistryWebhookStatus).optional(),
+    httpUrlSpec: z.lazy(() => marshalHttpUrlSpecSchema).optional(),
+    jobSpec: z.lazy(() => marshalJobSpecSchema).optional(),
+})
+    .transform(d => ({
+    model_name: d.modelName,
+    events: d.events,
+    description: d.description,
+    status: d.status,
+    http_url_spec: d.httpUrlSpec,
+    job_spec: d.jobSpec,
+}));
+export const marshalCreateTransitionRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    stage: z.string().optional(),
+    comment: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    stage: d.stage,
+    comment: d.comment,
+}));
+export const marshalHttpUrlSpecSchema = z
+    .object({
+    url: z.string().optional(),
+    enableSslVerification: z.boolean().optional(),
+    secret: z.string().optional(),
+    authorization: z.string().optional(),
+})
+    .transform(d => ({
+    url: d.url,
+    enable_ssl_verification: d.enableSslVerification,
+    secret: d.secret,
+    authorization: d.authorization,
+}));
+export const marshalJobSpecSchema = z
+    .object({
+    jobId: z.string().optional(),
+    workspaceUrl: z.string().optional(),
+    accessToken: z.string().optional(),
+})
+    .transform(d => ({
+    job_id: d.jobId,
+    workspace_url: d.workspaceUrl,
+    access_token: d.accessToken,
+}));
+export const marshalListLatestVersionsRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    stages: z.array(z.string()).optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    stages: d.stages,
+}));
+export const marshalModelVersionTagSchema = z
+    .object({
+    key: z.string().optional(),
+    value: z.string().optional(),
+})
+    .transform(d => ({
+    key: d.key,
+    value: d.value,
+}));
+export const marshalRegisteredModelTagSchema = z
+    .object({
+    key: z.string().optional(),
+    value: z.string().optional(),
+})
+    .transform(d => ({
+    key: d.key,
+    value: d.value,
+}));
+export const marshalRejectTransitionRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    stage: z.string().optional(),
+    comment: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    stage: d.stage,
+    comment: d.comment,
+}));
+export const marshalRenameRegisteredModelRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    newName: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    new_name: d.newName,
+}));
+export const marshalSetModelVersionTagRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    key: z.string().optional(),
+    value: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    key: d.key,
+    value: d.value,
+}));
+export const marshalSetRegisteredModelTagRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    key: z.string().optional(),
+    value: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    key: d.key,
+    value: d.value,
+}));
+export const marshalTestRegistryWebhookRequestSchema = z
+    .object({
+    id: z.string().optional(),
+    event: z.enum(RegistryWebhookEvent).optional(),
+})
+    .transform(d => ({
+    id: d.id,
+    event: d.event,
+}));
+export const marshalTransitionModelVersionStageDatabricksRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    stage: z.string().optional(),
+    archiveExistingVersions: z.boolean().optional(),
+    comment: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    stage: d.stage,
+    archive_existing_versions: d.archiveExistingVersions,
+    comment: d.comment,
+}));
+export const marshalUpdateCommentRequestSchema = z
+    .object({
+    id: z.string().optional(),
+    comment: z.string().optional(),
+})
+    .transform(d => ({
+    id: d.id,
+    comment: d.comment,
+}));
+export const marshalUpdateModelVersionRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    version: z.string().optional(),
+    description: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    version: d.version,
+    description: d.description,
+}));
+export const marshalUpdateRegisteredModelRequestSchema = z
+    .object({
+    name: z.string().optional(),
+    description: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    description: d.description,
+}));
+export const marshalUpdateRegistryWebhookRequestSchema = z
+    .object({
+    id: z.string().optional(),
+    events: z.array(z.enum(RegistryWebhookEvent)).optional(),
+    description: z.string().optional(),
+    status: z.enum(RegistryWebhookStatus).optional(),
+    httpUrlSpec: z.lazy(() => marshalHttpUrlSpecSchema).optional(),
+    jobSpec: z.lazy(() => marshalJobSpecSchema).optional(),
+})
+    .transform(d => ({
+    id: d.id,
+    events: d.events,
+    description: d.description,
+    status: d.status,
+    http_url_spec: d.httpUrlSpec,
+    job_spec: d.jobSpec,
+}));
+//# sourceMappingURL=model.js.map