gimlet-api 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl

gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py

@@ -34,6 +34,11 @@ class LogicalPipelineServiceStub(object):
                 request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.SerializeToString,
                 response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.FromString,
                 )
+        self.ShareLogicalPipeline = channel.unary_unary(
+                '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ShareLogicalPipeline',
+                request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineRequest.SerializeToString,
+                response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineResponse.FromString,
+                )
 
 
 class LogicalPipelineServiceServicer(object):
@@ -63,6 +68,12 @@ class LogicalPipelineServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ShareLogicalPipeline(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_LogicalPipelineServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -86,6 +97,11 @@ def add_LogicalPipelineServiceServicer_to_server(servicer, server):
                     request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.FromString,
                     response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.SerializeToString,
             ),
+            'ShareLogicalPipeline': grpc.unary_unary_rpc_method_handler(
+                    servicer.ShareLogicalPipeline,
+                    request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineRequest.FromString,
+                    response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService', rpc_method_handlers)
@@ -163,3 +179,20 @@ class LogicalPipelineService(object):
             src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ShareLogicalPipeline(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, '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ShareLogicalPipeline',
+            src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineRequest.SerializeToString,
+            src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ShareLogicalPipelineResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

gml/proto/src/controlplane/model/mpb/v1/mpb_pb2.py

@@ -16,7 +16,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
 from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'src/controlplane/model/mpb/v1/mpb.proto\x12\"gml.internal.controlplane.model.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\x81\x01\n\x0fGetModelRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x7f\n\x10GetModelResponse\x12\x42\n\nmodel_info\x18\x01 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\tmodelInfo\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x9f\x01\n\x12\x43reateModelRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x42\n\nmodel_info\x18\x03 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\tmodelInfo\">\n\x13\x43reateModelResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id2\x85\x02\n\x0cModelService\x12u\n\x08GetModel\x12\x33.gml.internal.controlplane.model.v1.GetModelRequest\x1a\x34.gml.internal.controlplane.model.v1.GetModelResponse\x12~\n\x0b\x43reateModel\x12\x36.gml.internal.controlplane.model.v1.CreateModelRequest\x1a\x37.gml.internal.controlplane.model.v1.CreateModelResponseB8Z6gimletlabs.ai/gimlet/src/controlplane/model/mpb/v1;mpbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'src/controlplane/model/mpb/v1/mpb.proto\x12\"gml.internal.controlplane.model.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\x9e\x01\n\x0fGetModelRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x33\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\x18\x01\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x19\n\x08org_name\x18\x04 \x01(\tR\x07orgName\"\x7f\n\x10GetModelResponse\x12\x42\n\nmodel_info\x18\x01 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\tmodelInfo\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x9f\x01\n\x12\x43reateModelRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x42\n\nmodel_info\x18\x03 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\tmodelInfo\">\n\x13\x43reateModelResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"\x94\x01\n\x11ShareModelRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x1a\n\x08relation\x18\x02 \x01(\tR\x08relation\x12\x1b\n\tall_users\x18\x03 \x01(\x08R\x08\x61llUsers\x12\x1d\n\nuser_email\x18\x04 \x01(\tR\tuserEmail\"\x14\n\x12ShareModelResponse2\x82\x03\n\x0cModelService\x12u\n\x08GetModel\x12\x33.gml.internal.controlplane.model.v1.GetModelRequest\x1a\x34.gml.internal.controlplane.model.v1.GetModelResponse\x12~\n\x0b\x43reateModel\x12\x36.gml.internal.controlplane.model.v1.CreateModelRequest\x1a\x37.gml.internal.controlplane.model.v1.CreateModelResponse\x12{\n\nShareModel\x12\x35.gml.internal.controlplane.model.v1.ShareModelRequest\x1a\x36.gml.internal.controlplane.model.v1.ShareModelResponseB8Z6gimletlabs.ai/gimlet/src/controlplane/model/mpb/v1;mpbb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.model.mpb.v1.mpb_pb2', globals())
@@ -27,21 +27,27 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _GETMODELREQUEST.fields_by_name['id']._options = None
   _GETMODELREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
   _GETMODELREQUEST.fields_by_name['org_id']._options = None
-  _GETMODELREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
+  _GETMODELREQUEST.fields_by_name['org_id']._serialized_options = b'\030\001\342\336\037\005OrgID'
   _GETMODELRESPONSE.fields_by_name['id']._options = None
   _GETMODELRESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
   _CREATEMODELREQUEST.fields_by_name['org_id']._options = None
   _CREATEMODELREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
   _CREATEMODELRESPONSE.fields_by_name['id']._options = None
   _CREATEMODELRESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
+  _SHAREMODELREQUEST.fields_by_name['id']._options = None
+  _SHAREMODELREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
   _GETMODELREQUEST._serialized_start=169
-  _GETMODELREQUEST._serialized_end=298
-  _GETMODELRESPONSE._serialized_start=300
-  _GETMODELRESPONSE._serialized_end=427
-  _CREATEMODELREQUEST._serialized_start=430
-  _CREATEMODELREQUEST._serialized_end=589
-  _CREATEMODELRESPONSE._serialized_start=591
-  _CREATEMODELRESPONSE._serialized_end=653
-  _MODELSERVICE._serialized_start=656
-  _MODELSERVICE._serialized_end=917
+  _GETMODELREQUEST._serialized_end=327
+  _GETMODELRESPONSE._serialized_start=329
+  _GETMODELRESPONSE._serialized_end=456
+  _CREATEMODELREQUEST._serialized_start=459
+  _CREATEMODELREQUEST._serialized_end=618
+  _CREATEMODELRESPONSE._serialized_start=620
+  _CREATEMODELRESPONSE._serialized_end=682
+  _SHAREMODELREQUEST._serialized_start=685
+  _SHAREMODELREQUEST._serialized_end=833
+  _SHAREMODELRESPONSE._serialized_start=835
+  _SHAREMODELRESPONSE._serialized_end=855
+  _MODELSERVICE._serialized_start=858
+  _MODELSERVICE._serialized_end=1244
 # @@protoc_insertion_point(module_scope)

gml/proto/src/controlplane/model/mpb/v1/mpb_pb2_grpc.py

@@ -24,6 +24,11 @@ class ModelServiceStub(object):
                 request_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelRequest.SerializeToString,
                 response_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelResponse.FromString,
                 )
+        self.ShareModel = channel.unary_unary(
+                '/gml.internal.controlplane.model.v1.ModelService/ShareModel',
+                request_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelRequest.SerializeToString,
+                response_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelResponse.FromString,
+                )
 
 
 class ModelServiceServicer(object):
@@ -41,6 +46,12 @@ class ModelServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ShareModel(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_ModelServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -54,6 +65,11 @@ def add_ModelServiceServicer_to_server(servicer, server):
                     request_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelRequest.FromString,
                     response_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelResponse.SerializeToString,
             ),
+            'ShareModel': grpc.unary_unary_rpc_method_handler(
+                    servicer.ShareModel,
+                    request_deserializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelRequest.FromString,
+                    response_serializer=src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'gml.internal.controlplane.model.v1.ModelService', rpc_method_handlers)
@@ -97,3 +113,20 @@ class ModelService(object):
             src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.CreateModelResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ShareModel(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, '/gml.internal.controlplane.model.v1.ModelService/ShareModel',
+            src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelRequest.SerializeToString,
+            src_dot_controlplane_dot_model_dot_mpb_dot_v1_dot_mpb__pb2.ShareModelResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

gml/version_utils.py

@@ -0,0 +1,43 @@
+# Copyright 2023- Gimlet Labs, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import functools
+import importlib.metadata
+
+from packaging import version
+
+
+def _is_version_greater_or_equal(pkg: str, target_version: str) -> bool:
+    """
+    Check if the installed package version is greater than or equal to the specified version.
+    Args:
+        pkg (str): The package name to check, e.g., 'torch', 'transformers'.
+        target_version (str): The version to compare against, in the format 'major.minor.patch'.
+    Returns:
+        bool: True if the installed package version is greater than or equal to `target_version`, False otherwise.
+    """
+    pkg_version_w_dev = importlib.metadata.version(pkg)
+    # ignore dev suffixes.
+    pkg_version = version.parse(version.parse(pkg_version_w_dev).base_version)
+    return pkg_version >= version.parse(target_version)
+
+
+is_torch_version_greater_or_equal = functools.partial(
+    _is_version_greater_or_equal, "torch"
+)
+is_transformers_version_greater_or_equal = functools.partial(
+    _is_version_greater_or_equal, "transformers"
+)