Bump examples

Author: michalbiesek

examples/example_cloud_worker_group.py

@@ -28,7 +28,7 @@
 
 
 import asyncio
-from cribl_control_plane.models import ConfigGroup, ConfigGroupCloud, CloudProvider, ProductsCore
+from cribl_control_plane.models import ConfigGroup, ConfigGroupCloud, CloudProvider, ProductsCore, ConfigGroupEstimatedIngestRate, GroupCreateRequestEstimatedIngestRate
 from auth import AuthCloud, CloudConfiguration
 
 
@@ -49,7 +49,7 @@
     provisioned=False,
     is_fleet=False,
     is_search=False,
-    estimated_ingest_rate=2048,  # Equivalent to 24 MB/s maximum estimated ingest rate with 9 Worker Processes
+    estimated_ingest_rate=ConfigGroupEstimatedIngestRate.RATE24_MB_PER_SEC,  # Equivalent to 24 MB/s maximum estimated ingest rate with 9 Worker Processes
     id=WORKER_GROUP_ID,
     name="my-aws-worker-group"
 )
@@ -83,13 +83,13 @@ async def main():
         provisioned=group.provisioned,
         is_fleet=group.is_fleet,
         is_search=group.is_search,
-        estimated_ingest_rate=group.estimated_ingest_rate,
+        estimated_ingest_rate=GroupCreateRequestEstimatedIngestRate(group.estimated_ingest_rate.value) if group.estimated_ingest_rate else None,
         name=group.name
     )
     print(f"✅ Worker Group created: {group.id}")
 
     # Scale and provision the Worker Group
-    group.estimated_ingest_rate = 4096  # Equivalent to 48 MB/s maximum estimated ingest rate with 21 Worker Processes
+    group.estimated_ingest_rate = ConfigGroupEstimatedIngestRate.RATE48_MB_PER_SEC  # Equivalent to 48 MB/s maximum estimated ingest rate with 21 Worker Processes
     group.provisioned = True
     cribl.groups.update(
         product=ProductsCore.STREAM,

examples/example_worker_group_replication.py

@@ -26,7 +26,7 @@
 import sys
 from typing import Optional
 from cribl_control_plane import CriblControlPlane
-from cribl_control_plane.models import ConfigGroup, ProductsCore
+from cribl_control_plane.models import ConfigGroup, ProductsCore, GroupCreateRequestEstimatedIngestRate, GroupCreateRequestType
 from auth import create_cribl_client
 
 
@@ -103,12 +103,12 @@ def replicate_worker_group(client: CriblControlPlane, source_id: str) -> Optiona
             provisioned=source.provisioned,
             is_fleet=source.is_fleet,
             is_search=source.is_search,
-            estimated_ingest_rate=source.estimated_ingest_rate,
+            estimated_ingest_rate=GroupCreateRequestEstimatedIngestRate(source.estimated_ingest_rate.value) if source.estimated_ingest_rate else None,
             inherits=source.inherits,
             max_worker_age=source.max_worker_age,
             streamtags=source.streamtags,
             tags=source.tags,
-            type_=source.type,
+            type_=GroupCreateRequestType(source.type.value) if source.type else None,
             upgrade_version=source.upgrade_version
         )