qwak-core 0.4.270__py3-none-any.whl → 0.4.272__py3-none-any.whl

_qwak_proto/qwak/model_group/model_group_pb2.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/model_group/model_group.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from _qwak_proto.qwak.model_group import model_group_repository_details_pb2 as qwak_dot_model__group_dot_model__group__repository__details__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"qwak/model_group/model_group.proto\x12\x1cqwak.model_groups.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x35qwak/model_group/model_group_repository_details.proto\"Y\n\"CreateIfNotExistsModelGroupRequest\x12\x18\n\x10model_group_name\x18\x01 \x01(\t\x12\x19\n\x11jfrog_project_key\x18\x02 \x01(\t\"\xdf\x01\n\x1bModelGroupBriefInfoResponse\x12\x16\n\x0emodel_group_id\x18\x01 \x01(\t\x12\x1f\n\x17model_group_description\x18\x02 \x01(\t\x12:\n\x16model_group_created_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12K\n\x12repository_details\x18\x04 \x03(\x0b\x32/.qwak.model_groups.management.RepositoryDetails2\xba\x01\n\x1bModelGroupManagementService\x12\x9a\x01\n\x1b\x43reateIfNotExistsModelGroup\x12@.qwak.model_groups.management.CreateIfNotExistsModelGroupRequest\x1a\x39.qwak.model_groups.management.ModelGroupBriefInfoResponseBL\n&com.qwak.ai.management.model.group.apiB ModelGroupManagementServiceProtoP\x01\x62\x06proto3')
+
+
+
+_CREATEIFNOTEXISTSMODELGROUPREQUEST = DESCRIPTOR.message_types_by_name['CreateIfNotExistsModelGroupRequest']
+_MODELGROUPBRIEFINFORESPONSE = DESCRIPTOR.message_types_by_name['ModelGroupBriefInfoResponse']
+CreateIfNotExistsModelGroupRequest = _reflection.GeneratedProtocolMessageType('CreateIfNotExistsModelGroupRequest', (_message.Message,), {
+  'DESCRIPTOR' : _CREATEIFNOTEXISTSMODELGROUPREQUEST,
+  '__module__' : 'qwak.model_group.model_group_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.CreateIfNotExistsModelGroupRequest)
+  })
+_sym_db.RegisterMessage(CreateIfNotExistsModelGroupRequest)
+
+ModelGroupBriefInfoResponse = _reflection.GeneratedProtocolMessageType('ModelGroupBriefInfoResponse', (_message.Message,), {
+  'DESCRIPTOR' : _MODELGROUPBRIEFINFORESPONSE,
+  '__module__' : 'qwak.model_group.model_group_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.ModelGroupBriefInfoResponse)
+  })
+_sym_db.RegisterMessage(ModelGroupBriefInfoResponse)
+
+_MODELGROUPMANAGEMENTSERVICE = DESCRIPTOR.services_by_name['ModelGroupManagementService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.management.model.group.apiB ModelGroupManagementServiceProtoP\001'
+  _CREATEIFNOTEXISTSMODELGROUPREQUEST._serialized_start=156
+  _CREATEIFNOTEXISTSMODELGROUPREQUEST._serialized_end=245
+  _MODELGROUPBRIEFINFORESPONSE._serialized_start=248
+  _MODELGROUPBRIEFINFORESPONSE._serialized_end=471
+  _MODELGROUPMANAGEMENTSERVICE._serialized_start=474
+  _MODELGROUPMANAGEMENTSERVICE._serialized_end=660
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/model_group/model_group_pb2.pyi

@@ -0,0 +1,68 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import qwak.model_group.model_group_repository_details_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class CreateIfNotExistsModelGroupRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
+    model_group_name: builtins.str
+    """Model group name"""
+    jfrog_project_key: builtins.str
+    """Artifactory project key used for repository organization"""
+    def __init__(
+        self,
+        *,
+        model_group_name: builtins.str = ...,
+        jfrog_project_key: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["jfrog_project_key", b"jfrog_project_key", "model_group_name", b"model_group_name"]) -> None: ...
+
+global___CreateIfNotExistsModelGroupRequest = CreateIfNotExistsModelGroupRequest
+
+class ModelGroupBriefInfoResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_ID_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_DESCRIPTION_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_CREATED_AT_FIELD_NUMBER: builtins.int
+    REPOSITORY_DETAILS_FIELD_NUMBER: builtins.int
+    model_group_id: builtins.str
+    """Unique identifier for the model group"""
+    model_group_description: builtins.str
+    """Description of the model group"""
+    @property
+    def model_group_created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """The model group creation timestamp"""
+    @property
+    def repository_details(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.model_group.model_group_repository_details_pb2.RepositoryDetails]:
+        """List of repository details associated with this model group"""
+    def __init__(
+        self,
+        *,
+        model_group_id: builtins.str = ...,
+        model_group_description: builtins.str = ...,
+        model_group_created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        repository_details: collections.abc.Iterable[qwak.model_group.model_group_repository_details_pb2.RepositoryDetails] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_group_created_at", b"model_group_created_at"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_group_created_at", b"model_group_created_at", "model_group_description", b"model_group_description", "model_group_id", b"model_group_id", "repository_details", b"repository_details"]) -> None: ...
+
+global___ModelGroupBriefInfoResponse = ModelGroupBriefInfoResponse

_qwak_proto/qwak/model_group/model_group_pb2_grpc.py

