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

_qwak_proto/qwak/model_group/model_group_pb2.pyi

@@ -18,6 +18,75 @@ else:
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+class GetModelGroupSystemPullImageSecretRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_SYSTEM_PULL_SECRET_IDENTIFIERS_FIELD_NUMBER: builtins.int
+    @property
+    def model_group_system_pull_secret_identifiers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ModelGroupSystemPullSecretIdentifier]: ...
+    def __init__(
+        self,
+        *,
+        model_group_system_pull_secret_identifiers: collections.abc.Iterable[global___ModelGroupSystemPullSecretIdentifier] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_group_system_pull_secret_identifiers", b"model_group_system_pull_secret_identifiers"]) -> None: ...
+
+global___GetModelGroupSystemPullImageSecretRequest = GetModelGroupSystemPullImageSecretRequest
+
+class ModelGroupSystemPullSecretIdentifier(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
+    ACCOUNT_ID_FIELD_NUMBER: builtins.int
+    model_group_name: builtins.str
+    account_id: builtins.str
+    def __init__(
+        self,
+        *,
+        model_group_name: builtins.str = ...,
+        account_id: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_group_identifier", b"model_group_identifier", "model_group_name", b"model_group_name"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["account_id", b"account_id", "model_group_identifier", b"model_group_identifier", "model_group_name", b"model_group_name"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["model_group_identifier", b"model_group_identifier"]) -> typing_extensions.Literal["model_group_name"] | None: ...
+
+global___ModelGroupSystemPullSecretIdentifier = ModelGroupSystemPullSecretIdentifier
+
+class SystemPullSecretInfoNamesResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SECRET_INFOS_FIELD_NUMBER: builtins.int
+    @property
+    def secret_infos(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SystemPullSecretInfo]: ...
+    def __init__(
+        self,
+        *,
+        secret_infos: collections.abc.Iterable[global___SystemPullSecretInfo] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["secret_infos", b"secret_infos"]) -> None: ...
+
+global___SystemPullSecretInfoNamesResponse = SystemPullSecretInfoNamesResponse
+
+class SystemPullSecretInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
+    SECRET_PULL_IMAGE_FIELD_NUMBER: builtins.int
+    ACCOUNT_ID_FIELD_NUMBER: builtins.int
+    model_group_name: builtins.str
+    secret_pull_image: builtins.str
+    account_id: builtins.str
+    def __init__(
+        self,
+        *,
+        model_group_name: builtins.str = ...,
+        secret_pull_image: builtins.str = ...,
+        account_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["account_id", b"account_id", "model_group_name", b"model_group_name", "secret_pull_image", b"secret_pull_image"]) -> None: ...
+
+global___SystemPullSecretInfo = SystemPullSecretInfo
+
 class CreateIfNotExistsModelGroupRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -102,3 +171,22 @@ class CreateRepositoriesForModelGroupResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["artifactory_repository_details", b"artifactory_repository_details"]) -> None: ...
 
 global___CreateRepositoriesForModelGroupResponse = CreateRepositoriesForModelGroupResponse
+
+class GetModelGroupInfoRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
+    model_group_name: builtins.str
+    """Model group name"""
+    jfrog_project_key: builtins.str
+    """Artifactory project key used for repository organization"""
+    def __init__(
+        self,
+        *,
+        model_group_name: builtins.str = ...,
+        jfrog_project_key: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["jfrog_project_key", b"jfrog_project_key", "model_group_name", b"model_group_name"]) -> None: ...
+
+global___GetModelGroupInfoRequest = GetModelGroupInfoRequest

_qwak_proto/qwak/model_group/model_group_pb2_grpc.py

@@ -24,6 +24,16 @@ class ModelGroupManagementServiceStub(object):
                 request_serializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupRequest.SerializeToString,
                 response_deserializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupResponse.FromString,
                 )
+        self.GetModelGroupInfo = channel.unary_unary(
+                '/qwak.model_groups.management.ModelGroupManagementService/GetModelGroupInfo',
+                request_serializer=qwak_dot_model__group_dot_model__group__pb2.GetModelGroupInfoRequest.SerializeToString,
+                response_deserializer=qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.FromString,
+                )
+        self.GetModelGroupSystemPullImageSecret = channel.unary_unary(
+                '/qwak.model_groups.management.ModelGroupManagementService/GetModelGroupSystemPullImageSecret',
+                request_serializer=qwak_dot_model__group_dot_model__group__pb2.GetModelGroupSystemPullImageSecretRequest.SerializeToString,
+                response_deserializer=qwak_dot_model__group_dot_model__group__pb2.SystemPullSecretInfoNamesResponse.FromString,
+                )
 
 
 class ModelGroupManagementServiceServicer(object):
@@ -41,6 +51,22 @@ class ModelGroupManagementServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetModelGroupInfo(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 GetModelGroupSystemPullImageSecret(self, request, context):
+        """*
+        This end point gets a list of model group names and returns a list of SystemPullSecretInfo
+        that contains model group name and the secret pull image with the "system-" addition to the pull secret name
+        For example:
+        """
+        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 = {
@@ -54,6 +80,16 @@ def add_ModelGroupManagementServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupRequest.FromString,
                     response_serializer=qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupResponse.SerializeToString,
             ),
+            'GetModelGroupInfo': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetModelGroupInfo,
+                    request_deserializer=qwak_dot_model__group_dot_model__group__pb2.GetModelGroupInfoRequest.FromString,
+                    response_serializer=qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.SerializeToString,
+            ),
+            'GetModelGroupSystemPullImageSecret': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetModelGroupSystemPullImageSecret,
+                    request_deserializer=qwak_dot_model__group_dot_model__group__pb2.GetModelGroupSystemPullImageSecretRequest.FromString,
+                    response_serializer=qwak_dot_model__group_dot_model__group__pb2.SystemPullSecretInfoNamesResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.model_groups.management.ModelGroupManagementService', rpc_method_handlers)
@@ -97,3 +133,37 @@ class ModelGroupManagementService(object):
             qwak_dot_model__group_dot_model__group__pb2.CreateRepositoriesForModelGroupResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetModelGroupInfo(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/GetModelGroupInfo',
+            qwak_dot_model__group_dot_model__group__pb2.GetModelGroupInfoRequest.SerializeToString,
+            qwak_dot_model__group_dot_model__group__pb2.ModelGroupBriefInfoResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetModelGroupSystemPullImageSecret(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/GetModelGroupSystemPullImageSecret',
+            qwak_dot_model__group_dot_model__group__pb2.GetModelGroupSystemPullImageSecretRequest.SerializeToString,
+            qwak_dot_model__group_dot_model__group__pb2.SystemPullSecretInfoNamesResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_qwak_proto/qwak/model_group/model_group_repository_details_pb2.py

@@ -4,9 +4,8 @@
 """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
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -14,82 +13,31 @@ _sym_db = _symbol_database.Default()
 
 
 
-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']
-_HUGGINGFACEREPOSITORY = DESCRIPTOR.message_types_by_name['HuggingFaceRepository']
-_REPOSITORYTYPE = DESCRIPTOR.message_types_by_name['RepositoryType']
-RepositoryDetails = _reflection.GeneratedProtocolMessageType('RepositoryDetails', (_message.Message,), {
-  'DESCRIPTOR' : _REPOSITORYDETAILS,
-  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.RepositoryDetails)
-  })
-_sym_db.RegisterMessage(RepositoryDetails)
-
-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'
-  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.DockerRepository)
-  })
-_sym_db.RegisterMessage(DockerRepository)
-
-DatasetRepository = _reflection.GeneratedProtocolMessageType('DatasetRepository', (_message.Message,), {
-  'DESCRIPTOR' : _DATASETREPOSITORY,
-  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.DatasetRepository)
-  })
-_sym_db.RegisterMessage(DatasetRepository)
-
-ArtifactRepository = _reflection.GeneratedProtocolMessageType('ArtifactRepository', (_message.Message,), {
-  'DESCRIPTOR' : _ARTIFACTREPOSITORY,
-  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.ArtifactRepository)
-  })
-_sym_db.RegisterMessage(ArtifactRepository)
-
-HuggingFaceRepository = _reflection.GeneratedProtocolMessageType('HuggingFaceRepository', (_message.Message,), {
-  'DESCRIPTOR' : _HUGGINGFACEREPOSITORY,
-  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.HuggingFaceRepository)
-  })
-_sym_db.RegisterMessage(HuggingFaceRepository)
-
-RepositoryType = _reflection.GeneratedProtocolMessageType('RepositoryType', (_message.Message,), {
-  'DESCRIPTOR' : _REPOSITORYTYPE,
-  '__module__' : 'qwak.model_group.model_group_repository_details_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.model_groups.management.RepositoryType)
-  })
-_sym_db.RegisterMessage(RepositoryType)
+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\"\x80\x03\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\x45\n\x0enpm_repository\x18\x04 \x01(\x0b\x32+.qwak.model_groups.management.NpmRepositoryH\x00\x12G\n\x0fpypi_repository\x18\x05 \x01(\x0b\x32,.qwak.model_groups.management.PypiRepositoryH\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\"\x0f\n\rNpmRepository\"\x10\n\x0ePypiRepository\"\xed\x03\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\x12\x45\n\x0enpm_repository\x18\x05 \x01(\x0b\x32+.qwak.model_groups.management.NpmRepositoryH\x00\x12G\n\x0fpypi_repository\x18\x06 \x01(\x0b\x32,.qwak.model_groups.management.PypiRepositoryH\x00\x42\x06\n\x04typeBN\n&com.qwak.ai.management.model.group.apiB\"ModelGroupRepositoriesDetailsProtoP\x01\x62\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.model_group.model_group_repository_details_pb2', _globals)
 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=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
+  _globals['_REPOSITORYDETAILS']._serialized_start=87
+  _globals['_REPOSITORYDETAILS']._serialized_end=201
+  _globals['_REMOTEREPOSITORYDETAILS']._serialized_start=204
+  _globals['_REMOTEREPOSITORYDETAILS']._serialized_end=588
+  _globals['_DOCKERREPOSITORY']._serialized_start=590
+  _globals['_DOCKERREPOSITORY']._serialized_end=608
+  _globals['_DATASETREPOSITORY']._serialized_start=610
+  _globals['_DATASETREPOSITORY']._serialized_end=629
+  _globals['_ARTIFACTREPOSITORY']._serialized_start=631
+  _globals['_ARTIFACTREPOSITORY']._serialized_end=651
+  _globals['_HUGGINGFACEREPOSITORY']._serialized_start=653
+  _globals['_HUGGINGFACEREPOSITORY']._serialized_end=676
+  _globals['_NPMREPOSITORY']._serialized_start=678
+  _globals['_NPMREPOSITORY']._serialized_end=693
+  _globals['_PYPIREPOSITORY']._serialized_start=695
+  _globals['_PYPIREPOSITORY']._serialized_end=711
+  _globals['_REPOSITORYTYPE']._serialized_start=714
+  _globals['_REPOSITORYTYPE']._serialized_end=1207
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/model_group/model_group_repository_details_pb2.pyi

@@ -40,6 +40,8 @@ class RemoteRepositoryDetails(google.protobuf.message.Message):
 
     DOCKER_REPOSITORY_FIELD_NUMBER: builtins.int
     HUGGING_FACE_REPOSITORY_FIELD_NUMBER: builtins.int
+    NPM_REPOSITORY_FIELD_NUMBER: builtins.int
+    PYPI_REPOSITORY_FIELD_NUMBER: builtins.int
     REPOSITORY_REMOTE_URL_FIELD_NUMBER: builtins.int
     @property
     def docker_repository(self) -> global___DockerRepository:
@@ -47,6 +49,12 @@ class RemoteRepositoryDetails(google.protobuf.message.Message):
     @property
     def hugging_face_repository(self) -> global___HuggingFaceRepository:
         """HuggingFace repository"""
+    @property
+    def npm_repository(self) -> global___NpmRepository:
+        """NPM repository"""
+    @property
+    def pypi_repository(self) -> global___PypiRepository:
+        """PyPI repository"""
     repository_remote_url: builtins.str
     """The remote repository URL"""
     def __init__(
@@ -54,11 +62,13 @@ class RemoteRepositoryDetails(google.protobuf.message.Message):
         *,
         docker_repository: global___DockerRepository | None = ...,
         hugging_face_repository: global___HuggingFaceRepository | None = ...,
+        npm_repository: global___NpmRepository | None = ...,
+        pypi_repository: global___PypiRepository | 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: ...
+    def HasField(self, field_name: typing_extensions.Literal["docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "npm_repository", b"npm_repository", "pypi_repository", b"pypi_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", "npm_repository", b"npm_repository", "pypi_repository", b"pypi_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", "npm_repository", "pypi_repository"] | None: ...
 
 global___RemoteRepositoryDetails = RemoteRepositoryDetails
 
@@ -98,6 +108,24 @@ class HuggingFaceRepository(google.protobuf.message.Message):
 
 global___HuggingFaceRepository = HuggingFaceRepository
 
+class NpmRepository(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___NpmRepository = NpmRepository
+
+class PypiRepository(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___PypiRepository = PypiRepository
+
 class RepositoryType(google.protobuf.message.Message):
     """The repository type"""
 
@@ -107,6 +135,8 @@ class RepositoryType(google.protobuf.message.Message):
     DATASET_REPOSITORY_FIELD_NUMBER: builtins.int
     ARTIFACT_REPOSITORY_FIELD_NUMBER: builtins.int
     HUGGING_FACE_REPOSITORY_FIELD_NUMBER: builtins.int
+    NPM_REPOSITORY_FIELD_NUMBER: builtins.int
+    PYPI_REPOSITORY_FIELD_NUMBER: builtins.int
     @property
     def docker_repository(self) -> global___DockerRepository:
         """Docker repository"""
@@ -119,6 +149,12 @@ class RepositoryType(google.protobuf.message.Message):
     @property
     def hugging_face_repository(self) -> global___HuggingFaceRepository:
         """HuggingFace repository"""
+    @property
+    def npm_repository(self) -> global___NpmRepository:
+        """NPM repository"""
+    @property
+    def pypi_repository(self) -> global___PypiRepository:
+        """PyPI repository"""
     def __init__(
         self,
         *,
@@ -126,9 +162,11 @@ class RepositoryType(google.protobuf.message.Message):
         dataset_repository: global___DatasetRepository | None = ...,
         artifact_repository: global___ArtifactRepository | None = ...,
         hugging_face_repository: global___HuggingFaceRepository | None = ...,
+        npm_repository: global___NpmRepository | None = ...,
+        pypi_repository: global___PypiRepository | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["artifact_repository", b"artifact_repository", "dataset_repository", b"dataset_repository", "docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["artifact_repository", b"artifact_repository", "dataset_repository", b"dataset_repository", "docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "type", b"type"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["docker_repository", "dataset_repository", "artifact_repository", "hugging_face_repository"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["artifact_repository", b"artifact_repository", "dataset_repository", b"dataset_repository", "docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "npm_repository", b"npm_repository", "pypi_repository", b"pypi_repository", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["artifact_repository", b"artifact_repository", "dataset_repository", b"dataset_repository", "docker_repository", b"docker_repository", "hugging_face_repository", b"hugging_face_repository", "npm_repository", b"npm_repository", "pypi_repository", b"pypi_repository", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["docker_repository", "dataset_repository", "artifact_repository", "hugging_face_repository", "npm_repository", "pypi_repository"] | None: ...
 
 global___RepositoryType = RepositoryType