duckrun 0.2.9.dev2__py3-none-any.whl → 0.2.9.dev4__py3-none-any.whl

duckrun/__init__.py

@@ -2,7 +2,7 @@
 
 from duckrun.core import Duckrun
 
-__version__ = "0.1.0"
+__version__ = "0.2.9.dev4"
 
 # Expose unified connect method at module level
 connect = Duckrun.connect

duckrun/core.py

@@ -727,8 +727,8 @@ class Duckrun:
         
         # Call the deployment function (DirectLake only)
         return deploy_semantic_model(
-            workspace_name=self.workspace,
-            lakehouse_name=self.lakehouse_name,
+            workspace_name_or_id=self.workspace,
+            lakehouse_name_or_id=self.lakehouse_name,
             schema_name=self.schema,
             dataset_name=dataset_name,
             bim_url=bim_url,

duckrun/semantic_model.py

@@ -129,9 +129,12 @@ def check_dataset_exists(dataset_name, workspace_id, client):
         return False
 
 
-def refresh_dataset(dataset_name, workspace_id, client):
+def refresh_dataset(dataset_name, workspace_id, client, dataset_id=None):
     """Refresh a dataset and monitor progress"""
-    dataset_id = get_dataset_id(dataset_name, workspace_id, client)
+    
+    # If dataset_id not provided, look it up by name
+    if not dataset_id:
+        dataset_id = get_dataset_id(dataset_name, workspace_id, client)
     
     payload = {
         "type": "full",
@@ -246,7 +249,7 @@ def update_bim_for_directlake(bim_content, workspace_id, lakehouse_id, schema_na
 
 
 def create_dataset_from_bim(dataset_name, bim_content, workspace_id, client):
-    """Create semantic model from BIM using Fabric REST API"""
+    """Create semantic model from BIM using Fabric REST API and return the dataset ID"""
     # Convert to base64
     bim_json = json.dumps(bim_content, indent=2)
     bim_base64 = base64.b64encode(bim_json.encode('utf-8')).decode('utf-8')
@@ -280,7 +283,7 @@ def create_dataset_from_bim(dataset_name, bim_content, workspace_id, client):
     
     print(f"✓ Semantic model created")
     
-    # Handle long-running operation
+    # Handle long-running operation and return the dataset ID
     if response.status_code == 202:
         operation_id = response.headers.get('x-ms-operation-id')
         print(f"   Waiting for operation to complete...")
@@ -288,17 +291,38 @@ def create_dataset_from_bim(dataset_name, bim_content, workspace_id, client):
         max_attempts = 30
         for attempt in range(max_attempts):
             time.sleep(2)
+            
+            # Get operation result (not just status)
+            result_response = client.get(f"/v1/operations/{operation_id}/result")
+            
+            # Check if operation is complete by getting the status
             status_response = client.get(f"/v1/operations/{operation_id}")
             status = status_response.json().get('status')
             
             if status == 'Succeeded':
                 print(f"✓ Operation completed")
-                break
+                # Return the created dataset ID from the result
+                result_data = result_response.json()
+                dataset_id = result_data.get('id')
+                if dataset_id:
+                    return dataset_id
+                else:
+                    # Fallback: search for the dataset by name
+                    return get_dataset_id(dataset_name, workspace_id, client)
             elif status == 'Failed':
                 error = status_response.json().get('error', {})
                 raise Exception(f"Operation failed: {error.get('message')}")
             elif attempt == max_attempts - 1:
                 raise Exception(f"Operation timed out")
+    
+    # For non-async responses (status 200/201)
+    result_data = response.json()
+    dataset_id = result_data.get('id')
+    if dataset_id:
+        return dataset_id
+    else:
+        # Fallback: search for the dataset by name
+        return get_dataset_id(dataset_name, workspace_id, client)
 
 
 def deploy_semantic_model(workspace_name_or_id, lakehouse_name_or_id, schema_name, dataset_name, 
@@ -365,17 +389,18 @@ def deploy_semantic_model(workspace_name_or_id, lakehouse_name_or_id, schema_nam
         modified_bim['name'] = dataset_name
         modified_bim['id'] = dataset_name
         
-        # Step 5: Deploy
+        # Step 5: Deploy and get the dataset ID
         print("\n[Step 5/6] Deploying semantic model...")
-        create_dataset_from_bim(dataset_name, modified_bim, workspace_id, client)
+        dataset_id = create_dataset_from_bim(dataset_name, modified_bim, workspace_id, client)
+        print(f"   Dataset ID: {dataset_id}")
         
         if wait_seconds > 0:
-            print(f"   Waiting {wait_seconds} seconds for permissions...")
+            print(f"   Waiting {wait_seconds} seconds before refresh...")
             time.sleep(wait_seconds)
         
-        # Step 6: Refresh
+        # Step 6: Refresh using the dataset ID returned from creation
         print("\n[Step 6/6] Refreshing semantic model...")
-        refresh_dataset(dataset_name, workspace_id, client)
+        refresh_dataset(dataset_name, workspace_id, client, dataset_id=dataset_id)
         
         print("\n" + "=" * 70)
         print("🎉 Deployment Completed!")

{duckrun-0.2.9.dev2.dist-info → duckrun-0.2.9.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: duckrun
-Version: 0.2.9.dev2
+Version: 0.2.9.dev4
 Summary: Lakehouse task runner powered by DuckDB for Microsoft Fabric
 Author: mim
 License: MIT

duckrun-0.2.9.dev4.dist-info/RECORD

@@ -0,0 +1,14 @@
+duckrun/__init__.py,sha256=VJAx606MLj6SVHu3nVePEO0BBp0WxCBtgk_U1olMU7g,235
+duckrun/auth.py,sha256=qPaLQ7InlV9leA9r6E6VEeYavFFoBi0zSN8m_l1aoQs,9545
+duckrun/core.py,sha256=CrWMgA1QHvVF2AAlTlBlQ7VfKsuakcqZa4VuX2WJmik,39279
+duckrun/files.py,sha256=Fvdjg3DyHJzIVzKo8M_j-eGz4zU61lOB38Y_onbQJkI,10137
+duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
+duckrun/runner.py,sha256=yrDxfy1RVkb8iK9GKGmIFZHzCvcO_0GVQlbng7Vw_iM,14171
+duckrun/semantic_model.py,sha256=4_VgsXAHaWhqxI2kOSB2UtRLa6CoBYFEXt418j5xce0,16739
+duckrun/stats.py,sha256=CXfb2DWF3PgOckelJooU0y-BAsNT9NFDfDYEmo0mUQQ,10473
+duckrun/writer.py,sha256=svUuPCYOhrz299NgnpTKhARKjfej0PxnoND2iPDSypk,8098
+duckrun-0.2.9.dev4.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
+duckrun-0.2.9.dev4.dist-info/METADATA,sha256=iBsF-oRskhqicNpd3i5NJq0XZxTepDERJ3i_LVV0rZ4,19277
+duckrun-0.2.9.dev4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+duckrun-0.2.9.dev4.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
+duckrun-0.2.9.dev4.dist-info/RECORD,,

duckrun-0.2.9.dev2.dist-info/RECORD

@@ -1,14 +0,0 @@
-duckrun/__init__.py,sha256=XA85pL2vK1AkmBic8e7WxeqNvcd6SjFX4zsQpImDO6E,230
-duckrun/auth.py,sha256=qPaLQ7InlV9leA9r6E6VEeYavFFoBi0zSN8m_l1aoQs,9545
-duckrun/core.py,sha256=ulKRnxTH8MfGtcKAAORBs-_vd-_jIyWmwv9Bims0TsQ,39267
-duckrun/files.py,sha256=Fvdjg3DyHJzIVzKo8M_j-eGz4zU61lOB38Y_onbQJkI,10137
-duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
-duckrun/runner.py,sha256=yrDxfy1RVkb8iK9GKGmIFZHzCvcO_0GVQlbng7Vw_iM,14171
-duckrun/semantic_model.py,sha256=y_E1VlpqSx9DHOGi--4ZccaODErthzty5CVN4TI-mQ0,15509
-duckrun/stats.py,sha256=CXfb2DWF3PgOckelJooU0y-BAsNT9NFDfDYEmo0mUQQ,10473
-duckrun/writer.py,sha256=svUuPCYOhrz299NgnpTKhARKjfej0PxnoND2iPDSypk,8098
-duckrun-0.2.9.dev2.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
-duckrun-0.2.9.dev2.dist-info/METADATA,sha256=4vDjuMN2L2Uiv6CLIP7UyaKss6-jjNE7ZUxJC5SLnT8,19277
-duckrun-0.2.9.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-duckrun-0.2.9.dev2.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
-duckrun-0.2.9.dev2.dist-info/RECORD,,