chalkpy 2.89.22__py3-none-any.whl → 2.95.3__py3-none-any.whl

chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.pyi

@@ -0,0 +1,212 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class MetadataJobStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    METADATA_JOB_STATUS_UNSPECIFIED: _ClassVar[MetadataJobStatus]
+    METADATA_JOB_STATUS_SCHEDULED: _ClassVar[MetadataJobStatus]
+    METADATA_JOB_STATUS_WORKING: _ClassVar[MetadataJobStatus]
+    METADATA_JOB_STATUS_RETRYABLE: _ClassVar[MetadataJobStatus]
+    METADATA_JOB_STATUS_SUCCESS: _ClassVar[MetadataJobStatus]
+    METADATA_JOB_STATUS_FAILED: _ClassVar[MetadataJobStatus]
+
+METADATA_JOB_STATUS_UNSPECIFIED: MetadataJobStatus
+METADATA_JOB_STATUS_SCHEDULED: MetadataJobStatus
+METADATA_JOB_STATUS_WORKING: MetadataJobStatus
+METADATA_JOB_STATUS_RETRYABLE: MetadataJobStatus
+METADATA_JOB_STATUS_SUCCESS: MetadataJobStatus
+METADATA_JOB_STATUS_FAILED: MetadataJobStatus
+
+class MetadataScheduledJob(_message.Message):
+    __slots__ = ("id", "metadata", "max_attempts", "created_at", "tag", "name", "custom_tags", "environment_id")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    MAX_ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    TAG_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    CUSTOM_TAGS_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    metadata: _struct_pb2.Struct
+    max_attempts: int
+    created_at: _timestamp_pb2.Timestamp
+    tag: str
+    name: str
+    custom_tags: _containers.RepeatedScalarFieldContainer[str]
+    environment_id: str
+    def __init__(
+        self,
+        id: _Optional[str] = ...,
+        metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        max_attempts: _Optional[int] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        tag: _Optional[str] = ...,
+        name: _Optional[str] = ...,
+        custom_tags: _Optional[_Iterable[str]] = ...,
+        environment_id: _Optional[str] = ...,
+    ) -> None: ...
+
+class MetadataJob(_message.Message):
+    __slots__ = (
+        "id",
+        "status",
+        "schedule_name",
+        "attempt",
+        "max_attempts",
+        "created_at",
+        "lease_start",
+        "lease_end",
+        "tag",
+        "metadata",
+        "attempts",
+        "custom_tags",
+        "environment_id",
+    )
+    ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULE_NAME_FIELD_NUMBER: _ClassVar[int]
+    ATTEMPT_FIELD_NUMBER: _ClassVar[int]
+    MAX_ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    LEASE_START_FIELD_NUMBER: _ClassVar[int]
+    LEASE_END_FIELD_NUMBER: _ClassVar[int]
+    TAG_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
+    CUSTOM_TAGS_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    id: int
+    status: MetadataJobStatus
+    schedule_name: str
+    attempt: int
+    max_attempts: int
+    created_at: _timestamp_pb2.Timestamp
+    lease_start: _timestamp_pb2.Timestamp
+    lease_end: _timestamp_pb2.Timestamp
+    tag: str
+    metadata: _struct_pb2.Struct
+    attempts: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
+    custom_tags: _containers.RepeatedScalarFieldContainer[str]
+    environment_id: str
+    def __init__(
+        self,
+        id: _Optional[int] = ...,
+        status: _Optional[_Union[MetadataJobStatus, str]] = ...,
+        schedule_name: _Optional[str] = ...,
+        attempt: _Optional[int] = ...,
+        max_attempts: _Optional[int] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        lease_start: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        lease_end: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        tag: _Optional[str] = ...,
+        metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        attempts: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ...,
+        custom_tags: _Optional[_Iterable[str]] = ...,
+        environment_id: _Optional[str] = ...,
+    ) -> None: ...
+
+class ListMetadataScheduledJobsRequest(_message.Message):
+    __slots__ = ("environment_id", "tag", "name", "limit", "offset")
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    TAG_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    environment_id: str
+    tag: str
+    name: str
+    limit: int
+    offset: int
+    def __init__(
+        self,
+        environment_id: _Optional[str] = ...,
+        tag: _Optional[str] = ...,
+        name: _Optional[str] = ...,
+        limit: _Optional[int] = ...,
+        offset: _Optional[int] = ...,
+    ) -> None: ...
+
+class ListMetadataScheduledJobsResponse(_message.Message):
+    __slots__ = ("scheduled_jobs", "total")
+    SCHEDULED_JOBS_FIELD_NUMBER: _ClassVar[int]
+    TOTAL_FIELD_NUMBER: _ClassVar[int]
+    scheduled_jobs: _containers.RepeatedCompositeFieldContainer[MetadataScheduledJob]
+    total: int
+    def __init__(
+        self,
+        scheduled_jobs: _Optional[_Iterable[_Union[MetadataScheduledJob, _Mapping]]] = ...,
+        total: _Optional[int] = ...,
+    ) -> None: ...
+
+class GetMetadataScheduledJobRequest(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    def __init__(self, id: _Optional[str] = ...) -> None: ...
+
+class GetMetadataScheduledJobResponse(_message.Message):
+    __slots__ = ("scheduled_job",)
+    SCHEDULED_JOB_FIELD_NUMBER: _ClassVar[int]
+    scheduled_job: MetadataScheduledJob
+    def __init__(self, scheduled_job: _Optional[_Union[MetadataScheduledJob, _Mapping]] = ...) -> None: ...
+
+class ListMetadataJobsRequest(_message.Message):
+    __slots__ = ("environment_id", "status", "tag", "schedule_name", "limit", "offset")
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    TAG_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULE_NAME_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    environment_id: str
+    status: MetadataJobStatus
+    tag: str
+    schedule_name: str
+    limit: int
+    offset: int
+    def __init__(
+        self,
+        environment_id: _Optional[str] = ...,
+        status: _Optional[_Union[MetadataJobStatus, str]] = ...,
+        tag: _Optional[str] = ...,
+        schedule_name: _Optional[str] = ...,
+        limit: _Optional[int] = ...,
+        offset: _Optional[int] = ...,
+    ) -> None: ...
+
+class ListMetadataJobsResponse(_message.Message):
+    __slots__ = ("jobs", "total")
+    JOBS_FIELD_NUMBER: _ClassVar[int]
+    TOTAL_FIELD_NUMBER: _ClassVar[int]
+    jobs: _containers.RepeatedCompositeFieldContainer[MetadataJob]
+    total: int
+    def __init__(
+        self, jobs: _Optional[_Iterable[_Union[MetadataJob, _Mapping]]] = ..., total: _Optional[int] = ...
+    ) -> None: ...
+
+class GetMetadataJobRequest(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: int
+    def __init__(self, id: _Optional[int] = ...) -> None: ...
+
+class GetMetadataJobResponse(_message.Message):
+    __slots__ = ("job",)
+    JOB_FIELD_NUMBER: _ClassVar[int]
+    job: MetadataJob
+    def __init__(self, job: _Optional[_Union[MetadataJob, _Mapping]] = ...) -> None: ...

chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.py

@@ -0,0 +1,217 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc
+
+from chalk._gen.chalk.server.v1 import (
+    metadataplanejobqueue_pb2 as chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2,
+)
+
+
+class MetadataPlaneJobQueueServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.ListMetadataScheduledJobs = channel.unary_unary(
+            "/chalk.server.v1.MetadataPlaneJobQueueService/ListMetadataScheduledJobs",
+            request_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataScheduledJobsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataScheduledJobsResponse.FromString,
+        )
+        self.GetMetadataScheduledJob = channel.unary_unary(
+            "/chalk.server.v1.MetadataPlaneJobQueueService/GetMetadataScheduledJob",
+            request_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataScheduledJobRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataScheduledJobResponse.FromString,
+        )
+        self.ListMetadataJobs = channel.unary_unary(
+            "/chalk.server.v1.MetadataPlaneJobQueueService/ListMetadataJobs",
+            request_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataJobsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataJobsResponse.FromString,
+        )
+        self.GetMetadataJob = channel.unary_unary(
+            "/chalk.server.v1.MetadataPlaneJobQueueService/GetMetadataJob",
+            request_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataJobRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataJobResponse.FromString,
+        )
+
+
+class MetadataPlaneJobQueueServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def ListMetadataScheduledJobs(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetMetadataScheduledJob(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def ListMetadataJobs(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetMetadataJob(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+
+def add_MetadataPlaneJobQueueServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "ListMetadataScheduledJobs": grpc.unary_unary_rpc_method_handler(
+            servicer.ListMetadataScheduledJobs,
+            request_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataScheduledJobsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataScheduledJobsResponse.SerializeToString,
+        ),
+        "GetMetadataScheduledJob": grpc.unary_unary_rpc_method_handler(
+            servicer.GetMetadataScheduledJob,
+            request_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataScheduledJobRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataScheduledJobResponse.SerializeToString,
+        ),
+        "ListMetadataJobs": grpc.unary_unary_rpc_method_handler(
+            servicer.ListMetadataJobs,
+            request_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataJobsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataJobsResponse.SerializeToString,
+        ),
+        "GetMetadataJob": grpc.unary_unary_rpc_method_handler(
+            servicer.GetMetadataJob,
+            request_deserializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataJobRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataJobResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+        "chalk.server.v1.MetadataPlaneJobQueueService", rpc_method_handlers
+    )
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class MetadataPlaneJobQueueService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def ListMetadataScheduledJobs(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.MetadataPlaneJobQueueService/ListMetadataScheduledJobs",
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataScheduledJobsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataScheduledJobsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetMetadataScheduledJob(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.MetadataPlaneJobQueueService/GetMetadataScheduledJob",
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataScheduledJobRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataScheduledJobResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def ListMetadataJobs(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.MetadataPlaneJobQueueService/ListMetadataJobs",
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataJobsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.ListMetadataJobsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetMetadataJob(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.MetadataPlaneJobQueueService/GetMetadataJob",
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataJobRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_metadataplanejobqueue__pb2.GetMetadataJobResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.pyi

@@ -0,0 +1,74 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.server.v1.metadataplanejobqueue_pb2 import (
+    GetMetadataJobRequest,
+    GetMetadataJobResponse,
+    GetMetadataScheduledJobRequest,
+    GetMetadataScheduledJobResponse,
+    ListMetadataJobsRequest,
+    ListMetadataJobsResponse,
+    ListMetadataScheduledJobsRequest,
+    ListMetadataScheduledJobsResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class MetadataPlaneJobQueueServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    ListMetadataScheduledJobs: UnaryUnaryMultiCallable[
+        ListMetadataScheduledJobsRequest,
+        ListMetadataScheduledJobsResponse,
+    ]
+    GetMetadataScheduledJob: UnaryUnaryMultiCallable[
+        GetMetadataScheduledJobRequest,
+        GetMetadataScheduledJobResponse,
+    ]
+    ListMetadataJobs: UnaryUnaryMultiCallable[
+        ListMetadataJobsRequest,
+        ListMetadataJobsResponse,
+    ]
+    GetMetadataJob: UnaryUnaryMultiCallable[
+        GetMetadataJobRequest,
+        GetMetadataJobResponse,
+    ]
+
+class MetadataPlaneJobQueueServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def ListMetadataScheduledJobs(
+        self,
+        request: ListMetadataScheduledJobsRequest,
+        context: ServicerContext,
+    ) -> ListMetadataScheduledJobsResponse: ...
+    @abstractmethod
+    def GetMetadataScheduledJob(
+        self,
+        request: GetMetadataScheduledJobRequest,
+        context: ServicerContext,
+    ) -> GetMetadataScheduledJobResponse: ...
+    @abstractmethod
+    def ListMetadataJobs(
+        self,
+        request: ListMetadataJobsRequest,
+        context: ServicerContext,
+    ) -> ListMetadataJobsResponse: ...
+    @abstractmethod
+    def GetMetadataJob(
+        self,
+        request: GetMetadataJobRequest,
+        context: ServicerContext,
+    ) -> GetMetadataJobResponse: ...
+
+def add_MetadataPlaneJobQueueServiceServicer_to_server(
+    servicer: MetadataPlaneJobQueueServiceServicer, server: Server
+) -> None: ...

chalk/_gen/chalk/server/v1/model_registry_pb2.py

@@ -23,7 +23,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n$chalk/server/v1/model_registry.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1a\x63halk/flags/v1/flags.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a$chalk/models/v1/model_artifact.proto\x1a google/protobuf/field_mask.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xf8\x02\n\rModelArtifact\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04path\x18\x02 \x01(\tR\x04path\x12\x36\n\x04spec\x18\x03 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\x04spec\x12H\n\x08metadata\x18\x06 \x03(\x0b\x32,.chalk.server.v1.ModelArtifact.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x04 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xc4\x03\n\x0cModelVersion\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12\x45\n\x0emodel_artifact\x18\x04 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x05 \x03(\tR\x07\x61liases\x12G\n\x08metadata\x18\x06 \x03(\x0b\x32+.chalk.server.v1.ModelVersion.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x07 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xed\x04\n\x05Model\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12@\n\x08metadata\x18\x04 \x03(\x0b\x32$.chalk.server.v1.Model.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x05 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12>\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x12@\n\x0b\x61rchived_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x12T\n\x14latest_model_version\x18\t \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x03R\x12latestModelVersion\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_byB\r\n\x0b_updated_atB\x0e\n\x0c_archived_atB\x17\n\x15_latest_model_version"`\n\x11ListModelsRequest\x12\x1b\n\x06\x63ursor\x18\x01 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limit"z\n\x12ListModelsResponse\x12.\n\x06models\x18\x01 \x03(\x0b\x32\x16.chalk.server.v1.ModelR\x06models\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"c\n\x0fGetModelRequest\x12\x1b\n\x08model_id\x18\x01 \x01(\tH\x00R\x07modelId\x12\x1f\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelNameB\x12\n\x10model_identifier"O\n\x10GetModelResponse\x12\x31\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelH\x00R\x05model\x88\x01\x01\x42\x08\n\x06_model"\xf9\x01\n\x12\x43reateModelRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x02 \x01(\tR\x0b\x64\x65scription\x12M\n\x08metadata\x18\x04 \x03(\x0b\x32\x31.chalk.server.v1.CreateModelRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"C\n\x13\x43reateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\xf8\x02\n\x14UpdateModelOperation\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x01R\x0b\x64\x65scription\x88\x01\x01\x12O\n\x08metadata\x18\x03 \x03(\x0b\x32\x33.chalk.server.v1.UpdateModelOperation.MetadataEntryR\x08metadata\x12@\n\x0b\x61rchived_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_model_nameB\x0e\n\x0c_descriptionB\x0e\n\x0c_archived_at"\xab\x01\n\x12UpdateModelRequest\x12\x19\n\x08model_id\x18\x01 \x01(\tR\x07modelId\x12=\n\x06update\x18\x02 \x01(\x0b\x32%.chalk.server.v1.UpdateModelOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"C\n\x13UpdateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\x9a\x01\n\x18ListModelVersionsRequest\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x01R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\t\n\x07_cursorB\x08\n\x06_limit"\x97\x01\n\x19ListModelVersionsResponse\x12\x44\n\x0emodel_versions\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\rmodelVersions\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"Q\n\x16GetModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"t\n\x17GetModelVersionResponse\x12G\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x00R\x0cmodelVersion\x88\x01\x01\x42\x10\n\x0e_model_version"\xbf\x02\n\x1a\x43reateModelArtifactRequest\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12U\n\x08metadata\x18\x04 \x03(\x0b\x32\x39.chalk.server.v1.CreateModelArtifactRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"d\n\x1b\x43reateModelArtifactResponse\x12\x45\n\x0emodel_artifact\x18\x01 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact"\xf6\x02\n\x19\x43reateModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x03 \x03(\tR\x07\x61liases\x12T\n\x08metadata\x18\x04 \x03(\x0b\x32\x38.chalk.server.v1.CreateModelVersionRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"`\n\x1a\x43reateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"J\n\x0fModelVersionKey\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"\xe4\x01\n\x1bUpdateModelVersionOperation\x12\x18\n\x07\x61liases\x18\x01 \x03(\tR\x07\x61liases\x12V\n\x08metadata\x18\x02 \x03(\x0b\x32:.chalk.server.v1.UpdateModelVersionOperation.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"\xec\x01\n\x19UpdateModelVersionRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey\x12\x44\n\x06update\x18\x02 \x01(\x0b\x32,.chalk.server.v1.UpdateModelVersionOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"`\n\x1aUpdateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"B\n!GetModelArtifactUploadUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"\xf5\x01\n"GetModelArtifactUploadUrlsResponse\x12\x64\n\x0bupload_urls\x18\x01 \x03(\x0b\x32\x43.chalk.server.v1.GetModelArtifactUploadUrlsResponse.UploadUrlsEntryR\nuploadUrls\x12*\n\x11model_artifact_id\x18\x02 \x01(\tR\x0fmodelArtifactId\x1a=\n\x0fUploadUrlsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"l\n\x1c\x44ownloadModelArtifactRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey"1\n\x1d\x44ownloadModelArtifactResponse\x12\x10\n\x03uri\x18\x01 \x01(\tR\x03uri"\xfc\x01\n\x19GetModelReferencesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12"\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12(\n\rmodel_version\x18\x03 \x01(\rH\x01R\x0cmodelVersion\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x04 \x01(\tH\x02R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x05 \x01(\x05H\x03R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0e_model_versionB\t\n\x07_cursorB\x08\n\x06_limit"]\n\rModelRelation\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures\x12%\n\x0eoutput_feature\x18\x02 \x01(\tR\routputFeature"\x8d\x03\n\x0eModelReference\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\trelations\x18\x05 \x03(\x0b\x32\x1e.chalk.server.v1.ModelRelationR\trelations\x12\x1c\n\tresolvers\x18\x06 \x03(\tR\tresolvers\x12\\\n\x15source_file_reference\x18\x07 \x01(\x0b\x32#.chalk.graph.v1.SourceFileReferenceH\x00R\x13sourceFileReference\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAtB\x18\n\x16_source_file_reference"\x9e\x01\n\x1aGetModelReferencesResponse\x12J\n\x10model_references\x18\x01 \x03(\x0b\x32\x1f.chalk.server.v1.ModelReferenceR\x0fmodelReferences\x12$\n\x0bnext_cursor\x18\x04 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"\xa2\x01\n\x18GetModelReferenceRequest\x12\x1d\n\x08model_id\x18\x01 \x01(\tB\x02\x18\x01R\x07modelId\x12\x1d\n\nmodel_name\x18\x04 \x01(\tR\tmodelName\x12#\n\rmodel_version\x18\x02 \x01(\x04R\x0cmodelVersion\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId"~\n\x19GetModelReferenceResponse\x12M\n\x0fmodel_reference\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.ModelReferenceH\x00R\x0emodelReference\x88\x01\x01\x42\x12\n\x10_model_reference"\xa6\x01\n\x0cRunCriterion\x12\x15\n\x06run_id\x18\x01 \x01(\tR\x05runId\x12\x1b\n\x06metric\x18\x02 \x01(\tH\x00R\x06metric\x88\x01\x01\x12I\n\tdirection\x18\x03 \x01(\x0e\x32&.chalk.server.v1.RunCriterionDirectionH\x01R\tdirection\x88\x01\x01\x42\t\n\x07_metricB\x0c\n\n_direction"\xe9\x01\n%CreateModelVersionFromArtifactRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12,\n\x11model_artifact_id\x18\x02 \x01(\tH\x00R\x0fmodelArtifactId\x12\x42\n\x0ctraining_run\x18\x03 \x01(\x0b\x32\x1d.chalk.server.v1.RunCriterionH\x00R\x0btrainingRun\x12\x18\n\x07\x61liases\x18\x04 \x03(\tR\x07\x61liasesB\x15\n\x13\x61rtifact_identifier"l\n&CreateModelVersionFromArtifactResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion*\x82\x01\n\x15RunCriterionDirection\x12\'\n#RUN_CRITERION_DIRECTION_UNSPECIFIED\x10\x00\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MAX\x10\x01\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MIN\x10\x02\x32\xec\x0f\n\x14ModelRegistryService\x12]\n\nListModels\x12".chalk.server.v1.ListModelsRequest\x1a#.chalk.server.v1.ListModelsResponse"\x06\x90\x02\x01\x80}\x0b\x12W\n\x08GetModel\x12 .chalk.server.v1.GetModelRequest\x1a!.chalk.server.v1.GetModelResponse"\x06\x90\x02\x01\x80}\x0b\x12\xb4\x01\n\x0b\x43reateModel\x12#.chalk.server.v1.CreateModelRequest\x1a$.chalk.server.v1.CreateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xb4\x01\n\x0bUpdateModel\x12#.chalk.server.v1.UpdateModelRequest\x1a$.chalk.server.v1.UpdateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x11ListModelVersions\x12).chalk.server.v1.ListModelVersionsRequest\x1a*.chalk.server.v1.ListModelVersionsResponse"\x06\x90\x02\x01\x80}\x0b\x12l\n\x0fGetModelVersion\x12\'.chalk.server.v1.GetModelVersionRequest\x1a(.chalk.server.v1.GetModelVersionResponse"\x06\x90\x02\x01\x80}\x0b\x12\xc9\x01\n\x12\x43reateModelVersion\x12*.chalk.server.v1.CreateModelVersionRequest\x1a+.chalk.server.v1.CreateModelVersionResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xcc\x01\n\x13\x43reateModelArtifact\x12+.chalk.server.v1.CreateModelArtifactRequest\x1a,.chalk.server.v1.CreateModelArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xed\x01\n\x1e\x43reateModelVersionFromArtifact\x12\x36.chalk.server.v1.CreateModelVersionFromArtifactRequest\x1a\x37.chalk.server.v1.CreateModelVersionFromArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x12UpdateModelVersion\x12*.chalk.server.v1.UpdateModelVersionRequest\x1a+.chalk.server.v1.UpdateModelVersionResponse"\x03\x80}\x0c\x12u\n\x12GetModelReferences\x12*.chalk.server.v1.GetModelReferencesRequest\x1a+.chalk.server.v1.GetModelReferencesResponse"\x06\x90\x02\x01\x80}\x0b\x12r\n\x11GetModelReference\x12).chalk.server.v1.GetModelReferenceRequest\x1a*.chalk.server.v1.GetModelReferenceResponse"\x06\x90\x02\x01\x80}\x0b\x12\xe1\x01\n\x1aGetModelArtifactUploadUrls\x12\x32.chalk.server.v1.GetModelArtifactUploadUrlsRequest\x1a\x33.chalk.server.v1.GetModelArtifactUploadUrlsResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.B\x9b\x01\n\x13\x63om.chalk.server.v1B\x12ModelRegistryProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
+    b'\n$chalk/server/v1/model_registry.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1a\x63halk/flags/v1/flags.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a$chalk/models/v1/model_artifact.proto\x1a google/protobuf/field_mask.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xf8\x02\n\rModelArtifact\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04path\x18\x02 \x01(\tR\x04path\x12\x36\n\x04spec\x18\x03 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\x04spec\x12H\n\x08metadata\x18\x06 \x03(\x0b\x32,.chalk.server.v1.ModelArtifact.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x04 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xc4\x03\n\x0cModelVersion\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12\x45\n\x0emodel_artifact\x18\x04 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x05 \x03(\tR\x07\x61liases\x12G\n\x08metadata\x18\x06 \x03(\x0b\x32+.chalk.server.v1.ModelVersion.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x07 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xed\x04\n\x05Model\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12@\n\x08metadata\x18\x04 \x03(\x0b\x32$.chalk.server.v1.Model.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x05 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12>\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x12@\n\x0b\x61rchived_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x12T\n\x14latest_model_version\x18\t \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x03R\x12latestModelVersion\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_byB\r\n\x0b_updated_atB\x0e\n\x0c_archived_atB\x17\n\x15_latest_model_version"`\n\x11ListModelsRequest\x12\x1b\n\x06\x63ursor\x18\x01 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limit"z\n\x12ListModelsResponse\x12.\n\x06models\x18\x01 \x03(\x0b\x32\x16.chalk.server.v1.ModelR\x06models\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"c\n\x0fGetModelRequest\x12\x1b\n\x08model_id\x18\x01 \x01(\tH\x00R\x07modelId\x12\x1f\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelNameB\x12\n\x10model_identifier"O\n\x10GetModelResponse\x12\x31\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelH\x00R\x05model\x88\x01\x01\x42\x08\n\x06_model"\xf9\x01\n\x12\x43reateModelRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x02 \x01(\tR\x0b\x64\x65scription\x12M\n\x08metadata\x18\x04 \x03(\x0b\x32\x31.chalk.server.v1.CreateModelRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"C\n\x13\x43reateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\xf8\x02\n\x14UpdateModelOperation\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x01R\x0b\x64\x65scription\x88\x01\x01\x12O\n\x08metadata\x18\x03 \x03(\x0b\x32\x33.chalk.server.v1.UpdateModelOperation.MetadataEntryR\x08metadata\x12@\n\x0b\x61rchived_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_model_nameB\x0e\n\x0c_descriptionB\x0e\n\x0c_archived_at"\xab\x01\n\x12UpdateModelRequest\x12\x19\n\x08model_id\x18\x01 \x01(\tR\x07modelId\x12=\n\x06update\x18\x02 \x01(\x0b\x32%.chalk.server.v1.UpdateModelOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"C\n\x13UpdateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\x9a\x01\n\x18ListModelVersionsRequest\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x01R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\t\n\x07_cursorB\x08\n\x06_limit"\x97\x01\n\x19ListModelVersionsResponse\x12\x44\n\x0emodel_versions\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\rmodelVersions\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"Q\n\x16GetModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"t\n\x17GetModelVersionResponse\x12G\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x00R\x0cmodelVersion\x88\x01\x01\x42\x10\n\x0e_model_version"\xbf\x02\n\x1a\x43reateModelArtifactRequest\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12U\n\x08metadata\x18\x04 \x03(\x0b\x32\x39.chalk.server.v1.CreateModelArtifactRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"d\n\x1b\x43reateModelArtifactResponse\x12\x45\n\x0emodel_artifact\x18\x01 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact"\xf6\x02\n\x19\x43reateModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x03 \x03(\tR\x07\x61liases\x12T\n\x08metadata\x18\x04 \x03(\x0b\x32\x38.chalk.server.v1.CreateModelVersionRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"`\n\x1a\x43reateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"J\n\x0fModelVersionKey\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"\xe4\x01\n\x1bUpdateModelVersionOperation\x12\x18\n\x07\x61liases\x18\x01 \x03(\tR\x07\x61liases\x12V\n\x08metadata\x18\x02 \x03(\x0b\x32:.chalk.server.v1.UpdateModelVersionOperation.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"\xec\x01\n\x19UpdateModelVersionRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey\x12\x44\n\x06update\x18\x02 \x01(\x0b\x32,.chalk.server.v1.UpdateModelVersionOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"`\n\x1aUpdateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"B\n!GetModelArtifactUploadUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"\xf5\x01\n"GetModelArtifactUploadUrlsResponse\x12\x64\n\x0bupload_urls\x18\x01 \x03(\x0b\x32\x43.chalk.server.v1.GetModelArtifactUploadUrlsResponse.UploadUrlsEntryR\nuploadUrls\x12*\n\x11model_artifact_id\x18\x02 \x01(\tR\x0fmodelArtifactId\x1a=\n\x0fUploadUrlsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"l\n\x1c\x44ownloadModelArtifactRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey"1\n\x1d\x44ownloadModelArtifactResponse\x12\x10\n\x03uri\x18\x01 \x01(\tR\x03uri"\xfc\x01\n\x19GetModelReferencesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12"\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12(\n\rmodel_version\x18\x03 \x01(\rH\x01R\x0cmodelVersion\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x04 \x01(\tH\x02R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x05 \x01(\x05H\x03R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0e_model_versionB\t\n\x07_cursorB\x08\n\x06_limit"]\n\rModelRelation\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures\x12%\n\x0eoutput_feature\x18\x02 \x01(\tR\routputFeature"\x8d\x03\n\x0eModelReference\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\trelations\x18\x05 \x03(\x0b\x32\x1e.chalk.server.v1.ModelRelationR\trelations\x12\x1c\n\tresolvers\x18\x06 \x03(\tR\tresolvers\x12\\\n\x15source_file_reference\x18\x07 \x01(\x0b\x32#.chalk.graph.v1.SourceFileReferenceH\x00R\x13sourceFileReference\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAtB\x18\n\x16_source_file_reference"\x9e\x01\n\x1aGetModelReferencesResponse\x12J\n\x10model_references\x18\x01 \x03(\x0b\x32\x1f.chalk.server.v1.ModelReferenceR\x0fmodelReferences\x12$\n\x0bnext_cursor\x18\x04 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"\xa2\x01\n\x18GetModelReferenceRequest\x12\x1d\n\x08model_id\x18\x01 \x01(\tB\x02\x18\x01R\x07modelId\x12\x1d\n\nmodel_name\x18\x04 \x01(\tR\tmodelName\x12#\n\rmodel_version\x18\x02 \x01(\x04R\x0cmodelVersion\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId"~\n\x19GetModelReferenceResponse\x12M\n\x0fmodel_reference\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.ModelReferenceH\x00R\x0emodelReference\x88\x01\x01\x42\x12\n\x10_model_reference"\xd7\x01\n\x0cRunCriterion\x12\x17\n\x06run_id\x18\x01 \x01(\tH\x00R\x05runId\x12\x1b\n\x08run_name\x18\x04 \x01(\tH\x00R\x07runName\x12\x1b\n\x06metric\x18\x02 \x01(\tH\x01R\x06metric\x88\x01\x01\x12I\n\tdirection\x18\x03 \x01(\x0e\x32&.chalk.server.v1.RunCriterionDirectionH\x02R\tdirection\x88\x01\x01\x42\x10\n\x0erun_identifierB\t\n\x07_metricB\x0c\n\n_direction"\xe9\x01\n%CreateModelVersionFromArtifactRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12,\n\x11model_artifact_id\x18\x02 \x01(\tH\x00R\x0fmodelArtifactId\x12\x42\n\x0ctraining_run\x18\x03 \x01(\x0b\x32\x1d.chalk.server.v1.RunCriterionH\x00R\x0btrainingRun\x12\x18\n\x07\x61liases\x18\x04 \x03(\tR\x07\x61liasesB\x15\n\x13\x61rtifact_identifier"l\n&CreateModelVersionFromArtifactResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion*\x82\x01\n\x15RunCriterionDirection\x12\'\n#RUN_CRITERION_DIRECTION_UNSPECIFIED\x10\x00\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MAX\x10\x01\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MIN\x10\x02\x32\xec\x0f\n\x14ModelRegistryService\x12]\n\nListModels\x12".chalk.server.v1.ListModelsRequest\x1a#.chalk.server.v1.ListModelsResponse"\x06\x90\x02\x01\x80}\x0b\x12W\n\x08GetModel\x12 .chalk.server.v1.GetModelRequest\x1a!.chalk.server.v1.GetModelResponse"\x06\x90\x02\x01\x80}\x0b\x12\xb4\x01\n\x0b\x43reateModel\x12#.chalk.server.v1.CreateModelRequest\x1a$.chalk.server.v1.CreateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xb4\x01\n\x0bUpdateModel\x12#.chalk.server.v1.UpdateModelRequest\x1a$.chalk.server.v1.UpdateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x11ListModelVersions\x12).chalk.server.v1.ListModelVersionsRequest\x1a*.chalk.server.v1.ListModelVersionsResponse"\x06\x90\x02\x01\x80}\x0b\x12l\n\x0fGetModelVersion\x12\'.chalk.server.v1.GetModelVersionRequest\x1a(.chalk.server.v1.GetModelVersionResponse"\x06\x90\x02\x01\x80}\x0b\x12\xc9\x01\n\x12\x43reateModelVersion\x12*.chalk.server.v1.CreateModelVersionRequest\x1a+.chalk.server.v1.CreateModelVersionResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xcc\x01\n\x13\x43reateModelArtifact\x12+.chalk.server.v1.CreateModelArtifactRequest\x1a,.chalk.server.v1.CreateModelArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xed\x01\n\x1e\x43reateModelVersionFromArtifact\x12\x36.chalk.server.v1.CreateModelVersionFromArtifactRequest\x1a\x37.chalk.server.v1.CreateModelVersionFromArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x12UpdateModelVersion\x12*.chalk.server.v1.UpdateModelVersionRequest\x1a+.chalk.server.v1.UpdateModelVersionResponse"\x03\x80}\x0c\x12u\n\x12GetModelReferences\x12*.chalk.server.v1.GetModelReferencesRequest\x1a+.chalk.server.v1.GetModelReferencesResponse"\x06\x90\x02\x01\x80}\x0b\x12r\n\x11GetModelReference\x12).chalk.server.v1.GetModelReferenceRequest\x1a*.chalk.server.v1.GetModelReferenceResponse"\x06\x90\x02\x01\x80}\x0b\x12\xe1\x01\n\x1aGetModelArtifactUploadUrls\x12\x32.chalk.server.v1.GetModelArtifactUploadUrlsRequest\x1a\x33.chalk.server.v1.GetModelArtifactUploadUrlsResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.B\x9b\x01\n\x13\x63om.chalk.server.v1B\x12ModelRegistryProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -98,8 +98,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_MODELREGISTRYSERVICE"].methods_by_name[
         "GetModelArtifactUploadUrls"
     ]._serialized_options = b"\200}\014\222\323\016S\n\026model_registry_enabled\0229This action is not enabled. Please contact Chalk Support."
-    _globals["_RUNCRITERIONDIRECTION"]._serialized_start = 7338
-    _globals["_RUNCRITERIONDIRECTION"]._serialized_end = 7468
+    _globals["_RUNCRITERIONDIRECTION"]._serialized_start = 7387
+    _globals["_RUNCRITERIONDIRECTION"]._serialized_end = 7517
     _globals["_MODELARTIFACT"]._serialized_start = 282
     _globals["_MODELARTIFACT"]._serialized_end = 658
     _globals["_MODELARTIFACT_METADATAENTRY"]._serialized_start = 560
@@ -187,11 +187,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GETMODELREFERENCERESPONSE"]._serialized_start = 6694
     _globals["_GETMODELREFERENCERESPONSE"]._serialized_end = 6820
     _globals["_RUNCRITERION"]._serialized_start = 6823
-    _globals["_RUNCRITERION"]._serialized_end = 6989
-    _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_start = 6992
-    _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_end = 7225
-    _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_start = 7227
-    _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_end = 7335
-    _globals["_MODELREGISTRYSERVICE"]._serialized_start = 7471
-    _globals["_MODELREGISTRYSERVICE"]._serialized_end = 9499
+    _globals["_RUNCRITERION"]._serialized_end = 7038
+    _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_start = 7041
+    _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_end = 7274
+    _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_start = 7276
+    _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_end = 7384
+    _globals["_MODELREGISTRYSERVICE"]._serialized_start = 7520
+    _globals["_MODELREGISTRYSERVICE"]._serialized_end = 9548
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/model_registry_pb2.pyi

@@ -566,16 +566,19 @@ class GetModelReferenceResponse(_message.Message):
     def __init__(self, model_reference: _Optional[_Union[ModelReference, _Mapping]] = ...) -> None: ...
 
 class RunCriterion(_message.Message):
-    __slots__ = ("run_id", "metric", "direction")
+    __slots__ = ("run_id", "run_name", "metric", "direction")
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    RUN_NAME_FIELD_NUMBER: _ClassVar[int]
     METRIC_FIELD_NUMBER: _ClassVar[int]
     DIRECTION_FIELD_NUMBER: _ClassVar[int]
     run_id: str
+    run_name: str
     metric: str
     direction: RunCriterionDirection
     def __init__(
         self,
         run_id: _Optional[str] = ...,
+        run_name: _Optional[str] = ...,
         metric: _Optional[str] = ...,
         direction: _Optional[_Union[RunCriterionDirection, str]] = ...,
     ) -> None: ...