import { Long, unknownFieldsSymbol } from "../../../../../runtime/protos/index.js"; import type { MessageDescriptor, MessageFns } from "../../../../../runtime/protos/index.js"; import { customJson } from "../../../../../runtime/util/logging.js"; import { ChannelCredentials, Client, ClientUnaryCall, ClientOptions, CallOptions, Metadata, ServiceError as GrpcServiceError, handleUnaryCall, UntypedServiceImplementation } from "@grpc/grpc-js"; import type { SDKInterface } from "../../../../../sdk.js"; import { Request as SDKRequestClass, type RetryOptions } from "../../../../../runtime/request.js"; import { Operation as OperationWrapper } from "../../../../../runtime/operation.js"; import { ResourceMetadata } from "../../../common/v1/index.js"; import { GetOperationRequest, Operation, OperationService } from "../../../common/v1alpha1/index.js"; import { ClusterStatus_Phase, ClusterStatus_State } from "../../v1alpha1/index.js"; /** Represents the `nebius.msp.mlflow.v1alpha1.GetClusterRequest` protobuf message. */ export interface GetClusterRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.GetClusterRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * ID of the cluster to retrieve. * */ id: string; } /** Encodes, decodes, converts, and creates {@link GetClusterRequest} messages. */ export declare const GetClusterRequest: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.GetClusterByNameRequest` protobuf message. */ export interface GetClusterByNameRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.GetClusterByNameRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * Identifier of IAM container to get cluster from. * */ parentId: string; /** * Name of the cluster. * */ name: string; } /** Encodes, decodes, converts, and creates {@link GetClusterByNameRequest} messages. */ export declare const GetClusterByNameRequest: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.ListClustersRequest` protobuf message. */ export interface ListClustersRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.ListClustersRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * Identifier of IAM container to list clusters from. * */ parentId: string; /** * Specifies the maximum number of items to return in the response. Default value is 100. * */ pageSize: Long; /** * Token for pagination, allowing the retrieval of the next set of results. * */ pageToken: string; } /** Encodes, decodes, converts, and creates {@link ListClustersRequest} messages. */ export declare const ListClustersRequest: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.ListClustersResponse` protobuf message. */ export interface ListClustersResponse { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.ListClustersResponse"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * List of clusters. * */ items: Cluster[]; /** * Token for pagination, indicating the next set of results can be retrieved using this token. * */ nextPageToken: string; } /** Encodes, decodes, converts, and creates {@link ListClustersResponse} messages. */ export declare const ListClustersResponse: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.CreateClusterRequest` protobuf message. */ export interface CreateClusterRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.CreateClusterRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * Metadata associated with the new cluster. Must include parent_id in which we create the cluster. * */ metadata?: ResourceMetadata | undefined; /** * Specification for the new cluster. * */ spec?: ClusterSpec | undefined; } /** Encodes, decodes, converts, and creates {@link CreateClusterRequest} messages. */ export declare const CreateClusterRequest: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.DeleteClusterRequest` protobuf message. */ export interface DeleteClusterRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.DeleteClusterRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * ID of the cluster to delete. * */ id: string; } /** Encodes, decodes, converts, and creates {@link DeleteClusterRequest} messages. */ export declare const DeleteClusterRequest: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.StopClusterRequest` protobuf message. */ export interface StopClusterRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.StopClusterRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** Contains the `id` protobuf field. */ id: string; } /** Encodes, decodes, converts, and creates {@link StopClusterRequest} messages. */ export declare const StopClusterRequest: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.StartClusterRequest` protobuf message. */ export interface StartClusterRequest { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.StartClusterRequest"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** Contains the `id` protobuf field. */ id: string; } /** Encodes, decodes, converts, and creates {@link StartClusterRequest} messages. */ export declare const StartClusterRequest: MessageFns; /** * Defines the gRPC methods for the `nebius.msp.mlflow.v1alpha1.ClusterService` service. */ export type ClusterServiceServiceDescription = typeof ClusterServiceServiceDescription; /** * Describes the gRPC methods for the `nebius.msp.mlflow.v1alpha1.ClusterService` service. */ export declare const ClusterServiceServiceDescription: { readonly get: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/Get"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: GetClusterRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => GetClusterRequest; readonly responseSerialize: (value: Cluster) => Buffer; readonly responseDeserialize: (value: Buffer) => Cluster; }; readonly getByName: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/GetByName"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: GetClusterByNameRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => GetClusterByNameRequest; readonly responseSerialize: (value: Cluster) => Buffer; readonly responseDeserialize: (value: Buffer) => Cluster; }; readonly list: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/List"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: ListClustersRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => ListClustersRequest; readonly responseSerialize: (value: ListClustersResponse) => Buffer; readonly responseDeserialize: (value: Buffer) => ListClustersResponse; }; readonly create: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/Create"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: CreateClusterRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => CreateClusterRequest; readonly responseSerialize: (value: Operation) => Buffer; readonly responseDeserialize: (value: Buffer) => Operation; }; readonly delete: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/Delete"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: DeleteClusterRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => DeleteClusterRequest; readonly responseSerialize: (value: Operation) => Buffer; readonly responseDeserialize: (value: Buffer) => Operation; }; readonly stop: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/Stop"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: StopClusterRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => StopClusterRequest; readonly responseSerialize: (value: Operation) => Buffer; readonly responseDeserialize: (value: Buffer) => Operation; }; readonly start: { readonly path: "/nebius.msp.mlflow.v1alpha1.ClusterService/Start"; readonly requestStream: false; readonly responseStream: false; readonly requestSerialize: (value: StartClusterRequest) => Buffer; readonly requestDescriptor: () => MessageDescriptor | undefined; readonly sendResetMask: false; readonly requestDeserialize: (value: Buffer) => StartClusterRequest; readonly responseSerialize: (value: Operation) => Buffer; readonly responseDeserialize: (value: Buffer) => Operation; }; }; /** * Handles server calls for the `nebius.msp.mlflow.v1alpha1.ClusterService` service. */ export interface ClusterServiceServer extends UntypedServiceImplementation { /** * Returns the specified cluster. * */ get: handleUnaryCall; /** * Returns the specified cluster. * */ getByName: handleUnaryCall; /** * Retrieves a list of clusters. * */ list: handleUnaryCall; /** * Creates a cluster. * */ create: handleUnaryCall; /** * Delete a cluster. * */ delete: handleUnaryCall; /** * Stops a cluster. * */ stop: handleUnaryCall; /** * Starts a cluster. * */ start: handleUnaryCall; } /** * Defines the low-level gRPC client for the `nebius.msp.mlflow.v1alpha1.ClusterService` service. */ export interface ClusterServiceBaseClient extends Client { /** * Returns the specified cluster. * */ get(request: GetClusterRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: Cluster) => void): ClientUnaryCall; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: Cluster) => void): ClientUnaryCall; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: ListClustersResponse) => void): ClientUnaryCall; /** * Creates a cluster. * */ create(request: CreateClusterRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: Operation) => void): ClientUnaryCall; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: Operation) => void): ClientUnaryCall; /** * Stops a cluster. * */ stop(request: StopClusterRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: Operation) => void): ClientUnaryCall; /** * Starts a cluster. * */ start(request: StartClusterRequest, metadata: Metadata, options: Partial, callback: (error: GrpcServiceError | null, response: Operation) => void): ClientUnaryCall; } /** * Creates low-level gRPC clients for the `nebius.msp.mlflow.v1alpha1.ClusterService` service. */ export declare const ClusterServiceBaseClient: { /** Creates a low-level client for the service address. */ new (address: string, credentials: ChannelCredentials, options?: Partial): ClusterServiceBaseClient; /** Contains the gRPC service description. */ service: typeof ClusterServiceServiceDescription; /** Contains the fully qualified protobuf service name. */ serviceName: string; }; /** * Calls the `nebius.msp.mlflow.v1alpha1.ClusterService` service. */ export interface ClusterService { /** Contains the fully qualified protobuf service name. */ $type: "nebius.msp.mlflow.v1alpha1.ClusterService"; /** * Returns the specified cluster. * */ get(request: GetClusterRequest): SDKRequestClass; /** * Returns the specified cluster. * */ get(request: GetClusterRequest, metadata: Metadata): SDKRequestClass; /** * Returns the specified cluster. * */ get(request: GetClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest): SDKRequestClass; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest, metadata: Metadata): SDKRequestClass; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest): SDKRequestClass; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest, metadata: Metadata): SDKRequestClass; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass; /** * Creates a cluster. * */ create(request: CreateClusterRequest): SDKRequestClass>; /** * Creates a cluster. * */ create(request: CreateClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Creates a cluster. * */ create(request: CreateClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest): SDKRequestClass>; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; /** * Stops a cluster. * */ stop(request: StopClusterRequest): SDKRequestClass>; /** * Stops a cluster. * */ stop(request: StopClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Stops a cluster. * */ stop(request: StopClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; /** * Starts a cluster. * */ start(request: StartClusterRequest): SDKRequestClass>; /** * Starts a cluster. * */ start(request: StartClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Starts a cluster. * */ start(request: StartClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; } /** * Calls the `nebius.msp.mlflow.v1alpha1.ClusterService` service through the Nebius SDK. */ export declare class ClusterService implements ClusterService { private sdk; $type: "nebius.msp.mlflow.v1alpha1.ClusterService"; private addr; private spec; private apiServiceName; /** Creates a client that uses the SDK service address. */ constructor(sdk: SDKInterface); /** Returns the operation service for this service address. */ getOperationService(): OperationService; /** * Returns the specified cluster. * */ get(request: GetClusterRequest): SDKRequestClass; /** * Returns the specified cluster. * */ get(request: GetClusterRequest, metadata: Metadata): SDKRequestClass; /** * Returns the specified cluster. * */ get(request: GetClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest): SDKRequestClass; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest, metadata: Metadata): SDKRequestClass; /** * Returns the specified cluster. * */ getByName(request: GetClusterByNameRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest): SDKRequestClass; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest, metadata: Metadata): SDKRequestClass; /** * Retrieves a list of clusters. * */ list(request: ListClustersRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass; /** * Creates a cluster. * */ create(request: CreateClusterRequest): SDKRequestClass>; /** * Creates a cluster. * */ create(request: CreateClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Creates a cluster. * */ create(request: CreateClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest): SDKRequestClass>; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Delete a cluster. * */ delete(request: DeleteClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; /** * Stops a cluster. * */ stop(request: StopClusterRequest): SDKRequestClass>; /** * Stops a cluster. * */ stop(request: StopClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Stops a cluster. * */ stop(request: StopClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; /** * Starts a cluster. * */ start(request: StartClusterRequest): SDKRequestClass>; /** * Starts a cluster. * */ start(request: StartClusterRequest, metadata: Metadata): SDKRequestClass>; /** * Starts a cluster. * */ start(request: StartClusterRequest, metadata: Metadata, options: Partial & RetryOptions): SDKRequestClass>; } /** Represents the `nebius.msp.mlflow.v1alpha1.Cluster` protobuf message. */ export interface Cluster { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.Cluster"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** Contains the `metadata` protobuf field. */ metadata?: ResourceMetadata | undefined; /** Contains the `spec` protobuf field. */ spec?: ClusterSpec | undefined; /** Contains the `status` protobuf field. */ status?: MlflowClusterStatus | undefined; } /** Encodes, decodes, converts, and creates {@link Cluster} messages. */ export declare const Cluster: MessageFns; /** * Cluster specification * */ export interface ClusterSpec { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.ClusterSpec"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * Description of the cluster. * */ description: string; /** * Either make cluster public accessible or accessible only via private VPC. * */ publicAccess: boolean; /** * MLflow admin username. * */ adminUsername: string; /** * MLflow admin password. * */ adminPassword: string; /** * Id of the service account that will be used to access S3 bucket (and create one if not provided). * */ serviceAccountId: string; /** * Name of the Nebius S3 bucket for MLflow artifacts. If not provided, will be created under the same parent. * */ storageBucketName: string; /** * ID of the vpc network. * */ networkId: string; /** * Size defines how much resources will be allocated to mlflow * See supported sizes in the documentation. Default size is the smallest available in the region. * */ size: string; } /** Encodes, decodes, converts, and creates {@link ClusterSpec} messages. */ export declare const ClusterSpec: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.MlflowClusterStatus` protobuf message. */ export interface MlflowClusterStatus { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.MlflowClusterStatus"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * Current phase of the cluster. * */ phase: ClusterStatus_Phase; /** * State reflects substatus of the phase to define whether it's healthy or not. * */ state: ClusterStatus_State; /** * Tracking endpoint url. * Will be removed soon in favor of private_tracking_endpoint and public_tracking_endpoint. * */ trackingEndpoint: string; /** * Name of the Nebius S3 bucket for MLflow artifacts. * */ effectiveStorageBucketName: string; /** * Count of experiments in the MLflow cluster * */ experimentsCount: number; /** * MLflow version * */ mlflowVersion: string; /** * Public and private tracking endpoints * */ trackingEndpoints?: Endpoints | undefined; } /** Encodes, decodes, converts, and creates {@link MlflowClusterStatus} messages. */ export declare const MlflowClusterStatus: MessageFns; /** Represents the `nebius.msp.mlflow.v1alpha1.Endpoints` protobuf message. */ export interface Endpoints { /** Contains the fully qualified protobuf type name. */ $type: "nebius.msp.mlflow.v1alpha1.Endpoints"; /** Preserves protobuf fields that this SDK version does not recognize. */ [unknownFieldsSymbol]?: Uint8Array | undefined; /** Returns a safe value for JSON logs. */ [customJson]?: () => unknown; /** * Private endpoint * */ private: string; /** * Public endpoint * */ public: string; } /** Encodes, decodes, converts, and creates {@link Endpoints} messages. */ export declare const Endpoints: MessageFns; //# sourceMappingURL=index.d.ts.map