kailash 0.8.6__py3-none-any.whl → 0.9.0__py3-none-any.whl

kailash/nodes/logic/operations.py

@@ -85,10 +85,14 @@ class SwitchNode(Node):
         ...     operator="==",
         ...     value=True
         ... ))
-        >>> workflow.connect("convergence", "switch")
-        >>> workflow.connect("switch", "processor",
-        ...     condition="false_output", cycle=True)
-        >>> workflow.connect("switch", "output", condition="true_output")
+        >>> workflow.add_connection("convergence", "result", "switch", "input_data")
+        >>> # Use CycleBuilder for cyclic connections
+        >>> cycle = workflow.create_cycle("convergence_loop")
+        >>> cycle.connect("switch", "false_output", "processor", "input")
+        >>> cycle.connect("processor", "result", "convergence", "data")
+        >>> cycle.max_iterations(50).build()
+        >>> # Non-cyclic output connection
+        >>> workflow.add_connection("switch", "true_output", "output", "data")
     """
 
     def get_parameters(self) -> dict[str, NodeParameter]:

kailash/runtime/local.py

@@ -636,6 +636,26 @@ class LocalRuntime:
                 if self.debug:
                     self.logger.debug(f"Node {node_id} inputs: {inputs}")
 
+                # CONDITIONAL EXECUTION: Skip nodes that only receive None inputs from conditional routing
+                if self._should_skip_conditional_node(workflow, node_id, inputs):
+                    self.logger.info(
+                        f"Skipping node {node_id} - all conditional inputs are None"
+                    )
+                    # Store None result to indicate the node was skipped
+                    results[node_id] = None
+                    node_outputs[node_id] = None
+
+                    # Update task status if tracking is enabled
+                    if task and task_manager:
+                        task_manager.update_task_status(
+                            task.task_id,
+                            TaskStatus.COMPLETED,
+                            result=None,
+                            ended_at=datetime.now(UTC),
+                            metadata={"skipped": True, "reason": "conditional_routing"},
+                        )
+                    continue
+
                 # Execute node with unified async/sync support and metrics collection
                 collector = MetricsCollector()
                 with collector.collect(node_id=node_id) as metrics_context:
@@ -1123,6 +1143,88 @@ class LocalRuntime:
         metrics_collector = get_metrics_collector()
         metrics_collector.reset_metrics()
 
+    def _should_skip_conditional_node(
+        self, workflow: Workflow, node_id: str, inputs: dict[str, Any]
+    ) -> bool:
+        """Determine if a node should be skipped due to conditional routing.
+
+        A node should be skipped if:
+        1. It has incoming connections from conditional nodes (like SwitchNode)
+        2. All of its connected inputs are None
+        3. It has no node-level configuration parameters that would make it run independently
+
+        Args:
+            workflow: The workflow being executed.
+            node_id: Node ID to check.
+            inputs: Prepared inputs for the node.
+
+        Returns:
+            True if the node should be skipped, False otherwise.
+        """
+        # Get all incoming edges for this node
+        incoming_edges = list(workflow.graph.in_edges(node_id, data=True))
+
+        # If the node has no incoming connections, don't skip it
+        # (it might be a source node or have configuration parameters)
+        if not incoming_edges:
+            return False
+
+        # Check if any incoming edges are from conditional nodes
+        has_conditional_inputs = False
+        for source_node_id, _, edge_data in incoming_edges:
+            source_node = workflow._node_instances.get(source_node_id)
+            if source_node and source_node.__class__.__name__ in ["SwitchNode"]:
+                has_conditional_inputs = True
+                break
+
+        # If no conditional inputs, don't skip
+        if not has_conditional_inputs:
+            return False
+
+        # Get the node instance to check for configuration parameters
+        node_instance = workflow._node_instances.get(node_id)
+        if not node_instance:
+            return False
+
+        # Check if the node has configuration parameters that would make it run independently
+        # (excluding standard parameters and None values)
+        node_config = getattr(node_instance, "config", {})
+        significant_config = {
+            k: v
+            for k, v in node_config.items()
+            if k not in ["metadata", "name", "id"] and v is not None
+        }
+
+        # If the node has significant configuration, it might still be valuable to run
+        if significant_config:
+            # Check if any connected inputs have actual data (not None)
+            connected_inputs = {}
+            for _, _, edge_data in incoming_edges:
+                mapping = edge_data.get("mapping", {})
+                for source_key, target_key in mapping.items():
+                    if target_key in inputs:
+                        connected_inputs[target_key] = inputs[target_key]
+
+            # If all connected inputs are None but node has config, still skip
+            # The user can configure the node to run with default values if needed
+            if all(v is None for v in connected_inputs.values()):
+                return True
+
+        # Check if all connected inputs are None
+        # This is the main condition for conditional routing
+        has_non_none_input = False
+        for _, _, edge_data in incoming_edges:
+            mapping = edge_data.get("mapping", {})
+            for source_key, target_key in mapping.items():
+                if target_key in inputs and inputs[target_key] is not None:
+                    has_non_none_input = True
+                    break
+            if has_non_none_input:
+                break
+
+        # Skip the node if all connected inputs are None
+        return not has_non_none_input
+
     def _should_stop_on_error(self, workflow: Workflow, node_id: str) -> bool:
         """Determine if execution should stop when a node fails.
 

