qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/models/models_pb2.pyi

@@ -13,6 +13,8 @@ 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
 
@@ -85,16 +87,21 @@ class CreateModelRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     MODEL_SPEC_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_SPEC_FIELD_NUMBER: builtins.int
     @property
     def model_spec(self) -> global___ModelSpec:
         """Model spec to add"""
+    @property
+    def jfrog_project_spec(self) -> qwak.projects.jfrog_project_spec_pb2.ModelRepositoryJFrogSpec:
+        """JFrog project spec to add in case we want to associate the created model to another project than default"""
     def __init__(
         self,
         *,
         model_spec: global___ModelSpec | None = ...,
+        jfrog_project_spec: qwak.projects.jfrog_project_spec_pb2.ModelRepositoryJFrogSpec | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["model_spec", b"model_spec"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["model_spec", b"model_spec"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["jfrog_project_spec", b"jfrog_project_spec", "model_spec", b"model_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["jfrog_project_spec", b"jfrog_project_spec", "model_spec", b"model_spec"]) -> None: ...
 
 global___CreateModelRequest = CreateModelRequest
 
@@ -144,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
 
@@ -348,24 +411,33 @@ class ModelSpec(google.protobuf.message.Message):
     MODEL_NAMED_ID_FIELD_NUMBER: builtins.int
     DISPLAY_NAME_FIELD_NUMBER: builtins.int
     PROJECT_ID_FIELD_NUMBER: builtins.int
+    PROJECT_NAME_FIELD_NUMBER: builtins.int
     MODEL_DESCRIPTION_FIELD_NUMBER: builtins.int
+    MODEL_BUILD_TYPE_FIELD_NUMBER: builtins.int
     model_named_id: builtins.str
     """The model id"""
     display_name: builtins.str
     """The display name"""
     project_id: builtins.str
-    """The project id"""
+    project_name: builtins.str
     model_description: builtins.str
     """The model description"""
+    @property
+    def model_build_type(self) -> global___ModelBuildType:
+        """The Model build type"""
     def __init__(
         self,
         *,
         model_named_id: builtins.str = ...,
         display_name: builtins.str = ...,
         project_id: builtins.str = ...,
+        project_name: builtins.str = ...,
         model_description: builtins.str = ...,
+        model_build_type: global___ModelBuildType | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["display_name", b"display_name", "model_description", b"model_description", "model_named_id", b"model_named_id", "project_id", b"project_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_build_type", b"model_build_type", "project_id", b"project_id", "project_identifier", b"project_identifier", "project_name", b"project_name"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["display_name", b"display_name", "model_build_type", b"model_build_type", "model_description", b"model_description", "model_named_id", b"model_named_id", "project_id", b"project_id", "project_identifier", b"project_identifier", "project_name", b"project_name"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["project_identifier", b"project_identifier"]) -> typing_extensions.Literal["project_id", "project_name"] | None: ...
 
 global___ModelSpec = ModelSpec
 
@@ -450,6 +522,9 @@ class Model(google.protobuf.message.Message):
     DEPLOYMENT_MODEL_STATUS_FIELD_NUMBER: builtins.int
     UUID_FIELD_NUMBER: builtins.int
     MODEL_TYPE_FIELD_NUMBER: builtins.int
+    PROJECT_NAME_FIELD_NUMBER: builtins.int
+    MODEL_BUILD_TYPE_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
     model_id: builtins.str
     """The model id"""
     display_name: builtins.str
@@ -479,6 +554,13 @@ class Model(google.protobuf.message.Message):
     """unique id for model."""
     model_type: qwak.deployment.deployment_pb2.KubeDeploymentType.ValueType
     """The deployment type of the model"""
+    project_name: builtins.str
+    """Project name"""
+    @property
+    def model_build_type(self) -> global___ModelBuildType:
+        """Model build type"""
+    jfrog_project_key: builtins.str
+    """Jfrog project key"""
     def __init__(
         self,
         *,
@@ -495,12 +577,59 @@ class Model(google.protobuf.message.Message):
         deployment_model_status: global___DeploymentModelStatus.ValueType = ...,
         uuid: builtins.str = ...,
         model_type: qwak.deployment.deployment_pb2.KubeDeploymentType.ValueType = ...,
+        project_name: builtins.str = ...,
+        model_build_type: global___ModelBuildType | None = ...,
+        jfrog_project_key: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "last_modified_at", b"last_modified_at"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["branches", b"branches", "created_at", b"created_at", "created_by", b"created_by", "deployment_model_status", b"deployment_model_status", "display_name", b"display_name", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by", "model_description", b"model_description", "model_id", b"model_id", "model_status", b"model_status", "model_type", b"model_type", "project_id", b"project_id", "uuid", b"uuid"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "last_modified_at", b"last_modified_at", "model_build_type", b"model_build_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["branches", b"branches", "created_at", b"created_at", "created_by", b"created_by", "deployment_model_status", b"deployment_model_status", "display_name", b"display_name", "jfrog_project_key", b"jfrog_project_key", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by", "model_build_type", b"model_build_type", "model_description", b"model_description", "model_id", b"model_id", "model_status", b"model_status", "model_type", b"model_type", "project_id", b"project_id", "project_name", b"project_name", "uuid", b"uuid"]) -> None: ...
 
 global___Model = Model
 
+class ModelBuildType(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOURCE_FIELD_NUMBER: builtins.int
+    MODEL_REGISTRY_IMAGE_FIELD_NUMBER: builtins.int
+    @property
+    def source(self) -> global___Source: ...
+    @property
+    def model_registry_image(self) -> global___ModelRegistryImage: ...
+    def __init__(
+        self,
+        *,
+        source: global___Source | None = ...,
+        model_registry_image: global___ModelRegistryImage | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_registry_image", b"model_registry_image", "source", b"source", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_registry_image", b"model_registry_image", "source", b"source", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["source", "model_registry_image"] | None: ...
+
+global___ModelBuildType = ModelBuildType
+
+class Source(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___Source = Source
+
+class ModelRegistryImage(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LAST_IMAGE_IDENTIFIER_FIELD_NUMBER: builtins.int
+    last_image_identifier: builtins.str
+    def __init__(
+        self,
+        *,
+        last_image_identifier: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["last_image_identifier", b"last_image_identifier"]) -> None: ...
+
+global___ModelRegistryImage = ModelRegistryImage
+
 class Branch(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -783,3 +912,89 @@ class ModelMetadata(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["audience_routes_by_environment", b"audience_routes_by_environment", "audience_routes_grouped_by_environment", b"audience_routes_grouped_by_environment", "build", b"build", "builds", b"builds", "deployment_details", b"deployment_details", "model", b"model"]) -> None: ...
 
 global___ModelMetadata = ModelMetadata
+
+class CheckModelCreationAllowedRequest(google.protobuf.message.Message):
+    """*
+    Checks if creation allowed for the following details
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    model_group_name: builtins.str
+    jfrog_project_key: builtins.str
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        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", "model_id", b"model_id"]) -> None: ...
+
+global___CheckModelCreationAllowedRequest = CheckModelCreationAllowedRequest
+
+class CheckModelCreationAllowedResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    IS_CREATABLE_FIELD_NUMBER: builtins.int
+    MODEL_EXISTS_FIELD_NUMBER: builtins.int
+    MODEL_NOT_EXISTS_FIELD_NUMBER: builtins.int
+    MODEL_EXISTS_WITH_MISMATCH_FIELD_NUMBER: builtins.int
+    is_creatable: builtins.bool
+    """set to true if allowed to create, otherwise false"""
+    @property
+    def model_exists(self) -> global___ModelExists: ...
+    @property
+    def model_not_exists(self) -> global___ModelNotExists: ...
+    @property
+    def model_exists_with_mismatch(self) -> global___ModelExistsWithMismatch: ...
+    def __init__(
+        self,
+        *,
+        is_creatable: builtins.bool = ...,
+        model_exists: global___ModelExists | None = ...,
+        model_not_exists: global___ModelNotExists | None = ...,
+        model_exists_with_mismatch: global___ModelExistsWithMismatch | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_exists", b"model_exists", "model_exists_with_mismatch", b"model_exists_with_mismatch", "model_not_exists", b"model_not_exists", "status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["is_creatable", b"is_creatable", "model_exists", b"model_exists", "model_exists_with_mismatch", b"model_exists_with_mismatch", "model_not_exists", b"model_not_exists", "status", b"status"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["status", b"status"]) -> typing_extensions.Literal["model_exists", "model_not_exists", "model_exists_with_mismatch"] | None: ...
+
+global___CheckModelCreationAllowedResponse = CheckModelCreationAllowedResponse
+
+class ModelExists(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ModelExists = ModelExists
+
+class ModelNotExists(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ModelNotExists = ModelNotExists
+
+class ModelExistsWithMismatch(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DETAILS_FIELD_NUMBER: builtins.int
+    @property
+    def details(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        details: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["details", b"details"]) -> None: ...
+
+global___ModelExistsWithMismatch = ModelExistsWithMismatch

_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,
@@ -69,6 +74,11 @@ class ModelsManagementServiceStub(object):
                 request_serializer=qwak_dot_models_dot_models__pb2.GetModelMetadataRequest.SerializeToString,
                 response_deserializer=qwak_dot_models_dot_models__pb2.GetModelMetadataResponse.FromString,
                 )
+        self.CheckModelCreationAllowed = channel.unary_unary(
+                '/qwak.models.management.ModelsManagementService/CheckModelCreationAllowed',
+                request_serializer=qwak_dot_models_dot_models__pb2.CheckModelCreationAllowedRequest.SerializeToString,
+                response_deserializer=qwak_dot_models_dot_models__pb2.CheckModelCreationAllowedResponse.FromString,
+                )
 
 
 class ModelsManagementServiceServicer(object):
@@ -88,6 +98,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
         """
@@ -146,6 +163,13 @@ class ModelsManagementServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CheckModelCreationAllowed(self, request, context):
+        """Check if model creation allowed for the provided information: model group name, jfrog project key and model id
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_ModelsManagementServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -159,6 +183,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,
@@ -204,6 +233,11 @@ def add_ModelsManagementServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_models_dot_models__pb2.GetModelMetadataRequest.FromString,
                     response_serializer=qwak_dot_models_dot_models__pb2.GetModelMetadataResponse.SerializeToString,
             ),
+            'CheckModelCreationAllowed': grpc.unary_unary_rpc_method_handler(
+                    servicer.CheckModelCreationAllowed,
+                    request_deserializer=qwak_dot_models_dot_models__pb2.CheckModelCreationAllowedRequest.FromString,
+                    response_serializer=qwak_dot_models_dot_models__pb2.CheckModelCreationAllowedResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.models.management.ModelsManagementService', rpc_method_handlers)
@@ -248,6 +282,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,
@@ -400,3 +451,20 @@ class ModelsManagementService(object):
             qwak_dot_models_dot_models__pb2.GetModelMetadataResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def CheckModelCreationAllowed(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/CheckModelCreationAllowed',
+            qwak_dot_models_dot_models__pb2.CheckModelCreationAllowedRequest.SerializeToString,
+            qwak_dot_models_dot_models__pb2.CheckModelCreationAllowedResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_qwak_proto/qwak/models/models_query_pb2.py

@@ -0,0 +1,37 @@
+# -*- 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 import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.models.models_query_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!com.qwak.ai.management.models.apiB\017ModelQueryProtoP\001'
+  _globals['_SORTBY']._serialized_start=479
+  _globals['_SORTBY']._serialized_end=553
+  _globals['_SORTORDER']._serialized_start=555
+  _globals['_SORTORDER']._serialized_end=631
+  _globals['_DEPLOYMENTMODELSTATUSFILTER']._serialized_start=634
+  _globals['_DEPLOYMENTMODELSTATUSFILTER']._serialized_end=799
+  _globals['_SORTMODELS']._serialized_start=59
+  _globals['_SORTMODELS']._serialized_end=212
+  _globals['_FILTER']._serialized_start=215
+  _globals['_FILTER']._serialized_end=415
+  _globals['_PAGINATIONMETADATA']._serialized_start=417
+  _globals['_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. Returns models that belong to any of the specified model groups.
+        Optional field. If not specified, models from all groups are included.
+        The maximum number of model group names allowed is 100, with each model group name having a maximum length of 100 characters.
+        """
+    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 models returned in the current page."""
+    next_cursor: builtins.str
+    """Cursor for the next page.
+    If there are no more pages, an empty string will be returned as the next cursor.
+    """
+    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
+