frogml-core 0.0.97__py3-none-any.whl → 0.0.99__py3-none-any.whl

frogml_core/__init__.py

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

frogml_core/clients/batch_job_management/client.py

@@ -36,6 +36,10 @@ from frogml_proto.qwak.batch_job.v1.batch_job_service_pb2 import (
     StartBatchJobResponse,
     StartWarmupJobRequest,
     StartWarmupJobResponse,
+    UpdateTasksDetailsResponse,
+    UpdateTasksDetailsRequest,
+    BatchTaskDetails,
+    InputFileDetails,
 )
 from frogml_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
     BatchJobManagementServiceStub,
@@ -67,6 +71,7 @@ from .results import (
     GetExecutionReportResult,
     StartExecutionResult,
 )
+from typing import List
 
 CLIENT_TIMEOUT = 180  # Seconds
 
@@ -542,3 +547,37 @@ class BatchJobManagerClient:
 
         except grpc.RpcError as e:
             raise FrogmlException(f"Failed to get batch job details, error is: {e}")
+
+    def update_task_details(self, task_id: str, input_file_paths: List[str]):
+        """
+        Update task details for a specific task using the provided list of input file paths.
+
+        Args:
+            task_id: The ID of the task to update.
+            input_file_paths: A list of paths for the input files associated with the task.
+        """
+        # Create InputFileDetails messages for each input file path
+        input_files_details = [
+            InputFileDetails(path=file_path) for file_path in input_file_paths
+        ]
+
+        # Create BatchTaskDetails message for the single task
+        batch_task_details = BatchTaskDetails(
+            task_id=task_id, input_files_details=input_files_details
+        )
+
+        update_request = UpdateTasksDetailsRequest(
+            tasks_details=[batch_task_details]  # Wrap in a list
+        )
+
+        try:
+            response: UpdateTasksDetailsResponse = (
+                self.batch_job_management.UpdateTasksDetails(
+                    update_request, timeout=CLIENT_TIMEOUT
+                )
+            )
+            return response
+        except grpc.RpcError as e:
+            raise FrogmlException(
+                f"Failed to update task details for task '{task_id}', error is: {e}"
+            )

