kailash 0.7.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

kailash/workflow/visualization.py

@@ -18,7 +18,7 @@ class WorkflowVisualizer:
 
     def __init__(
         self,
-        workflow: Workflow,
+        workflow: Workflow | None = None,
         node_colors: dict[str, str] | None = None,
         edge_colors: dict[str, str] | None = None,
         layout: str = "hierarchical",
@@ -26,7 +26,7 @@ class WorkflowVisualizer:
         """Initialize visualizer.
 
         Args:
-            workflow: Workflow to visualize
+            workflow: Workflow to visualize (can be set later)
             node_colors: Custom node color map
             edge_colors: Custom edge color map
             layout: Layout algorithm to use
@@ -70,9 +70,12 @@ class WorkflowVisualizer:
         """Get colors for all nodes in workflow."""
         colors = []
         for node_id in self.workflow.graph.nodes():
-            node_instance = self.workflow.nodes[node_id]
-            node_type = node_instance.node_type
-            colors.append(self._get_node_color(node_type))
+            node_instance = target_workflow.nodes.get(node_id)
+            if node_instance:
+                node_type = node_instance.node_type
+                colors.append(self._get_node_color(node_type))
+            else:
+                colors.append(self.node_colors["default"])
         return colors
 
     def _get_node_labels(self) -> dict[str, str]:
@@ -119,11 +122,15 @@ class WorkflowVisualizer:
 
         return edge_labels
 
-    def _calculate_layout(self) -> dict[str, tuple[float, float]]:
+    def _calculate_layout(self, workflow: 'Workflow' = None) -> dict[str, tuple[float, float]]:
         """Calculate node positions for visualization."""
+        target_workflow = workflow or self.workflow
+        if not target_workflow:
+            return {}
+            
         # Try to use stored positions first
         pos = {}
-        for node_id, node_instance in self.workflow.nodes.items():
+        for node_id, node_instance in target_workflow.nodes.items():
             if node_instance.position != (0, 0):
                 pos[node_id] = node_instance.position
 
@@ -133,32 +140,69 @@ class WorkflowVisualizer:
                 # Use hierarchical layout for DAGs
                 try:
                     # Create layers based on topological order
-                    layers = self._create_layers()
+                    layers = self._create_layers(target_workflow)
                     pos = self._hierarchical_layout(layers)
                 except Exception:
                     # Fallback to spring layout
-                    pos = nx.spring_layout(self.workflow.graph, k=3, iterations=50)
+                    pos = nx.spring_layout(target_workflow.graph, k=3, iterations=50)
             elif self.layout == "circular":
-                pos = nx.circular_layout(self.workflow.graph)
+                pos = nx.circular_layout(target_workflow.graph)
             elif self.layout == "spring":
-                pos = nx.spring_layout(self.workflow.graph, k=2, iterations=100)
+                pos = nx.spring_layout(target_workflow.graph, k=2, iterations=100)
             else:
                 # Default to spring layout
-                pos = nx.spring_layout(self.workflow.graph)
+                pos = nx.spring_layout(target_workflow.graph)
 
         return pos
+    
+    def _get_layout_positions(self, workflow: Workflow) -> dict[str, tuple[float, float]]:
+        """Get layout positions for workflow nodes."""
+        # Temporarily store workflow and calculate layout
+        original_workflow = self.workflow
+        self.workflow = workflow
+        try:
+            return self._calculate_layout()
+        finally:
+            self.workflow = original_workflow
+    
+    def _get_node_colors(self, workflow: Workflow) -> list[str]:
+        """Get node colors for workflow."""
+        colors = []
+        for node_id in workflow.graph.nodes():
+            node_instance = workflow.get_node(node_id)
+            if node_instance:
+                node_type = node_instance.__class__.__name__.lower()
+                # Map node types to color categories
+                if "data" in node_type or "csv" in node_type or "json" in node_type:
+                    color_key = "data"
+                elif "transform" in node_type or "python" in node_type:
+                    color_key = "transform"
+                elif "switch" in node_type or "merge" in node_type:
+                    color_key = "logic"
+                elif "llm" in node_type or "ai" in node_type:
+                    color_key = "ai"
+                else:
+                    color_key = "default"
+                colors.append(self.node_colors.get(color_key, self.node_colors["default"]))
+            else:
+                colors.append(self.node_colors["default"])
+        return colors
 
-    def _create_layers(self) -> dict[int, list]:
+    def _create_layers(self, workflow: 'Workflow' = None) -> dict[int, list]:
         """Create layers of nodes for hierarchical layout."""
+        target_workflow = workflow or self.workflow
+        if not target_workflow:
+            return {}
+            
         layers = {}
-        remaining = set(self.workflow.graph.nodes())
+        remaining = set(target_workflow.graph.nodes())
         layer = 0
 
         while remaining:
             # Find nodes with no dependencies in remaining set
             current_layer = []
             for node in remaining:
-                predecessors = set(self.workflow.graph.predecessors(node))
+                predecessors = set(target_workflow.graph.predecessors(node))
                 if not predecessors.intersection(remaining):
                     current_layer.append(node)
 
@@ -196,20 +240,34 @@ class WorkflowVisualizer:
 
     def _draw_graph(
         self,
-        pos: dict[str, tuple[float, float]],
-        node_colors: list[str],
-        show_labels: bool,
-        show_connections: bool,
+        workflow: Workflow | None = None,
+        pos: dict[str, tuple[float, float]] | None = None,
+        node_colors: list[str] | None = None,
+        show_labels: bool = True,
+        show_connections: bool = True,
     ) -> None:
         """Draw the graph with given positions and options."""
+        # Use provided workflow or fall back to instance workflow
+        target_workflow = workflow or self.workflow
+        if not target_workflow:
+            raise ValueError("No workflow provided to draw")
+        
+        # Use default position if not provided
+        if pos is None:
+            pos = self._get_layout_positions(target_workflow)
+        
+        # Use default colors if not provided
+        if node_colors is None:
+            node_colors = self._get_node_colors(target_workflow)
+        
         # Draw nodes
         nx.draw_networkx_nodes(
-            self.workflow.graph, pos, node_color=node_colors, node_size=3000, alpha=0.9
+            target_workflow.graph, pos, node_color=node_colors, node_size=3000, alpha=0.9
         )
 
         # Draw edges
         nx.draw_networkx_edges(
-            self.workflow.graph,
+            target_workflow.graph,
             pos,
             edge_color=self.edge_colors["default"],
             width=2,
@@ -221,16 +279,26 @@ class WorkflowVisualizer:
 
         # Draw labels
         if show_labels:
+            # Temporarily set workflow for label generation
+            old_workflow = self.workflow
+            self.workflow = target_workflow
             labels = self._get_node_labels()
+            self.workflow = old_workflow
+            
             nx.draw_networkx_labels(
-                self.workflow.graph, pos, labels, font_size=10, font_weight="bold"
+                target_workflow.graph, pos, labels, font_size=10, font_weight="bold"
             )
 
         # Draw connection labels
         if show_connections:
+            # Temporarily set workflow for edge label generation
+            old_workflow = self.workflow
+            self.workflow = target_workflow
             edge_labels = self._get_edge_labels()
+            self.workflow = old_workflow
+            
             nx.draw_networkx_edge_labels(
-                self.workflow.graph, pos, edge_labels, font_size=8
+                target_workflow.graph, pos, edge_labels, font_size=8
             )
 
     def visualize(
@@ -255,10 +323,14 @@ class WorkflowVisualizer:
             **kwargs: Additional options passed to plt.savefig
         """
         try:
+            # Check if workflow is available
+            if not self.workflow:
+                raise ValueError("No workflow to visualize. Set workflow property or create visualizer with workflow.")
+                
             plt.figure(figsize=figsize)
 
             # Calculate node positions
-            pos = self._calculate_layout()
+            pos = self._calculate_layout(self.workflow)
 
             # Handle empty workflow case
             if not self.workflow.graph.nodes():
@@ -266,11 +338,11 @@ class WorkflowVisualizer:
                 node_colors = []
             else:
                 # Draw the graph with colors
-                node_colors = self._get_node_colors()
+                node_colors = self._get_node_colors(self.workflow)
 
             # Draw the graph components
-            if pos and node_colors:
-                self._draw_graph(pos, node_colors, show_labels, show_connections)
+            if pos:
+                self._draw_graph(workflow=self.workflow, pos=pos, node_colors=node_colors, show_labels=show_labels, show_connections=show_connections)
 
             # Set title
             if title is None:

{kailash-0.7.0.dist-info → kailash-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.7.0
+Version: 0.8.0
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum
@@ -21,7 +21,7 @@ Requires-Dist: matplotlib>=3.5
 Requires-Dist: pyyaml>=6.0
 Requires-Dist: click>=8.0
 Requires-Dist: pytest>=8.3.5
-Requires-Dist: mcp[cli]>=1.9.2
+Requires-Dist: mcp[cli]==1.11.0
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: numpy>=2.2.5
 Requires-Dist: scipy>=1.15.3
@@ -78,6 +78,7 @@ Requires-Dist: passlib>=1.7.4
 Requires-Dist: pyotp>=2.9.0
 Requires-Dist: opentelemetry-instrumentation-fastapi>=0.55b1
 Requires-Dist: seaborn>=0.13.2
+Requires-Dist: sqlparse>=0.5.3
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-cov>=3.0; extra == "dev"

{kailash-0.7.0.dist-info → kailash-0.8.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
-kailash/__init__.py,sha256=p4a_FOvSFXBcUpOZgCnsoETI17SOvpc5J1aYyN49K34,2878
+kailash/__init__.py,sha256=yCfV0dNXfj0xBIsqAbdMl31CxaSma9AtltRW6rI2AhU,2878
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
-kailash/access_control.py,sha256=2ctdRFeSeu-d7DU04Aovxh6Rt_4t3IyQfkKEjTeQiMM,25519
+kailash/access_control.py,sha256=MjKtkoQ2sg1Mgfe7ovGxVwhAbpJKvaepPWr8dxOueMA,26058
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
 kailash/manifest.py,sha256=qzOmeMGWz20Sp4IJitSH9gTVbGng7hlimc96VTW4KI8,24814
 kailash/sdk_exceptions.py,sha256=dueSUxUYqKXmHS5mwl6QtEzP6a0rMGXcBFGCh5sT0tg,10179
-kailash/security.py,sha256=qISSfqoue7bjqudHcUUdEUR9sG-Rj9p3zG0J_Ean4Ek,27063
+kailash/security.py,sha256=GBU93tUXcTZGEjl0f9XXRlF3W-sSulhiatMZzv1dNqc,27228
 kailash/access_control/__init__.py,sha256=ykR_zGJXQoCU_4gGNFbYzhVdUemxeAAoDQLhKIPVBGE,4018
 kailash/access_control/managers.py,sha256=Vg2inaZqR2GBXsySvPZcEqQtFHgF94z7A_wUHMtA3qA,14431
 kailash/access_control/rule_evaluators.py,sha256=niguhjThBjA0jIXvdKdGAXzdSM_bAd0ebphGgRrDFKU,15337
@@ -16,7 +16,7 @@ kailash/api/custom_nodes_secure.py,sha256=cMh1FbEtUAEoLmTb5ew_CeX5kjikBzmBsat1GO
 kailash/api/gateway.py,sha256=BVEKyC53JRnBkOyg3YXAPGDzLm3zon0vbO1xiM6yNxU,12540
 kailash/api/mcp_integration.py,sha256=xY5VjYXh4bFuNyyWBXEuTm4jjwUn8_9QxZpa4hh6a0Q,14521
 kailash/api/studio.py,sha256=8Gj3R3p_-EpJerkWYLyrqVzJprKOQmTmK-1EgiLoXMc,34892
-kailash/api/workflow_api.py,sha256=8g0yOtlartpuVpJpR92xpqjSt-sPohcw9CUp-J_uOus,13147
+kailash/api/workflow_api.py,sha256=S1O_PRtTo2SSk47iUh4-uTh0WY1Tjbl1Gt54QIOLDmE,14330
 kailash/channels/__init__.py,sha256=DP3PxbGDRCLrJkr6gGf1BBJ0MaHsfp_zUSbkyc04ieE,610
 kailash/channels/api_channel.py,sha256=stuJxNJRUobzj1CavwsHWVZWr2kYqyVLjv_ynt6Wy6A,14541
 kailash/channels/base.py,sha256=8-cUYAVMlO67BMZCZscShy0FPYAK31s9bxk0JXA9wIM,7825
@@ -60,7 +60,7 @@ kailash/mcp_server/ai_registry_server.py,sha256=vMNMvWLegKBVp7YAHVKgltWa2vTXKNvV
 kailash/mcp_server/auth.py,sha256=Y2BztTL2IILCLSHzZIDSXsIKyUTR8gUXwum5rKffUxs,25251
 kailash/mcp_server/client.py,sha256=uJXOP7Z4q3CFnX_RvRL5NM6PBcEfR7fUkPQIxfa8c8I,25710
 kailash/mcp_server/client_new.py,sha256=YU671JvAM0uvuX0uhGZCIKI8co3fqz0cs6HqLZ59Xyo,10285
-kailash/mcp_server/discovery.py,sha256=wtSQivxCTUv1M_o_EsXUgxmAChbT2U1TBFCLWCGq9BM,56473
+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
@@ -83,11 +83,11 @@ kailash/middleware/auth/models.py,sha256=sMQ1JnEDKPId1kLNCi2fDfhgLZX-YguHPnSMfS_
 kailash/middleware/auth/utils.py,sha256=AW4fYJ2opQvOr_1YgSQi_MsC_RoFPq0XVOEgbxoXWtw,6555
 kailash/middleware/communication/__init__.py,sha256=keQ2db4WI2-oZ_nJ5sLE1Tum_RkUt7M2VLTmqOlt0zA,778
 kailash/middleware/communication/ai_chat.py,sha256=2XmnJB2Nz2xm2darsF2DYnSLGLAYdyYooHok5GrHKb8,35994
-kailash/middleware/communication/api_gateway.py,sha256=-y2gUyeWXWvn5PO2GXkaDYFIEZb6kM1iHTiKHrm8WYc,31363
+kailash/middleware/communication/api_gateway.py,sha256=hjs3Ued5YhFR1LcGJEB2d-F96imIuS7W7bxctQG3blA,32272
 kailash/middleware/communication/events.py,sha256=MEjgcibNyjA4tSFK8CeXDn1oCE75My7K_saxdCBz2HY,15367
-kailash/middleware/communication/realtime.py,sha256=__34kf61akSibvKYz-myw0BnIxWfMOns-wLh-oRfI-A,24955
+kailash/middleware/communication/realtime.py,sha256=xm0i-TUs82XUtnFfHpdeZcx4uMt_V-fmS_uTuvXqDV0,28647
 kailash/middleware/core/__init__.py,sha256=4yQkOWC4b88zSogs1YVqtKG1PugbncqNCwNNWxTdIZA,545
-kailash/middleware/core/agent_ui.py,sha256=_2mIsseJjAm118iOhPeGrdE8_lmccrv4K29uC3tlPrc,36357
+kailash/middleware/core/agent_ui.py,sha256=Ki5QpsYNM3o3h9KGor4YEXs9Xg8ehXe3w_M8SJnYhxI,36358
 kailash/middleware/core/schema.py,sha256=uVF-5ZJlLYHOQdsKrG46FnTO1bq_QtDjhUSkIIDL9dY,23584
 kailash/middleware/core/workflows.py,sha256=kjwwP69-T6eCY7kWIMLUBwVy2CapoPR34cqCETquq0s,12480
 kailash/middleware/database/__init__.py,sha256=UMws94L-vja94AjfzPWIgn0h4_5BGQzI3YaSdqtzeLk,1682
@@ -104,14 +104,18 @@ kailash/middleware/gateway/deduplicator.py,sha256=CblV3fwc7s4wg6KIvypwyNMYL5AuQi
 kailash/middleware/gateway/durable_gateway.py,sha256=EsIgMNxS_no2W40AXDyE7FmVdnGNU26kBRC5rfnSzoQ,14626
 kailash/middleware/gateway/durable_request.py,sha256=SCnp-bF0tQX9oahr9reqcZjJ_YhyJkeYYl-un9rJ6lo,15437
 kailash/middleware/gateway/event_store.py,sha256=A3Kh2MhVVPbXWvjeo550SqEGPiJYyspAfu6Gv7UZzo4,16131
+kailash/middleware/gateway/storage_backends.py,sha256=HJTi6zU6oBI3R3ffcm_U-Xp9qqb9yOHS7JbRawP8AV8,12974
 kailash/middleware/mcp/__init__.py,sha256=EdZB8zOMSBEEmudRzs8ksz9QZJYWQMEx7Tm1MOwIWnI,922
 kailash/middleware/mcp/client_integration.py,sha256=dY1RmX-g5E6JzUFuWxk7viuOYIh8bMwoUSvHQMVEsYk,18265
 kailash/middleware/mcp/enhanced_server.py,sha256=RUVS7jWHn0ma4F3F23UvuFwUdu7OkSsIRNQmGtkG9I8,18547
 kailash/nexus/__init__.py,sha256=zaOXOHmmxsDDuyVDA7P-EWkXHLsLHcnrZPSlNQ74M1k,463
 kailash/nexus/factory.py,sha256=oqtPJoVvwQUdafs_h92MDpjt6xSO1vNLZHizZ1-XLks,12296
 kailash/nexus/gateway.py,sha256=QzvJVFVmgNI_Z4VHIRpUQ9F8_HNrhINjAiJQ1kTpqJU,19105
-kailash/nodes/__init__.py,sha256=J8u4EPAqNWYLDTm6gQGmbIck94fB8cSeqUvc_AgYsko,996
-kailash/nodes/base.py,sha256=g8hVZYd-XDbyLaJlKHRYb-xJ9wMGresI34UsCrgvURE,55876
+kailash/nexus/cli/__init__.py,sha256=XbGcZSytNg5lJCOZ981prJpzwTXYJcR6_d6Fw9cXUTo,106
+kailash/nexus/cli/__main__.py,sha256=GCIzeLFbdtBoRwfB1r0gxME078qd1e3IIFrGHHZQfAc,117
+kailash/nexus/cli/main.py,sha256=2GqJP5fvP8DgPaeWzOjAG9yS0CbDWhzEl_tI38jO3VQ,5550
+kailash/nodes/__init__.py,sha256=0uT1nGAIZNzk_tkQo1gmEEkroE8qiVWSjDv1hpoKqQA,1009
+kailash/nodes/base.py,sha256=LqxnXG1hl33GY4caZiOCshagq-FfHe_nDHVAww1Dkmg,57125
 kailash/nodes/base_async.py,sha256=whxepCiVplrltfzEQuabmnGCpEV5WgfqwgxbLdCyiDk,8864
 kailash/nodes/base_cycle_aware.py,sha256=Xpze9xZzLepWeLpi9Y3tMn1dm2LVv-omr5TSQuGTtWo,13377
 kailash/nodes/base_with_acl.py,sha256=ZfrkLPgrEBcNbG0LKvtq6glDxyOYOMRw3VXX4vWX6bI,11852
@@ -162,13 +166,13 @@ kailash/nodes/cache/cache_invalidation.py,sha256=IUvxrRj3K5EF29Z2EaKl7t6Uze_cssn
 kailash/nodes/cache/redis_pool_manager.py,sha256=GR82GCWxo_gAzRE-091OB6AhKre8CTwM3OoePLb2gvE,21574
 kailash/nodes/code/__init__.py,sha256=yhEwuMjUEPFfe6hMGMd4E4gZdLUuf2JEQ7knYapiM4o,1283
 kailash/nodes/code/async_python.py,sha256=Ai-iMpmz-sAori73JBk0wZtqmwtmF2GNPDxqB04I2Ck,37058
-kailash/nodes/code/python.py,sha256=lH3GHorFZL4Uz7pkaXliTtLi4rUog5ni5EFip8cNqmY,58509
+kailash/nodes/code/python.py,sha256=u-5lSoxc5ul7zGGSanrcVwXLxlGcDlirdCXT6mLSm40,60331
 kailash/nodes/compliance/__init__.py,sha256=6a_FL4ofc8MAVuZ-ARW5uYenZLS4mBFVM9AI2QsnoF8,214
 kailash/nodes/compliance/data_retention.py,sha256=90bH_eGwlcDzUdklAJeXQM-RcuLUGQFQ5fgHOK8a4qk,69443
 kailash/nodes/compliance/gdpr.py,sha256=ZMoHZjAo4QtGwtFCzGMrAUBFV3TbZOnJ5DZGZS87Bas,70548
 kailash/nodes/data/__init__.py,sha256=f0h4ysvXxlyFcNJLvDyXrgJ0ixwDF1cS0pJ2QNPakhg,5213
 kailash/nodes/data/async_connection.py,sha256=wfArHs9svU48bxGZIiixSV2YVn9cukNgEjagwTRu6J4,17250
-kailash/nodes/data/async_sql.py,sha256=bb6Qc-QwoTJwzAOjwT_ooQikk4JWoS2ntb71O7KzXbw,99228
+kailash/nodes/data/async_sql.py,sha256=k4zgeHxgZLB06bl8Aos4WdFrHkYvhtAHX3en9wJv1Hw,103160
 kailash/nodes/data/async_vector.py,sha256=HtwQLO25IXu8Vq80qzU8rMkUAKPQ2qM0x8YxjXHlygU,21005
 kailash/nodes/data/bulk_operations.py,sha256=WVopmosVkIlweFxVt3boLdCPc93EqpYyQ1Ez9mCIt0c,34453
 kailash/nodes/data/directory.py,sha256=fbfLqD_ijRubk-4xew3604QntPsyDxqaF4k6TpfyjDg,9923
@@ -237,7 +241,7 @@ kailash/nodes/rag/workflows.py,sha256=PL9d3Lk15drwo2QVdNR9B61EpZ8Kc1crjnqFXc10O1
 kailash/nodes/security/__init__.py,sha256=oaoZdbKhjEZ8vtUHGRPZPDixInzFDmhUZg4ou_Lz6mI,611
 kailash/nodes/security/abac_evaluator.py,sha256=MAVJrIIQGHdOI-OfzGJFImVaHSq4vO8avEB0bSTprWk,50313
 kailash/nodes/security/audit_log.py,sha256=zjwLoWHJLhkJiD5rRmE8H2mSEE7-6EHh_FR0FT91uLU,3130
-kailash/nodes/security/behavior_analysis.py,sha256=4R2NNC3CKGnePAeVP1gjjyAL7ZjWDjsEXdHebRVNRig,71563
+kailash/nodes/security/behavior_analysis.py,sha256=zFR5Yny0C85JFnR4I4nfWgHX5PI1bmpB1i-cLZjSvq4,90007
 kailash/nodes/security/credential_manager.py,sha256=kqetmEBUnOOtSJyCpf0o_IRlwzpCv5FeInB4enBfk5E,14968
 kailash/nodes/security/rotating_credentials.py,sha256=yEDSraBi_M3DKAaodmkkVPXHreUvrLU8xuTky_Z-tT4,28928
 kailash/nodes/security/security_event.py,sha256=yVP8foaki-LDD16mlRapaPsbaWMwsum5J_V0n1bnqvw,4150
@@ -266,7 +270,7 @@ kailash/resources/health.py,sha256=i6XS0HdL6pUYq8SBdRjvchf-oj0sy3JoRLszNylfQJc,9
 kailash/resources/reference.py,sha256=RKfXzJFIdid0cLOwsXSmlXq4NhSEci4GO-q3M3qcjA8,7526
 kailash/resources/registry.py,sha256=5978e9VcUq0XBi9LRN89swptBSOAOAyNmwe2pxFHMxM,13789
 kailash/runtime/__init__.py,sha256=CvU-qBMESYYISqFOlYlLsYJrXJu0Gqr4x6yr4Ob_Rng,278
-kailash/runtime/access_controlled.py,sha256=Td8Fa8oPxEqlb74bDiH_HtT6J-ku0AudvN7mrUZOacc,17219
+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=-xqgydrSpg0eiC3MIVfWK3S6H4dUNUojrXFKttN_YGc,46422
@@ -274,13 +278,13 @@ kailash/runtime/parallel.py,sha256=mz_wPD13-YVc3Q_8HkOs4nPQPdTjnjCcnRL7ZRM70lo,2
 kailash/runtime/parallel_cyclic.py,sha256=yANZHnePjhCPuCFbq3lFQA1K6jbCv5Of5-vIKbCsmZk,19863
 kailash/runtime/parameter_injection.py,sha256=kG4GhmarsRr5t3VDFbc2G1HSbsZJg6UmienHCE2Ru7o,14852
 kailash/runtime/parameter_injector.py,sha256=4l-OOKEBzM_cQ17Zz5o24QSV9a1Zqfu3D2qyq6Uy7hE,30810
-kailash/runtime/runner.py,sha256=Lt9ugsk56ICD9XLKJjnhejtqdiz0mJNdlqH7sSxIlWU,3214
-kailash/runtime/testing.py,sha256=nHTYKdVCPZpyC5h2J6cvtgsG4lXEZnMYn7VtKDSlutI,19081
+kailash/runtime/runner.py,sha256=t_SEumtVn9uXlSzJicb50MpUu6xdqjW0uZ5b2phjVPY,3318
+kailash/runtime/testing.py,sha256=hTrgGnqxwSBYoztVqnZpxzFwm0DwUnaJdChRHikgoio,19089
 kailash/servers/__init__.py,sha256=-2nkB5pIkBjc-klGJcKCuzWRAl_G7hlepvyV_Es64DI,976
 kailash/servers/durable_workflow_server.py,sha256=lMnf-rTUKrjiS9NhNo3qx2lWYC0NR7EEF-4WBUPZ9zI,16588
-kailash/servers/enterprise_workflow_server.py,sha256=y0fFS1S7NuWpurCyMNtvjFUcFaiJwFWu8JN2oAv5rB8,17658
+kailash/servers/enterprise_workflow_server.py,sha256=HxArcgu5fQwV2KxKkf1n0Yu75zXy_tS8oxd4-1DM5UY,19785
 kailash/servers/gateway.py,sha256=faZTHsHL-3GwX-um3iLT4MaJmub2QIMNPnlkLAJe7Oc,6552
-kailash/servers/workflow_server.py,sha256=mqqYmpZBWpaKtME_6Ykclz3mCDRArjcRV1wGCrqnZZ0,9373
+kailash/servers/workflow_server.py,sha256=nkx8qpZ4xJD767L3M2dTj68SXv8uLUNht6tZ0v2eB5c,9538
 kailash/testing/__init__.py,sha256=4cqLSbcFz9bDY09vuK5Vpg3UtoAAa93R5tc9zIUvGAE,796
 kailash/testing/async_test_case.py,sha256=3Pdi13zT-_LduJ5Tfe7yuhz8Qz7KQpcER6G2nMe-t5A,12815
 kailash/testing/async_utils.py,sha256=gSHcArDJf98mDi--NPFpCoGmAgKHDZ0__tyKAjDjpm0,10745
@@ -312,8 +316,7 @@ kailash/visualization/reports.py,sha256=D7kJ0flHr16d-qSEq8vnw20N8u_dgTrXtKVSXVm8
 kailash/workflow/__init__.py,sha256=DDQDE9K6RmbX6479guNLLgjiVVV-gQERRvCEJWSVlsM,1836
 kailash/workflow/async_builder.py,sha256=iv8bDJHdWAUZ77SyMo6sucd92dTdtXesdxycrSE7mM4,20613
 kailash/workflow/async_patterns.py,sha256=X0ZDXwr6UAu0WC1xnCB7-0V1-tRbKs9UI4JqaBCB6tE,22824
-kailash/workflow/builder.py,sha256=9TaX9PUle5Cl4W8ksQMYdpR-P9XYLTAsMnujdZ94cPA,27335
-kailash/workflow/builder_improvements.py,sha256=MYAf-CJkrXTP8Q6leapnfVZATQxCJJ1cAymNOqXCh2w,7374
+kailash/workflow/builder.py,sha256=3l5wVRtMB1Om_QmeoQ6jsrc_Lm8Hna_pH47UjnfMb9I,31184
 kailash/workflow/convergence.py,sha256=vfIDR-uNaQE-LVUEzrRtfgKPgX9gL0nLNH-nTg5ra-c,10031
 kailash/workflow/cycle_analyzer.py,sha256=BGBpgdB-g0-KRI65sVAvHV4lxfoCzMt4uKOHbw8GXT4,32596
 kailash/workflow/cycle_builder.py,sha256=uWAx8K4ZKMtFC3mWylK4gHT03xeu0xChJlhw50hVqEE,20883
@@ -322,7 +325,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=NZqK_GTNbvf-pp08_brvzOlDCytrKscr1XRDy8PD8Is,40816
+kailash/workflow/cyclic_runner.py,sha256=0IlmghTrwYb3ivqP975DprP3aj45-8_Sn5Wq9tEutG0,43888
 kailash/workflow/graph.py,sha256=zRpGLXeuwtuxFBvE7_16c_bB9yqZirM_uwtfD1_MY4g,59272
 kailash/workflow/input_handling.py,sha256=HrW--AmelYC8F18nkfmYlF_wXycA24RuNbDRjvM8rqk,6561
 kailash/workflow/mermaid_visualizer.py,sha256=UsQDvxgIAhy-Th7ZzFnbuziaTx1Cd5yUh6S_25DffoQ,22294
@@ -334,10 +337,10 @@ kailash/workflow/safety.py,sha256=pS5GKu7UdkzFZcb16Dn-0jBxjULDU-59_M0CbUVMVyw,11
 kailash/workflow/state.py,sha256=UTZxs5-Ona6uvBhx1__i6-RX8gB4qazkBIWE7uyRmWQ,7600
 kailash/workflow/templates.py,sha256=98EN5H4fO9b4xeczk20Hu5L8hNcAuRQNGayT6vnZYCw,48540
 kailash/workflow/validation.py,sha256=rWZNJYA_XAfk_og6Cxz8p1gZ3j5CylPvMTDXg2SfjgM,38574
-kailash/workflow/visualization.py,sha256=ICMWCWqh5fOQ7eJygbvu2PMWHxe-H5_0epwdZuz8cMw,19737
-kailash-0.7.0.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
-kailash-0.7.0.dist-info/METADATA,sha256=V9gLCHDGwMu7vagtRV1DryE2sylSebDKkNWePmUqgAE,26409
-kailash-0.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kailash-0.7.0.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
-kailash-0.7.0.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
-kailash-0.7.0.dist-info/RECORD,,
+kailash/workflow/visualization.py,sha256=_oWiLrd6PjoAnNvCnQtFjxx2SVwhDeuYkuPz_ptIQvQ,23013
+kailash-0.8.0.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.8.0.dist-info/METADATA,sha256=xsd4eOqs4Lt3pigjT2SoOKnoQ0-P6YjSXf9zYu66LVY,26441
+kailash-0.8.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.8.0.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.8.0.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.8.0.dist-info/RECORD,,

kailash/workflow/builder_improvements.py

@@ -1,207 +0,0 @@
-"""
-WorkflowBuilder Improvements for Parameter Passing
-
-This module contains improvements to the WorkflowBuilder to handle
-parameter passing to nodes without incoming connections.
-"""
-
-from typing import Any, Dict, List, Optional
-
-from kailash.workflow.builder import WorkflowBuilder
-from kailash.workflow.graph import Workflow
-
-
-class ImprovedWorkflowBuilder(WorkflowBuilder):
-    """Enhanced WorkflowBuilder that automatically handles parameter passing"""
-
-    def __init__(self):
-        super().__init__()
-        self.workflow_parameters: Dict[str, Any] = {}
-        self.parameter_mappings: Dict[str, Dict[str, str]] = {}
-
-    def set_workflow_parameters(self, **parameters) -> "ImprovedWorkflowBuilder":
-        """
-        Set default parameters that will be passed to all nodes.
-
-        Args:
-            **parameters: Key-value pairs of workflow-level parameters
-
-        Returns:
-            Self for chaining
-        """
-        self.workflow_parameters.update(parameters)
-        return self
-
-    def add_parameter_mapping(
-        self, node_id: str, mappings: Dict[str, str]
-    ) -> "ImprovedWorkflowBuilder":
-        """
-        Add parameter mappings for a specific node.
-
-        Args:
-            node_id: Node to configure
-            mappings: Dict mapping workflow param names to node param names
-
-        Returns:
-            Self for chaining
-        """
-        if node_id not in self.parameter_mappings:
-            self.parameter_mappings[node_id] = {}
-        self.parameter_mappings[node_id].update(mappings)
-        return self
-
-    def add_input_connection(
-        self, to_node: str, to_input: str, from_workflow_param: str
-    ) -> "ImprovedWorkflowBuilder":
-        """
-        Connect a workflow parameter directly to a node input.
-
-        Args:
-            to_node: Target node ID
-            to_input: Input parameter name on the node
-            from_workflow_param: Workflow parameter name
-
-        Returns:
-            Self for chaining
-        """
-        # Add a special connection type for workflow inputs
-        connection = {
-            "from_node": "__workflow_input__",
-            "from_output": from_workflow_param,
-            "to_node": to_node,
-            "to_input": to_input,
-            "is_workflow_input": True,
-        }
-        self.connections.append(connection)
-        return self
-
-    def build(self, workflow_id: str | None = None, **kwargs) -> Workflow:
-        """
-        Build the workflow with automatic parameter injection.
-
-        Returns:
-            Enhanced Workflow instance
-        """
-        # First, build the base workflow
-        workflow = super().build(workflow_id, **kwargs)
-
-        # Find nodes without incoming connections
-        nodes_with_inputs = set()
-        for conn in self.connections:
-            if not conn.get("is_workflow_input"):
-                nodes_with_inputs.add(conn["to_node"])
-
-        nodes_without_inputs = set(self.nodes.keys()) - nodes_with_inputs
-
-        # For each node without inputs, check if it needs workflow parameters
-        for node_id in nodes_without_inputs:
-            node = self.nodes[node_id]
-            node_instance = workflow.get_node(node_id)
-
-            if hasattr(node_instance, "get_parameters"):
-                params = node_instance.get_parameters()
-
-                # Check which required parameters are missing from config
-                for param_name, param_def in params.items():
-                    if param_def.required and param_name not in node["config"]:
-                        # Check if this parameter should come from workflow parameters
-                        if param_name in self.workflow_parameters:
-                            # Add to node config
-                            node["config"][param_name] = self.workflow_parameters[
-                                param_name
-                            ]
-                        elif node_id in self.parameter_mappings:
-                            # Check parameter mappings
-                            mapping = self.parameter_mappings[node_id]
-                            if param_name in mapping:
-                                workflow_param = mapping[param_name]
-                                if workflow_param in self.workflow_parameters:
-                                    node["config"][param_name] = (
-                                        self.workflow_parameters[workflow_param]
-                                    )
-
-        # Store workflow parameters in metadata for runtime reference
-        workflow._metadata["workflow_parameters"] = self.workflow_parameters
-        workflow._metadata["parameter_mappings"] = self.parameter_mappings
-
-        return workflow
-
-
-def create_user_login_workflow_improved(config: Dict[str, Any]) -> Workflow:
-    """
-    Example of creating a login workflow with proper parameter handling.
-    """
-    workflow = ImprovedWorkflowBuilder()
-
-    # Set workflow-level parameters that will be shared
-    workflow.set_workflow_parameters(
-        tenant_id="default", database_config=config["database_config"]
-    )
-
-    # Add user fetcher node
-    workflow.add_node(
-        "UserManagementNode",
-        "user_fetcher",
-        {
-            "operation": "get_user",
-            "identifier": "$.email",
-            "identifier_type": "email",
-            # tenant_id and database_config will be auto-injected
-        },
-    )
-
-    # Map workflow inputs to the first node
-    workflow.add_input_connection("user_fetcher", "email", "email")
-    workflow.add_input_connection("user_fetcher", "password", "password")
-
-    # Add other nodes...
-    workflow.add_node(
-        "PythonCodeNode",
-        "password_verifier",
-        {"code": "# Password verification code here"},
-    )
-
-    # Connect nodes
-    workflow.add_connection("user_fetcher", "result", "password_verifier", "input")
-
-    return workflow.build(name="user_login_improved")
-
-
-# Alternative approach: Fix in the existing WorkflowBuilder
-def patch_workflow_builder():
-    """
-    Monkey patch the existing WorkflowBuilder to handle parameters better.
-    """
-    original_build = WorkflowBuilder.build
-
-    def enhanced_build(self, workflow_id: str | None = None, **kwargs) -> Workflow:
-        # Build the workflow normally
-        workflow = original_build(self, workflow_id, **kwargs)
-
-        # Enhanced parameter handling
-        # Find nodes without incoming connections and inject common parameters
-        nodes_with_inputs = set()
-        for edge in workflow._graph.edges():
-            nodes_with_inputs.add(edge[1])  # target node
-
-        # Get all nodes
-        all_nodes = set(workflow._graph.nodes())
-        nodes_without_inputs = all_nodes - nodes_with_inputs
-
-        # Common parameters that should be injected
-        common_params = {
-            "tenant_id": "default",
-            "database_config": kwargs.get("database_config", {}),
-        }
-
-        for node_id in nodes_without_inputs:
-            if node_id in workflow._nodes:
-                node_instance = workflow._nodes[node_id]
-                # Update node config with common parameters if not already set
-                for param, value in common_params.items():
-                    if param not in node_instance.config:
-                        node_instance.config[param] = value
-
-        return workflow
-
-    WorkflowBuilder.build = enhanced_build