duckrun 0.2.9.dev2__tar.gz → 0.2.9.dev4__tar.gz

{duckrun-0.2.9.dev2 → duckrun-0.2.9.dev4}/PKG-INFO

@@ -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.dev2 → duckrun-0.2.9.dev4}/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-0.2.9.dev2 → duckrun-0.2.9.dev4}/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-0.2.9.dev2 → duckrun-0.2.9.dev4}/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 → duckrun-0.2.9.dev4}/duckrun.egg-info/PKG-INFO

@@ -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.dev2 → duckrun-0.2.9.dev4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "duckrun"
-version = "0.2.9.dev2"
+version = "0.2.9.dev4"
 description = "Lakehouse task runner powered by DuckDB for Microsoft Fabric"
 readme = "README.md"
 license = {text = "MIT"}