kailash/workflow/cyclic_runner.py

@@ -220,6 +220,60 @@ class CyclicWorkflowExecutor:
         else:
             return obj
 
+    def _should_skip_conditional_node_cyclic(
+        self, workflow: Workflow, node_id: str, merged_inputs: dict[str, Any]
+    ) -> bool:
+        """Determine if a node should be skipped due to conditional routing in cyclic execution.
+
+        This is similar to LocalRuntime._should_skip_conditional_node but adapted for cyclic execution.
+
+        Args:
+            workflow: The workflow being executed.
+            node_id: Node ID to check.
+            merged_inputs: Merged inputs for the node.
+
+        Returns:
+            True if the node should be skipped, False otherwise.
+        """
+        # Get all incoming edges for this node
+        incoming_edges = list(workflow.graph.in_edges(node_id, data=True))
+
+        # If the node has no incoming connections, don't skip it
+        if not incoming_edges:
+            return False
+
+        # Check if any incoming edges are from conditional nodes
+        has_conditional_inputs = False
+        for source_node_id, _, edge_data in incoming_edges:
+            try:
+                source_node = workflow.get_node(source_node_id)
+                if source_node and source_node.__class__.__name__ in ["SwitchNode"]:
+                    has_conditional_inputs = True
+                    break
+            except:
+                continue
+
+        # If no conditional inputs, don't skip
+        if not has_conditional_inputs:
+            return False
+
+        # Check if all connected inputs are None
+        has_non_none_input = False
+        for _, _, edge_data in incoming_edges:
+            mapping = edge_data.get("mapping", {})
+            for source_key, target_key in mapping.items():
+                if (
+                    target_key in merged_inputs
+                    and merged_inputs[target_key] is not None
+                ):
+                    has_non_none_input = True
+                    break
+            if has_non_none_input:
+                break
+
+        # Skip the node if all connected inputs are None
+        return not has_non_none_input
+
     def _execute_with_cycles(
         self,
         workflow: Workflow,
@@ -863,6 +917,20 @@ class CyclicWorkflowExecutor:
             except Exception as e:
                 logger.warning(f"Failed to create task for node '{node_id}': {e}")
 
+        # CONDITIONAL EXECUTION: Skip nodes that only receive None inputs from conditional routing
+        if self._should_skip_conditional_node_cyclic(workflow, node_id, merged_inputs):
+            logger.info(f"Skipping node {node_id} - all conditional inputs are None")
+            # Store None result to indicate the node was skipped
+            if task and task_manager:
+                task_manager.update_task_status(
+                    task.task_id,
+                    TaskStatus.COMPLETED,
+                    result=None,
+                    ended_at=datetime.now(UTC),
+                    metadata={"skipped": True, "reason": "conditional_routing"},
+                )
+            return None
+
         # Execute node with metrics collection
         collector = MetricsCollector()
         logger.debug(

{kailash-0.8.6.dist-info → kailash-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.8.6
+Version: 0.9.0
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum
@@ -81,6 +81,7 @@ Requires-Dist: opentelemetry-instrumentation-fastapi>=0.55b1
 Requires-Dist: seaborn>=0.13.2
 Requires-Dist: sqlparse>=0.5.3
 Requires-Dist: jsonschema>=4.24.0
+Requires-Dist: openai>=1.97.1
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-cov>=3.0; extra == "dev"

{kailash-0.8.6.dist-info → kailash-0.9.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kailash/__init__.py,sha256=OnMcVOw8TgoGUth6Z_dyWPrlXQV4pECdOA5GYsfelEI,2628
+kailash/__init__.py,sha256=c9Upj65mN4RX9QCF8NliTUtNaZJJW8jiJqBvR4c6vmA,2537
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
 kailash/access_control.py,sha256=MjKtkoQ2sg1Mgfe7ovGxVwhAbpJKvaepPWr8dxOueMA,26058
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
@@ -88,9 +88,10 @@ kailash/mcp_server/client_new.py,sha256=YU671JvAM0uvuX0uhGZCIKI8co3fqz0cs6HqLZ59
 kailash/mcp_server/discovery.py,sha256=D8vcwVkbgQCNp0_BlkGeU_dnqgIXN2g0s3_GpndlQu0,57828
 kailash/mcp_server/errors.py,sha256=_lycwudWP_AJ_KwLO5N3VCKbG1ikfaTyzA2PBF8UAYU,21181
 kailash/mcp_server/oauth.py,sha256=GFC2O2ueiTTI6V-91Huevhc3K8CxrHe22knuHfuCTqY,56493
-kailash/mcp_server/protocol.py,sha256=gXeJ-GvSf39WymfS6433SqLKBA40PdeDoBQqu7DUbJE,35129
+kailash/mcp_server/protocol.py,sha256=NIdEwJT21JT9ItajXniPNvCbZtTbpqyOC_ZezqsguGE,35694
 kailash/mcp_server/registry_integration.py,sha256=B8CSLq_O1ea3cXrbVjC3bB_OFgHIP-KS9dk77mNM02I,19791
-kailash/mcp_server/server.py,sha256=RQchBc-yaz5Mf_ZRrHukv8IGGgwKKXTg0hEtaVLGeo4,80246
+kailash/mcp_server/server.py,sha256=yFp1F4QQl6gkTY_9JJWmiMiwfT-zACLJLubz-NR5sCw,108675
+kailash/mcp_server/subscriptions.py,sha256=J0FXg6_8lOffQ5SSwm2_DmqIS2pOii6ikxORANLCy1o,58589
 kailash/mcp_server/transports.py,sha256=fBa7CTVYTDb0ZbBQTsZ2d8rKvcVuqBIteczq8eqarr4,49919
 kailash/mcp_server/servers/ai_registry.py,sha256=IdF_keUuJlMsvjLjSAykxxbm46K4qA7eCj7T-lYSrzk,10007
 kailash/mcp_server/utils/__init__.py,sha256=R20N-iiKXUPxc9MOh6vPO1vIfkPmwhEQ5KNFgGd4xSs,771
@@ -247,7 +248,7 @@ kailash/nodes/logic/__init__.py,sha256=JKGFXwBDfY3s1MWQkx3ivdvCMm3b3HIXCn-wH9uMo
 kailash/nodes/logic/async_operations.py,sha256=bpCc-t5uKi4DVJiVdXUcIZ6BWJOLsxQu-4UCEnwpw6M,27509
 kailash/nodes/logic/convergence.py,sha256=ooNMdEJs8B0SeMQrzU1Uo_iPiuPXi8j-9IyRDSl9sYM,25080
 kailash/nodes/logic/loop.py,sha256=34hnrcfeigcpsVcomsd-ZLE2x7f3irAd_-Q89vZzW9w,5756
-kailash/nodes/logic/operations.py,sha256=dC594biBTdiw3vbyeu3TgPZs7UYAfFjbXGwcYbN_5ec,28722
+kailash/nodes/logic/operations.py,sha256=eJpTCWXqOIyP7jc057CXkhW2CBpTIVZXGkV74LUgza4,29002
 kailash/nodes/logic/workflow.py,sha256=p2ED6tOWGVC50iNyUInSpJI41eBXmSF8Tb_w0h7NeD0,17136
 kailash/nodes/mixins/__init__.py,sha256=0WYfu5kj-lHbFwP9g5vmlbsG8UzvI-vhOyHMEUzXbz4,558
 kailash/nodes/mixins/event_emitter.py,sha256=xTeNrBWmuWIf8qYA5DZekymjjrTAD1sboW9dKbAP40w,7492
@@ -316,7 +317,7 @@ kailash/runtime/__init__.py,sha256=CvU-qBMESYYISqFOlYlLsYJrXJu0Gqr4x6yr4Ob_Rng,2
 kailash/runtime/access_controlled.py,sha256=HtNJZylaB-2FuPsfEOfQ-4ny4HzwJfHaHNMu2xS1Nzs,17324
 kailash/runtime/async_local.py,sha256=sYNggSU0R-oo8cCvU5ayodDBqASzUhxu994ZvZxDSC0,34010
 kailash/runtime/docker.py,sha256=sZknVl1PCGfAZeyc0-exTuKlllSyjYlFIgJoiB3CRNs,23500
-kailash/runtime/local.py,sha256=lDE36rIL1Pl1SjDkDStWMlF3qeaZRQzGugkWtcoMQRc,66013
+kailash/runtime/local.py,sha256=gWHuSgbPWqIHYvirRo80LV8RjDvOR2PQoUs737_pAh0,70485
 kailash/runtime/parallel.py,sha256=mz_wPD13-YVc3Q_8HkOs4nPQPdTjnjCcnRL7ZRM70lo,21070
 kailash/runtime/parallel_cyclic.py,sha256=yANZHnePjhCPuCFbq3lFQA1K6jbCv5Of5-vIKbCsmZk,19863
 kailash/runtime/parameter_injection.py,sha256=kG4GhmarsRr5t3VDFbc2G1HSbsZJg6UmienHCE2Ru7o,14852
@@ -379,7 +380,7 @@ kailash/workflow/cycle_debugger.py,sha256=eG-Q_kakqyhr1Ts-q4pRnO0EI7mVO9ao1s9WOx
 kailash/workflow/cycle_exceptions.py,sha256=4_OLnbEXqIiXKzOc3uh8DzFik4wEHwl8bRZhY9Xhf2A,21838
 kailash/workflow/cycle_profiler.py,sha256=aEWSCm0Xy15SjgLTpPooVJMzpFhtJWt4livR-3Me4N8,28547
 kailash/workflow/cycle_state.py,sha256=hzRUvciRreWfS56Cf7ZLQPit_mlPTQDoNTawh8yi-2s,10747
-kailash/workflow/cyclic_runner.py,sha256=0IlmghTrwYb3ivqP975DprP3aj45-8_Sn5Wq9tEutG0,43888
+kailash/workflow/cyclic_runner.py,sha256=CFr4eucYrN_9K1XwiTE_ZmtbpdwDJv1OA7g_Pc0hEAI,46616
 kailash/workflow/edge_infrastructure.py,sha256=lQDzs0-KdoCMqI4KAXAGbhHbwadM6t-ffJEWLlRuSNo,12448
 kailash/workflow/graph.py,sha256=zRpGLXeuwtuxFBvE7_16c_bB9yqZirM_uwtfD1_MY4g,59272
 kailash/workflow/input_handling.py,sha256=HrW--AmelYC8F18nkfmYlF_wXycA24RuNbDRjvM8rqk,6561
@@ -394,9 +395,9 @@ kailash/workflow/templates.py,sha256=XQMAKZXC2dlxgMMQhSEOWAF3hIbe9JJt9j_THchhAm8
 kailash/workflow/type_inference.py,sha256=i1F7Yd_Z3elTXrthsLpqGbOnQBIVVVEjhRpI0HrIjd0,24492
 kailash/workflow/validation.py,sha256=r2zApGiiG8UEn7p5Ji842l8OR1_KftzDkWc7gg0cac0,44675
 kailash/workflow/visualization.py,sha256=nHBW-Ai8QBMZtn2Nf3EE1_aiMGi9S6Ui_BfpA5KbJPU,23187
-kailash-0.8.6.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
-kailash-0.8.6.dist-info/METADATA,sha256=8CMVRW3iAN5K_8U_1GN8OxLJF4HGjoo8G0vzKMq7kRc,21703
-kailash-0.8.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kailash-0.8.6.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
-kailash-0.8.6.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
-kailash-0.8.6.dist-info/RECORD,,
+kailash-0.9.0.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.9.0.dist-info/METADATA,sha256=h2b6X5_Qrk6f-iadmgTO-muDLzhiH2Huf1uCMBq1ANY,21733
+kailash-0.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.9.0.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.9.0.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.9.0.dist-info/RECORD,,