@@ -0,0 +1,66 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from _qwak_proto.qwak.model_group import model_group_pb2 as qwak_dot_model__group_dot_model__group__pb2
+
+
+class ModelGroupManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateIfNotExistsModelGroup = channel.unary_unary(
+                '/qwak.model_groups.management.ModelGroupManagementService/CreateIfNotExistsModelGroup',
+                request_serializer=qwak_dot_model__group_dot_model__group__pb2.CreateIfNotExistsModelGroupRequest.SerializeToString,
+                response_deserializer=qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.FromString,
+                )
+
+
+class ModelGroupManagementServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def CreateIfNotExistsModelGroup(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_ModelGroupManagementServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateIfNotExistsModelGroup': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateIfNotExistsModelGroup,
+                    request_deserializer=qwak_dot_model__group_dot_model__group__pb2.CreateIfNotExistsModelGroupRequest.FromString,
+                    response_serializer=qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.model_groups.management.ModelGroupManagementService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class ModelGroupManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def CreateIfNotExistsModelGroup(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, '/qwak.model_groups.management.ModelGroupManagementService/CreateIfNotExistsModelGroup',
+            qwak_dot_model__group_dot_model__group__pb2.CreateIfNotExistsModelGroupRequest.SerializeToString,
+            qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_qwak_proto/qwak/model_group/model_group_repository_details_pb2.py

@@ -0,0 +1,86 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/model_group/model_group_repository_details.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5qwak/model_group/model_group_repository_details.proto\x12\x1cqwak.model_groups.management\x1a\x1fgoogle/protobuf/timestamp.proto\"r\n\x11RepositoryDetails\x12\x45\n\x0frepository_type\x18\x01 \x01(\x0b\x32,.qwak.model_groups.management.RepositoryType\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\"\x12\n\x10\x44ockerRepository\"\x13\n\x11\x44\x61tasetRepository\"\x14\n\x12\x41rtifactRepository\"\x17\n\x15HuggingFaceRepository\"\xdd\x02\n\x0eRepositoryType\x12K\n\x11\x64ocker_repository\x18\x01 \x01(\x0b\x32..qwak.model_groups.management.DockerRepositoryH\x00\x12M\n\x12\x64\x61taset_repository\x18\x02 \x01(\x0b\x32/.qwak.model_groups.management.DatasetRepositoryH\x00\x12O\n\x13\x61rtifact_repository\x18\x03 \x01(\x0b\x32\x30.qwak.model_groups.management.ArtifactRepositoryH\x00\x12V\n\x17hugging_face_repository\x18\x04 \x01(\x0b\x32\x33.qwak.model_groups.management.HuggingFaceRepositoryH\x00\x42\x06\n\x04typeBN\n&com.qwak.ai.management.model.group.apiB\"ModelGroupRepositoriesDetailsProtoP\x01\x62\x06proto3')
+
+
+
+_REPOSITORYDETAILS = DESCRIPTOR.message_types_by_name['RepositoryDetails']
+_DOCKERREPOSITORY = DESCRIPTOR.message_types_by_name['DockerRepository']
+_DATASETREPOSITORY = DESCRIPTOR.message_types_by_name['DatasetRepository']
+_ARTIFACTREPOSITORY = DESCRIPTOR.message_types_by_name['ArtifactRepository']
+_HUGGINGFACEREPOSITORY = DESCRIPTOR.message_types_by_name['HuggingFaceRepository']
+_REPOSITORYTYPE = DESCRIPTOR.message_types_by_name['RepositoryType']
+RepositoryDetails = _reflection.GeneratedProtocolMessageType('RepositoryDetails', (_message.Message,), {
+  'DESCRIPTOR' : _REPOSITORYDETAILS,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.RepositoryDetails)
+  })
+_sym_db.RegisterMessage(RepositoryDetails)
+
+DockerRepository = _reflection.GeneratedProtocolMessageType('DockerRepository', (_message.Message,), {
+  'DESCRIPTOR' : _DOCKERREPOSITORY,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.DockerRepository)
+  })
+_sym_db.RegisterMessage(DockerRepository)
+
+DatasetRepository = _reflection.GeneratedProtocolMessageType('DatasetRepository', (_message.Message,), {
+  'DESCRIPTOR' : _DATASETREPOSITORY,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.DatasetRepository)
+  })
+_sym_db.RegisterMessage(DatasetRepository)
+
+ArtifactRepository = _reflection.GeneratedProtocolMessageType('ArtifactRepository', (_message.Message,), {
+  'DESCRIPTOR' : _ARTIFACTREPOSITORY,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.ArtifactRepository)
+  })
+_sym_db.RegisterMessage(ArtifactRepository)
+
+HuggingFaceRepository = _reflection.GeneratedProtocolMessageType('HuggingFaceRepository', (_message.Message,), {
+  'DESCRIPTOR' : _HUGGINGFACEREPOSITORY,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.HuggingFaceRepository)
+  })
+_sym_db.RegisterMessage(HuggingFaceRepository)
+
+RepositoryType = _reflection.GeneratedProtocolMessageType('RepositoryType', (_message.Message,), {
+  'DESCRIPTOR' : _REPOSITORYTYPE,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.RepositoryType)
+  })
+_sym_db.RegisterMessage(RepositoryType)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.management.model.group.apiB\"ModelGroupRepositoriesDetailsProtoP\001'
+  _REPOSITORYDETAILS._serialized_start=120
+  _REPOSITORYDETAILS._serialized_end=234
+  _DOCKERREPOSITORY._serialized_start=236
+  _DOCKERREPOSITORY._serialized_end=254
+  _DATASETREPOSITORY._serialized_start=256
+  _DATASETREPOSITORY._serialized_end=275
+  _ARTIFACTREPOSITORY._serialized_start=277
+  _ARTIFACTREPOSITORY._serialized_end=297
+  _HUGGINGFACEREPOSITORY._serialized_start=299
+  _HUGGINGFACEREPOSITORY._serialized_end=322
+  _REPOSITORYTYPE._serialized_start=325
+  _REPOSITORYTYPE._serialized_end=674
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/model_group/model_group_repository_details_pb2.pyi

