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

kailash/__init__.py

@@ -52,7 +52,7 @@ except ImportError:
 # For backward compatibility
 WorkflowGraph = Workflow
 
-__version__ = "0.8.0"
+__version__ = "0.8.1"
 
 __all__ = [
     # Core workflow components

kailash/middleware/communication/realtime.py

@@ -177,87 +177,108 @@ class ConnectionManager:
             "active_users": len(self.user_connections),
             "total_messages_sent": total_messages,
         }
-    
-    def filter_events(self, events: List[BaseEvent], event_filter: EventFilter = None) -> List[BaseEvent]:
+
+    def filter_events(
+        self, events: List[BaseEvent], event_filter: EventFilter = None
+    ) -> List[BaseEvent]:
         """Filter events based on event filter criteria."""
         if not event_filter:
             return events
-        
+
         filtered = []
         for event in events:
             # Apply session filter
-            if event_filter.session_id and hasattr(event, 'session_id'):
+            if event_filter.session_id and hasattr(event, "session_id"):
                 if event.session_id != event_filter.session_id:
                     continue
-            
+
             # Apply user filter
-            if event_filter.user_id and hasattr(event, 'user_id'):
+            if event_filter.user_id and hasattr(event, "user_id"):
                 if event.user_id != event_filter.user_id:
                     continue
-            
+
             # Apply event type filter
-            if event_filter.event_types and event.event_type not in event_filter.event_types:
+            if (
+                event_filter.event_types
+                and event.event_type not in event_filter.event_types
+            ):
                 continue
-            
+
             filtered.append(event)
-        
+
         return filtered
-    
+
     def set_event_filter(self, connection_id: str, event_filter: EventFilter):
         """Set event filter for a specific connection."""
         if connection_id in self.connections:
             self.connections[connection_id]["event_filter"] = event_filter
-    
+
     def get_event_filter(self, connection_id: str) -> Optional[EventFilter]:
         """Get event filter for a specific connection."""
         if connection_id in self.connections:
             return self.connections[connection_id].get("event_filter")
         return None
-    
+
     # Alias methods for compatibility
-    def event_filter(self, events: List[BaseEvent], filter_criteria: EventFilter = None) -> List[BaseEvent]:
+    def event_filter(
+        self, events: List[BaseEvent], filter_criteria: EventFilter = None
+    ) -> List[BaseEvent]:
         """Alias for filter_events method."""
         return self.filter_events(events, filter_criteria)
-    
+
     async def on_event(self, event: BaseEvent):
         """Handle incoming event - route to appropriate connections."""
         await self.handle_event(event)
-    
+
     async def handle_event(self, event: BaseEvent):
         """Handle and route event to matching connections."""
         await self.process_event(event)
-    
+
     async def process_event(self, event: BaseEvent):
         """Process event and broadcast to matching connections."""
         message = {
             "type": "event",
-            "event_type": event.event_type.value if hasattr(event.event_type, 'value') else str(event.event_type),
+            "event_type": (
+                event.event_type.value
+                if hasattr(event.event_type, "value")
+                else str(event.event_type)
+            ),
             "data": event.data,
-            "timestamp": event.timestamp.isoformat() if hasattr(event, 'timestamp') else datetime.now(timezone.utc).isoformat(),
-            "session_id": getattr(event, 'session_id', None),
-            "user_id": getattr(event, 'user_id', None),
+            "timestamp": (
+                event.timestamp.isoformat()
+                if hasattr(event, "timestamp")
+                else datetime.now(timezone.utc).isoformat()
+            ),
+            "session_id": getattr(event, "session_id", None),
+            "user_id": getattr(event, "user_id", None),
         }
-        
+
         # Broadcast to all matching connections
         for connection_id, connection in self.connections.items():
             event_filter = connection.get("event_filter")
-            
+
             # Check if this connection should receive this event
             should_send = True
             if event_filter:
                 # Apply session filter
-                if event_filter.session_id and connection["session_id"] != event_filter.session_id:
+                if (
+                    event_filter.session_id
+                    and connection["session_id"] != event_filter.session_id
+                ):
                     should_send = False
-                
+
                 # Apply user filter
-                if event_filter.user_id and connection["user_id"] != event_filter.user_id:
+                if (
+                    event_filter.user_id
+                    and connection["user_id"] != event_filter.user_id
+                ):
                     should_send = False
-                
+
                 # Apply event type filter
-                if hasattr(event_filter, 'event_types') and event_filter.event_types:
+                if hasattr(event_filter, "event_types") and event_filter.event_types:
                     if event.event_type not in event_filter.event_types:
                         should_send = False
-            
+
             if should_send:
                 await self.send_to_connection(connection_id, message)
 

kailash/nodes/code/python.py

@@ -50,6 +50,7 @@ import ast
 import importlib.util
 import inspect
 import logging
+import os
 import resource
 import traceback
 from collections.abc import Callable
@@ -465,6 +466,10 @@ class CodeExecutor:
             # Normal operation - eagerly load all modules
             for module_name in self.allowed_modules:
                 try:
+                    # Skip scipy in CI due to version conflicts
+                    if module_name == "scipy" and os.environ.get("CI"):
+                        logger.warning("Skipping scipy import in CI environment")
+                        continue
                     module = importlib.import_module(module_name)
                     namespace[module_name] = module
                 except ImportError:

kailash/workflow/visualization.py

@@ -70,7 +70,7 @@ class WorkflowVisualizer:
         """Get colors for all nodes in workflow."""
         colors = []
         for node_id in self.workflow.graph.nodes():
