frogml-core 0.0.44__py3-none-any.whl → 0.0.46__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.44"
+__version__ = "0.0.46"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

{frogml_core-0.0.44.dist-info → frogml_core-0.0.46.dist-info}/METADATA

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

{frogml_core-0.0.44.dist-info → frogml_core-0.0.46.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=mpUvgThoXhvUkPKkiIiydDsyjs7hizVka4Qna_Yu-6w,777
+frogml_core/__init__.py,sha256=Myxbpn00uVlTUP0-qUzYEoj7uaUAUcal20Jn_wQOysM,777
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -431,6 +431,12 @@ frogml_core/vector_store/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 frogml_core/vector_store/utils/filter_utils.py,sha256=n80IlmS7lGelCNEpZ8o3rAHzrZpHc-3EykEd8tZoeAE,854
 frogml_core/vector_store/utils/upsert_utils.py,sha256=vqnyAJOl46WqpZMUiOa4I8Zf2CDIXQxR7wM4Z8rgooc,7517
 frogml_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+frogml_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py,sha256=dbxw7rgdWh38JDI7F1X6PFei4xvjAgsqKWihemdFyCs,2449
+frogml_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.pyi,sha256=HFrUIrEJFdKls7PhmadzZ-daVPV1i062w-71wPMMPa4,3116
+frogml_proto/jfml/hosting_gateway/v1/build_upload_url_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.py,sha256=yOxiJNcuXeCaeTBLm5KlIpvp18S0RglzdJRFcHpTawc,3075
+frogml_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.pyi,sha256=XV6h6X3zJnqXocbRvXqG_hr9AYAJJJVjExCgC5l7QBs,1729
+frogml_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2_grpc.py,sha256=tAHU1JMZ-5BODK4it2-yNAG6XzMGAmhoWcWoyeDjNAs,3277
 frogml_proto/jfml/model_version/v1/artifact_pb2.py,sha256=HjSD0Tp2XGxwMBnDzFAYhCBftaJwiJRoYSkTcr1pJyA,1973
 frogml_proto/jfml/model_version/v1/artifact_pb2.pyi,sha256=5CGU58qXOSbe0KSu0ovjIbXWrpS9r-PIvdq4qhqPQIc,1673
 frogml_proto/jfml/model_version/v1/artifact_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -575,8 +581,8 @@ frogml_proto/qwak/build_settings/build_settings_pb2_grpc.py,sha256=1oboBPFxaTEXt
 frogml_proto/qwak/builds/build_pb2.py,sha256=e1fecTsRZgEWlBtRoFP93yMPp-kbui0xozBCRPwptLE,15189
 frogml_proto/qwak/builds/build_pb2.pyi,sha256=YGGVicG6kMMQpYDT67WCoy4NJN5-iB1tLPnlIiDR4aw,25381
 frogml_proto/qwak/builds/build_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/builds/build_url_pb2.py,sha256=anKHqrZRH82uplaN_LvxRm3wIIaf9DDjh9AHQ1J1u70,5565
-frogml_proto/qwak/builds/build_url_pb2.pyi,sha256=IC0L3D7zg5vf_4U9Ubc-V4lLqviPknUaBKCCrVK6w9k,7299
+frogml_proto/qwak/builds/build_url_pb2.py,sha256=fGRVl0cMS2U7oEFMGmj_4VxDNX2ZXV1TfdugG5sHY3s,5663
+frogml_proto/qwak/builds/build_url_pb2.pyi,sha256=S-G1wACYvNpCrjiK64BOu1zNo9qQU94RYnpFQjjNKgA,8008
 frogml_proto/qwak/builds/build_url_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/builds/builds_orchestrator_service_pb2.py,sha256=1b9kaFQ3b5KXVZe6fnuxN9l3OdRqm3EZBrtSxWm9X-k,15687
 frogml_proto/qwak/builds/builds_orchestrator_service_pb2.pyi,sha256=3GGNRcoviu3BppOv8wWOkfYVvC96HYzOEu3Kqzhzj-A,12923
@@ -1019,6 +1025,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.44.dist-info/METADATA,sha256=k3xkBof6b6DjS8G3Mcwi21e662VqxxMrTpJGudZlo68,2004
-frogml_core-0.0.44.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.44.dist-info/RECORD,,
+frogml_core-0.0.46.dist-info/METADATA,sha256=3vdwJyAVS5VG3_4KExHO5zCeL-51RnSngakZREPI-o8,2004
+frogml_core-0.0.46.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.46.dist-info/RECORD,,

frogml_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.jfml.hosting_gateway/v1/build_upload_url.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.jfml/hosting_gateway/v1/build_upload_url.proto\x12\x17jfml.hosting_gateway.v1\"\xd3\x01\n\x18\x42uildUploadUrlParameters\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x11\n\ttag_value\x18\x03 \x01(\t\x12\x41\n\rartifact_type\x18\x04 \x01(\x0e\x32*.jfml.hosting_gateway.v1.BuildArtifactType\x12$\n\x1ause_cloud_provider_storage\x18\x05 \x01(\x08H\x00\x42\x17\n\x15\x63loud_provider_option*v\n\x11\x42uildArtifactType\x12\x19\n\x15INVALID_ARTIFACT_TYPE\x10\x00\x12\x16\n\x12\x44\x41TA_ARTIFACT_TYPE\x10\x01\x12\x16\n\x12\x46ILE_ARTIFACT_TYPE\x10\x02\x12\x16\n\x12\x43ODE_ARTIFACT_TYPE\x10\x03\x42\'\n#com.jfrog.ml.hosting_gateway.api.v1P\x01\x62\x06proto3')
+
+_BUILDARTIFACTTYPE = DESCRIPTOR.enum_types_by_name['BuildArtifactType']
+BuildArtifactType = enum_type_wrapper.EnumTypeWrapper(_BUILDARTIFACTTYPE)
+INVALID_ARTIFACT_TYPE = 0
+DATA_ARTIFACT_TYPE = 1
+FILE_ARTIFACT_TYPE = 2
+CODE_ARTIFACT_TYPE = 3
+
+
+_BUILDUPLOADURLPARAMETERS = DESCRIPTOR.message_types_by_name['BuildUploadUrlParameters']
+BuildUploadUrlParameters = _reflection.GeneratedProtocolMessageType('BuildUploadUrlParameters', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDUPLOADURLPARAMETERS,
+  '__module__' : 'jfml.hosting_gateway.v1.build_upload_url_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.hosting_gateway.v1.BuildUploadUrlParameters)
+  })
+_sym_db.RegisterMessage(BuildUploadUrlParameters)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
+  _BUILDARTIFACTTYPE._serialized_start=289
+  _BUILDARTIFACTTYPE._serialized_end=407
+  _BUILDUPLOADURLPARAMETERS._serialized_start=76
+  _BUILDUPLOADURLPARAMETERS._serialized_end=287
+# @@protoc_insertion_point(module_scope)