{frogml_core-0.0.97.dist-info → frogml_core-0.0.99.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.97
+Version: 0.0.99
 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.97.dist-info → frogml_core-0.0.99.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=qVJvp0Th1tj8hSuSvHfQlESs2X9saUoSnods3O0RX1M,777
+frogml_core/__init__.py,sha256=QdMe5Qmm9qmXOskppK40Rzg8cfe58ugR6WtvaypKK5I,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
@@ -34,7 +34,7 @@ frogml_core/clients/automation_management/client.py,sha256=rG1tvdtKoSBCSG5IdzmSP
 frogml_core/clients/autoscaling/__init__.py,sha256=A-zuZOWaZf16NjnpX38204m730c5Lde491KaXneGirE,38
 frogml_core/clients/autoscaling/client.py,sha256=Izl_Ko31ECsVEadTcthru8GGujsHpwWN5E_VSCKiymM,1296
 frogml_core/clients/batch_job_management/__init__.py,sha256=zywxYf2JupkIWnGVB2C6ugZs5CrOdxTPrksD5P6uQu4,211
-frogml_core/clients/batch_job_management/client.py,sha256=mXGyIQLlEOVcrmESje_rEOdlDLWIB8Ndi6N38cXWe74,21120
+frogml_core/clients/batch_job_management/client.py,sha256=V8sqltMAsoHAoXHxXZsoLmY7L1XykbBvod1xyhCmHHA,22528
 frogml_core/clients/batch_job_management/executions_config.py,sha256=eEbT_ldIEVtwe85oThJ-Jv3qFBIfvLpZ-w6aJCLlqlk,6910
 frogml_core/clients/batch_job_management/results.py,sha256=nrFGZ3y538FBhej94CYV4cwqQssK1aUQOLGubYdQVrc,1846
 frogml_core/clients/build_management/__init__.py,sha256=oXurHX6Kho5X-fOuuNHB3B9oMX6ejYUqh6GWv5qJyjw,43
@@ -947,11 +947,11 @@ frogml_proto/qwak/logging/log_source_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2.py,sha256=yArUUX9SsaOuTmFfhxGIxe_ZEFC6u39oqYBAiPro5mo,6764
 frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi,sha256=G5CaQGczrZ9kGsMbmJXMFWAwnqMZ4jPSl9KQpSq0yMo,12345
 frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/model_group/model_group_pb2.py,sha256=Clk4w9yFYetBnrk04LVS6DCVsodGCPD69msJS9_FrGk,3441
-frogml_proto/qwak/model_group/model_group_pb2.pyi,sha256=AfUK0LLa1uzTDQvOncQUu6t8GJHYTfdlc2R5ehHhaVc,3051
-frogml_proto/qwak/model_group/model_group_pb2_grpc.py,sha256=DWBD3YffezJW8ov0UO6dYn7Ofx1sww-LjPGBKHfUf2U,3074
-frogml_proto/qwak/model_group/model_group_repository_details_pb2.py,sha256=iZ7Psb7aUSGWG0IB6igZDHbsYbZratpqx5Dtzkd0v9A,5035
-frogml_proto/qwak/model_group/model_group_repository_details_pb2.pyi,sha256=h5ciIby9peqjZqHosajfrHdmX5kosmzDJiadoCVpTmc,4119
+frogml_proto/qwak/model_group/model_group_pb2.py,sha256=1A5FJtSSiL4rfAbqzV00xln0PM42hLq3v7cDyFQt4H8,5388
+frogml_proto/qwak/model_group/model_group_pb2.pyi,sha256=sOHTXNJjDNGUpIcgqzrCcV_lmwaUt2MJiW6RN0YOAEA,5086
+frogml_proto/qwak/model_group/model_group_pb2_grpc.py,sha256=10iHVXBKc3FXdBh5u5SqQFIn5qZaOsVGuKVaI4kHMZk,5126
+frogml_proto/qwak/model_group/model_group_repository_details_pb2.py,sha256=HaWF7QWEDWZJm-IEV25xT8XbVWgqwVU4pqFKt7t3sRk,5812
+frogml_proto/qwak/model_group/model_group_repository_details_pb2.pyi,sha256=GgYhrSXu8L2RdEV0clooKgB9Gye3_no4q0YrMy8QrZw,5701
 frogml_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/models/models_pb2.py,sha256=Q-XUigDskzrltE8l1x3r268kLOq7SgUJMD-wg-r6_Bw,32112
 frogml_proto/qwak/models/models_pb2.pyi,sha256=277mGQradMyhof3J54ilaW_SKN0d-dan7DAy_6Zkmo8,39261
@@ -1070,7 +1070,7 @@ frogml_services_mock/mocks/audience_service_api.py,sha256=x96adhF05NYqEtI8lJaf5l
 frogml_services_mock/mocks/authentication_service.py,sha256=M3cVfnQ2XcxQAWZCm9uR9cAVu_0BahEP4BdymYQfihI,1180
 frogml_services_mock/mocks/automation_management_service.py,sha256=-7kncqM3R-_pLmppzFpuv_TniXCVGVuESculVuQ3EbY,8238
 frogml_services_mock/mocks/autoscaling_service_api.py,sha256=u3RkwHxACPFSk9DwhLFQSI7_ULoHN9uWFL-zyyo-sNw,1032
-frogml_services_mock/mocks/batch_job_manager_service.py,sha256=QY3HA0MlA8eg2jkQEzjyGx8UsSnlGbOUopDLOYLacL4,13146
+frogml_services_mock/mocks/batch_job_manager_service.py,sha256=AJNM5yJDD7fGRSeHoxC8AdGdx6iwwxUWXNhEFDcD3QE,14098
 frogml_services_mock/mocks/build_management.py,sha256=IECGXfSEDOUhwMcHtI3pd9PF0aXqZL2R98gl3wS0joY,3697
 frogml_services_mock/mocks/build_orchestrator_build_api.py,sha256=fnB6x7W_UWoOxkCm7AgtrRYJjjm1QCotkf-2hPGk75c,5026
 frogml_services_mock/mocks/build_orchestrator_build_settings_api.py,sha256=5V15s2QUur4CPMNZPyXIIt_NJ2W_DwugfreMDA6WbfM,1394
@@ -1114,7 +1114,7 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=sgKgwhu2W0YOHtzil8x7f1znK_sZr_i27XSeiF4xqVE,21200
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_storage/__init__.py,sha256=93lvSWnHyZhOl_RAxkMtdi_akHt37ijuw8MdOyJYdKM,23
+frogml_storage/__init__.py,sha256=3svy--sM0f1CbY-gO0A9g6zDgvjpar3s5vNWbJeQ08Q,23
 frogml_storage/_environment.py,sha256=zuzOJBtBwFaguwn_JkKjfhXStZoustgP30KzOP3mYv8,707
 frogml_storage/artifactory/__init__.py,sha256=C02rcm7kqsZBVA6c6Gztxamj96hn8Aj6BuzYWFRmWbQ,71
 frogml_storage/artifactory/_artifactory_api.py,sha256=Oz0HOpQPSNwWIVAy94UJUyPhLetc7sdZjoTfSXtrFug,11200
@@ -1153,6 +1153,6 @@ frogml_storage/utils/__init__.py,sha256=HQUWfuGUIPZY7kfS795TRW8BQ4WmNqrNjS7lUrbx
 frogml_storage/utils/_input_checks_utility.py,sha256=CFiJOdTBS9piJMtR3lemEz27wZcQ6_-7XESu8iy-mrw,3221
 frogml_storage/utils/_storage_utils.py,sha256=HB2g7uY5A3b33yIcAUM1OjHb5jWsnpESsiDrEviQwrI,366
 frogml_storage/utils/_url_utils.py,sha256=NUEfz9Fp1iE8b676-A5wrMlSTsJVRKrUhcUItOFAJD8,821
-frogml_core-0.0.97.dist-info/METADATA,sha256=RUgAcVvSjFGkMrXZ0k4tJRYXa1aLyXlZJX14uE8J5XQ,14975
-frogml_core-0.0.97.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml_core-0.0.97.dist-info/RECORD,,
+frogml_core-0.0.99.dist-info/METADATA,sha256=omvhGi0BYg-wKqHXHm1JY6Zn52ChbLZkaSDoYfF_XMc,14975
+frogml_core-0.0.99.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml_core-0.0.99.dist-info/RECORD,,

frogml_proto/qwak/model_group/model_group_pb2.py

@@ -16,12 +16,14 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 from frogml_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')
+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.RepositoryDetails\"\x95\x01\n&CreateRepositoriesForModelGroupRequest\x12\x18\n\x10model_group_uuid\x18\x01 \x01(\t\x12Q\n\x12repository_details\x18\x02 \x03(\x0b\x32\x35.qwak.model_groups.management.RemoteRepositoryDetails\"\x82\x01\n\'CreateRepositoriesForModelGroupResponse\x12W\n\x1e\x61rtifactory_repository_details\x18\x01 \x03(\x0b\x32/.qwak.model_groups.management.RepositoryDetails2\xeb\x02\n\x1bModelGroupManagementService\x12\x9a\x01\n\x1b\x43reateIfNotExistsModelGroup\x12@.qwak.model_groups.management.CreateIfNotExistsModelGroupRequest\x1a\x39.qwak.model_groups.management.ModelGroupBriefInfoResponse\x12\xae\x01\n\x1f\x43reateRepositoriesForModelGroup\x12\x44.qwak.model_groups.management.CreateRepositoriesForModelGroupRequest\x1a\x45.qwak.model_groups.management.CreateRepositoriesForModelGroupResponseBL\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']
+_CREATEREPOSITORIESFORMODELGROUPREQUEST = DESCRIPTOR.message_types_by_name['CreateRepositoriesForModelGroupRequest']
+_CREATEREPOSITORIESFORMODELGROUPRESPONSE = DESCRIPTOR.message_types_by_name['CreateRepositoriesForModelGroupResponse']
 CreateIfNotExistsModelGroupRequest = _reflection.GeneratedProtocolMessageType('CreateIfNotExistsModelGroupRequest', (_message.Message,), {
   'DESCRIPTOR' : _CREATEIFNOTEXISTSMODELGROUPREQUEST,
   '__module__' : 'qwak.model_group.model_group_pb2'
@@ -36,6 +38,20 @@ ModelGroupBriefInfoResponse = _reflection.GeneratedProtocolMessageType('ModelGro
   })
 _sym_db.RegisterMessage(ModelGroupBriefInfoResponse)
 
+CreateRepositoriesForModelGroupRequest = _reflection.GeneratedProtocolMessageType('CreateRepositoriesForModelGroupRequest', (_message.Message,), {
+  'DESCRIPTOR' : _CREATEREPOSITORIESFORMODELGROUPREQUEST,
+  '__module__' : 'qwak.model_group.model_group_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.CreateRepositoriesForModelGroupRequest)
+  })
+_sym_db.RegisterMessage(CreateRepositoriesForModelGroupRequest)
+
+CreateRepositoriesForModelGroupResponse = _reflection.GeneratedProtocolMessageType('CreateRepositoriesForModelGroupResponse', (_message.Message,), {
+  'DESCRIPTOR' : _CREATEREPOSITORIESFORMODELGROUPRESPONSE,
+  '__module__' : 'qwak.model_group.model_group_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.CreateRepositoriesForModelGroupResponse)
+  })
+_sym_db.RegisterMessage(CreateRepositoriesForModelGroupResponse)
+
 _MODELGROUPMANAGEMENTSERVICE = DESCRIPTOR.services_by_name['ModelGroupManagementService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -45,6 +61,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _CREATEIFNOTEXISTSMODELGROUPREQUEST._serialized_end=245
   _MODELGROUPBRIEFINFORESPONSE._serialized_start=248
   _MODELGROUPBRIEFINFORESPONSE._serialized_end=471
-  _MODELGROUPMANAGEMENTSERVICE._serialized_start=474
-  _MODELGROUPMANAGEMENTSERVICE._serialized_end=660
+  _CREATEREPOSITORIESFORMODELGROUPREQUEST._serialized_start=474
+  _CREATEREPOSITORIESFORMODELGROUPREQUEST._serialized_end=623
+  _CREATEREPOSITORIESFORMODELGROUPRESPONSE._serialized_start=626
+  _CREATEREPOSITORIESFORMODELGROUPRESPONSE._serialized_end=756
+  _MODELGROUPMANAGEMENTSERVICE._serialized_start=759
+  _MODELGROUPMANAGEMENTSERVICE._serialized_end=1122
 # @@protoc_insertion_point(module_scope)

frogml_proto/qwak/model_group/model_group_pb2.pyi

@@ -66,3 +66,39 @@ class ModelGroupBriefInfoResponse(google.protobuf.message.Message):
     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
+
+class CreateRepositoriesForModelGroupRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_UUID_FIELD_NUMBER: builtins.int
+    REPOSITORY_DETAILS_FIELD_NUMBER: builtins.int
+    model_group_uuid: builtins.str
+    """the model group UUID for which to create repositories"""
+    @property
+    def repository_details(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.model_group.model_group_repository_details_pb2.RemoteRepositoryDetails]:
+        """List of remote repository details to create for the model group"""
+    def __init__(
+        self,
+        *,
+        model_group_uuid: builtins.str = ...,
+        repository_details: collections.abc.Iterable[qwak.model_group.model_group_repository_details_pb2.RemoteRepositoryDetails] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_group_uuid", b"model_group_uuid", "repository_details", b"repository_details"]) -> None: ...
+
+global___CreateRepositoriesForModelGroupRequest = CreateRepositoriesForModelGroupRequest
+
+class CreateRepositoriesForModelGroupResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ARTIFACTORY_REPOSITORY_DETAILS_FIELD_NUMBER: builtins.int
+    @property
+    def artifactory_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,
+        *,
+        artifactory_repository_details: collections.abc.Iterable[qwak.model_group.model_group_repository_details_pb2.RepositoryDetails] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["artifactory_repository_details", b"artifactory_repository_details"]) -> None: ...
+
+global___CreateRepositoriesForModelGroupResponse = CreateRepositoriesForModelGroupResponse

frogml_proto/qwak/model_group/model_group_pb2_grpc.py

@@ -19,6 +19,11 @@ class ModelGroupManagementServiceStub(object):
                 request_serializer=qwak_dot_model__group_dot_model__group__pb2.CreateIfNotExistsModelGroupRequest.SerializeToString,
                 response_deserializer=qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.FromString,
                 )
+        self.CreateRepositoriesForModelGroup = channel.unary_unary(
+                '/qwak.model_groups.management.ModelGroupManagementService/CreateRepositoriesForModelGroup',
+                request_serializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupRequest.SerializeToString,
+                response_deserializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupResponse.FromString,
+                )
 
 
 class ModelGroupManagementServiceServicer(object):
@@ -30,6 +35,12 @@ class ModelGroupManagementServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CreateRepositoriesForModelGroup(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 = {
@@ -38,6 +49,11 @@ def add_ModelGroupManagementServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_model__group_dot_model__group__pb2.CreateIfNotExistsModelGroupRequest.FromString,
                     response_serializer=qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.SerializeToString,
             ),
+            'CreateRepositoriesForModelGroup': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateRepositoriesForModelGroup,
+                    request_deserializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupRequest.FromString,
+                    response_serializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.model_groups.management.ModelGroupManagementService', rpc_method_handlers)
@@ -64,3 +80,20 @@ class ModelGroupManagementService(object):
             qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def CreateRepositoriesForModelGroup(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/CreateRepositoriesForModelGroup',
+            qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupRequest.SerializeToString,
+            qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_proto/qwak/model_group/model_group_repository_details_pb2.py

@@ -12,14 +12,14 @@ from google.protobuf import symbol_database as _symbol_database
 _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')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5qwak/model_group/model_group_repository_details.proto\x12\x1cqwak.model_groups.management\"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\"\xf0\x01\n\x17RemoteRepositoryDetails\x12K\n\x11\x64ocker_repository\x18\x01 \x01(\x0b\x32..qwak.model_groups.management.DockerRepositoryH\x00\x12V\n\x17hugging_face_repository\x18\x02 \x01(\x0b\x32\x33.qwak.model_groups.management.HuggingFaceRepositoryH\x00\x12\x1d\n\x15repository_remote_url\x18\x03 \x01(\tB\x11\n\x0frepository_type\"\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']
+_REMOTEREPOSITORYDETAILS = DESCRIPTOR.message_types_by_name['RemoteRepositoryDetails']
 _DOCKERREPOSITORY = DESCRIPTOR.message_types_by_name['DockerRepository']
 _DATASETREPOSITORY = DESCRIPTOR.message_types_by_name['DatasetRepository']
 _ARTIFACTREPOSITORY = DESCRIPTOR.message_types_by_name['ArtifactRepository']
@@ -32,6 +32,13 @@ RepositoryDetails = _reflection.GeneratedProtocolMessageType('RepositoryDetails'
   })
 _sym_db.RegisterMessage(RepositoryDetails)
 
+RemoteRepositoryDetails = _reflection.GeneratedProtocolMessageType('RemoteRepositoryDetails', (_message.Message,), {
+  'DESCRIPTOR' : _REMOTEREPOSITORYDETAILS,
+  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.RemoteRepositoryDetails)
+  })
+_sym_db.RegisterMessage(RemoteRepositoryDetails)
+
 DockerRepository = _reflection.GeneratedProtocolMessageType('DockerRepository', (_message.Message,), {
   'DESCRIPTOR' : _DOCKERREPOSITORY,
   '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
@@ -71,16 +78,18 @@ 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
+  _REPOSITORYDETAILS._serialized_start=87
+  _REPOSITORYDETAILS._serialized_end=201
+  _REMOTEREPOSITORYDETAILS._serialized_start=204
+  _REMOTEREPOSITORYDETAILS._serialized_end=444
+  _DOCKERREPOSITORY._serialized_start=446
+  _DOCKERREPOSITORY._serialized_end=464
+  _DATASETREPOSITORY._serialized_start=466
+  _DATASETREPOSITORY._serialized_end=485
+  _ARTIFACTREPOSITORY._serialized_start=487
+  _ARTIFACTREPOSITORY._serialized_end=507
+  _HUGGINGFACEREPOSITORY._serialized_start=509
+  _HUGGINGFACEREPOSITORY._serialized_end=532
+  _REPOSITORYTYPE._serialized_start=535
+  _REPOSITORYTYPE._serialized_end=884
 # @@protoc_insertion_point(module_scope)

frogml_proto/qwak/model_group/model_group_repository_details_pb2.pyi

@@ -35,6 +35,33 @@ class RepositoryDetails(google.protobuf.message.Message):
 
 global___RepositoryDetails = RepositoryDetails
 
+class RemoteRepositoryDetails(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DOCKER_REPOSITORY_FIELD_NUMBER: builtins.int
+    HUGGING_FACE_REPOSITORY_FIELD_NUMBER: builtins.int
+    REPOSITORY_REMOTE_URL_FIELD_NUMBER: builtins.int
+    @property
+    def docker_repository(self) -> global___DockerRepository:
+        """Docker repository"""
+    @property
+    def hugging_face_repository(self) -> global___HuggingFaceRepository:
+        """HuggingFace repository"""
+    repository_remote_url: builtins.str
+    """The remote repository URL"""
+    def __init__(
+        self,
+        *,
+        docker_repository: global___DockerRepository | None = ...,
+        hugging_face_repository: global___HuggingFaceRepository | None = ...,
+        repository_remote_url: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "repository_type", b"repository_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "repository_remote_url", b"repository_remote_url", "repository_type", b"repository_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["repository_type", b"repository_type"]) -> typing_extensions.Literal["docker_repository", "hugging_face_repository"] | None: ...
+
+global___RemoteRepositoryDetails = RemoteRepositoryDetails
+
 class DockerRepository(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml_services_mock/mocks/batch_job_manager_service.py

@@ -38,6 +38,9 @@ from frogml_proto.qwak.batch_job.v1.batch_job_service_pb2 import (
     StartWarmupJobRequest,
     StartWarmupJobResponse,
     TaskExecutionDetails,
+    UpdateTasksDetailsRequest,
+    UpdateTasksDetailsResponse,
+    BatchTaskDetails,
 )
 from frogml_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
     BatchJobManagementServiceServicer,
@@ -54,6 +57,7 @@ class BatchJobManagerService(BatchJobManagementServiceServicer):
         self.id_to_batch_job: Dict[str, MockBatchJob] = dict()
         self.model_id_active_warmup: Set[str] = set()
         self.models_to_fail: Set[str] = set()
+        self.task_id_to_update_task_details: Dict[str, BatchTaskDetails] = dict()
 
     def StartBatchJob(
         self, request: StartBatchJobRequest, context
@@ -268,6 +272,25 @@ class BatchJobManagerService(BatchJobManagementServiceServicer):
         except Exception as e:
             raise_internal_grpc_error(context, e)
 
+    def UpdateTasksDetails(
+            self, request: UpdateTasksDetailsRequest, context
+    ) -> UpdateTasksDetailsResponse:
+        for task in request.tasks_details:
+            task_id = task.task_id
+            if not task_id :
+                raise ValueError("Task ID cannot be empty")
+            input_files = task.input_files_details
+
+            print(f"Updating task with ID: {task_id} and input files: {[file.path for file in input_files]}")
+            self.task_id_to_update_task_details[task_id] = task
+        return UpdateTasksDetailsResponse()
+
+    def get_task_details(self,task_id: str
+        ) -> Optional[BatchTaskDetails]:
+            """
+            Get task details by task ID
+            """
+            return self.task_id_to_update_task_details[task_id]
 
 class MockBatchJob:
     def __init__(self, request: StartBatchJobRequest):

frogml_storage/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.97"
+__version__ = "0.0.99"