-            node_instance = target_workflow.nodes.get(node_id)
+            node_instance = self.workflow.nodes.get(node_id)
             if node_instance:
                 node_type = node_instance.node_type
                 colors.append(self._get_node_color(node_type))
@@ -122,12 +122,14 @@ class WorkflowVisualizer:
 
         return edge_labels
 
-    def _calculate_layout(self, workflow: 'Workflow' = None) -> 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 target_workflow.nodes.items():
@@ -154,8 +156,10 @@ class WorkflowVisualizer:
                 pos = nx.spring_layout(target_workflow.graph)
 
         return pos
-    
-    def _get_layout_positions(self, workflow: Workflow) -> dict[str, tuple[float, float]]:
+
+    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
@@ -164,7 +168,7 @@ class WorkflowVisualizer:
             return self._calculate_layout()
         finally:
             self.workflow = original_workflow
-    
+
     def _get_node_colors(self, workflow: Workflow) -> list[str]:
         """Get node colors for workflow."""
         colors = []
@@ -183,17 +187,19 @@ class WorkflowVisualizer:
                     color_key = "ai"
                 else:
                     color_key = "default"
-                colors.append(self.node_colors.get(color_key, self.node_colors["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, workflow: 'Workflow' = None) -> 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(target_workflow.graph.nodes())
         layer = 0
@@ -251,18 +257,22 @@ class WorkflowVisualizer:
         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(
-            target_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
@@ -284,7 +294,7 @@ class WorkflowVisualizer:
             self.workflow = target_workflow
             labels = self._get_node_labels()
             self.workflow = old_workflow
-            
+
             nx.draw_networkx_labels(
                 target_workflow.graph, pos, labels, font_size=10, font_weight="bold"
             )
@@ -296,7 +306,7 @@ class WorkflowVisualizer:
             self.workflow = target_workflow
             edge_labels = self._get_edge_labels()
             self.workflow = old_workflow
-            
+
             nx.draw_networkx_edge_labels(
                 target_workflow.graph, pos, edge_labels, font_size=8
             )
@@ -325,8 +335,10 @@ class WorkflowVisualizer:
         try:
             # Check if workflow is available
             if not self.workflow:
-                raise ValueError("No workflow to visualize. Set workflow property or create visualizer with workflow.")
-                
+                raise ValueError(
+                    "No workflow to visualize. Set workflow property or create visualizer with workflow."
+                )
+
             plt.figure(figsize=figsize)
 
             # Calculate node positions
@@ -342,7 +354,13 @@ class WorkflowVisualizer:
 
             # Draw the graph components
             if pos:
-                self._draw_graph(workflow=self.workflow, pos=pos, node_colors=node_colors, show_labels=show_labels, show_connections=show_connections)
+                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.8.0.dist-info → kailash-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.8.0
+Version: 0.8.1
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum
@@ -58,6 +58,7 @@ Requires-Dist: python-jose>=3.5.0
 Requires-Dist: pytest-xdist>=3.6.0
 Requires-Dist: pytest-timeout>=2.3.0
 Requires-Dist: pytest-split>=0.9.0
+Requires-Dist: pytest-forked>=1.6.0
 Requires-Dist: asyncpg>=0.30.0
 Requires-Dist: aiomysql>=0.2.0
 Requires-Dist: twilio>=9.6.3

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

@@ -1,4 +1,4 @@
-kailash/__init__.py,sha256=yCfV0dNXfj0xBIsqAbdMl31CxaSma9AtltRW6rI2AhU,2878
+kailash/__init__.py,sha256=_Mwxl0OoLmLW4_EJ4sxICIUWNEA2gG4iFinzIxnweFk,2878
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
 kailash/access_control.py,sha256=MjKtkoQ2sg1Mgfe7ovGxVwhAbpJKvaepPWr8dxOueMA,26058
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
@@ -85,7 +85,7 @@ kailash/middleware/communication/__init__.py,sha256=keQ2db4WI2-oZ_nJ5sLE1Tum_RkU
 kailash/middleware/communication/ai_chat.py,sha256=2XmnJB2Nz2xm2darsF2DYnSLGLAYdyYooHok5GrHKb8,35994
 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=xm0i-TUs82XUtnFfHpdeZcx4uMt_V-fmS_uTuvXqDV0,28647
+kailash/middleware/communication/realtime.py,sha256=lg2SCASOfEk00aoXq6lCYScpb7YZYCpmLKLxiE3RHAA,28827
 kailash/middleware/core/__init__.py,sha256=4yQkOWC4b88zSogs1YVqtKG1PugbncqNCwNNWxTdIZA,545
 kailash/middleware/core/agent_ui.py,sha256=Ki5QpsYNM3o3h9KGor4YEXs9Xg8ehXe3w_M8SJnYhxI,36358
 kailash/middleware/core/schema.py,sha256=uVF-5ZJlLYHOQdsKrG46FnTO1bq_QtDjhUSkIIDL9dY,23584
@@ -166,7 +166,7 @@ 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=u-5lSoxc5ul7zGGSanrcVwXLxlGcDlirdCXT6mLSm40,60331
+kailash/nodes/code/python.py,sha256=AfKNL4B-OTiFQoNuCzZ2mLiJV61TaHLXhHmXudl20eg,60592
 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
@@ -337,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=_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/visualization.py,sha256=nHBW-Ai8QBMZtn2Nf3EE1_aiMGi9S6Ui_BfpA5KbJPU,23187
+kailash-0.8.1.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.8.1.dist-info/METADATA,sha256=KBxycJ3rYfE1ygfMR8LKrzA0AoKo1C37cU3fM66UXRw,26477
+kailash-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.8.1.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.8.1.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.8.1.dist-info/RECORD,,