@@ -0,0 +1,107 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class RepositoryDetails(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REPOSITORY_TYPE_FIELD_NUMBER: builtins.int
+    REPOSITORY_KEY_FIELD_NUMBER: builtins.int
+    @property
+    def repository_type(self) -> global___RepositoryType:
+        """The repository type"""
+    repository_key: builtins.str
+    """The repository key is the full repository path, e.g. <jfrog_project_key><model>-docker-local"""
+    def __init__(
+        self,
+        *,
+        repository_type: global___RepositoryType | None = ...,
+        repository_key: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["repository_type", b"repository_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["repository_key", b"repository_key", "repository_type", b"repository_type"]) -> None: ...
+
+global___RepositoryDetails = RepositoryDetails
+
+class DockerRepository(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___DockerRepository = DockerRepository
+
+class DatasetRepository(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___DatasetRepository = DatasetRepository
+
+class ArtifactRepository(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ArtifactRepository = ArtifactRepository
+
+class HuggingFaceRepository(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___HuggingFaceRepository = HuggingFaceRepository
+
+class RepositoryType(google.protobuf.message.Message):
+    """The repository type"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DOCKER_REPOSITORY_FIELD_NUMBER: builtins.int
+    DATASET_REPOSITORY_FIELD_NUMBER: builtins.int
+    ARTIFACT_REPOSITORY_FIELD_NUMBER: builtins.int
+    HUGGING_FACE_REPOSITORY_FIELD_NUMBER: builtins.int
+    @property
+    def docker_repository(self) -> global___DockerRepository:
+        """Docker repository"""
+    @property
+    def dataset_repository(self) -> global___DatasetRepository:
+        """Dataset repository"""
+    @property
+    def artifact_repository(self) -> global___ArtifactRepository:
+        """Artifact repository"""
+    @property
+    def hugging_face_repository(self) -> global___HuggingFaceRepository:
+        """HuggingFace repository"""
+    def __init__(
+        self,
+        *,
+        docker_repository: global___DockerRepository | None = ...,
+        dataset_repository: global___DatasetRepository | None = ...,
+        artifact_repository: global___ArtifactRepository | None = ...,
+        hugging_face_repository: global___HuggingFaceRepository | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["artifact_repository", b"artifact_repository", "dataset_repository", b"dataset_repository", "docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["artifact_repository", b"artifact_repository", "dataset_repository", b"dataset_repository", "docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["docker_repository", "dataset_repository", "artifact_repository", "hugging_face_repository"] | None: ...
+
+global___RepositoryType = RepositoryType

_qwak_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_qwak_proto/qwak/models/models_pb2.py

@@ -16,12 +16,13 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from _qwak_proto.qwak.builds import builds_pb2 as qwak_dot_builds_dot_builds__pb2
 from _qwak_proto.qwak.projects import jfrog_project_spec_pb2 as qwak_dot_projects_dot_jfrog__project__spec__pb2
+from _qwak_proto.qwak.models import models_query_pb2 as qwak_dot_models_dot_models__query__pb2
 from _qwak_proto.qwak.audience.v1 import audience_pb2 as qwak_dot_audience_dot_v1_dot_audience__pb2
 from _qwak_proto.qwak.deployment import deployment_pb2 as qwak_dot_deployment_dot_deployment__pb2
 from _qwak_proto.qwak.build.v1 import build_pb2 as qwak_dot_build_dot_v1_dot_build__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xe0\t\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1eqwak/models/models_query.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"\xf0\x01\n\x1aListModelsPaginatedRequest\x12\x13\n\tpage_size\x18\x01 \x01(\x05H\x00\x12\x10\n\x06\x63ursor\x18\x02 \x01(\tH\x01\x12\x39\n\x0bsort_models\x18\x03 \x01(\x0b\x32\".qwak.models.management.SortModelsH\x02\x12\x30\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x1e.qwak.models.management.FilterH\x03\x42\x0f\n\rpage_size_optB\x0c\n\ncursor_optB\x11\n\x0fsort_models_optB\x0c\n\nfilter_opt\"\x8a\x01\n\x1bListModelsPaginatedResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\x12<\n\x08metadata\x18\x02 \x01(\x0b\x32*.qwak.models.management.PaginationMetadata\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xe0\n\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12~\n\x13ListModelsPaginated\x12\x32.qwak.models.management.ListModelsPaginatedRequest\x1a\x33.qwak.models.management.ListModelsPaginatedResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
 
 _DEPLOYMENTMODELSTATUS = DESCRIPTOR.enum_types_by_name['DeploymentModelStatus']
 DeploymentModelStatus = enum_type_wrapper.EnumTypeWrapper(_DEPLOYMENTMODELSTATUS)
@@ -45,6 +46,8 @@ _CREATEMODELREQUEST = DESCRIPTOR.message_types_by_name['CreateModelRequest']
 _CREATEMODELRESPONSE = DESCRIPTOR.message_types_by_name['CreateModelResponse']
 _LISTMODELSREQUEST = DESCRIPTOR.message_types_by_name['ListModelsRequest']
 _LISTMODELSRESPONSE = DESCRIPTOR.message_types_by_name['ListModelsResponse']
+_LISTMODELSPAGINATEDREQUEST = DESCRIPTOR.message_types_by_name['ListModelsPaginatedRequest']
+_LISTMODELSPAGINATEDRESPONSE = DESCRIPTOR.message_types_by_name['ListModelsPaginatedResponse']
 _GETMODELREQUEST = DESCRIPTOR.message_types_by_name['GetModelRequest']
 _GETMODELRESPONSE = DESCRIPTOR.message_types_by_name['GetModelResponse']
 _DELETEMODELREQUEST = DESCRIPTOR.message_types_by_name['DeleteModelRequest']
@@ -105,6 +108,20 @@ ListModelsResponse = _reflection.GeneratedProtocolMessageType('ListModelsRespons
   })
 _sym_db.RegisterMessage(ListModelsResponse)
 
+ListModelsPaginatedRequest = _reflection.GeneratedProtocolMessageType('ListModelsPaginatedRequest', (_message.Message,), {
+  'DESCRIPTOR' : _LISTMODELSPAGINATEDREQUEST,
+  '__module__' : 'qwak.models.models_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.ListModelsPaginatedRequest)
+  })
+_sym_db.RegisterMessage(ListModelsPaginatedRequest)
+
+ListModelsPaginatedResponse = _reflection.GeneratedProtocolMessageType('ListModelsPaginatedResponse', (_message.Message,), {
+  'DESCRIPTOR' : _LISTMODELSPAGINATEDRESPONSE,
+  '__module__' : 'qwak.models.models_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.ListModelsPaginatedResponse)
+  })
+_sym_db.RegisterMessage(ListModelsPaginatedResponse)
+
 GetModelRequest = _reflection.GeneratedProtocolMessageType('GetModelRequest', (_message.Message,), {
   'DESCRIPTOR' : _GETMODELREQUEST,
   '__module__' : 'qwak.models.models_pb2'
@@ -345,84 +362,88 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _MODELSMANAGEMENTSERVICE.methods_by_name['DeleteBranch']._serialized_options = b'\210\002\001'
   _MODELSMANAGEMENTSERVICE.methods_by_name['GetDeploymentDetails']._options = None
   _MODELSMANAGEMENTSERVICE.methods_by_name['GetDeploymentDetails']._serialized_options = b'\210\002\001'
-  _DEPLOYMENTMODELSTATUS._serialized_start=4484
-  _DEPLOYMENTMODELSTATUS._serialized_end=4563
-  _MODELSTATUS._serialized_start=4565
-  _MODELSTATUS._serialized_end=4632
-  _MEMORYUNIT._serialized_start=4634
-  _MEMORYUNIT._serialized_end=4689
-  _CREATEMODELREQUEST._serialized_start=246
-  _CREATEMODELREQUEST._serialized_end=401
-  _CREATEMODELRESPONSE._serialized_start=403
-  _CREATEMODELRESPONSE._serialized_end=442
-  _LISTMODELSREQUEST._serialized_start=444
-  _LISTMODELSREQUEST._serialized_end=483
-  _LISTMODELSRESPONSE._serialized_start=485
-  _LISTMODELSRESPONSE._serialized_end=552
-  _GETMODELREQUEST._serialized_start=554
-  _GETMODELREQUEST._serialized_end=627
-  _GETMODELRESPONSE._serialized_start=629
-  _GETMODELRESPONSE._serialized_end=693
-  _DELETEMODELREQUEST._serialized_start=695
-  _DELETEMODELREQUEST._serialized_end=753
-  _DELETEMODELRESPONSE._serialized_start=755
-  _DELETEMODELRESPONSE._serialized_end=776
-  _UPDATEMODELREQUEST._serialized_start=778
-  _UPDATEMODELREQUEST._serialized_end=891
-  _UPDATEMODELRESPONSE._serialized_start=893
-  _UPDATEMODELRESPONSE._serialized_end=969
-  _CREATEBRANCHREQUEST._serialized_start=971
-  _CREATEBRANCHREQUEST._serialized_end=1031
-  _CREATEBRANCHRESPONSE._serialized_start=1033
-  _CREATEBRANCHRESPONSE._serialized_end=1074
-  _DELETEBRANCHREQUEST._serialized_start=1076
-  _DELETEBRANCHREQUEST._serialized_end=1116
-  _DELETEBRANCHRESPONSE._serialized_start=1118
-  _DELETEBRANCHRESPONSE._serialized_end=1140
-  _ISMODELEXISTSREQUEST._serialized_start=1142
-  _ISMODELEXISTSREQUEST._serialized_end=1220
-  _ISMODELEXISTSRESPONSE._serialized_start=1222
-  _ISMODELEXISTSRESPONSE._serialized_end=1261
-  _MODELSPEC._serialized_start=1264
-  _MODELSPEC._serialized_end=1482
-  _MODELBRIEF._serialized_start=1485
-  _MODELBRIEF._serialized_end=1812
-  _MODELSPECCONFIGURATION._serialized_start=1814
-  _MODELSPECCONFIGURATION._serialized_end=1865
-  _MODEL._serialized_start=1868
-  _MODEL._serialized_end=2496
-  _MODELBUILDTYPE._serialized_start=2499
-  _MODELBUILDTYPE._serialized_end=2649
-  _SOURCE._serialized_start=2651
-  _SOURCE._serialized_end=2659
-  _MODELREGISTRYIMAGE._serialized_start=2661
-  _MODELREGISTRYIMAGE._serialized_end=2712
-  _BRANCH._serialized_start=2715
-  _BRANCH._serialized_end=2950
-  _BRANCH_BRANCHSTATUS._serialized_start=2882
-  _BRANCH_BRANCHSTATUS._serialized_end=2950
-  _GETDEPLOYMENTDETAILSREQUEST._serialized_start=2952
-  _GETDEPLOYMENTDETAILSREQUEST._serialized_end=3018
-  _GETDEPLOYMENTDETAILSRESPONSE._serialized_start=3020
-  _GETDEPLOYMENTDETAILSRESPONSE._serialized_end=3121
-  _DEPLOYMENTDETAILS._serialized_start=3124
-  _DEPLOYMENTDETAILS._serialized_end=3399
-  _LISTMODELSMETADATAREQUEST._serialized_start=3401
-  _LISTMODELSMETADATAREQUEST._serialized_end=3448
-  _LISTMODELSMETADATARESPONSE._serialized_start=3450
-  _LISTMODELSMETADATARESPONSE._serialized_end=3541
-  _GETMODELMETADATAREQUEST._serialized_start=3543
-  _GETMODELMETADATAREQUEST._serialized_end=3586
-  _GETMODELMETADATARESPONSE._serialized_start=3588
-  _GETMODELMETADATARESPONSE._serialized_end=3677
-  _ENVIRONMENTROUTES._serialized_start=3679
-  _ENVIRONMENTROUTES._serialized_end=3783
-  _MODELMETADATA._serialized_start=3786
-  _MODELMETADATA._serialized_end=4482
-  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_start=4259
-  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_end=4364
-  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_start=4366
-  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_end=4482
-  _MODELSMANAGEMENTSERVICE._serialized_start=4692
-  _MODELSMANAGEMENTSERVICE._serialized_end=5940
+  _DEPLOYMENTMODELSTATUS._serialized_start=4900
+  _DEPLOYMENTMODELSTATUS._serialized_end=4979
+  _MODELSTATUS._serialized_start=4981
+  _MODELSTATUS._serialized_end=5048
+  _MEMORYUNIT._serialized_start=5050
+  _MEMORYUNIT._serialized_end=5105
+  _CREATEMODELREQUEST._serialized_start=278
+  _CREATEMODELREQUEST._serialized_end=433
+  _CREATEMODELRESPONSE._serialized_start=435
+  _CREATEMODELRESPONSE._serialized_end=474
+  _LISTMODELSREQUEST._serialized_start=476
+  _LISTMODELSREQUEST._serialized_end=515
+  _LISTMODELSRESPONSE._serialized_start=517
+  _LISTMODELSRESPONSE._serialized_end=584
+  _LISTMODELSPAGINATEDREQUEST._serialized_start=587
+  _LISTMODELSPAGINATEDREQUEST._serialized_end=827
+  _LISTMODELSPAGINATEDRESPONSE._serialized_start=830
+  _LISTMODELSPAGINATEDRESPONSE._serialized_end=968
+  _GETMODELREQUEST._serialized_start=970
+  _GETMODELREQUEST._serialized_end=1043
+  _GETMODELRESPONSE._serialized_start=1045
+  _GETMODELRESPONSE._serialized_end=1109
+  _DELETEMODELREQUEST._serialized_start=1111
+  _DELETEMODELREQUEST._serialized_end=1169
+  _DELETEMODELRESPONSE._serialized_start=1171
+  _DELETEMODELRESPONSE._serialized_end=1192
+  _UPDATEMODELREQUEST._serialized_start=1194
+  _UPDATEMODELREQUEST._serialized_end=1307
+  _UPDATEMODELRESPONSE._serialized_start=1309
+  _UPDATEMODELRESPONSE._serialized_end=1385
+  _CREATEBRANCHREQUEST._serialized_start=1387
+  _CREATEBRANCHREQUEST._serialized_end=1447
+  _CREATEBRANCHRESPONSE._serialized_start=1449
+  _CREATEBRANCHRESPONSE._serialized_end=1490
+  _DELETEBRANCHREQUEST._serialized_start=1492
+  _DELETEBRANCHREQUEST._serialized_end=1532
+  _DELETEBRANCHRESPONSE._serialized_start=1534
+  _DELETEBRANCHRESPONSE._serialized_end=1556
+  _ISMODELEXISTSREQUEST._serialized_start=1558
+  _ISMODELEXISTSREQUEST._serialized_end=1636
+  _ISMODELEXISTSRESPONSE._serialized_start=1638
+  _ISMODELEXISTSRESPONSE._serialized_end=1677
+  _MODELSPEC._serialized_start=1680
+  _MODELSPEC._serialized_end=1898
+  _MODELBRIEF._serialized_start=1901
+  _MODELBRIEF._serialized_end=2228
+  _MODELSPECCONFIGURATION._serialized_start=2230
+  _MODELSPECCONFIGURATION._serialized_end=2281
+  _MODEL._serialized_start=2284
+  _MODEL._serialized_end=2912
+  _MODELBUILDTYPE._serialized_start=2915
+  _MODELBUILDTYPE._serialized_end=3065
+  _SOURCE._serialized_start=3067
+  _SOURCE._serialized_end=3075
+  _MODELREGISTRYIMAGE._serialized_start=3077
+  _MODELREGISTRYIMAGE._serialized_end=3128
+  _BRANCH._serialized_start=3131
+  _BRANCH._serialized_end=3366
+  _BRANCH_BRANCHSTATUS._serialized_start=3298
+  _BRANCH_BRANCHSTATUS._serialized_end=3366
+  _GETDEPLOYMENTDETAILSREQUEST._serialized_start=3368
+  _GETDEPLOYMENTDETAILSREQUEST._serialized_end=3434
+  _GETDEPLOYMENTDETAILSRESPONSE._serialized_start=3436
+  _GETDEPLOYMENTDETAILSRESPONSE._serialized_end=3537
+  _DEPLOYMENTDETAILS._serialized_start=3540
+  _DEPLOYMENTDETAILS._serialized_end=3815
+  _LISTMODELSMETADATAREQUEST._serialized_start=3817
+  _LISTMODELSMETADATAREQUEST._serialized_end=3864
+  _LISTMODELSMETADATARESPONSE._serialized_start=3866
+  _LISTMODELSMETADATARESPONSE._serialized_end=3957
+  _GETMODELMETADATAREQUEST._serialized_start=3959
+  _GETMODELMETADATAREQUEST._serialized_end=4002
+  _GETMODELMETADATARESPONSE._serialized_start=4004
+  _GETMODELMETADATARESPONSE._serialized_end=4093
+  _ENVIRONMENTROUTES._serialized_start=4095
+  _ENVIRONMENTROUTES._serialized_end=4199
+  _MODELMETADATA._serialized_start=4202
+  _MODELMETADATA._serialized_end=4898
+  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_start=4675
+  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_end=4780
+  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_start=4782
+  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_end=4898
+  _MODELSMANAGEMENTSERVICE._serialized_start=5108
+  _MODELSMANAGEMENTSERVICE._serialized_end=6484
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/models/models_pb2.pyi

@@ -13,6 +13,7 @@ import qwak.audience.v1.audience_pb2
 import qwak.build.v1.build_pb2
 import qwak.builds.builds_pb2
 import qwak.deployment.deployment_pb2
+import qwak.models.models_query_pb2
 import qwak.projects.jfrog_project_spec_pb2
 import sys
 import typing
@@ -150,6 +151,62 @@ class ListModelsResponse(google.protobuf.message.Message):
 
 global___ListModelsResponse = ListModelsResponse
 
+class ListModelsPaginatedRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PAGE_SIZE_FIELD_NUMBER: builtins.int
+    CURSOR_FIELD_NUMBER: builtins.int
+    SORT_MODELS_FIELD_NUMBER: builtins.int
+    FILTER_FIELD_NUMBER: builtins.int
+    page_size: builtins.int
+    cursor: builtins.str
+    @property
+    def sort_models(self) -> qwak.models.models_query_pb2.SortModels: ...
+    @property
+    def filter(self) -> qwak.models.models_query_pb2.Filter: ...
+    def __init__(
+        self,
+        *,
+        page_size: builtins.int = ...,
+        cursor: builtins.str = ...,
+        sort_models: qwak.models.models_query_pb2.SortModels | None = ...,
+        filter: qwak.models.models_query_pb2.Filter | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["cursor", b"cursor", "cursor_opt", b"cursor_opt", "filter", b"filter", "filter_opt", b"filter_opt", "page_size", b"page_size", "page_size_opt", b"page_size_opt", "sort_models", b"sort_models", "sort_models_opt", b"sort_models_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cursor", b"cursor", "cursor_opt", b"cursor_opt", "filter", b"filter", "filter_opt", b"filter_opt", "page_size", b"page_size", "page_size_opt", b"page_size_opt", "sort_models", b"sort_models", "sort_models_opt", b"sort_models_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["cursor_opt", b"cursor_opt"]) -> typing_extensions.Literal["cursor"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["filter_opt", b"filter_opt"]) -> typing_extensions.Literal["filter"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["page_size_opt", b"page_size_opt"]) -> typing_extensions.Literal["page_size"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["sort_models_opt", b"sort_models_opt"]) -> typing_extensions.Literal["sort_models"] | None: ...
+
+global___ListModelsPaginatedRequest = ListModelsPaginatedRequest
+
+class ListModelsPaginatedResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODELS_FIELD_NUMBER: builtins.int
+    METADATA_FIELD_NUMBER: builtins.int
+    @property
+    def models(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Model]:
+        """List of models matching the query criteria."""
+    @property
+    def metadata(self) -> qwak.models.models_query_pb2.PaginationMetadata:
+        """Pagination information about the result set."""
+    def __init__(
+        self,
+        *,
+        models: collections.abc.Iterable[global___Model] | None = ...,
+        metadata: qwak.models.models_query_pb2.PaginationMetadata | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["metadata", b"metadata"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["metadata", b"metadata", "models", b"models"]) -> None: ...
+
+global___ListModelsPaginatedResponse = ListModelsPaginatedResponse
+
 class GetModelRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

_qwak_proto/qwak/models/models_pb2_grpc.py

@@ -24,6 +24,11 @@ class ModelsManagementServiceStub(object):
                 request_serializer=qwak_dot_models_dot_models__pb2.ListModelsRequest.SerializeToString,
                 response_deserializer=qwak_dot_models_dot_models__pb2.ListModelsResponse.FromString,
                 )
+        self.ListModelsPaginated = channel.unary_unary(
+                '/qwak.models.management.ModelsManagementService/ListModelsPaginated',
+                request_serializer=qwak_dot_models_dot_models__pb2.ListModelsPaginatedRequest.SerializeToString,
+                response_deserializer=qwak_dot_models_dot_models__pb2.ListModelsPaginatedResponse.FromString,
+                )
         self.GetModel = channel.unary_unary(
                 '/qwak.models.management.ModelsManagementService/GetModel',
                 request_serializer=qwak_dot_models_dot_models__pb2.GetModelRequest.SerializeToString,
@@ -88,6 +93,13 @@ class ModelsManagementServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListModelsPaginated(self, request, context):
+        """Retrieves a paginated list of models based on the provided filters and pagination parameters.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def GetModel(self, request, context):
         """Get model
         """
@@ -159,6 +171,11 @@ def add_ModelsManagementServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_models_dot_models__pb2.ListModelsRequest.FromString,
                     response_serializer=qwak_dot_models_dot_models__pb2.ListModelsResponse.SerializeToString,
             ),
+            'ListModelsPaginated': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListModelsPaginated,
+                    request_deserializer=qwak_dot_models_dot_models__pb2.ListModelsPaginatedRequest.FromString,
+                    response_serializer=qwak_dot_models_dot_models__pb2.ListModelsPaginatedResponse.SerializeToString,
+            ),
             'GetModel': grpc.unary_unary_rpc_method_handler(
                     servicer.GetModel,
                     request_deserializer=qwak_dot_models_dot_models__pb2.GetModelRequest.FromString,
@@ -248,6 +265,23 @@ class ModelsManagementService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListModelsPaginated(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, '/qwak.models.management.ModelsManagementService/ListModelsPaginated',
+            qwak_dot_models_dot_models__pb2.ListModelsPaginatedRequest.SerializeToString,
+            qwak_dot_models_dot_models__pb2.ListModelsPaginatedResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def GetModel(request,
             target,

_qwak_proto/qwak/models/models_query_pb2.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/models/models_query.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eqwak/models/models_query.proto\x12\x16qwak.models.management\"\x99\x01\n\nSortModels\x12\x31\n\x07sort_by\x18\x01 \x01(\x0e\x32\x1e.qwak.models.management.SortByH\x00\x12\x37\n\nsort_order\x18\x02 \x01(\x0e\x32!.qwak.models.management.SortOrderH\x01\x42\r\n\x0bsort_by_optB\x10\n\x0esort_order_opt\"\xc8\x01\n\x06\x46ilter\x12\x1e\n\x14model_name_substring\x18\x01 \x01(\tH\x00\x12\x19\n\x11model_group_names\x18\x02 \x03(\t\x12V\n\x17\x64\x65ployment_model_status\x18\x03 \x01(\x0e\x32\x33.qwak.models.management.DeploymentModelStatusFilterH\x01\x42\x0c\n\nsearch_optB\x1d\n\x1b\x64\x65ployment_model_status_opt\"<\n\x12PaginationMetadata\x12\x11\n\tpage_size\x18\x01 \x01(\x05\x12\x13\n\x0bnext_cursor\x18\x02 \x01(\t*J\n\x06SortBy\x12\x13\n\x0fSORT_BY_INVALID\x10\x00\x12\x10\n\x0cSORT_BY_NAME\x10\x01\x12\x19\n\x15SORT_BY_CREATION_DATE\x10\x02*L\n\tSortOrder\x12\x16\n\x12SORT_ORDER_INVALID\x10\x00\x12\x12\n\x0eSORT_ORDER_ASC\x10\x01\x12\x13\n\x0fSORT_ORDER_DESC\x10\x02*\xa5\x01\n\x1b\x44\x65ploymentModelStatusFilter\x12*\n&DEPLOYMENT_MODEL_STATUS_FILTER_INVALID\x10\x00\x12+\n\'DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED\x10\x01\x12-\n)DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED\x10\x02\x42\x36\n!com.qwak.ai.management.models.apiB\x0fModelQueryProtoP\x01\x62\x06proto3')
+
+_SORTBY = DESCRIPTOR.enum_types_by_name['SortBy']
+SortBy = enum_type_wrapper.EnumTypeWrapper(_SORTBY)
+_SORTORDER = DESCRIPTOR.enum_types_by_name['SortOrder']
+SortOrder = enum_type_wrapper.EnumTypeWrapper(_SORTORDER)
+_DEPLOYMENTMODELSTATUSFILTER = DESCRIPTOR.enum_types_by_name['DeploymentModelStatusFilter']
+DeploymentModelStatusFilter = enum_type_wrapper.EnumTypeWrapper(_DEPLOYMENTMODELSTATUSFILTER)
+SORT_BY_INVALID = 0
+SORT_BY_NAME = 1
+SORT_BY_CREATION_DATE = 2
+SORT_ORDER_INVALID = 0
+SORT_ORDER_ASC = 1
+SORT_ORDER_DESC = 2
+DEPLOYMENT_MODEL_STATUS_FILTER_INVALID = 0
+DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED = 1
+DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED = 2
+
+
+_SORTMODELS = DESCRIPTOR.message_types_by_name['SortModels']
+_FILTER = DESCRIPTOR.message_types_by_name['Filter']
+_PAGINATIONMETADATA = DESCRIPTOR.message_types_by_name['PaginationMetadata']
+SortModels = _reflection.GeneratedProtocolMessageType('SortModels', (_message.Message,), {
+  'DESCRIPTOR' : _SORTMODELS,
+  '__module__' : 'qwak.models.models_query_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.SortModels)
+  })
+_sym_db.RegisterMessage(SortModels)
+
+Filter = _reflection.GeneratedProtocolMessageType('Filter', (_message.Message,), {
+  'DESCRIPTOR' : _FILTER,
+  '__module__' : 'qwak.models.models_query_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.Filter)
+  })
+_sym_db.RegisterMessage(Filter)
+
+PaginationMetadata = _reflection.GeneratedProtocolMessageType('PaginationMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _PAGINATIONMETADATA,
+  '__module__' : 'qwak.models.models_query_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.PaginationMetadata)
+  })
+_sym_db.RegisterMessage(PaginationMetadata)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!com.qwak.ai.management.models.apiB\017ModelQueryProtoP\001'
+  _SORTBY._serialized_start=479
+  _SORTBY._serialized_end=553
+  _SORTORDER._serialized_start=555
+  _SORTORDER._serialized_end=631
+  _DEPLOYMENTMODELSTATUSFILTER._serialized_start=634
+  _DEPLOYMENTMODELSTATUSFILTER._serialized_end=799
+  _SORTMODELS._serialized_start=59
+  _SORTMODELS._serialized_end=212
+  _FILTER._serialized_start=215
+  _FILTER._serialized_end=415
+  _PAGINATIONMETADATA._serialized_start=417
+  _PAGINATIONMETADATA._serialized_end=477
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/models/models_query_pb2.pyi

@@ -0,0 +1,157 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _SortBy:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _SortByEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SortBy.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    SORT_BY_INVALID: _SortBy.ValueType  # 0
+    SORT_BY_NAME: _SortBy.ValueType  # 1
+    """Sort models by name (default)."""
+    SORT_BY_CREATION_DATE: _SortBy.ValueType  # 2
+    """Sort models by their creation date."""
+
+class SortBy(_SortBy, metaclass=_SortByEnumTypeWrapper): ...
+
+SORT_BY_INVALID: SortBy.ValueType  # 0
+SORT_BY_NAME: SortBy.ValueType  # 1
+"""Sort models by name (default)."""
+SORT_BY_CREATION_DATE: SortBy.ValueType  # 2
+"""Sort models by their creation date."""
+global___SortBy = SortBy
+
+class _SortOrder:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _SortOrderEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SortOrder.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    SORT_ORDER_INVALID: _SortOrder.ValueType  # 0
+    SORT_ORDER_ASC: _SortOrder.ValueType  # 1
+    """Sort in ascending order (default)."""
+    SORT_ORDER_DESC: _SortOrder.ValueType  # 2
+    """Sort in descending order."""
+
+class SortOrder(_SortOrder, metaclass=_SortOrderEnumTypeWrapper): ...
+
+SORT_ORDER_INVALID: SortOrder.ValueType  # 0
+SORT_ORDER_ASC: SortOrder.ValueType  # 1
+"""Sort in ascending order (default)."""
+SORT_ORDER_DESC: SortOrder.ValueType  # 2
+"""Sort in descending order."""
+global___SortOrder = SortOrder
+
+class _DeploymentModelStatusFilter:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _DeploymentModelStatusFilterEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DeploymentModelStatusFilter.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DEPLOYMENT_MODEL_STATUS_FILTER_INVALID: _DeploymentModelStatusFilter.ValueType  # 0
+    DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED: _DeploymentModelStatusFilter.ValueType  # 1
+    DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED: _DeploymentModelStatusFilter.ValueType  # 2
+
+class DeploymentModelStatusFilter(_DeploymentModelStatusFilter, metaclass=_DeploymentModelStatusFilterEnumTypeWrapper): ...
+
+DEPLOYMENT_MODEL_STATUS_FILTER_INVALID: DeploymentModelStatusFilter.ValueType  # 0
+DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED: DeploymentModelStatusFilter.ValueType  # 1
+DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED: DeploymentModelStatusFilter.ValueType  # 2
+global___DeploymentModelStatusFilter = DeploymentModelStatusFilter
+
+class SortModels(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SORT_BY_FIELD_NUMBER: builtins.int
+    SORT_ORDER_FIELD_NUMBER: builtins.int
+    sort_by: global___SortBy.ValueType
+    sort_order: global___SortOrder.ValueType
+    def __init__(
+        self,
+        *,
+        sort_by: global___SortBy.ValueType = ...,
+        sort_order: global___SortOrder.ValueType = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["sort_by", b"sort_by", "sort_by_opt", b"sort_by_opt", "sort_order", b"sort_order", "sort_order_opt", b"sort_order_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sort_by", b"sort_by", "sort_by_opt", b"sort_by_opt", "sort_order", b"sort_order", "sort_order_opt", b"sort_order_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["sort_by_opt", b"sort_by_opt"]) -> typing_extensions.Literal["sort_by"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["sort_order_opt", b"sort_order_opt"]) -> typing_extensions.Literal["sort_order"] | None: ...
+
+global___SortModels = SortModels
+
+class Filter(google.protobuf.message.Message):
+    """Filter models by specific properties using a filter expression.
+    The following fields are used to filter the models returned by the query. An AND operation is applied between each of these fields.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_NAME_SUBSTRING_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_NAMES_FIELD_NUMBER: builtins.int
+    DEPLOYMENT_MODEL_STATUS_FIELD_NUMBER: builtins.int
+    model_name_substring: builtins.str
+    @property
+    def model_group_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Filter models by one or more model group names.
+        Return models that belong to one of the specified model groups.
+        Optional field. If not specified, models from all groups are included.
+        """
+    deployment_model_status: global___DeploymentModelStatusFilter.ValueType
+    def __init__(
+        self,
+        *,
+        model_name_substring: builtins.str = ...,
+        model_group_names: collections.abc.Iterable[builtins.str] | None = ...,
+        deployment_model_status: global___DeploymentModelStatusFilter.ValueType = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["deployment_model_status", b"deployment_model_status", "deployment_model_status_opt", b"deployment_model_status_opt", "model_name_substring", b"model_name_substring", "search_opt", b"search_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["deployment_model_status", b"deployment_model_status", "deployment_model_status_opt", b"deployment_model_status_opt", "model_group_names", b"model_group_names", "model_name_substring", b"model_name_substring", "search_opt", b"search_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["deployment_model_status_opt", b"deployment_model_status_opt"]) -> typing_extensions.Literal["deployment_model_status"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["search_opt", b"search_opt"]) -> typing_extensions.Literal["model_name_substring"] | None: ...
+
+global___Filter = Filter
+
+class PaginationMetadata(google.protobuf.message.Message):
+    """Metadata for pagination."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PAGE_SIZE_FIELD_NUMBER: builtins.int
+    NEXT_CURSOR_FIELD_NUMBER: builtins.int
+    page_size: builtins.int
+    """Number of items per page."""
+    next_cursor: builtins.str
+    """Cursor for the next page.
+    If there are no more pages, this field will be empty.
+    """
+    def __init__(
+        self,
+        *,
+        page_size: builtins.int = ...,
+        next_cursor: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["next_cursor", b"next_cursor", "page_size", b"page_size"]) -> None: ...
+
+global___PaginationMetadata = PaginationMetadata

_qwak_proto/qwak/models/models_query_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

qwak/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for qwak-core."""
 
 __author__ = "Qwak.ai"
-__version__ = "0.4.270"
+__version__ = "0.4.272"
 
 from qwak.inner.di_configuration import wire_dependencies
 from qwak.model.experiment_tracking import log_metric, log_param

{qwak_core-0.4.270.dist-info → qwak_core-0.4.272.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qwak-core
-Version: 0.4.270
+Version: 0.4.272
 Summary: Qwak Core contains the necessary objects and communication tools for using the Qwak Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{qwak_core-0.4.270.dist-info → qwak_core-0.4.272.dist-info}/RECORD

@@ -489,9 +489,18 @@ _qwak_proto/qwak/logging/log_source_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHD
 _qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2.py,sha256=HCM2-Ah66CkK5Io1jCouxgDHT5HmSgAYcHKvsfTZw48,6751
 _qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi,sha256=G5CaQGczrZ9kGsMbmJXMFWAwnqMZ4jPSl9KQpSq0yMo,12345
 _qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-_qwak_proto/qwak/models/models_pb2.py,sha256=GfzTIawDHmJS0wUwHkgB_6tQiwObZFz7FQw1mwjkUtg,30090
-_qwak_proto/qwak/models/models_pb2.pyi,sha256=ALci3H03JCyHhZM3QYr43xwshfLZvAnujwhL-Tr8XQQ,35787
-_qwak_proto/qwak/models/models_pb2_grpc.py,sha256=Iv1z-S7Ju2artlN-LOT6_plKq2aCjB2F7KqPW9LMSRw,19830
+_qwak_proto/qwak/model_group/model_group_pb2.py,sha256=fP198ECc5Ca3WccDvRlcaNavZgWeXKrzKyFKZXL52So,3427
+_qwak_proto/qwak/model_group/model_group_pb2.pyi,sha256=6xP_BE9OccBWV784jfKS5BcBabUcDL22XtrtWq2_hlU,3038
+_qwak_proto/qwak/model_group/model_group_pb2_grpc.py,sha256=UyRBABlYr-W3uUKUnStENcfnc0mXOtShh8_mjpzpro0,3073
+_qwak_proto/qwak/model_group/model_group_repository_details_pb2.py,sha256=t7oUXXLQvgxNuLyrA0qzuyfazRXK-uCIArXD_4s_1J4,5022
+_qwak_proto/qwak/model_group/model_group_repository_details_pb2.pyi,sha256=h5ciIby9peqjZqHosajfrHdmX5kosmzDJiadoCVpTmc,4119
+_qwak_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+_qwak_proto/qwak/models/models_pb2.py,sha256=dypKmBnjqzAmSLkXRPx0QMXiB1D-mUNxRu5pGqxVj68,32093
+_qwak_proto/qwak/models/models_pb2.pyi,sha256=6h6PCqMOwZgSf77CUEx_SiRWX0uTxjiFMysVG_Z0Qvc,39027
+_qwak_proto/qwak/models/models_pb2_grpc.py,sha256=2V7rVTiSaJcjw38OjUgtvYkouHMt6nAVJb0DE22ujE0,21680
+_qwak_proto/qwak/models/models_query_pb2.py,sha256=GlkN0jKFqX2x4fbX1zLnk0Jbr3nlfLNINmSh5_JGqHo,4322
+_qwak_proto/qwak/models/models_query_pb2.pyi,sha256=cZO7EA_JetDIQbC-9hg42G07DJMBC6Adgn6IMhnhhNQ,7456
+_qwak_proto/qwak/models/models_query_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 _qwak_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.py,sha256=Wj1nO5GzAigK3axuY2Qwc3WeENLLwLvCj3FcQ6Zuhpo,10745
 _qwak_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.pyi,sha256=0l-75-C7DniOm1DOY8y9DbxKuU4_930V6KS6bjCrM2c,6790
 _qwak_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2_grpc.py,sha256=YuYQU-wACkHV0h9aBlXJ8qFFT2p85LGXswfkme74_s0,13657
@@ -594,7 +603,7 @@ _qwak_proto/qwak/workspace/workspace_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 _qwak_proto/qwak/workspace/workspace_service_pb2.py,sha256=AB3C9S_AbOD7Nx1Ni4j1rW6PNtYTV1zjiqFQk-goQ74,21429
 _qwak_proto/qwak/workspace/workspace_service_pb2.pyi,sha256=nKKCHwnovZhsy8TSVmdz-Vtl0nviOOoX56HD-41Xo08,13726
 _qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py,sha256=yKGuexxTBza99Ihe0DSTniV2ZSd_AG47inHenqfi890,27193
-qwak/__init__.py,sha256=8jXgmVlbysc3jtH1DoAf_UgYwa9JJ1xrkbhLapZho4A,587
+qwak/__init__.py,sha256=8jN3thTv7G9Gdxw5tA7nuf_YCiVmgBUIxZDTgvG2Mew,587
 qwak/automations/__init__.py,sha256=qFZRvCxUUn8gcxkJR0v19ulHW2oJ0x6-Rif7HiheDP4,1522
 qwak/automations/automation_executions.py,sha256=5MeH_epYYWb8NKXgAozwT_jPyyUDednBHG7izloi7RY,3228
 qwak/automations/automations.py,sha256=3yx8e2v0uSKDnXbqyknasyEoQ5vxGni6K40Hbi1_zkk,12599
@@ -1068,6 +1077,6 @@ qwak_services_mock/mocks/workspace_manager_service_mock.py,sha256=O9ZSwln4T4kHVk
 qwak_services_mock/services_mock.py,sha256=zXtHcX8a_acz7ynxuCBxxVpHpde7aAGjIn6Uw52LY1s,19593
 qwak_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-qwak_core-0.4.270.dist-info/METADATA,sha256=jSxv4Gj_3ccDrKxH_j4k2TT3Y541H8fkqfIBcaAMbU0,2549
-qwak_core-0.4.270.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-qwak_core-0.4.270.dist-info/RECORD,,
+qwak_core-0.4.272.dist-info/METADATA,sha256=6QwKVVwymTNEAzfNvoarcqgRNGXTLFTD_K_GJ_ZSP7I,2549
+qwak_core-0.4.272.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+qwak_core-0.4.272.dist-info/RECORD,,