frogml-core 0.0.110__py3-none-any.whl → 0.0.112__py3-none-any.whl

frogml_core/__init__.py

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

frogml_core/automations/automations.py

@@ -293,6 +293,7 @@ class Automation:
     create_audit: AutomationAudit = field(default_factory=AutomationAudit)
     on_error: Notification = field(default=None)
     on_success: Notification = field(default=None)
+    jfrog_token_id: str = field(default="")
 
     def to_proto(self):
         return AutomationProto(
@@ -315,10 +316,11 @@ class Automation:
             qwak_environment_id=self.environment,
             create_audit=self.create_audit.to_proto(),
             is_deleted=self.is_deleted,
+            jfrog_token_id=self.jfrog_token_id,
         )
 
-    @staticmethod
-    def from_proto(message: AutomationProto):
+    @classmethod
+    def from_proto(cls, message: AutomationProto):
         action = map_action_name_to_class(
             message.automation_spec.action.WhichOneof("action")
         )
@@ -332,7 +334,7 @@ class Automation:
             message.automation_spec.on_success.WhichOneof("notification")
         )
 
-        return Automation(
+        return cls(
             id=message.automation_id,
             name=message.automation_spec.automation_name,
             description=message.automation_spec.automation_description,
@@ -359,6 +361,7 @@ class Automation:
                 if on_success_notification
                 else None
             ),
+            jfrog_token_id=message.jfrog_token_id,
         )
 
     def __str__(self):

frogml_core/feature_store/data_sources/batch/_jdbc.py

@@ -13,6 +13,7 @@ class JdbcSource(BaseBatchSource, ABC):
     url: Optional[str] = None
     db_table: Optional[str] = None
     query: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         self._validate()

frogml_core/feature_store/data_sources/batch/athena.py

@@ -52,6 +52,7 @@ class AthenaSource(JdbcSource):
     aws_authentication: Optional[AwsAuthentication] = None
     time_partition_columns: Optional[TimePartitionColumns] = None
     workgroup: Optional[str] = None
+    repository: Optional[str] = None
 
     def _validate(self):
         super()._validate()
@@ -67,7 +68,7 @@ class AthenaSource(JdbcSource):
                 "At least one of: `s3_output_location`, `workgroup` must be set"
             )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         proto_date_partition_columns = None
         proto_time_fragment_partition_columns = None
 
@@ -97,6 +98,7 @@ class AthenaSource(JdbcSource):
         )
 
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -116,7 +118,7 @@ class AthenaSource(JdbcSource):
                         workgroup=self.workgroup,
                     ),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/big_query.py

@@ -25,6 +25,7 @@ class BigQuerySource(BaseBatchSource):
     views_enabled: Optional[bool] = False
     materialization_dataset: Optional[str] = None
     materialization_project: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         if not (self.sql or (self.table and self.dataset)):
@@ -59,8 +60,9 @@ class BigQuerySource(BaseBatchSource):
             materialization_dataset=bigquery.materialization_dataset,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -76,5 +78,5 @@ class BigQuerySource(BaseBatchSource):
                     materialization_dataset=self.materialization_dataset,
                     materialization_project=self.materialization_project,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/clickhouse.py

@@ -21,6 +21,7 @@ class ClickhouseSource(BaseBatchSource):
     url: Optional[str] = None
     db_table: Optional[str] = None
     query: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         self._validate()
@@ -29,8 +30,9 @@ class ClickhouseSource(BaseBatchSource):
         if not (bool(self.db_table) ^ bool(self.query)):
             raise FrogmlException("Only one of query and db_table must be set")
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -42,7 +44,7 @@ class ClickhouseSource(BaseBatchSource):
                     table=self.db_table,
                     sql=self.query,
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/csv.py