frogml_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.pyi

@@ -0,0 +1,70 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+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 _BuildArtifactType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _BuildArtifactTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BuildArtifactType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_ARTIFACT_TYPE: _BuildArtifactType.ValueType  # 0
+    DATA_ARTIFACT_TYPE: _BuildArtifactType.ValueType  # 1
+    FILE_ARTIFACT_TYPE: _BuildArtifactType.ValueType  # 2
+    CODE_ARTIFACT_TYPE: _BuildArtifactType.ValueType  # 3
+
+class BuildArtifactType(_BuildArtifactType, metaclass=_BuildArtifactTypeEnumTypeWrapper):
+    """build artifact Types"""
+
+INVALID_ARTIFACT_TYPE: BuildArtifactType.ValueType  # 0
+DATA_ARTIFACT_TYPE: BuildArtifactType.ValueType  # 1
+FILE_ARTIFACT_TYPE: BuildArtifactType.ValueType  # 2
+CODE_ARTIFACT_TYPE: BuildArtifactType.ValueType  # 3
+global___BuildArtifactType = BuildArtifactType
+
+class BuildUploadUrlParameters(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    TAG_VALUE_FIELD_NUMBER: builtins.int
+    ARTIFACT_TYPE_FIELD_NUMBER: builtins.int
+    USE_CLOUD_PROVIDER_STORAGE_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    """Model ID"""
+    build_id: builtins.str
+    """Build Id"""
+    tag_value: builtins.str
+    """Tag to save the artifact by"""
+    artifact_type: global___BuildArtifactType.ValueType
+    """The type of the artifact to upload"""
+    use_cloud_provider_storage: builtins.bool
+    """When true, will use the cloud provider URL"""
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        build_id: builtins.str = ...,
+        tag_value: builtins.str = ...,
+        artifact_type: global___BuildArtifactType.ValueType = ...,
+        use_cloud_provider_storage: builtins.bool = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["cloud_provider_option", b"cloud_provider_option", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["artifact_type", b"artifact_type", "build_id", b"build_id", "cloud_provider_option", b"cloud_provider_option", "model_id", b"model_id", "tag_value", b"tag_value", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["cloud_provider_option", b"cloud_provider_option"]) -> typing_extensions.Literal["use_cloud_provider_storage"] | None: ...
+
+global___BuildUploadUrlParameters = BuildUploadUrlParameters

frogml_proto/jfml/hosting_gateway/v1/build_upload_url_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
+

frogml_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.jfml.hosting_gateway/v1/hosting_gateway_service.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 frogml_proto.jfml.hosting_gateway.v1 import build_upload_url_pb2 as jfml_dot_hosting__gateway_dot_v1_dot_build__upload__url__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5jfml/hosting_gateway/v1/hosting_gateway_service.proto\x12\x17jfml.hosting_gateway.v1\x1a.jfml/hosting_gateway/v1/build_upload_url.proto\"g\n\"GetBuildVersioningUploadURLRequest\x12\x41\n\x06params\x18\x01 \x01(\x0b\x32\x31.jfml.hosting_gateway.v1.BuildUploadUrlParameters\"9\n#GetBuildVersioningUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t2\xb2\x01\n\x15HostingGatewayService\x12\x98\x01\n\x1bGetBuildVersioningUploadURL\x12;.jfml.hosting_gateway.v1.GetBuildVersioningUploadURLRequest\x1a<.jfml.hosting_gateway.v1.GetBuildVersioningUploadURLResponseB\'\n#com.jfrog.ml.hosting_gateway.api.v1P\x01\x62\x06proto3')
+
+
+
+_GETBUILDVERSIONINGUPLOADURLREQUEST = DESCRIPTOR.message_types_by_name['GetBuildVersioningUploadURLRequest']
+_GETBUILDVERSIONINGUPLOADURLRESPONSE = DESCRIPTOR.message_types_by_name['GetBuildVersioningUploadURLResponse']
+GetBuildVersioningUploadURLRequest = _reflection.GeneratedProtocolMessageType('GetBuildVersioningUploadURLRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETBUILDVERSIONINGUPLOADURLREQUEST,
+  '__module__' : 'jfml.hosting_gateway.v1.hosting_gateway_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.hosting_gateway.v1.GetBuildVersioningUploadURLRequest)
+  })
+_sym_db.RegisterMessage(GetBuildVersioningUploadURLRequest)
+
+GetBuildVersioningUploadURLResponse = _reflection.GeneratedProtocolMessageType('GetBuildVersioningUploadURLResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETBUILDVERSIONINGUPLOADURLRESPONSE,
+  '__module__' : 'jfml.hosting_gateway.v1.hosting_gateway_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.hosting_gateway.v1.GetBuildVersioningUploadURLResponse)
+  })
+_sym_db.RegisterMessage(GetBuildVersioningUploadURLResponse)
+
+_HOSTINGGATEWAYSERVICE = DESCRIPTOR.services_by_name['HostingGatewayService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
+  _GETBUILDVERSIONINGUPLOADURLREQUEST._serialized_start=130
+  _GETBUILDVERSIONINGUPLOADURLREQUEST._serialized_end=233
+  _GETBUILDVERSIONINGUPLOADURLRESPONSE._serialized_start=235
+  _GETBUILDVERSIONINGUPLOADURLRESPONSE._serialized_end=292
+  _HOSTINGGATEWAYSERVICE._serialized_start=295
+  _HOSTINGGATEWAYSERVICE._serialized_end=473
+# @@protoc_insertion_point(module_scope)

frogml_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.pyi

@@ -0,0 +1,48 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import frogml_proto.jfml.hosting_gateway.v1.build_upload_url_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetBuildVersioningUploadURLRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PARAMS_FIELD_NUMBER: builtins.int
+    @property
+    def params(self) -> frogml_proto.jfml.hosting_gateway.v1.build_upload_url_pb2.BuildUploadUrlParameters:
+        """The params to define the url."""
+    def __init__(
+        self,
+        *,
+        params: frogml_proto.jfml.hosting_gateway.v1.build_upload_url_pb2.BuildUploadUrlParameters | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["params", b"params"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["params", b"params"]) -> None: ...
+
+global___GetBuildVersioningUploadURLRequest = GetBuildVersioningUploadURLRequest
+
+class GetBuildVersioningUploadURLResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    UPLOAD_URL_FIELD_NUMBER: builtins.int
+    upload_url: builtins.str
+    """Upload url intended for the upload"""
+    def __init__(
+        self,
+        *,
+        upload_url: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["upload_url", b"upload_url"]) -> None: ...
+
+global___GetBuildVersioningUploadURLResponse = GetBuildVersioningUploadURLResponse

frogml_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2_grpc.py

@@ -0,0 +1,67 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml_proto.jfml.hosting_gateway.v1 import hosting_gateway_service_pb2 as jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2
+
+
+class HostingGatewayServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetBuildVersioningUploadURL = channel.unary_unary(
+                '/jfml.hosting_gateway.v1.HostingGatewayService/GetBuildVersioningUploadURL',
+                request_serializer=jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2.GetBuildVersioningUploadURLRequest.SerializeToString,
+                response_deserializer=jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2.GetBuildVersioningUploadURLResponse.FromString,
+                )
+
+
+class HostingGatewayServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetBuildVersioningUploadURL(self, request, context):
+        """Request the service for a path ready for upload in purpose of uploading local model code or data
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_HostingGatewayServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetBuildVersioningUploadURL': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetBuildVersioningUploadURL,
+                    request_deserializer=jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2.GetBuildVersioningUploadURLRequest.FromString,
+                    response_serializer=jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2.GetBuildVersioningUploadURLResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'jfml.hosting_gateway.v1.HostingGatewayService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class HostingGatewayService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetBuildVersioningUploadURL(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, '/jfml.hosting_gateway.v1.HostingGatewayService/GetBuildVersioningUploadURL',
+            jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2.GetBuildVersioningUploadURLRequest.SerializeToString,
+            jfml_dot_hosting__gateway_dot_v1_dot_hosting__gateway__service__pb2.GetBuildVersioningUploadURLResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_proto/qwak/builds/build_url_pb2.py

@@ -15,7 +15,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bqwak/builds/build_url.proto\x12\x18qwak.builds.orchestrator\"\xa8\x01\n\x18\x42uildVersioningUrlParams\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x04 \x01(\t\x12\x43\n\x08tag_type\x18\x05 \x01(\x0e\x32\x31.qwak.builds.orchestrator.BuildVersioningTagsType\"\xb5\x01\n\x13\x42uildVersioningTags\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x10\n\x04tags\x18\x03 \x03(\tB\x02\x18\x01\x12P\n\x0ftags_definition\x18\x04 \x03(\x0b\x32\x37.qwak.builds.orchestrator.BuildVersioningTagsDefinition\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\"n\n\x1d\x42uildVersioningTagsDefinition\x12\x0c\n\x04tags\x18\x01 \x03(\t\x12?\n\x04type\x18\x02 \x01(\x0e\x32\x31.qwak.builds.orchestrator.BuildVersioningTagsType\"\xf8\x01\n\x1d\x42uildVersioningTagsProperties\x12\x0b\n\x03tag\x18\x01 \x01(\t\x12\x14\n\x0c\x64ownload_url\x18\x02 \x01(\t\x12\x11\n\tfile_size\x18\x03 \x01(\x03\x12\x43\n\x08tag_type\x18\x04 \x01(\x0e\x32\x31.qwak.builds.orchestrator.BuildVersioningTagsType\x12\x44\n\x15\x65xternal_download_url\x18\x05 \x01(\x0b\x32%.qwak.builds.orchestrator.ExternalUrl\x12\x16\n\x0eui_direct_link\x18\x06 \x01(\t\"5\n\x0b\x45xternalUrl\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x19\n\x11original_hostname\x18\x02 \x01(\t*h\n\x17\x42uildVersioningTagsType\x12\x14\n\x10INVALID_TAG_TYPE\x10\x00\x12\x11\n\rDATA_TAG_TYPE\x10\x01\x12\x11\n\rFILE_TAG_TYPE\x10\x02\x12\x11\n\rCODE_TAG_TYPE\x10\x03\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bqwak/builds/build_url.proto\x12\x18qwak.builds.orchestrator\"\xe7\x01\n\x18\x42uildVersioningUrlParams\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x04 \x01(\t\x12\x43\n\x08tag_type\x18\x05 \x01(\x0e\x32\x31.qwak.builds.orchestrator.BuildVersioningTagsType\x12$\n\x1ause_cloud_provider_storage\x18\x06 \x01(\x08H\x00\x42\x17\n\x15\x63loud_provider_option\"\xb5\x01\n\x13\x42uildVersioningTags\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x10\n\x04tags\x18\x03 \x03(\tB\x02\x18\x01\x12P\n\x0ftags_definition\x18\x04 \x03(\x0b\x32\x37.qwak.builds.orchestrator.BuildVersioningTagsDefinition\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\"n\n\x1d\x42uildVersioningTagsDefinition\x12\x0c\n\x04tags\x18\x01 \x03(\t\x12?\n\x04type\x18\x02 \x01(\x0e\x32\x31.qwak.builds.orchestrator.BuildVersioningTagsType\"\xf8\x01\n\x1d\x42uildVersioningTagsProperties\x12\x0b\n\x03tag\x18\x01 \x01(\t\x12\x14\n\x0c\x64ownload_url\x18\x02 \x01(\t\x12\x11\n\tfile_size\x18\x03 \x01(\x03\x12\x43\n\x08tag_type\x18\x04 \x01(\x0e\x32\x31.qwak.builds.orchestrator.BuildVersioningTagsType\x12\x44\n\x15\x65xternal_download_url\x18\x05 \x01(\x0b\x32%.qwak.builds.orchestrator.ExternalUrl\x12\x16\n\x0eui_direct_link\x18\x06 \x01(\t\"5\n\x0b\x45xternalUrl\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x19\n\x11original_hostname\x18\x02 \x01(\t*h\n\x17\x42uildVersioningTagsType\x12\x14\n\x10INVALID_TAG_TYPE\x10\x00\x12\x11\n\rDATA_TAG_TYPE\x10\x01\x12\x11\n\rFILE_TAG_TYPE\x10\x02\x12\x11\n\rCODE_TAG_TYPE\x10\x03\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
 
 _BUILDVERSIONINGTAGSTYPE = DESCRIPTOR.enum_types_by_name['BuildVersioningTagsType']
 BuildVersioningTagsType = enum_type_wrapper.EnumTypeWrapper(_BUILDVERSIONINGTAGSTYPE)
@@ -71,16 +71,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._serialized_options = b'\n#com.qwak.ai.builds.orchestrator.apiP\001'
   _BUILDVERSIONINGTAGS.fields_by_name['tags']._options = None
   _BUILDVERSIONINGTAGS.fields_by_name['tags']._serialized_options = b'\030\001'
-  _BUILDVERSIONINGTAGSTYPE._serialized_start=830
-  _BUILDVERSIONINGTAGSTYPE._serialized_end=934
+  _BUILDVERSIONINGTAGSTYPE._serialized_start=893
+  _BUILDVERSIONINGTAGSTYPE._serialized_end=997
   _BUILDVERSIONINGURLPARAMS._serialized_start=58
-  _BUILDVERSIONINGURLPARAMS._serialized_end=226
-  _BUILDVERSIONINGTAGS._serialized_start=229
-  _BUILDVERSIONINGTAGS._serialized_end=410
-  _BUILDVERSIONINGTAGSDEFINITION._serialized_start=412
-  _BUILDVERSIONINGTAGSDEFINITION._serialized_end=522
-  _BUILDVERSIONINGTAGSPROPERTIES._serialized_start=525
-  _BUILDVERSIONINGTAGSPROPERTIES._serialized_end=773
-  _EXTERNALURL._serialized_start=775
-  _EXTERNALURL._serialized_end=828
+  _BUILDVERSIONINGURLPARAMS._serialized_end=289
+  _BUILDVERSIONINGTAGS._serialized_start=292
+  _BUILDVERSIONINGTAGS._serialized_end=473
+  _BUILDVERSIONINGTAGSDEFINITION._serialized_start=475
+  _BUILDVERSIONINGTAGSDEFINITION._serialized_end=585
+  _BUILDVERSIONINGTAGSPROPERTIES._serialized_start=588
+  _BUILDVERSIONINGTAGSPROPERTIES._serialized_end=836
+  _EXTERNALURL._serialized_start=838
+  _EXTERNALURL._serialized_end=891
 # @@protoc_insertion_point(module_scope)

frogml_proto/qwak/builds/build_url_pb2.pyi

@@ -46,6 +46,7 @@ class BuildVersioningUrlParams(google.protobuf.message.Message):
     TAG_FIELD_NUMBER: builtins.int
     ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
     TAG_TYPE_FIELD_NUMBER: builtins.int
+    USE_CLOUD_PROVIDER_STORAGE_FIELD_NUMBER: builtins.int
     model_id: builtins.str
     """Model ID"""
     build_id: builtins.str
@@ -56,6 +57,8 @@ class BuildVersioningUrlParams(google.protobuf.message.Message):
     """Environment to save the artifact under"""
     tag_type: global___BuildVersioningTagsType.ValueType
     """The type of the tags"""
+    use_cloud_provider_storage: builtins.bool
+    """When true, will use the cloud provider URL"""
     def __init__(
         self,
         *,
@@ -64,8 +67,11 @@ class BuildVersioningUrlParams(google.protobuf.message.Message):
         tag: builtins.str = ...,
         environment_id: builtins.str = ...,
         tag_type: global___BuildVersioningTagsType.ValueType = ...,
+        use_cloud_provider_storage: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "environment_id", b"environment_id", "model_id", b"model_id", "tag", b"tag", "tag_type", b"tag_type"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["cloud_provider_option", b"cloud_provider_option", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "cloud_provider_option", b"cloud_provider_option", "environment_id", b"environment_id", "model_id", b"model_id", "tag", b"tag", "tag_type", b"tag_type", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["cloud_provider_option", b"cloud_provider_option"]) -> typing_extensions.Literal["use_cloud_provider_storage"] | None: ...
 
 global___BuildVersioningUrlParams = BuildVersioningUrlParams