sqlmesh-dag-generator 0.9.7__tar.gz → 0.9.8__tar.gz

{sqlmesh_dag_generator-0.9.7/sqlmesh_dag_generator.egg-info → sqlmesh_dag_generator-0.9.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh-dag-generator
-Version: 0.9.7
+Version: 0.9.8
 Summary: Open-source Airflow DAG generator for SQLMesh projects
 Home-page: https://github.com/kubolko/sqlmesh-dag-generator
 Author: Jakub Sumionka

{sqlmesh_dag_generator-0.9.7 → sqlmesh_dag_generator-0.9.8}/setup.py

@@ -20,7 +20,7 @@ if requirements_file.exists():
 
 setup(
     name="sqlmesh-dag-generator",
-    version="0.9.7",
+    version="0.9.8",
     description="Open-source Airflow DAG generator for SQLMesh projects",
     long_description=long_description,
     long_description_content_type="text/markdown",

{sqlmesh_dag_generator-0.9.7 → sqlmesh_dag_generator-0.9.8}/sqlmesh_dag_generator/__init__.py

@@ -2,7 +2,7 @@
 SQLMesh DAG Generator - Open Source Airflow Integration for SQLMesh
 """
 
-__version__ = "0.9.7"
+__version__ = "0.9.8"
 
 
 from sqlmesh_dag_generator.generator import SQLMeshDAGGenerator

{sqlmesh_dag_generator-0.9.7 → sqlmesh_dag_generator-0.9.8}/sqlmesh_dag_generator/config.py

@@ -138,7 +138,7 @@ class GenerationConfig:
     include_source_tables: bool = True  # Include upstream source tables as dummy tasks
     return_value: bool = True  # Whether to return execution result (XCom)
     auto_replan_on_change: bool = True  # Automatically plan+apply before running models
-    replan_timeout_hours: int = 6  # Timeout for replan task (initial backfills can be long!)
+    replan_timeout_hours: Optional[int] = 6  # None disables the replan task timeout.
     skip_audits: bool = False  # Skip audit checks during execution
     enable_health_check: bool = False  # Add a pre-flight health check task
     # Tag-based filtering - only include models with any of these tags

{sqlmesh_dag_generator-0.9.7 → sqlmesh_dag_generator-0.9.8}/sqlmesh_dag_generator/generator.py

@@ -183,7 +183,7 @@ class SQLMeshDAGGenerator:
                 include_source_tables=kwargs.get("include_source_tables", True),  # Default: enabled
                 return_value=kwargs.get("return_value", True),
                 auto_replan_on_change=kwargs.get("auto_replan_on_change", True),
-                replan_timeout_hours=kwargs.get("replan_timeout_hours", 6),  # Default: 6 hours for backfills
+                replan_timeout_hours=kwargs.get("replan_timeout_hours", 6),  # None disables the replan timeout.
                 skip_audits=kwargs.get("skip_audits", False),
                 enable_health_check=kwargs.get("enable_health_check", False),
                 include_tags=kwargs.get("include_tags"),
@@ -991,14 +991,25 @@ class SQLMeshDAGGenerator:
                 }
 
             from datetime import timedelta as td
+
+            replan_timeout_hours = self.config.generation.replan_timeout_hours
             replan_task = PythonOperator(
                 task_id="sqlmesh_plan_apply",
                 python_callable=run_replan,
                 dag=dag,
-                # Use custom timeout for replan task - initial backfills can take hours!
-                execution_timeout=td(hours=self.config.generation.replan_timeout_hours),
+                execution_timeout=(
+                    td(hours=replan_timeout_hours)
+                    if replan_timeout_hours is not None
+                    else None
+                ),
             )
-            logger.info(f"Created auto-replan task: sqlmesh_plan_apply (timeout: {self.config.generation.replan_timeout_hours}h)")
+            if replan_timeout_hours is None:
+                logger.info("Created auto-replan task: sqlmesh_plan_apply (timeout: disabled)")
+            else:
+                logger.info(
+                    "Created auto-replan task: sqlmesh_plan_apply (timeout: %sh)",
+                    replan_timeout_hours,
+                )
 
             # Link health check to replan if both exist
             if health_check_task:

{sqlmesh_dag_generator-0.9.7 → sqlmesh_dag_generator-0.9.8/sqlmesh_dag_generator.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh-dag-generator
-Version: 0.9.7
+Version: 0.9.8
 Summary: Open-source Airflow DAG generator for SQLMesh projects
 Home-page: https://github.com/kubolko/sqlmesh-dag-generator
 Author: Jakub Sumionka