@@ -25,6 +25,7 @@ class CsvSource(BaseBatchSource):
     quote_character: str = '"'
     escape_character: str = '"'
     filesystem_configuration: Optional[FileSystemConfiguration] = None
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -42,12 +43,13 @@ class CsvSource(BaseBatchSource):
             filesystem_configuration=fs_conf,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         fs_conf = None
         if self.filesystem_configuration:
             fs_conf = self.filesystem_configuration._to_proto()
 
         return ProtoDataSourceSpec(
+            featureset_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -58,5 +60,5 @@ class CsvSource(BaseBatchSource):
                     escape_character=self.escape_character,
                     filesystem_configuration=fs_conf,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/elastic_search.py

@@ -23,6 +23,7 @@ class ElasticSearchSource(BaseBatchSource):
     password_secret_name: Optional[str] = None
     exclude_fields: Optional[str] = None
     parse_dates: Optional[bool] = True
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -41,8 +42,9 @@ class ElasticSearchSource(BaseBatchSource):
             parse_dates=elastic.parse_dates,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -57,5 +59,5 @@ class ElasticSearchSource(BaseBatchSource):
                     exclude_fields=self.exclude_fields,
                     parse_dates=self.parse_dates,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/mongodb.py

@@ -22,6 +22,7 @@ class MongoDbSource(BaseBatchSource):
     collection: str
     connection_params: str
     protocol: str = "mongodb"
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -39,8 +40,9 @@ class MongoDbSource(BaseBatchSource):
             protocol=mongo.protocol,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoBatchSource:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -54,5 +56,5 @@ class MongoDbSource(BaseBatchSource):
                     connection_params=self.connection_params,
                     protocol=self.protocol,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/mysql.py

@@ -32,8 +32,9 @@ class MysqlSource(JdbcSource):
             query=mysql.query,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoJdbcSource:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -46,5 +47,5 @@ class MysqlSource(JdbcSource):
                     query=self.query,
                     mysqlSource=ProtoMysqlSource(),
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/parquet.py

@@ -26,6 +26,7 @@ class ParquetSource(BaseBatchSource):
     filesystem_configuration: FileSystemConfiguration = field(
         default_factory=lambda: AnonymousS3Configuration()
     )
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -40,12 +41,13 @@ class ParquetSource(BaseBatchSource):
             filesystem_configuration=fs_conf,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         fs_conf = None
         if self.filesystem_configuration:
             fs_conf = self.filesystem_configuration._to_proto()
 
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -53,5 +55,5 @@ class ParquetSource(BaseBatchSource):
                 parquetSource=ProtoParquetSource(
                     path=self.path, filesystem_configuration=fs_conf
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/postgres.py

@@ -18,8 +18,9 @@ from frogml_core.feature_store.data_sources.batch._jdbc import JdbcSource
 
 @dataclass
 class PostgresSource(JdbcSource):
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -32,7 +33,7 @@ class PostgresSource(JdbcSource):
                     query=self.query,
                     postgresqlSource=ProtoPostgresqlSource(),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/redshift.py

@@ -27,6 +27,8 @@ class RedshiftSource(JdbcSource):
     secret_access_key: Optional[str] = None
     query_group: Optional[str] = "_qwak_featurestore"
 
+    repository: Optional[str] = None
+
     def _validate(self):
         authentication_methods = sum(
             [
@@ -46,8 +48,9 @@ class RedshiftSource(JdbcSource):
                 "or user id secret name and password secret name"
             )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -65,7 +68,7 @@ class RedshiftSource(JdbcSource):
                         secret_access_key=self.secret_access_key,
                     ),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/snowflake.py

@@ -24,6 +24,7 @@ class SnowflakeSource(BaseBatchSource):
     warehouse: str
     table: Optional[str] = None
     query: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         self._validate()
@@ -40,8 +41,9 @@ class SnowflakeSource(BaseBatchSource):
         if not self.password_secret_name:
             raise FrogmlException("password_secret_name must be set!")
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -56,7 +58,7 @@ class SnowflakeSource(BaseBatchSource):
                     table=self.table,
                     query=self.query,
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/vertica.py

@@ -22,6 +22,7 @@ class VerticaSource(BaseBatchSource):
     table: str
     username_secret_name: str
     password_secret_name: str
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -39,8 +40,9 @@ class VerticaSource(BaseBatchSource):
             table=vertica.table,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -54,5 +56,5 @@ class VerticaSource(BaseBatchSource):
                     port=self.port,
                     table=self.table,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/streaming/kafka/kafka.py

@@ -52,6 +52,8 @@ class KafkaSource(BaseStreamingSource):
         default_factory=lambda: SslAuthentication()
     )
 
+    repository: Optional[str] = None
+
     def __post_init__(self):
         self._validate()
 
@@ -84,6 +86,7 @@ class KafkaSource(BaseStreamingSource):
 
     def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             stream_source=ProtoStreamingSource(
                 name=self.name,
                 description=self.description,
@@ -99,7 +102,7 @@ class KafkaSource(BaseStreamingSource):
                         artifact_path=artifact_url
                     ),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/feature_sets/base_feature_set.py

@@ -33,6 +33,7 @@ class BaseFeatureSet(ABC):
     data_sources: List[str]
     entity: str = str()
     key: str = str()
+    repository: Optional[str] = None
     __instance_module_path__: Optional[str] = None
 
     def _validate(self):

frogml_core/feature_store/feature_sets/batch.py

@@ -72,6 +72,7 @@ def feature_set(
     name: str = None,
     entity: Optional[str] = None,
     key: Optional[str] = None,
+    repository: Optional[str] = None,
 ):
     """
     Define a batch scheduled feature set. Default scheduling policy is every 4 hours.
@@ -116,6 +117,7 @@ def feature_set(
             name=fs_name,
             entity=entity if entity else None,
             key=key if key else None,
+            repository=repository if repository else None,
             data_sources=data_sources,
             timestamp_column_name=timestamp_column_name,
             transformation=user_transformation,
@@ -323,6 +325,7 @@ class BatchFeatureSet(BaseFeatureSet):
             data_sources=[
                 ds.data_source.name for ds in batch_v1_def.feature_set_batch_sources
             ],
+            repository=proto.featureset_repository_name,
             timestamp_column_name=batch_v1_def.timestamp_column_name,
             scheduling_policy=batch_v1_def.scheduling_policy,
             transformation=BaseTransformation._from_proto(batch_v1_def.transformation),
@@ -404,6 +407,7 @@ class BatchFeatureSet(BaseFeatureSet):
                 git_commit=git_commit,
                 features=features,
                 entity=self._get_entity_definition(feature_registry),
+                featureset_repository_name=self.repository,
                 feature_set_type=ProtoFeatureSetType(
                     batch_feature_set_v1=ProtoBatchFeatureSetV1(
                         online_sink=True,

frogml_core/feature_store/feature_sets/streaming.py

@@ -86,6 +86,7 @@ def feature_set(
     entity: Optional[str] = None,
     key: Optional[str] = None,
     auxiliary_sinks: List[BaseSink] = [],
+    repository: Optional[str] = None,
 ):
     """
     Creates a streaming feature set for the specified entity using the given streaming data sources.
@@ -140,6 +141,7 @@ def feature_set(
             name=fs_name,
             entity=entity if entity else None,
             key=key if key else None,
+            repository=repository,
             data_sources=data_sources,
             timestamp_column_name=timestamp_column_name,
             transformation=user_transformation,
@@ -340,6 +342,7 @@ class StreamingFeatureSet(BaseFeatureSet):
 
         return cls(
             name=proto.name,
+            repository=proto.featureset_repository_name,
             entity=Entity._from_proto(proto.entity).name,
             data_sources=[ds.name for ds in streaming_def.data_sources],
             timestamp_column_name=streaming_def.timestamp_column_name,
@@ -420,6 +423,7 @@ class StreamingFeatureSet(BaseFeatureSet):
                 features=features,
                 entity=self._get_entity_definition(feature_registry),
                 feature_set_type=proto_featureset_type,
+                featureset_repository_name=self.repository,
             ),
             artifact_url,
         )
@@ -429,10 +433,12 @@ class StreamingFeatureSet(BaseFeatureSet):
 
         super()._validate()
 
+        # verify offline_scheduling_policy was set
+        if not self.offline_scheduling_policy:
+            raise FrogmlException("'offline_scheduling_policy' field must be set")
+
         # verify the cron expression is valid
-        if not self.offline_scheduling_policy or not croniter.croniter.is_valid(
-            self.offline_scheduling_policy
-        ):
+        if not croniter.croniter.is_valid(self.offline_scheduling_policy):
             raise FrogmlException(
                 f"offline scheduling policy "
                 f"'{self.offline_scheduling_policy}'"

frogml_core/feature_store/online/client.py

@@ -1,5 +1,6 @@
 import logging
 import math
+from io import StringIO
 from typing import TYPE_CHECKING, List, Optional, Sequence, Tuple
 
 from frogml_proto.qwak.ecosystem.v0.ecosystem_pb2 import (
@@ -221,7 +222,7 @@ class OnlineClient:
                     request, metadata=self._metadata
                 )
                 features_df: pd.DataFrame = pd.read_json(
-                    response_df_json.pandas_df_as_json, orient="split"
+                    StringIO(response_df_json.pandas_df_as_json), orient="split"
                 )
                 results.append(
                     pd.concat(

frogml_core/model_loggers/artifact_logger.py

@@ -1,5 +1,8 @@
-import urllib.request
+import logging
+from datetime import timedelta
 from typing import Optional
+import requests
+from requests import Response
 
 from frogml_proto.qwak.builds.build_url_pb2 import BuildVersioningTagsType
 from frogml_core.clients.build_orchestrator.client import BuildOrchestratorClient
@@ -13,8 +16,14 @@ from frogml_core.inner.model_loggers_utils import (
 )
 from frogml_proto.qwak.builds.builds_orchestrator_service_pb2 import (
     GetBuildVersioningUploadURLResponse,
+    GetBuildVersioningDownloadURLResponse,
 )
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+MAX_CHUNK_SIZE = 8 * 1_024
+
 
 def log_file(
     from_path: str,
@@ -80,21 +89,40 @@ def load_file(
     Returns:
         the path to the newly created data file
     """
+    logger.info(f"Loading file to {to_path}")
     if not validate_tag(tag):
         raise FrogmlException(
             "Tag should contain only letters, numbers, underscore or hyphen"
         )
 
     model_id = validate_model(model_id)
-    download_url_response = BuildOrchestratorClient().get_build_versioning_download_url(
-        build_id=build_id, model_id=model_id, tag=tag
+    download_url_response: (
+        GetBuildVersioningDownloadURLResponse
+    ) = BuildOrchestratorClient().get_build_versioning_download_url(
+        build_id=build_id,
+        model_id=model_id,
+        tag=tag,
+        tag_type=BuildVersioningTagsType.FILE_TAG_TYPE,
     )
 
     try:
-        filename, headers = urllib.request.urlretrieve(  # nosec B310
-            url=download_url_response.download_url, filename=to_path
+        response: Response = requests.get(
+            download_url_response.download_url,
+            headers=download_url_response.headers,
+            stream=True,
+            timeout=(
+                timedelta(seconds=10).total_seconds(),  # timeout to connect
+                timedelta(minutes=20).total_seconds(),  # timeout to read
+            ),
         )
+        logger.info(f"Downloading file finished with status {response.status_code}")
+        response.raise_for_status()
+
+        with open(to_path, "wb") as f:
+            for chunk in response.iter_content(chunk_size=MAX_CHUNK_SIZE):
+                if chunk:
+                    f.write(chunk)
 
-        return filename
+        return to_path
     except Exception as error:
         raise FrogmlException(f"Unable to load save artifact locally: {str(error)}")

{frogml_core-0.0.110.dist-info → frogml_core-0.0.112.dist-info}/METADATA

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

{frogml_core-0.0.110.dist-info → frogml_core-0.0.112.dist-info}/RECORD

@@ -1,7 +1,7 @@
-frogml_core/__init__.py,sha256=Q10let7zj7bZvhVQ91wAHfK90sOgGYDqArvf7wpigpk,778
+frogml_core/__init__.py,sha256=SNMYZzIxL9iYw_75tt9Zjp8e606xQbK4AT4rRIpY-BU,778
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
-frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
+frogml_core/automations/automations.py,sha256=srG8S5Z20hUsk6uehz75dmCs8mEgNCCwuCXfj3CG8hY,13210
 frogml_core/automations/batch_execution_action.py,sha256=tFGQWI16CGNfo-DZJ0n86SBkKWbNt1Q8712wF0_OgUI,13327
 frogml_core/automations/build_and_deploy_action.py,sha256=WK9CJ0oYr6u6Sy794OMjTvLohA_aHaofx0ectWN-K3o,31962
 frogml_core/automations/common.py,sha256=98YNX-jCrmQ97k1IxRqzdsnet-rohcXp4yIdVNwpyh8,2752
@@ -124,31 +124,31 @@ frogml_core/feature_store/data_sources/attributes.py,sha256=SsiPoJkqSONhK8mnn6ft
 frogml_core/feature_store/data_sources/base.py,sha256=sCJ1CzbhRX-fgsw_Y3ucUu-tKC3i93upasJfGw3EmtE,4287
 frogml_core/feature_store/data_sources/batch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/data_sources/batch/_batch.py,sha256=xnrC1SZ4OH4b7gYLRajEBFQ2U00cA3JxKT8bKNTNHLg,204
-frogml_core/feature_store/data_sources/batch/_jdbc.py,sha256=lC-m3ZypER5vLtYo4h6ghYWeXvdfGSOdTadqS-sTb58,676
-frogml_core/feature_store/data_sources/batch/athena.py,sha256=fKKdKE1B59dbCiOQQ5amJECoyETwxswkBhDLQ-Pctsg,11292
-frogml_core/feature_store/data_sources/batch/big_query.py,sha256=h9iscw9I-WyCj1BeyesASNoIiNwPty9acQ37KGD0Ulo,3112
-frogml_core/feature_store/data_sources/batch/clickhouse.py,sha256=mUSE139-ACnInctQNZqETbnWgFyZidvffVJpwQu0T3Q,2149
-frogml_core/feature_store/data_sources/batch/csv.py,sha256=saxFeP--CL4h5j2toyxSuOl4kN_WtEEg8pumaYOjCHg,2070
-frogml_core/feature_store/data_sources/batch/elastic_search.py,sha256=GJKEsOlaxHbBYSor47bofNtV9kJnVVYtq8BSies7hjM,2205
+frogml_core/feature_store/data_sources/batch/_jdbc.py,sha256=u-o7BF2W66ZVLbOhYsL91f1LWThNy8hGog1I-bwyHsU,713
+frogml_core/feature_store/data_sources/batch/athena.py,sha256=VNy6gbIV87vn6TIWDwTBkfcsX0awLCJ5dFZMv448-F0,11410
+frogml_core/feature_store/data_sources/batch/big_query.py,sha256=V5zKuuESbM_0bR_z8FqT-EMWu10GxR3JMdXCZAER8mc,3230
+frogml_core/feature_store/data_sources/batch/clickhouse.py,sha256=HNhTl21BDJ--Bs_5texT_lmTQufBpowJP4ONi-H6Xbw,2267
+frogml_core/feature_store/data_sources/batch/csv.py,sha256=FMeCEVaBWDZKSEmsgF3nk3TRnWp82QzE9DXCYUAEFJs,2187
+frogml_core/feature_store/data_sources/batch/elastic_search.py,sha256=5P9hGg63iV2DWvNWTq-wCXGGJbYzXRs3zU9fPDSfCaY,2323
 frogml_core/feature_store/data_sources/batch/filesystem/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/data_sources/batch/filesystem/aws.py,sha256=4mj34C16mH1JeU2dEKFIiV-ubCGHwILlgFNyDuEHHQY,3051
 frogml_core/feature_store/data_sources/batch/filesystem/base_config.py,sha256=fqvqkM_0QY4rcZNa1qB-KLl72o8PhrQQnak8-Bk4tHM,245
 frogml_core/feature_store/data_sources/batch/filesystem/gcp.py,sha256=_UU1a0u1DeTh7okstWiM1e2nD8TJLrEigPIFLQnSkuE,1723
 frogml_core/feature_store/data_sources/batch/filesystem/utils.py,sha256=u2FyAh91JY8j1EzX5fJSgVJqwiwpMvt43mfMeyabVdE,2171
-frogml_core/feature_store/data_sources/batch/mongodb.py,sha256=98Y67LdRgIzfcPNw3CEPV08BOa-HMgrox3Pw-rW883A,1996
-frogml_core/feature_store/data_sources/batch/mysql.py,sha256=MOYgsosXIUf3nap83X4lAn-afFE_tydK5_GU0SW8La8,1736
-frogml_core/feature_store/data_sources/batch/parquet.py,sha256=uOAmTSWn9plN96UuhSqFs_7R9nkrb6DPnJfc7qzcFfY,1963
-frogml_core/feature_store/data_sources/batch/postgres.py,sha256=bpvYg7bgZKG4Mhdh7pp5aN23i9d1GffX5yBSxkQ4Eho,1789
-frogml_core/feature_store/data_sources/batch/redshift.py,sha256=f0xH1Nzg3MMjj-WUFR7DzDLNgawv1KmJQzGSw_-eetE,3268
-frogml_core/feature_store/data_sources/batch/snowflake.py,sha256=T5b9_mRrq_7en-70KgELmgkiKnS3CI9IYw1sBcmSJlw,2671
-frogml_core/feature_store/data_sources/batch/vertica.py,sha256=J9-1KvzWL5JlCqxRYo7MGfF3yXt4wEQ2GzB5RvuS8OM,1905
+frogml_core/feature_store/data_sources/batch/mongodb.py,sha256=sVR38ZBZ0Y8cLng5RVm5ahag0rfABRZ_mngMzJk9YiQ,2111
+frogml_core/feature_store/data_sources/batch/mysql.py,sha256=Hr6urfu6tAa54L-oxTCYugwmuXi2gOtOIIWh-3vBBHU,1813
+frogml_core/feature_store/data_sources/batch/parquet.py,sha256=_KfsDbwJheCtqg1BebmAj6MJTTKlGo9Kd_xpG2upeCk,2081
+frogml_core/feature_store/data_sources/batch/postgres.py,sha256=PoDl3B_gg1wQvLB0NI1vS5hen7rOfd7ShwduWKE93Lk,1870
+frogml_core/feature_store/data_sources/batch/redshift.py,sha256=c5M0L7gmMnmxAvd39D4Eawyx9ZrHfx5YI4p5eMGTbSU,3387
+frogml_core/feature_store/data_sources/batch/snowflake.py,sha256=HWiJoTSe6osG_fITggMjgjttlIOOZxTSRlQ-NGGlbOg,2789
+frogml_core/feature_store/data_sources/batch/vertica.py,sha256=KZo9oIn8vVAYhoE1OWvU_tvcS7vkOr8I2Yj9iDXVd3k,2023
 frogml_core/feature_store/data_sources/source_authentication.py,sha256=AhMxuCYqjmYeaPc5OF_K2YnHq44GcN46AtT3_4nAAIA,962
 frogml_core/feature_store/data_sources/streaming/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/data_sources/streaming/_streaming.py,sha256=LM7vcBeuwlpj9JZjAppuy04mtRoaoQtGixv1oaFcJxI,188
 frogml_core/feature_store/data_sources/streaming/kafka/__init__.py,sha256=ISoTTLW8e8hNWJHVekWT0yWhjHEwaNvc3Pyzkd23UAE,670
 frogml_core/feature_store/data_sources/streaming/kafka/authentication.py,sha256=z1VuiHRvALbSFcC2Oa4of2Emy1FHtd6df_-QE6xCR7Q,4305
 frogml_core/feature_store/data_sources/streaming/kafka/deserialization.py,sha256=uKdYTcVOkq8r9rFlimtVqbfCAROQv66Zs1ZQh1yx9jI,3736
-frogml_core/feature_store/data_sources/streaming/kafka/kafka.py,sha256=CcfzXgY6lcxTIEJ7bb3Z8rQM693LtBhd_FSHb-l0xvA,4527
+frogml_core/feature_store/data_sources/streaming/kafka/kafka.py,sha256=UJos6wt_F-DbkvmXAoxZSP8kMlXvF03e9YbyLBR7IkI,4623
 frogml_core/feature_store/data_sources/time_partition_columns.py,sha256=Glt5aMLKY6fCRiNupgxyliKOev_y9fg6c3DYRNDb5hA,6381
 frogml_core/feature_store/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/entities/entity.py,sha256=ktMQUgd4Wmfcko9g9MB0KXk7UkLXlJFZpBFD5NRZE_Q,2325
@@ -159,13 +159,13 @@ frogml_core/feature_store/execution/execution_query.py,sha256=xWTVcgfXAxxTHuQyrH
 frogml_core/feature_store/feature_sets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/feature_sets/_utils/_featureset_utils.py,sha256=mnznpQ0RAk5ONRdbp9m4Y95Z2zWUj0bl4ztamtczO1U,1658
 frogml_core/feature_store/feature_sets/backfill.py,sha256=oZqGb8PA1BQxVOxf9-MwzeZyjJqgwulbA-flmBmXAj4,1981
-frogml_core/feature_store/feature_sets/base_feature_set.py,sha256=1niCItXApA6JZB7tviXJJVaw91sFwSBwiA1f2xOQnwM,5378
-frogml_core/feature_store/feature_sets/batch.py,sha256=VADm8KVYhyxV7Ibt2luOlnrbrqd774_ICG0iKPs82cU,17422
+frogml_core/feature_store/feature_sets/base_feature_set.py,sha256=zYJsn0iDBPSS4BmkDyWBKmW-LWJPTWePq9McEu7pNxQ,5415
+frogml_core/feature_store/feature_sets/batch.py,sha256=xnVvONl2sDnOGOiJMYk_uNWACm8oKHl6-CPCZckn-Wc,17636
 frogml_core/feature_store/feature_sets/context.py,sha256=zV6r0O70cfM4pmxlfC6xxAtro-wBhenXWwYwF3KwfTY,263
 frogml_core/feature_store/feature_sets/execution_spec.py,sha256=zKQd7U-PdYkZMqBpA9eIRhhWff-8xxKB_Qo4IDolwGI,2348
 frogml_core/feature_store/feature_sets/metadata.py,sha256=Vv2pyBbwaJZRFhWKRhxdFyN3AsV-DvTQzLs9nyRMWK0,1888
 frogml_core/feature_store/feature_sets/read_policies.py,sha256=BQu6B6IZuKJt8Ff5RYeADdqpHmSkec790RIYeSl6Ulo,6844
-frogml_core/feature_store/feature_sets/streaming.py,sha256=78ItVSojMfLoTY3k0SH6UD40XS7dnVZFn9cKfYUcS3Y,25295
+frogml_core/feature_store/feature_sets/streaming.py,sha256=CDdhFALqWbrU8qAl1iFf0OPF2vWTp53t_hhaWkWMah0,25607
 frogml_core/feature_store/feature_sets/streaming_backfill.py,sha256=u-tjq86AaXAusLUwPYtdCKhTuySQFAtRrUvbPrY3CtI,9834
 frogml_core/feature_store/feature_sets/transformations/__init__.py,sha256=ozc50AI9RBY71nhNiJDu1-vSWJL2Bdgstyh7GGUW2ig,902
 frogml_core/feature_store/feature_sets/transformations/aggregations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -181,7 +181,7 @@ frogml_core/feature_store/offline/_offline_serving_validations.py,sha256=yfkV8UV
 frogml_core/feature_store/offline/client_v2.py,sha256=kTFyHAYIsKBe3wcuE1S_LyD9CLQ_yCorOLcHVR3Emms,14966
 frogml_core/feature_store/offline/feature_set_features.py,sha256=MjrQrXNhzk7QBdCojdpLfy1fuGdP3GcpOgcc7n7H0G8,740
 frogml_core/feature_store/online/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml_core/feature_store/online/client.py,sha256=_NCVJngFgh8D3IVBW8u8estPqZ7mSwg5IxmuTndFY0A,13572
+frogml_core/feature_store/online/client.py,sha256=U1lpnNE_UGYC2Tvi2y6CayQShYNP0YR8roNI1gxmAWA,13606
 frogml_core/feature_store/online/endpoint_utils.py,sha256=lGssZR-r8kJpcSozVxQAk1_JpVXgRLqOVrK6fw8flPg,2242
 frogml_core/feature_store/sinks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/sinks/base.py,sha256=QdIutDlO_8IBRr-zKfHBRHJ1-DjDmFfR_Yuad193kg0,361
@@ -405,7 +405,7 @@ frogml_core/model/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 frogml_core/model/utils/extract_wrapped_function.py,sha256=uIle1zL8vbmeS3PGAuaNFLIUQAsvpuzk3LlH-Teba94,320
 frogml_core/model/utils/feature_utils.py,sha256=ObIU4jLKOMEa3DWEyrXm1m3wC2wseAVwr-zxlSiojYk,2525
 frogml_core/model_loggers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml_core/model_loggers/artifact_logger.py,sha256=7T06QL1keo6BTpIh76JtXRcNlZuIbqECz343kEw0Hfc,3128
+frogml_core/model_loggers/artifact_logger.py,sha256=F3Xhw9O82b0_cx24DHoc13SaJ06rqO5-c4Tw4yIMR74,4033
 frogml_core/model_loggers/data_logger.py,sha256=I3o2TkAGAnKuVWccPMlGL5-rg1e2UDHRc9cRSS8kqmc,5678
 frogml_core/model_loggers/model_logger.py,sha256=xDUIwRzeFAorzzR47ovyag5ohwuYZWp9fRWpLpMzcg4,863
 frogml_core/testing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -957,7 +957,7 @@ frogml_proto/qwak/models/models_pb2.py,sha256=CTvYhYuPcZcPgxm2W_ZjcU1w3NSKwO195e
 frogml_proto/qwak/models/models_pb2.pyi,sha256=277mGQradMyhof3J54ilaW_SKN0d-dan7DAy_6Zkmo8,39261
 frogml_proto/qwak/models/models_pb2_grpc.py,sha256=dE-D9iLNznyqNcn3nWMrNd1kAYd8hqXbezOqDP41as4,21681
 frogml_proto/qwak/models/models_query_pb2.py,sha256=1QDjgF0hd_sN3O501f7qfudVz2Qz7EzA80IPcuzOcnM,2822
-frogml_proto/qwak/models/models_query_pb2.pyi,sha256=cZO7EA_JetDIQbC-9hg42G07DJMBC6Adgn6IMhnhhNQ,7456
+frogml_proto/qwak/models/models_query_pb2.pyi,sha256=MYheg8TQtQuDv4BqHOWinzemOKhj35ZIVMHQy3YrAQA,7631
 frogml_proto/qwak/models/models_query_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.py,sha256=ZbLDlHJ-2GboL9D_kojialc7YaX8z7LaZNtilpJWRiI,5061
 frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.pyi,sha256=enqAQBr8RW_FLVL1BkZ_MGjeklQkhIZZ2OU2jczePdk,6907
@@ -1114,7 +1114,7 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=sgKgwhu2W0YOHtzil8x7f1znK_sZr_i27XSeiF4xqVE,21200
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_storage/__init__.py,sha256=00B0bSjQW8FClEUTZrcxxlOoilJS1Cz7oACh4nBKyM4,24
+frogml_storage/__init__.py,sha256=oUJHw8__aJg8lCcQGVNPbGho_Gk1e9FvvH2adAponBo,24
 frogml_storage/_environment.py,sha256=zuzOJBtBwFaguwn_JkKjfhXStZoustgP30KzOP3mYv8,707
 frogml_storage/artifactory/__init__.py,sha256=C02rcm7kqsZBVA6c6Gztxamj96hn8Aj6BuzYWFRmWbQ,71
 frogml_storage/artifactory/_artifactory_api.py,sha256=Oz0HOpQPSNwWIVAy94UJUyPhLetc7sdZjoTfSXtrFug,11200
@@ -1153,6 +1153,6 @@ frogml_storage/utils/__init__.py,sha256=HQUWfuGUIPZY7kfS795TRW8BQ4WmNqrNjS7lUrbx
 frogml_storage/utils/_input_checks_utility.py,sha256=CFiJOdTBS9piJMtR3lemEz27wZcQ6_-7XESu8iy-mrw,3221
 frogml_storage/utils/_storage_utils.py,sha256=HB2g7uY5A3b33yIcAUM1OjHb5jWsnpESsiDrEviQwrI,366
 frogml_storage/utils/_url_utils.py,sha256=NUEfz9Fp1iE8b676-A5wrMlSTsJVRKrUhcUItOFAJD8,821
-frogml_core-0.0.110.dist-info/METADATA,sha256=kfGIqB-6PU6_my0GsK_S8TBn5K63qCE6xnfknO4DN6Q,14898
-frogml_core-0.0.110.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml_core-0.0.110.dist-info/RECORD,,
+frogml_core-0.0.112.dist-info/METADATA,sha256=q_FfTqj6cc_ch7srz-2ga5qcX-aB2qFJexyE8qtAKts,14898
+frogml_core-0.0.112.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml_core-0.0.112.dist-info/RECORD,,

frogml_proto/qwak/models/models_query_pb2.pyi

@@ -112,9 +112,9 @@ class Filter(google.protobuf.message.Message):
     model_name_substring: builtins.str
     @property
     def model_group_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
-        """Filter models by one or more model group names.
-        Return models that belong to one of the specified model groups.
+        """Filter models by one or more model group names. Returns models that belong to any of the specified model groups.
         Optional field. If not specified, models from all groups are included.
+        The maximum number of model group names allowed is 100, with each model group name having a maximum length of 100 characters.
         """
     deployment_model_status: global___DeploymentModelStatusFilter.ValueType
     def __init__(
@@ -141,10 +141,10 @@ class PaginationMetadata(google.protobuf.message.Message):
     PAGE_SIZE_FIELD_NUMBER: builtins.int
     NEXT_CURSOR_FIELD_NUMBER: builtins.int
     page_size: builtins.int
-    """Number of items per page."""
+    """Number of models returned in the current page."""
     next_cursor: builtins.str
     """Cursor for the next page.
-    If there are no more pages, this field will be empty.
+    If there are no more pages, an empty string will be returned as the next cursor.
     """
     def __init__(
         self,

frogml_storage/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.110"
+__version__ = "0.0.112"