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

_qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi

@@ -0,0 +1,215 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import google.protobuf.struct_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class OpenAIProviderDescriptor(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODELS_FIELD_NUMBER: builtins.int
+    @property
+    def models(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpenAIModelDescriptor]: ...
+    def __init__(
+        self,
+        *,
+        models: collections.abc.Iterable[global___OpenAIModelDescriptor] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["models", b"models"]) -> None: ...
+
+global___OpenAIProviderDescriptor = OpenAIProviderDescriptor
+
+class OpenAIModelDescriptor(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    CAPABILITIES_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    @property
+    def capabilities(self) -> global___OpenAICapabilities: ...
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        capabilities: global___OpenAICapabilities | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["capabilities", b"capabilities"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["capabilities", b"capabilities", "model_id", b"model_id"]) -> None: ...
+
+global___OpenAIModelDescriptor = OpenAIModelDescriptor
+
+class OpenAIChatModelParams(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FREQUENCY_PENALTY_FIELD_NUMBER: builtins.int
+    LOGIT_BIAS_FIELD_NUMBER: builtins.int
+    LOGPROBS_FIELD_NUMBER: builtins.int
+    MAX_TOKENS_FIELD_NUMBER: builtins.int
+    N_FIELD_NUMBER: builtins.int
+    PRESENCE_PENALTY_FIELD_NUMBER: builtins.int
+    RESPONSE_FORMAT_FIELD_NUMBER: builtins.int
+    SEED_FIELD_NUMBER: builtins.int
+    STOP_FIELD_NUMBER: builtins.int
+    TEMPERATURE_FIELD_NUMBER: builtins.int
+    TOOL_CHOICE_FIELD_NUMBER: builtins.int
+    TOOLS_SPEC_FIELD_NUMBER: builtins.int
+    TOP_LOGPROBS_FIELD_NUMBER: builtins.int
+    TOP_P_FIELD_NUMBER: builtins.int
+    USER_FIELD_NUMBER: builtins.int
+    frequency_penalty: builtins.float
+    @property
+    def logit_bias(self) -> google.protobuf.struct_pb2.Struct:
+        """json object of string to int"""
+    logprobs: builtins.bool
+    max_tokens: builtins.int
+    n: builtins.int
+    presence_penalty: builtins.float
+    response_format: builtins.str
+    seed: builtins.int
+    @property
+    def stop(self) -> google.protobuf.struct_pb2.ListValue:
+        """Json list of strings"""
+    temperature: builtins.float
+    @property
+    def tool_choice(self) -> global___ToolChoice: ...
+    @property
+    def tools_spec(self) -> global___Tools: ...
+    top_logprobs: builtins.int
+    top_p: builtins.float
+    user: builtins.str
+    def __init__(
+        self,
+        *,
+        frequency_penalty: builtins.float = ...,
+        logit_bias: google.protobuf.struct_pb2.Struct | None = ...,
+        logprobs: builtins.bool = ...,
+        max_tokens: builtins.int = ...,
+        n: builtins.int = ...,
+        presence_penalty: builtins.float = ...,
+        response_format: builtins.str = ...,
+        seed: builtins.int = ...,
+        stop: google.protobuf.struct_pb2.ListValue | None = ...,
+        temperature: builtins.float = ...,
+        tool_choice: global___ToolChoice | None = ...,
+        tools_spec: global___Tools | None = ...,
+        top_logprobs: builtins.int = ...,
+        top_p: builtins.float = ...,
+        user: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["frequency_penalty", b"frequency_penalty", "logit_bias", b"logit_bias", "logprobs", b"logprobs", "max_tokens", b"max_tokens", "n", b"n", "optional_frequency_penalty", b"optional_frequency_penalty", "optional_logit_bias", b"optional_logit_bias", "optional_logprobs", b"optional_logprobs", "optional_max_tokens", b"optional_max_tokens", "optional_n", b"optional_n", "optional_presence_penalty", b"optional_presence_penalty", "optional_response_format", b"optional_response_format", "optional_seed", b"optional_seed", "optional_stop", b"optional_stop", "optional_temperature", b"optional_temperature", "optional_tool_choice", b"optional_tool_choice", "optional_tools", b"optional_tools", "optional_top_logprobs", b"optional_top_logprobs", "optional_top_p", b"optional_top_p", "optional_user", b"optional_user", "presence_penalty", b"presence_penalty", "response_format", b"response_format", "seed", b"seed", "stop", b"stop", "temperature", b"temperature", "tool_choice", b"tool_choice", "tools_spec", b"tools_spec", "top_logprobs", b"top_logprobs", "top_p", b"top_p", "user", b"user"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["frequency_penalty", b"frequency_penalty", "logit_bias", b"logit_bias", "logprobs", b"logprobs", "max_tokens", b"max_tokens", "n", b"n", "optional_frequency_penalty", b"optional_frequency_penalty", "optional_logit_bias", b"optional_logit_bias", "optional_logprobs", b"optional_logprobs", "optional_max_tokens", b"optional_max_tokens", "optional_n", b"optional_n", "optional_presence_penalty", b"optional_presence_penalty", "optional_response_format", b"optional_response_format", "optional_seed", b"optional_seed", "optional_stop", b"optional_stop", "optional_temperature", b"optional_temperature", "optional_tool_choice", b"optional_tool_choice", "optional_tools", b"optional_tools", "optional_top_logprobs", b"optional_top_logprobs", "optional_top_p", b"optional_top_p", "optional_user", b"optional_user", "presence_penalty", b"presence_penalty", "response_format", b"response_format", "seed", b"seed", "stop", b"stop", "temperature", b"temperature", "tool_choice", b"tool_choice", "tools_spec", b"tools_spec", "top_logprobs", b"top_logprobs", "top_p", b"top_p", "user", b"user"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_frequency_penalty", b"optional_frequency_penalty"]) -> typing_extensions.Literal["frequency_penalty"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_logit_bias", b"optional_logit_bias"]) -> typing_extensions.Literal["logit_bias"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_logprobs", b"optional_logprobs"]) -> typing_extensions.Literal["logprobs"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_max_tokens", b"optional_max_tokens"]) -> typing_extensions.Literal["max_tokens"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_n", b"optional_n"]) -> typing_extensions.Literal["n"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_presence_penalty", b"optional_presence_penalty"]) -> typing_extensions.Literal["presence_penalty"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_response_format", b"optional_response_format"]) -> typing_extensions.Literal["response_format"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_seed", b"optional_seed"]) -> typing_extensions.Literal["seed"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_stop", b"optional_stop"]) -> typing_extensions.Literal["stop"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_temperature", b"optional_temperature"]) -> typing_extensions.Literal["temperature"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tool_choice", b"optional_tool_choice"]) -> typing_extensions.Literal["tool_choice"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tools", b"optional_tools"]) -> typing_extensions.Literal["tools_spec"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_top_logprobs", b"optional_top_logprobs"]) -> typing_extensions.Literal["top_logprobs"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_top_p", b"optional_top_p"]) -> typing_extensions.Literal["top_p"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_user", b"optional_user"]) -> typing_extensions.Literal["user"] | None: ...
+
+global___OpenAIChatModelParams = OpenAIChatModelParams
+
+class Tools(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TOOLS_FIELD_NUMBER: builtins.int
+    @property
+    def tools(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.struct_pb2.Struct]: ...
+    def __init__(
+        self,
+        *,
+        tools: collections.abc.Iterable[google.protobuf.struct_pb2.Struct] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["tools", b"tools"]) -> None: ...
+
+global___Tools = Tools
+
+class ToolChoice(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    JSON_FIELD_NUMBER: builtins.int
+    LITERAL_FIELD_NUMBER: builtins.int
+    @property
+    def json(self) -> google.protobuf.struct_pb2.Struct: ...
+    literal: builtins.str
+    def __init__(
+        self,
+        *,
+        json: google.protobuf.struct_pb2.Struct | None = ...,
+        literal: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["json", b"json", "literal", b"literal", "value_type", b"value_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["json", b"json", "literal", b"literal", "value_type", b"value_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["value_type", b"value_type"]) -> typing_extensions.Literal["json", "literal"] | None: ...
+
+global___ToolChoice = ToolChoice
+
+class OpenAICapabilities(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CHAT_API_FIELD_NUMBER: builtins.int
+    @property
+    def chat_api(self) -> global___OpenAIChatAPI: ...
+    def __init__(
+        self,
+        *,
+        chat_api: global___OpenAIChatAPI | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["chat_api", b"chat_api", "optional_chat", b"optional_chat"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["chat_api", b"chat_api", "optional_chat", b"optional_chat"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_chat", b"optional_chat"]) -> typing_extensions.Literal["chat_api"] | None: ...
+
+global___OpenAICapabilities = OpenAICapabilities
+
+class OpenAIChatAPI(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CHAT_PARAMS_FIELD_NUMBER: builtins.int
+    @property
+    def chat_params(self) -> global___OpenAIChatModelParams: ...
+    def __init__(
+        self,
+        *,
+        chat_params: global___OpenAIChatModelParams | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["chat_params", b"chat_params", "optional_chat_params", b"optional_chat_params"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["chat_params", b"chat_params", "optional_chat_params", b"optional_chat_params"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_chat_params", b"optional_chat_params"]) -> typing_extensions.Literal["chat_params"] | None: ...
+
+global___OpenAIChatAPI = OpenAIChatAPI

_qwak_proto/qwak/model_descriptor/open_ai_descriptor_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
+

_qwak_proto/qwak/model_group/model_group_pb2.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/model_group/model_group.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from _qwak_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\"\xa3\x01\n)GetModelGroupSystemPullImageSecretRequest\x12v\n*model_group_system_pull_secret_identifiers\x18\x01 \x03(\x0b\x32\x42.qwak.model_groups.management.ModelGroupSystemPullSecretIdentifier\"p\n$ModelGroupSystemPullSecretIdentifier\x12\x1a\n\x10model_group_name\x18\x01 \x01(\tH\x00\x12\x12\n\naccount_id\x18\x02 \x01(\tB\x18\n\x16model_group_identifier\"m\n!SystemPullSecretInfoNamesResponse\x12H\n\x0csecret_infos\x18\x01 \x03(\x0b\x32\x32.qwak.model_groups.management.SystemPullSecretInfo\"_\n\x14SystemPullSecretInfo\x12\x18\n\x10model_group_name\x18\x01 \x01(\t\x12\x19\n\x11secret_pull_image\x18\x02 \x01(\t\x12\x12\n\naccount_id\x18\x03 \x01(\t\"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.RepositoryDetails\"O\n\x18GetModelGroupInfoRequest\x12\x18\n\x10model_group_name\x18\x01 \x01(\t\x12\x19\n\x11jfrog_project_key\x18\x02 \x01(\t2\xa5\x05\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.CreateRepositoriesForModelGroupResponse\x12\x86\x01\n\x11GetModelGroupInfo\x12\x36.qwak.model_groups.management.GetModelGroupInfoRequest\x1a\x39.qwak.model_groups.management.ModelGroupBriefInfoResponse\x12\xae\x01\n\"GetModelGroupSystemPullImageSecret\x12G.qwak.model_groups.management.GetModelGroupSystemPullImageSecretRequest\x1a?.qwak.model_groups.management.SystemPullSecretInfoNamesResponseBL\n&com.qwak.ai.management.model.group.apiB ModelGroupManagementServiceProtoP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.model_group.model_group_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.management.model.group.apiB ModelGroupManagementServiceProtoP\001'
+  _globals['_GETMODELGROUPSYSTEMPULLIMAGESECRETREQUEST']._serialized_start=157
+  _globals['_GETMODELGROUPSYSTEMPULLIMAGESECRETREQUEST']._serialized_end=320
+  _globals['_MODELGROUPSYSTEMPULLSECRETIDENTIFIER']._serialized_start=322
+  _globals['_MODELGROUPSYSTEMPULLSECRETIDENTIFIER']._serialized_end=434
+  _globals['_SYSTEMPULLSECRETINFONAMESRESPONSE']._serialized_start=436
+  _globals['_SYSTEMPULLSECRETINFONAMESRESPONSE']._serialized_end=545
+  _globals['_SYSTEMPULLSECRETINFO']._serialized_start=547
+  _globals['_SYSTEMPULLSECRETINFO']._serialized_end=642
+  _globals['_CREATEIFNOTEXISTSMODELGROUPREQUEST']._serialized_start=644
+  _globals['_CREATEIFNOTEXISTSMODELGROUPREQUEST']._serialized_end=733
+  _globals['_MODELGROUPBRIEFINFORESPONSE']._serialized_start=736
+  _globals['_MODELGROUPBRIEFINFORESPONSE']._serialized_end=959
+  _globals['_CREATEREPOSITORIESFORMODELGROUPREQUEST']._serialized_start=962
+  _globals['_CREATEREPOSITORIESFORMODELGROUPREQUEST']._serialized_end=1111
+  _globals['_CREATEREPOSITORIESFORMODELGROUPRESPONSE']._serialized_start=1114
+  _globals['_CREATEREPOSITORIESFORMODELGROUPRESPONSE']._serialized_end=1244
+  _globals['_GETMODELGROUPINFOREQUEST']._serialized_start=1246
+  _globals['_GETMODELGROUPINFOREQUEST']._serialized_end=1325
+  _globals['_MODELGROUPMANAGEMENTSERVICE']._serialized_start=1328
+  _globals['_MODELGROUPMANAGEMENTSERVICE']._serialized_end=2005
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/model_group/model_group_pb2.pyi

@@ -0,0 +1,192 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import qwak.model_group.model_group_repository_details_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+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
+
+    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___CreateIfNotExistsModelGroupRequest = CreateIfNotExistsModelGroupRequest
+
+class ModelGroupBriefInfoResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_ID_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_DESCRIPTION_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_CREATED_AT_FIELD_NUMBER: builtins.int
+    REPOSITORY_DETAILS_FIELD_NUMBER: builtins.int
+    model_group_id: builtins.str
+    """Unique identifier for the model group"""
+    model_group_description: builtins.str
+    """Description of the model group"""
+    @property
+    def model_group_created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """The model group creation timestamp"""
+    @property
+    def 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,
+        *,
+        model_group_id: builtins.str = ...,
+        model_group_description: builtins.str = ...,
+        model_group_created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        repository_details: collections.abc.Iterable[qwak.model_group.model_group_repository_details_pb2.RepositoryDetails] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_group_created_at", b"model_group_created_at"]) -> builtins.bool: ...
+    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
+
+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

@@ -0,0 +1,169 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from _qwak_proto.qwak.model_group import model_group_pb2 as qwak_dot_model__group_dot_model__group__pb2
+
+
+class ModelGroupManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateIfNotExistsModelGroup = channel.unary_unary(
+                '/qwak.model_groups.management.ModelGroupManagementService/CreateIfNotExistsModelGroup',
+                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,
+                )
+        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):
+    """Missing associated documentation comment in .proto file."""
+
+    def CreateIfNotExistsModelGroup(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 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 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 = {
+            'CreateIfNotExistsModelGroup': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateIfNotExistsModelGroup,
+                    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,
+            ),
+            '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)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class ModelGroupManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def CreateIfNotExistsModelGroup(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/CreateIfNotExistsModelGroup',
+            qwak_dot_model__group_dot_model__group__pb2.CreateIfNotExistsModelGroupRequest.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 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)
+
+    @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)