temp

singlestore-labs · Nov 21, 2024 · 0c1e0f2 · 0c1e0f2
1 parent 074cb03
commit 0c1e0f2
Show file tree

Hide file tree

Showing 5 changed files with 494 additions and 143 deletions.
diff --git a/singlestoredb/fusion/handlers/export.py b/singlestoredb/fusion/handlers/export.py
@@ -11,7 +11,7 @@
 from ...management.export import Link
 from ..handler import SQLHandler
 from ..result import FusionSQLResult
-from .utils import get_workspace_group
+from .utils import get_deployment_group
 
 
 class CreateClusterIdentity(SQLHandler):
@@ -80,7 +80,7 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         storage_config = json.loads(params['storage'].get('link_config', '{}') or '{}')
         storage_creds = json.loads(params['storage'].get('link_creds', '{}') or '{}')
 
-        wsg = get_workspace_group({})
+        wsg = get_deployment_group({})
 
         if wsg._manager is None:
             raise TypeError('no workspace manager is associated with workspace group')
@@ -179,7 +179,7 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         storage_config = json.loads(params['storage'].get('link_config', '{}') or '{}')
         storage_creds = json.loads(params['storage'].get('link_creds', '{}') or '{}')
 
-        wsg = get_workspace_group({})
+        wsg = get_deployment_group({})
 
         if from_database is None:
             raise ValueError('database name must be specified for source table')
@@ -216,7 +216,7 @@ class ShowExport(SQLHandler):
     """
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
-        wsg = get_workspace_group({})
+        wsg = get_deployment_group({})
         out = ExportStatus(params['export_id'], wsg)
 
         status = out._info()