frogml 1.1.22__tar.gz → 1.2.22__tar.gz

frogml-1.2.22/PKG-INFO

@@ -0,0 +1,117 @@
+Metadata-Version: 2.3
+Name: frogml
+Version: 1.2.22
+Summary: frogml contains the necessary objects and communication tools for using the JFrog ml Platform
+License: Apache-2.0
+Keywords: mlops,ml,deployment,serving,model
+Author: JFrog
+Requires-Python: >=3.9,<3.14
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Provides-Extra: catboost
+Provides-Extra: feature-store
+Provides-Extra: onnx
+Provides-Extra: scikit-learn
+Provides-Extra: torch
+Provides-Extra: transformers
+Requires-Dist: PyYAML (>=6.0.2)
+Requires-Dist: cachetools (>=5.2,<7)
+Requires-Dist: catboost (>=1.2.8,<2.0.0) ; extra == "catboost"
+Requires-Dist: chevron (==0.14.0)
+Requires-Dist: cloudpickle (==2.2.1)
+Requires-Dist: dacite (==1.9.2)
+Requires-Dist: dependency-injector (>=4.0)
+Requires-Dist: filelock
+Requires-Dist: grpcio (>=1.71.2,<1.75)
+Requires-Dist: joblib (>=1.3.2,<2.0.0)
+Requires-Dist: libcst (>=1.8.2,<2.0.0)
+Requires-Dist: marshmallow-dataclass (>=8.5.8,<9.0.0)
+Requires-Dist: onnx (>=1.18.0,<2.0.0) ; extra == "onnx"
+Requires-Dist: pandas (>=1.4.0) ; extra == "feature-store"
+Requires-Dist: protobuf (>=4.25.8,<5) ; python_version < "3.12"
+Requires-Dist: protobuf (>=6.33.1,<7) ; python_version >= "3.12"
+Requires-Dist: pyarrow (>=20.0.0) ; extra == "feature-store"
+Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
+Requires-Dist: pydantic
+Requires-Dist: pyjwt (>=2.10.1,<3)
+Requires-Dist: pyspark (==3.5.7) ; (python_version < "3.12") and (extra == "feature-store")
+Requires-Dist: pyspark (>=4.0.1,<5) ; (python_version >= "3.12") and (extra == "feature-store")
+Requires-Dist: python-json-logger (>=2.0.2)
+Requires-Dist: requests
+Requires-Dist: retrying (==1.4.2)
+Requires-Dist: scikit-learn (>=1.5,<2.0) ; extra == "scikit-learn"
+Requires-Dist: strenum (>=0.4.15,<0.5.0)
+Requires-Dist: torch (>=2.7.1,<3.0.0) ; extra == "torch"
+Requires-Dist: tqdm
+Requires-Dist: transformers (>=4.53.0,<5.0.0) ; extra == "transformers"
+Requires-Dist: typeguard (>=2,<3)
+Project-URL: Home page, https://www.jfrog.com/
+Description-Content-Type: text/markdown
+
+# Frogml
+
+Frogml is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Frogml Core contains all the objects and tools necessary to use the Frogml Platform
+
+## Table of contents:
+
+- [Overview](#overview)
+- [Working with Artifactory](#Working-with-Artifactory)
+- [Upload ML model to Artifactory](#Upload-ML-model-to-Artifactory)
+- [Local Development Setup](#local-development-setup)
+
+## Overview
+
+JFrog ML Storage is a smart python client library providing a simple and efficient method of storing and downloading models, model data and datasets from the JFrog platform, utilizing the advanced capabilities of the JFrog platform.
+
+## Working with Artifactory
+
+FrogML Storage Library support is available from Artifactory version 7.84.x.
+
+To be able to use FrogML Storage with Artifactory, you should authenticate the frogml storage client against Artifactory.
+JFrog implements a credentials provider chain. It sequentially checks each place where you can set the credentials to authenticate with FrogML, and then selects the first one you set.
+
+### Upload ML model to Artifactory
+
+You can upload a model to a FrogML repository using the upload_model_version() function. 
+You can upload a single file or an entire folder.
+This function uses checksum upload, assigning a SHA2 value to each model for retrieval from storage. If the binary content cannot be reused, the smart upload mechanism performs regular upload instead.
+After uploading the model, FrogML generates a file named model-info.json which contains the model name and its related files and dependencies.
+
+The version parameter is optional. If not specified, Artifactory will set the version as the timestamp of the time you uploaded the model in your time zone, in UTC format:  yyyy-MM-dd-HH-mm-ss.
+Additionally, you can add properties to the model in Artifactory to categorize and label it.
+The function upload_model_version returns an instance of FrogMlModelVersion, which includes the model's name, version, and namespace.
+
+## Local Development Setup
+
+To install FrogML locally with development dependencies, you must authenticate with **Repo21** (a private JFrog repository) to fetch the `QwakBentoML` dependency.
+### 1. Generate Credentials
+1. Log in to **Repo 21** via JFrog Okta.
+2. Go to **User Profile** (top right) → **Set Me Up**.
+3. Select **PyPI** and choose the repository `artifactory-pypi-virtual`.
+4. Click **Generate Token & Create Instructions**. Your **username** and **token** will be displayed there.
+
+### 2. Configure Poetry
+Choose **one** of the following methods to authenticate:
+
+#### Option A: Global Configuration
+Run the following command to persist your credentials:
+```bash
+poetry config http-basic.jfrog <your_username> <your_token>
+```
+
+#### Option B: Environment Variables
+Export the credentials as environment variables:
+```bash
+export POETRY_HTTP_BASIC_JFROG_USERNAME=<your_username>
+export POETRY_HTTP_BASIC_JFROG_PASSWORD=<your_token>
+```
+

frogml-1.2.22/README.md

@@ -0,0 +1,58 @@
+# Frogml
+
+Frogml is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Frogml Core contains all the objects and tools necessary to use the Frogml Platform
+
+## Table of contents:
+
+- [Overview](#overview)
+- [Working with Artifactory](#Working-with-Artifactory)
+- [Upload ML model to Artifactory](#Upload-ML-model-to-Artifactory)
+- [Local Development Setup](#local-development-setup)
+
+## Overview
+
+JFrog ML Storage is a smart python client library providing a simple and efficient method of storing and downloading models, model data and datasets from the JFrog platform, utilizing the advanced capabilities of the JFrog platform.
+
+## Working with Artifactory
+
+FrogML Storage Library support is available from Artifactory version 7.84.x.
+
+To be able to use FrogML Storage with Artifactory, you should authenticate the frogml storage client against Artifactory.
+JFrog implements a credentials provider chain. It sequentially checks each place where you can set the credentials to authenticate with FrogML, and then selects the first one you set.
+
+### Upload ML model to Artifactory
+
+You can upload a model to a FrogML repository using the upload_model_version() function. 
+You can upload a single file or an entire folder.
+This function uses checksum upload, assigning a SHA2 value to each model for retrieval from storage. If the binary content cannot be reused, the smart upload mechanism performs regular upload instead.
+After uploading the model, FrogML generates a file named model-info.json which contains the model name and its related files and dependencies.
+
+The version parameter is optional. If not specified, Artifactory will set the version as the timestamp of the time you uploaded the model in your time zone, in UTC format:  yyyy-MM-dd-HH-mm-ss.
+Additionally, you can add properties to the model in Artifactory to categorize and label it.
+The function upload_model_version returns an instance of FrogMlModelVersion, which includes the model's name, version, and namespace.
+
+## Local Development Setup
+
+To install FrogML locally with development dependencies, you must authenticate with **Repo21** (a private JFrog repository) to fetch the `QwakBentoML` dependency.
+### 1. Generate Credentials
+1. Log in to **Repo 21** via JFrog Okta.
+2. Go to **User Profile** (top right) → **Set Me Up**.
+3. Select **PyPI** and choose the repository `artifactory-pypi-virtual`.
+4. Click **Generate Token & Create Instructions**. Your **username** and **token** will be displayed there.
+
+### 2. Configure Poetry
+Choose **one** of the following methods to authenticate:
+
+#### Option A: Global Configuration
+Run the following command to persist your credentials:
+```bash
+poetry config http-basic.jfrog <your_username> <your_token>
+```
+
+#### Option B: Environment Variables
+Export the credentials as environment variables:
+```bash
+export POETRY_HTTP_BASIC_JFROG_USERNAME=<your_username>
+export POETRY_HTTP_BASIC_JFROG_PASSWORD=<your_token>
+```

frogml-1.2.22/frogml/__init__.py

@@ -0,0 +1,23 @@
+__author__ = "jfrog"
+__version__ = "1.2.22"
+
+from frogml.sdk.model.decorators.api import api_decorator as api
+from frogml.sdk.model_loggers.model_logger import load_model, log_model
+from frogml.sdk.frogml_client.client import FrogMLClient
+from frogml.sdk.model.base import BaseModel as FrogMlModel
+
+from frogml.core.model_loggers.artifact_logger import (
+    load_file,
+    log_file,
+)
+from frogml.core.model_loggers.data_logger import load_data, log_data
+from frogml.sdk.model.decorators.timer import frogml_timer
+from frogml.sdk.model.model_version_tracking import (
+    log_metric,
+    log_param,
+)
+from frogml.sdk.model_version import *
+
+from frogml.core.inner.di_configuration import wire_dependencies
+
+_container = wire_dependencies()

frogml-1.2.22/frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.jfml.hosting_gateway/v1/build_upload_url.proto
+# Protobuf Python Version: 4.25.1
+"""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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.jfml/hosting_gateway/v1/build_upload_url.proto\x12\x17jfml.hosting_gateway.v1\"\xd3\x01\n\x18\x42uildUploadUrlParameters\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x11\n\ttag_value\x18\x03 \x01(\t\x12\x41\n\rartifact_type\x18\x04 \x01(\x0e\x32*.jfml.hosting_gateway.v1.BuildArtifactType\x12$\n\x1ause_cloud_provider_storage\x18\x05 \x01(\x08H\x00\x42\x17\n\x15\x63loud_provider_option*v\n\x11\x42uildArtifactType\x12\x19\n\x15INVALID_ARTIFACT_TYPE\x10\x00\x12\x16\n\x12\x44\x41TA_ARTIFACT_TYPE\x10\x01\x12\x16\n\x12\x46ILE_ARTIFACT_TYPE\x10\x02\x12\x16\n\x12\x43ODE_ARTIFACT_TYPE\x10\x03\x42\'\n#com.jfrog.ml.hosting_gateway.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.hosting_gateway.v1.build_upload_url_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
+  _globals['_BUILDARTIFACTTYPE']._serialized_start=289
+  _globals['_BUILDARTIFACTTYPE']._serialized_end=407
+  _globals['_BUILDUPLOADURLPARAMETERS']._serialized_start=76
+  _globals['_BUILDUPLOADURLPARAMETERS']._serialized_end=287
+# @@protoc_insertion_point(module_scope)

frogml-1.2.22/frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.pyi

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

frogml-1.2.22/frogml/_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.py

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

frogml-1.2.22/frogml/_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.pyi

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

frogml-1.2.22/frogml/_proto/jfml/model_version/v1/artifact_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.jfml.model_version/v1/artifact.proto
+# Protobuf Python Version: 4.25.1
+"""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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$jfml/model_version/v1/artifact.proto\x12\x15jfml.model_version.v1\"d\n\x08\x41rtifact\x12\x15\n\rartifact_path\x18\x01 \x01(\t\x12\x0c\n\x04size\x18\x02 \x01(\x04\x12\x33\n\tchecksums\x18\x03 \x01(\x0b\x32 .jfml.model_version.v1.Checksums\"\x19\n\tChecksums\x12\x0c\n\x04sha2\x18\x01 \x01(\tB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.artifact_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['_ARTIFACT']._serialized_start=63
+  _globals['_ARTIFACT']._serialized_end=163
+  _globals['_CHECKSUMS']._serialized_start=165
+  _globals['_CHECKSUMS']._serialized_end=190
+# @@protoc_insertion_point(module_scope)

frogml-1.2.22/frogml/_proto/jfml/model_version/v1/artifact_pb2.pyi

@@ -0,0 +1,54 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class Artifact(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ARTIFACT_PATH_FIELD_NUMBER: builtins.int
+    SIZE_FIELD_NUMBER: builtins.int
+    CHECKSUMS_FIELD_NUMBER: builtins.int
+    artifact_path: builtins.str
+    """Path to the artifact"""
+    size: builtins.int
+    """Size of the artifact"""
+    @property
+    def checksums(self) -> global___Checksums:
+        """Checksums of the artifact"""
+
+    def __init__(
+        self,
+        *,
+        artifact_path: builtins.str = ...,
+        size: builtins.int = ...,
+        checksums: global___Checksums | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["checksums", b"checksums"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["artifact_path", b"artifact_path", "checksums", b"checksums", "size", b"size"]) -> None: ...
+
+global___Artifact = Artifact
+
+@typing.final
+class Checksums(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SHA2_FIELD_NUMBER: builtins.int
+    sha2: builtins.str
+    """SHA2 checksum"""
+    def __init__(
+        self,
+        *,
+        sha2: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["sha2", b"sha2"]) -> None: ...
+
+global___Checksums = Checksums

frogml-1.2.22/frogml/_proto/jfml/model_version/v1/build_spec_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.jfml.model_version/v1/build_spec.proto
+# Protobuf Python Version: 4.25.1
+"""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 frogml._proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
+from frogml._proto.qwak.fitness_service import fitness_pb2 as qwak_dot_fitness__service_dot_fitness__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&jfml/model_version/v1/build_spec.proto\x12\x15jfml.model_version.v1\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\"\xbc\x02\n\tBuildSpec\x12@\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32&.jfml.model_version.v1.BuildProperties\x12\x32\n\tbuild_env\x18\x02 \x01(\x0b\x32\x1f.jfml.model_version.v1.BuildEnv\x12Q\n\rpod_resources\x18\x03 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\x12\x0f\n\x07verbose\x18\x04 \x01(\r\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\x12=\n\x0fpurchase_option\x18\x06 \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\"K\n\x0f\x42uildProperties\x12\x12\n\nbuild_name\x18\x01 \x01(\t\x12\x0c\n\x04tags\x18\x02 \x03(\t\x12\x16\n\x0egpu_compatible\x18\x03 \x01(\x08\"@\n\x08\x42uildEnv\x12\x34\n\ndocker_env\x18\x01 \x01(\x0b\x32 .jfml.model_version.v1.DockerEnv\"1\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08no_cache\x18\x02 \x01(\x08\x42%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.build_spec_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['_BUILDSPEC']._serialized_start=151
+  _globals['_BUILDSPEC']._serialized_end=467
+  _globals['_BUILDPROPERTIES']._serialized_start=469
+  _globals['_BUILDPROPERTIES']._serialized_end=544
+  _globals['_BUILDENV']._serialized_start=546
+  _globals['_BUILDENV']._serialized_end=610
+  _globals['_DOCKERENV']._serialized_start=612
+  _globals['_DOCKERENV']._serialized_end=661
+# @@protoc_insertion_point(module_scope)

frogml-1.2.22/frogml/_proto/jfml/model_version/v1/build_spec_pb2.pyi

@@ -0,0 +1,124 @@
+"""
+@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 frogml._proto.qwak.fitness_service.fitness_pb2
+import frogml._proto.qwak.user_application.common.v0.resources_pb2
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class BuildSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_PROPERTIES_FIELD_NUMBER: builtins.int
+    BUILD_ENV_FIELD_NUMBER: builtins.int
+    POD_RESOURCES_FIELD_NUMBER: builtins.int
+    VERBOSE_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    verbose: builtins.int
+    """The build verbosity level (0: warning, 1: info, 2: debug)"""
+    environment_id: builtins.str
+    """The environment to build on"""
+    purchase_option: frogml._proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType
+    """Purchase option for the build"""
+    @property
+    def build_properties(self) -> global___BuildProperties:
+        """The build properties"""
+
+    @property
+    def build_env(self) -> global___BuildEnv:
+        """The build environment"""
+
+    @property
+    def pod_resources(self) -> frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources: ...
+    def __init__(
+        self,
+        *,
+        build_properties: global___BuildProperties | None = ...,
+        build_env: global___BuildEnv | None = ...,
+        pod_resources: frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources | None = ...,
+        verbose: builtins.int = ...,
+        environment_id: builtins.str = ...,
+        purchase_option: frogml._proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["build_env", b"build_env", "build_properties", b"build_properties", "pod_resources", b"pod_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["build_env", b"build_env", "build_properties", b"build_properties", "environment_id", b"environment_id", "pod_resources", b"pod_resources", "purchase_option", b"purchase_option", "verbose", b"verbose"]) -> None: ...
+
+global___BuildSpec = BuildSpec
+
+@typing.final
+class BuildProperties(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_NAME_FIELD_NUMBER: builtins.int
+    TAGS_FIELD_NUMBER: builtins.int
+    GPU_COMPATIBLE_FIELD_NUMBER: builtins.int
+    build_name: builtins.str
+    """Optional - The name of the build we want"""
+    gpu_compatible: builtins.bool
+    """Optional - Should the build be GPU compatible"""
+    @property
+    def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Optional - The Build's tags"""
+
+    def __init__(
+        self,
+        *,
+        build_name: builtins.str = ...,
+        tags: collections.abc.Iterable[builtins.str] | None = ...,
+        gpu_compatible: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_name", b"build_name", "gpu_compatible", b"gpu_compatible", "tags", b"tags"]) -> None: ...
+
+global___BuildProperties = BuildProperties
+
+@typing.final
+class BuildEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DOCKER_ENV_FIELD_NUMBER: builtins.int
+    @property
+    def docker_env(self) -> global___DockerEnv:
+        """Optional - The docker specs to use for the build"""
+
+    def __init__(
+        self,
+        *,
+        docker_env: global___DockerEnv | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["docker_env", b"docker_env"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["docker_env", b"docker_env"]) -> None: ...
+
+global___BuildEnv = BuildEnv
+
+@typing.final
+class DockerEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BASE_IMAGE_FIELD_NUMBER: builtins.int
+    NO_CACHE_FIELD_NUMBER: builtins.int
+    base_image: builtins.str
+    """Optional - Base image to use for serving"""
+    no_cache: builtins.bool
+    """Optional -
+    Don't use remote cache to build the images, defaults to use cache,
+    if the model dependencies changed significantly it could be faster and better to run with no cache
+    """
+    def __init__(
+        self,
+        *,
+        base_image: builtins.str = ...,
+        no_cache: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["base_image", b"base_image", "no_cache", b"no_cache"]) -> None: ...
+
+global___DockerEnv = DockerEnv

frogml-1.2.22/frogml/_proto/jfml/model_version/v1/model_repository_spec_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.jfml.model_version/v1/model_repository_spec.proto
+# Protobuf Python Version: 4.25.1
+"""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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1jfml/model_version/v1/model_repository_spec.proto\x12\x15jfml.model_version.v1\"\x80\x01\n\x13ModelRepositorySpec\x12\x15\n\x0bproject_key\x18\x01 \x01(\tH\x00\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x12\n\nmodel_name\x18\x04 \x01(\tB\x14\n\x12project_identifierB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.model_repository_spec_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['_MODELREPOSITORYSPEC']._serialized_start=77
+  _globals['_MODELREPOSITORYSPEC']._serialized_end=205
+# @@protoc_insertion_point(module_scope)