Skip to content

Commit 32aea6a

Browse files
kesmit13claude
andcommitted
feat: complete management API integration
- Add missing imports for manage_storage_dr and manage_metrics in __init__.py - Add missing Workspace fields: auto_scale, kai_enabled, scale_factor - Ensure complete OpenAPI specification alignment for all management modules 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent ec7c4c4 commit 32aea6a

File tree

2 files changed

+20
-0
lines changed

2 files changed

+20
-0
lines changed

singlestoredb/management/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,10 @@
33
from .cluster import manage_cluster
44
from .files import manage_files
55
from .manager import get_token
6+
from .metrics import manage_metrics
67
from .private_connections import manage_private_connections
78
from .region import manage_regions
9+
from .storage_dr import manage_storage_dr
810
from .teams import manage_teams
911
from .users import manage_users
1012
from .workspace import get_organization

singlestoredb/management/workspace.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -685,6 +685,9 @@ class Workspace(object):
685685
resume_attachments: Optional[List[Dict[str, Any]]]
686686
scaling_progress: Optional[int]
687687
last_resumed_at: Optional[datetime.datetime]
688+
auto_scale: Optional[Dict[str, Any]]
689+
kai_enabled: Optional[bool]
690+
scale_factor: Optional[int]
688691

689692
def __init__(
690693
self,
@@ -702,6 +705,9 @@ def __init__(
702705
resume_attachments: Optional[List[Dict[str, Any]]] = None,
703706
scaling_progress: Optional[int] = None,
704707
last_resumed_at: Optional[Union[str, datetime.datetime]] = None,
708+
auto_scale: Optional[Dict[str, Any]] = None,
709+
kai_enabled: Optional[bool] = None,
710+
scale_factor: Optional[int] = None,
705711
):
706712
#: Name of the workspace
707713
self.name = name
@@ -753,6 +759,15 @@ def __init__(
753759
#: Timestamp when workspace was last resumed
754760
self.last_resumed_at = to_datetime(last_resumed_at)
755761

762+
#: Autoscaling configuration
763+
self.auto_scale = camel_to_snake_dict(auto_scale)
764+
765+
#: Whether SingleStore Kai is enabled
766+
self.kai_enabled = kai_enabled
767+
768+
#: Current scale factor
769+
self.scale_factor = scale_factor
770+
756771
self._manager: Optional[WorkspaceManager] = None
757772

758773
def __str__(self) -> str:
@@ -795,6 +810,9 @@ def from_dict(cls, obj: Dict[str, Any], manager: 'WorkspaceManager') -> 'Workspa
795810
last_resumed_at=obj.get('lastResumedAt'),
796811
resume_attachments=obj.get('resumeAttachments'),
797812
scaling_progress=obj.get('scalingProgress'),
813+
auto_scale=obj.get('autoScale'),
814+
kai_enabled=obj.get('kaiEnabled'),
815+
scale_factor=obj.get('scaleFactor'),
798816
)
799817
out._manager = manager
800818
return out

0 commit comments

Comments
 (0)