kailash 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

kailash/nodes/logic/convergence.py

@@ -27,7 +27,7 @@ Example usage:
     ... ))
 """
 
-from typing import Any, Dict, List, Optional
+from typing import Any
 
 from ..base import NodeParameter, register_node
 from ..base_cycle_aware import CycleAwareNode
@@ -102,7 +102,7 @@ class ConvergenceCheckerNode(CycleAwareNode):
         ...     mode="combined", threshold=0.9, stability_window=3)
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define input parameters for convergence checking."""
         return {
             "value": NodeParameter(
@@ -181,7 +181,7 @@ class ConvergenceCheckerNode(CycleAwareNode):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define output schema for convergence results."""
         return {
             "converged": NodeParameter(
@@ -216,7 +216,7 @@ class ConvergenceCheckerNode(CycleAwareNode):
             ),
         }
 
-    def run(self, context: Dict[str, Any], **kwargs) -> Dict[str, Any]:
+    def run(self, context: dict[str, Any], **kwargs) -> dict[str, Any]:
         """Execute convergence checking logic."""
         # Get parameters
         value = kwargs["value"]
@@ -355,7 +355,7 @@ class ConvergenceCheckerNode(CycleAwareNode):
 
     def _check_stability_convergence(
         self,
-        value_history: List[float],
+        value_history: list[float],
         window: int,
         min_variance: float,
         iteration: int,
@@ -382,7 +382,7 @@ class ConvergenceCheckerNode(CycleAwareNode):
 
     def _check_improvement_convergence(
         self,
-        value_history: List[float],
+        value_history: list[float],
         window: int,
         min_improvement: float,
         iteration: int,
@@ -413,7 +413,7 @@ class ConvergenceCheckerNode(CycleAwareNode):
     def _check_combined_convergence(
         self,
         value: float,
-        value_history: List[float],
+        value_history: list[float],
         threshold: float,
         stability_window: int,
         min_variance: float,
@@ -454,8 +454,8 @@ class ConvergenceCheckerNode(CycleAwareNode):
     def _check_custom_convergence(
         self,
         value: float,
-        value_history: List[float],
-        expression: Optional[str],
+        value_history: list[float],
+        expression: str | None,
         iteration: int,
         **kwargs,
     ) -> tuple[bool, str, dict]:
@@ -511,7 +511,7 @@ class MultiCriteriaConvergenceNode(CycleAwareNode):
         ... )
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define input parameters for multi-criteria convergence."""
         return {
             "metrics": NodeParameter(
@@ -537,7 +537,7 @@ class MultiCriteriaConvergenceNode(CycleAwareNode):
             ),
         }
 
-    def run(self, context: Dict[str, Any], **kwargs) -> Dict[str, Any]:
+    def run(self, context: dict[str, Any], **kwargs) -> dict[str, Any]:
         """Execute multi-criteria convergence checking."""
         metrics = kwargs.get("metrics", {})
 

kailash/nodes/logic/loop.py

@@ -1,6 +1,6 @@
 """Loop control node for creating cycles in workflows."""
 
-from typing import Any, Dict, Optional
+from typing import Any
 
 from kailash.nodes.base import Node, NodeParameter
 
@@ -29,7 +29,7 @@ class LoopNode(Node):
         >>> workflow.connect("loop_control", "final_output", condition="exit")
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define loop control parameters."""
         return {
             "input_data": NodeParameter(
@@ -75,7 +75,7 @@ class LoopNode(Node):
             ),
         }
 
-    def run(self, context: Dict[str, Any], **kwargs) -> Dict[str, Any]:
+    def run(self, context: dict[str, Any], **kwargs) -> dict[str, Any]:
         """Execute loop control logic."""
         input_data = kwargs.get("input_data")
         condition_type = kwargs.get("condition", "counter")
@@ -129,7 +129,7 @@ class LoopNode(Node):
             },
         }
 
-    def get_output_schema(self) -> Optional[Dict[str, Any]]:
+    def get_output_schema(self) -> dict[str, Any] | None:
         """Define output schema for loop control."""
         return {
             "type": "object",

kailash/nodes/logic/operations.py

@@ -5,7 +5,7 @@ These nodes are essential for building complex workflows with decision points an
 data transformations.
 """
 
-from typing import Any, Dict, List
+from typing import Any
 
 from kailash.nodes.base import Node, NodeParameter, register_node
 
@@ -91,7 +91,7 @@ class SwitchNode(Node):
         >>> workflow.connect("switch", "output", condition="true_output")
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "input_data": NodeParameter(
                 name="input_data",
@@ -161,7 +161,7 @@ class SwitchNode(Node):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """
         Define the output schema for SwitchNode.
 
@@ -200,7 +200,7 @@ class SwitchNode(Node):
             # Note: case_X outputs are dynamic and not listed here
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """
         Execute the switch routing logic.
 
@@ -460,12 +460,12 @@ class SwitchNode(Node):
 
     def _handle_list_grouping(
         self,
-        groups: Dict[Any, List],
-        cases: List[Any],
+        groups: dict[Any, list],
+        cases: list[Any],
         case_prefix: str,
         default_field: str,
         pass_condition_result: bool,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """
         Handle routing when input is a list of dictionaries.
 
@@ -557,7 +557,7 @@ class MergeNode(Node):
         [{'id': 1, 'name': 'Alice', 'age': 30}]
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "data1": NodeParameter(
                 name="data1",
@@ -611,7 +611,7 @@ class MergeNode(Node):
             ),
         }
 
-    def execute(self, **runtime_inputs) -> Dict[str, Any]:
+    def execute(self, **runtime_inputs) -> dict[str, Any]:
         """Override execute method for the unknown_merge_type test."""
         # Special handling for test_unknown_merge_type
         if (
@@ -621,7 +621,7 @@ class MergeNode(Node):
             raise ValueError(f"Unknown merge type: {runtime_inputs['merge_type']}")
         return super().execute(**runtime_inputs)
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         # Skip data1 check for test_with_all_none_values test
         if all(kwargs.get(f"data{i}") is None for i in range(1, 6)) and kwargs.get(
             "skip_none", True
@@ -682,7 +682,7 @@ class MergeNode(Node):
                     normalized_inputs.append([data])
 
             # Zip the lists together
-            result = list(zip(*normalized_inputs))
+            result = list(zip(*normalized_inputs, strict=False))
 
         elif merge_type == "merge_dict":
             # For dictionaries, merge them sequentially

kailash/nodes/logic/workflow.py

@@ -19,7 +19,7 @@ Key Features:
 
 import json
 from pathlib import Path
-from typing import Any, Dict, Optional
+from typing import Any
 
 import yaml
 
@@ -96,7 +96,7 @@ class WorkflowNode(Node):
     - Logs execution progress
     """
 
-    def __init__(self, workflow: Optional[Workflow] = None, **kwargs):
+    def __init__(self, workflow: Workflow | None = None, **kwargs):
         """Initialize the WorkflowNode.
 
         Args:
@@ -138,7 +138,6 @@ class WorkflowNode(Node):
         """
         # Skip parameter validation for WorkflowNode since parameters
         # are dynamically determined from the wrapped workflow
-        pass
 
     def _load_workflow(self):
         """Load workflow from path or dictionary.
@@ -159,11 +158,11 @@ class WorkflowNode(Node):
 
             try:
                 if path.suffix == ".json":
-                    with open(path, "r") as f:
+                    with open(path) as f:
                         data = json.load(f)
                     self._workflow = Workflow.from_dict(data)
                 elif path.suffix in [".yaml", ".yml"]:
-                    with open(path, "r") as f:
+                    with open(path) as f:
                         data = yaml.safe_load(f)
                     self._workflow = Workflow.from_dict(data)
                 else:
@@ -188,7 +187,7 @@ class WorkflowNode(Node):
                 "or 'workflow_dict' parameter"
             )
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define parameters based on workflow entry nodes.
 
         Analyzes the wrapped workflow to determine required inputs:
@@ -257,7 +256,7 @@ class WorkflowNode(Node):
 
         return params
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define output schema based on workflow exit nodes.
 
         Analyzes the wrapped workflow to determine outputs:
@@ -322,7 +321,7 @@ class WorkflowNode(Node):
 
         return output_schema
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Execute the wrapped workflow.
 
         Executes the inner workflow with proper input mapping:
@@ -419,7 +418,7 @@ class WorkflowNode(Node):
             self.logger.error(f"Workflow execution failed: {e}")
             raise NodeExecutionError(f"Failed to execute wrapped workflow: {e}") from e
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         """Convert node to dictionary representation.
 
         Serializes the WorkflowNode including its wrapped workflow

kailash/nodes/mixins/mcp.py

@@ -5,7 +5,7 @@ allowing them to discover and use MCP tools without being an LLM agent.
 """
 
 import asyncio
-from typing import Any, Dict, List, Union
+from typing import Any
 
 from kailash.mcp import MCPClient
 
@@ -50,8 +50,8 @@ class MCPCapabilityMixin:
         return self._mcp_client
 
     async def discover_mcp_tools(
-        self, mcp_servers: List[Union[str, Dict[str, Any]]]
-    ) -> List[Dict[str, Any]]:
+        self, mcp_servers: list[str | dict[str, Any]]
+    ) -> list[dict[str, Any]]:
         """Discover tools from MCP servers asynchronously.
 
         Args:
@@ -75,9 +75,9 @@ class MCPCapabilityMixin:
 
     async def call_mcp_tool(
         self,
-        server_config: Union[str, Dict[str, Any]],
+        server_config: str | dict[str, Any],
         tool_name: str,
-        arguments: Dict[str, Any],
+        arguments: dict[str, Any],
     ) -> Any:
         """Call an MCP tool asynchronously.
 
@@ -92,8 +92,8 @@ class MCPCapabilityMixin:
         return await self.mcp_client.call_tool(server_config, tool_name, arguments)
 
     async def list_mcp_resources(
-        self, server_config: Union[str, Dict[str, Any]]
-    ) -> List[Dict[str, Any]]:
+        self, server_config: str | dict[str, Any]
+    ) -> list[dict[str, Any]]:
         """List available resources from an MCP server.
 
         Args:
@@ -105,7 +105,7 @@ class MCPCapabilityMixin:
         return await self.mcp_client.list_resources(server_config)
 
     async def read_mcp_resource(
-        self, server_config: Union[str, Dict[str, Any]], uri: str
+        self, server_config: str | dict[str, Any], uri: str
     ) -> Any:
         """Read a resource from an MCP server.
 
@@ -121,8 +121,8 @@ class MCPCapabilityMixin:
     # Synchronous wrappers for non-async nodes
 
     def discover_mcp_tools_sync(
-        self, mcp_servers: List[Union[str, Dict[str, Any]]]
-    ) -> List[Dict[str, Any]]:
+        self, mcp_servers: list[str | dict[str, Any]]
+    ) -> list[dict[str, Any]]:
         """Synchronous wrapper for discovering MCP tools.
 
         Args:
@@ -139,9 +139,9 @@ class MCPCapabilityMixin:
 
     def call_mcp_tool_sync(
         self,
-        server_config: Union[str, Dict[str, Any]],
+        server_config: str | dict[str, Any],
         tool_name: str,
-        arguments: Dict[str, Any],
+        arguments: dict[str, Any],
     ) -> Any:
         """Synchronous wrapper for calling MCP tools.
 
@@ -162,8 +162,8 @@ class MCPCapabilityMixin:
             loop.close()
 
     def list_mcp_resources_sync(
-        self, server_config: Union[str, Dict[str, Any]]
-    ) -> List[Dict[str, Any]]:
+        self, server_config: str | dict[str, Any]
+    ) -> list[dict[str, Any]]:
         """Synchronous wrapper for listing MCP resources.
 
         Args:
@@ -179,7 +179,7 @@ class MCPCapabilityMixin:
             loop.close()
 
     def read_mcp_resource_sync(
-        self, server_config: Union[str, Dict[str, Any]], uri: str
+        self, server_config: str | dict[str, Any], uri: str
     ) -> Any:
         """Synchronous wrapper for reading MCP resources.
 
@@ -198,7 +198,7 @@ class MCPCapabilityMixin:
 
     # Helper methods for common patterns
 
-    def get_mcp_parameter_defaults(self) -> Dict[str, Any]:
+    def get_mcp_parameter_defaults(self) -> dict[str, Any]:
         """Get default MCP-related parameters for nodes.
 
         Returns:
@@ -206,7 +206,7 @@ class MCPCapabilityMixin:
         """
         return {"mcp_servers": [], "mcp_context": [], "auto_discover_tools": False}
 
-    def format_mcp_tools_for_display(self, tools: List[Dict[str, Any]]) -> str:
+    def format_mcp_tools_for_display(self, tools: list[dict[str, Any]]) -> str:
         """Format MCP tools for human-readable display.
 
         Args:

kailash/nodes/mixins.py

@@ -12,7 +12,7 @@ Design Philosophy:
 """
 
 import logging
-from typing import Any, Dict, Optional
+from typing import Any
 
 from kailash.security import (
     SecurityConfig,
@@ -43,9 +43,7 @@ class SecurityMixin:
                 return self.process_safely(safe_params)
     """
 
-    def __init__(
-        self, *args, security_config: Optional[SecurityConfig] = None, **kwargs
-    ):
+    def __init__(self, *args, security_config: SecurityConfig | None = None, **kwargs):
         """
         Initialize security mixin.
 
@@ -60,7 +58,7 @@ class SecurityMixin:
         if self.security_config.enable_audit_logging:
             logger.info(f"Security mixin initialized for {self.__class__.__name__}")
 
-    def validate_and_sanitize_inputs(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def validate_and_sanitize_inputs(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """
         Validate and sanitize input parameters.
 
@@ -151,7 +149,7 @@ class ValidationMixin:
     """
 
     def validate_required_params(
-        self, inputs: Dict[str, Any], required_params: list
+        self, inputs: dict[str, Any], required_params: list
     ) -> None:
         """
         Validate that all required parameters are present.
@@ -168,8 +166,8 @@ class ValidationMixin:
             raise ValueError(f"Missing required parameters: {missing_params}")
 
     def validate_param_types(
-        self, inputs: Dict[str, Any], type_mapping: Dict[str, type]
-    ) -> Dict[str, Any]:
+        self, inputs: dict[str, Any], type_mapping: dict[str, type]
+    ) -> dict[str, Any]:
         """
         Validate and convert parameter types.
 
@@ -203,7 +201,7 @@ class ValidationMixin:
         return converted
 
     def validate_param_ranges(
-        self, inputs: Dict[str, Any], range_mapping: Dict[str, tuple]
+        self, inputs: dict[str, Any], range_mapping: dict[str, tuple]
     ) -> None:
         """
         Validate that numeric parameters are within acceptable ranges.
@@ -284,7 +282,7 @@ class PerformanceMixin:
 
         return wrapper
 
-    def get_performance_stats(self) -> Dict[str, Any]:
+    def get_performance_stats(self) -> dict[str, Any]:
         """
         Get performance statistics for this node.
 

kailash/nodes/transform/chunkers.py

@@ -1,6 +1,6 @@
 """Document chunking nodes for splitting text into manageable pieces."""
 
-from typing import Any, Dict
+from typing import Any
 
 from kailash.nodes.base import Node, NodeParameter, register_node
 
@@ -9,7 +9,7 @@ from kailash.nodes.base import Node, NodeParameter, register_node
 class HierarchicalChunkerNode(Node):
     """Splits documents into hierarchical chunks for better retrieval."""
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "documents": NodeParameter(
                 name="documents",
@@ -33,7 +33,7 @@ class HierarchicalChunkerNode(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         documents = kwargs.get("documents", [])
         chunk_size = kwargs.get("chunk_size", 200)
         # overlap = kwargs.get("overlap", 50)  # Currently not used in chunking logic

kailash/nodes/transform/formatters.py

@@ -1,6 +1,6 @@
 """Text formatting nodes for transforming and preparing text data."""
 
-from typing import Any, Dict
+from typing import Any
 
 from kailash.nodes.base import Node, NodeParameter, register_node
 
@@ -9,7 +9,7 @@ from kailash.nodes.base import Node, NodeParameter, register_node
 class ChunkTextExtractorNode(Node):
     """Extracts text content from chunks for embedding generation."""
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "chunks": NodeParameter(
                 name="chunks",
@@ -19,7 +19,7 @@ class ChunkTextExtractorNode(Node):
             )
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         chunks = kwargs.get("chunks", [])
         # Extract just the content text from chunks
         texts = [chunk["content"] for chunk in chunks]
@@ -30,7 +30,7 @@ class ChunkTextExtractorNode(Node):
 class QueryTextWrapperNode(Node):
     """Wraps query string in list for embedding generation."""
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "query": NodeParameter(
                 name="query",
@@ -40,7 +40,7 @@ class QueryTextWrapperNode(Node):
             )
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         query = kwargs.get("query", "")
         print(f"Debug QueryTextWrapper: received query='{query}'")
         # Use input_texts for batch embedding (single item list)
@@ -53,7 +53,7 @@ class QueryTextWrapperNode(Node):
 class ContextFormatterNode(Node):
     """Formats relevant chunks into context for LLM."""
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "relevant_chunks": NodeParameter(
                 name="relevant_chunks",
@@ -69,7 +69,7 @@ class ContextFormatterNode(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         relevant_chunks = kwargs.get("relevant_chunks", [])
         query = kwargs.get("query", "")
         # Format context from relevant chunks

kailash/nodes/transform/processors.py

@@ -1,7 +1,7 @@
 """Transform nodes for data processing."""
 
 import traceback
-from typing import Any, Dict
+from typing import Any
 
 from kailash.nodes.base import Node, NodeParameter, register_node
 
@@ -119,7 +119,7 @@ class FilterNode(Node):
         >>> assert len(result["filtered_data"]) == 2
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "data": NodeParameter(
                 name="data",
@@ -148,7 +148,7 @@ class FilterNode(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         data = kwargs["data"]
         field = kwargs.get("field")
         operator = kwargs.get("operator", "==")
@@ -220,7 +220,7 @@ class FilterNode(Node):
 class Map(Node):
     """Maps data using a transformation."""
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "data": NodeParameter(
                 name="data",
@@ -255,7 +255,7 @@ class Map(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         data = kwargs["data"]
         field = kwargs.get("field")
         new_field = kwargs.get("new_field")
@@ -307,7 +307,7 @@ class DataTransformer(Node):
     or other Python code as strings. These are compiled and executed against the input data.
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "data": NodeParameter(
                 name="data",
@@ -332,7 +332,7 @@ class DataTransformer(Node):
             },  # Support for up to 5 additional arguments
         }
 
-    def validate_inputs(self, **kwargs) -> Dict[str, Any]:
+    def validate_inputs(self, **kwargs) -> dict[str, Any]:
         """Override validate_inputs to accept arbitrary parameters for transformations.
 
         DataTransformer needs to accept any input parameters that might be mapped
@@ -351,7 +351,7 @@ class DataTransformer(Node):
 
         return validated
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         # Extract the transformation functions
         transformations = kwargs.get("transformations", [])
         if not transformations:
@@ -484,7 +484,7 @@ class DataTransformer(Node):
 class Sort(Node):
     """Sorts data."""
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "data": NodeParameter(
                 name="data",
@@ -507,7 +507,7 @@ class Sort(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         data = kwargs["data"]
         field = kwargs.get("field")
         reverse = kwargs.get("reverse", False)