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

kailash/workflow/state.py

@@ -6,7 +6,7 @@ making it easier to handle state transitions in a predictable manner.
 
 import logging
 from copy import deepcopy
-from typing import Any, Generic, List, Tuple, TypeVar
+from typing import Any, Generic, TypeVar
 
 from pydantic import BaseModel
 
@@ -24,7 +24,7 @@ class StateManager:
     """
 
     @staticmethod
-    def update_in(state_obj: BaseModel, path: List[str], value: Any) -> BaseModel:
+    def update_in(state_obj: BaseModel, path: list[str], value: Any) -> BaseModel:
         """Update a nested property in the state and return a new state object.
 
         Args:
@@ -61,10 +61,7 @@ class StateManager:
             if isinstance(next_obj, BaseModel):
                 next_obj = next_obj.model_copy(deep=True)
                 setattr(current, key, next_obj)
-            elif isinstance(next_obj, dict):
-                next_obj = deepcopy(next_obj)
-                setattr(current, key, next_obj)
-            elif isinstance(next_obj, list):
+            elif isinstance(next_obj, dict) or isinstance(next_obj, list):
                 next_obj = deepcopy(next_obj)
                 setattr(current, key, next_obj)
 
@@ -80,7 +77,7 @@ class StateManager:
 
     @staticmethod
     def batch_update(
-        state_obj: BaseModel, updates: List[Tuple[List[str], Any]]
+        state_obj: BaseModel, updates: list[tuple[list[str], Any]]
     ) -> BaseModel:
         """Apply multiple updates to the state atomically.
 
@@ -108,7 +105,7 @@ class StateManager:
         return new_state
 
     @staticmethod
-    def get_in(state_obj: BaseModel, path: List[str]) -> Any:
+    def get_in(state_obj: BaseModel, path: list[str]) -> Any:
         """Get the value at a nested path.
 
         Args:
@@ -175,7 +172,7 @@ class WorkflowStateWrapper(Generic[StateT]):
         """
         self._state = state
 
-    def update_in(self, path: List[str], value: Any) -> "WorkflowStateWrapper[StateT]":
+    def update_in(self, path: list[str], value: Any) -> "WorkflowStateWrapper[StateT]":
         """Update state at path and return new wrapper.
 
         Args:
@@ -189,7 +186,7 @@ class WorkflowStateWrapper(Generic[StateT]):
         return WorkflowStateWrapper(new_state)
 
     def batch_update(
-        self, updates: List[Tuple[List[str], Any]]
+        self, updates: list[tuple[list[str], Any]]
     ) -> "WorkflowStateWrapper[StateT]":
         """Apply multiple updates to the state atomically.
 
@@ -202,7 +199,7 @@ class WorkflowStateWrapper(Generic[StateT]):
         new_state = StateManager.batch_update(self._state, updates)
         return WorkflowStateWrapper(new_state)
 
-    def get_in(self, path: List[str]) -> Any:
+    def get_in(self, path: list[str]) -> Any:
         """Get the value at a nested path.
 
         Args:

kailash/workflow/templates.py

@@ -113,7 +113,7 @@ See Also:
 import math
 import time
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional
+from typing import Any
 
 from ..nodes.code import PythonCodeNode
 from . import Workflow
@@ -125,11 +125,11 @@ class CycleTemplate:
 
     name: str
     description: str
-    nodes: List[str]
-    convergence_condition: Optional[str] = None
+    nodes: list[str]
+    convergence_condition: str | None = None
     max_iterations: int = 100
-    timeout: Optional[float] = None
-    parameters: Optional[Dict[str, Any]] = None
+    timeout: float | None = None
+    parameters: dict[str, Any] | None = None
 
 
 class CycleTemplates:
@@ -142,7 +142,7 @@ class CycleTemplates:
         evaluator_node: str,
         convergence: str = "quality > 0.9",
         max_iterations: int = 50,
-        cycle_id: Optional[str] = None,
+        cycle_id: str | None = None,
     ) -> str:
         """
         Add an optimization cycle pattern to workflow.
@@ -195,7 +195,7 @@ class CycleTemplates:
         max_retries: int = 3,
         backoff_strategy: str = "exponential",
         success_condition: str = "success == True",
-        cycle_id: Optional[str] = None,
+        cycle_id: str | None = None,
     ) -> str:
         """
         Add a retry cycle pattern to workflow.
@@ -300,7 +300,7 @@ result = {{
         validator_node: str,
         quality_threshold: float = 0.95,
         max_iterations: int = 10,
-        cycle_id: Optional[str] = None,
+        cycle_id: str | None = None,
     ) -> str:
         """
         Add a data quality improvement cycle to workflow.
@@ -353,7 +353,7 @@ result = {{
         target_accuracy: float = 0.95,
         max_epochs: int = 100,
         early_stopping_patience: int = 10,
-        cycle_id: Optional[str] = None,
+        cycle_id: str | None = None,
     ) -> str:
         """
         Add a machine learning training cycle to workflow.
@@ -462,7 +462,7 @@ result = {{
         processor_node: str,
         tolerance: float = 0.001,
         max_iterations: int = 1000,
-        cycle_id: Optional[str] = None,
+        cycle_id: str | None = None,
     ) -> str:
         """
         Add a numerical convergence cycle to workflow.
@@ -560,8 +560,8 @@ previous_value = current_value
         workflow: Workflow,
         processor_node: str,
         batch_size: int = 100,
-        total_items: Optional[int] = None,
-        cycle_id: Optional[str] = None,
+        total_items: int | None = None,
+        cycle_id: str | None = None,
     ) -> str:
         """
         Add a batch processing cycle to workflow.
@@ -671,7 +671,7 @@ def add_optimization_cycle(
     evaluator_node: str,
     convergence: str = "quality > 0.9",
     max_iterations: int = 50,
-    cycle_id: Optional[str] = None,
+    cycle_id: str | None = None,
 ) -> str:
     """Add an optimization cycle pattern to this workflow."""
     return CycleTemplates.optimization_cycle(
@@ -685,7 +685,7 @@ def add_retry_cycle(
     max_retries: int = 3,
     backoff_strategy: str = "exponential",
     success_condition: str = "success == True",
-    cycle_id: Optional[str] = None,
+    cycle_id: str | None = None,
 ) -> str:
     """Add a retry cycle pattern to this workflow."""
     return CycleTemplates.retry_cycle(
@@ -699,7 +699,7 @@ def add_data_quality_cycle(
     validator_node: str,
     quality_threshold: float = 0.95,
     max_iterations: int = 10,
-    cycle_id: Optional[str] = None,
+    cycle_id: str | None = None,
 ) -> str:
     """Add a data quality improvement cycle to this workflow."""
     return CycleTemplates.data_quality_cycle(
@@ -714,7 +714,7 @@ def add_learning_cycle(
     target_accuracy: float = 0.95,
     max_epochs: int = 100,
     early_stopping_patience: int = 10,
-    cycle_id: Optional[str] = None,
+    cycle_id: str | None = None,
 ) -> str:
     """Add a machine learning training cycle to this workflow."""
     return CycleTemplates.learning_cycle(
@@ -733,7 +733,7 @@ def add_convergence_cycle(
     processor_node: str,
     tolerance: float = 0.001,
     max_iterations: int = 1000,
-    cycle_id: Optional[str] = None,
+    cycle_id: str | None = None,
 ) -> str:
     """Add a numerical convergence cycle to this workflow."""
     return CycleTemplates.convergence_cycle(
@@ -745,8 +745,8 @@ def add_batch_processing_cycle(
     self,
     processor_node: str,
     batch_size: int = 100,
-    total_items: Optional[int] = None,
-    cycle_id: Optional[str] = None,
+    total_items: int | None = None,
+    cycle_id: str | None = None,
 ) -> str:
     """Add a batch processing cycle to this workflow."""
     return CycleTemplates.batch_processing_cycle(

kailash/workflow/validation.py

@@ -116,7 +116,7 @@ See Also:
 import re
 from dataclasses import dataclass
 from enum import Enum
-from typing import Any, Dict, List, Optional
+from typing import Any
 
 from . import Workflow
 
@@ -137,10 +137,10 @@ class ValidationIssue:
     category: str
     code: str
     message: str
-    node_id: Optional[str] = None
-    cycle_id: Optional[str] = None
-    suggestion: Optional[str] = None
-    documentation_link: Optional[str] = None
+    node_id: str | None = None
+    cycle_id: str | None = None
+    suggestion: str | None = None
+    documentation_link: str | None = None
 
 
 class CycleLinter:
@@ -160,9 +160,9 @@ class CycleLinter:
         """
         self.workflow = workflow
         self.graph = workflow.graph
-        self.issues: List[ValidationIssue] = []
+        self.issues: list[ValidationIssue] = []
 
-    def check_all(self) -> List[ValidationIssue]:
+    def check_all(self) -> list[ValidationIssue]:
         """
         Run all validation checks on the workflow.
 
@@ -536,7 +536,7 @@ class CycleLinter:
                             )
                         )
 
-    def _get_cycle_id(self, cycle_nodes: List[str]) -> str:
+    def _get_cycle_id(self, cycle_nodes: list[str]) -> str:
         """Generate a cycle identifier from cycle nodes."""
         return f"cycle_{'-'.join(sorted(cycle_nodes))}"
 
@@ -658,23 +658,23 @@ class CycleLinter:
         node_id_lower = node_id.lower()
         return any(keyword in node_id_lower for keyword in file_keywords)
 
-    def get_issues_by_severity(self, severity: IssueSeverity) -> List[ValidationIssue]:
+    def get_issues_by_severity(self, severity: IssueSeverity) -> list[ValidationIssue]:
         """Get all issues of a specific severity level."""
         return [issue for issue in self.issues if issue.severity == severity]
 
-    def get_issues_by_category(self, category: str) -> List[ValidationIssue]:
+    def get_issues_by_category(self, category: str) -> list[ValidationIssue]:
         """Get all issues of a specific category."""
         return [issue for issue in self.issues if issue.category == category]
 
-    def get_issues_for_cycle(self, cycle_id: str) -> List[ValidationIssue]:
+    def get_issues_for_cycle(self, cycle_id: str) -> list[ValidationIssue]:
         """Get all issues for a specific cycle."""
         return [issue for issue in self.issues if issue.cycle_id == cycle_id]
 
-    def get_issues_for_node(self, node_id: str) -> List[ValidationIssue]:
+    def get_issues_for_node(self, node_id: str) -> list[ValidationIssue]:
         """Get all issues for a specific node."""
         return [issue for issue in self.issues if issue.node_id == node_id]
 
-    def generate_report(self) -> Dict[str, Any]:
+    def generate_report(self) -> dict[str, Any]:
         """
         Generate comprehensive validation report.
 
@@ -724,7 +724,7 @@ class CycleLinter:
             "recommendations": self._generate_recommendations(),
         }
 
-    def _generate_recommendations(self) -> List[str]:
+    def _generate_recommendations(self) -> list[str]:
         """Generate high-level recommendations based on found issues."""
         recommendations = []
 

kailash/workflow/visualization.py

@@ -4,7 +4,7 @@ import matplotlib
 
 matplotlib.use("Agg")  # Use non-interactive backend
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any
 
 import matplotlib.pyplot as plt
 import networkx as nx
@@ -19,8 +19,8 @@ class WorkflowVisualizer:
     def __init__(
         self,
         workflow: Workflow,
-        node_colors: Optional[Dict[str, str]] = None,
-        edge_colors: Optional[Dict[str, str]] = None,
+        node_colors: dict[str, str] | None = None,
+        edge_colors: dict[str, str] | None = None,
         layout: str = "hierarchical",
     ):
         """Initialize visualizer.
@@ -36,7 +36,7 @@ class WorkflowVisualizer:
         self.edge_colors = edge_colors or self._default_edge_colors()
         self.layout = layout
 
-    def _default_node_colors(self) -> Dict[str, str]:
+    def _default_node_colors(self) -> dict[str, str]:
         """Get default node color map."""
         return {
             "data": "lightblue",
@@ -46,7 +46,7 @@ class WorkflowVisualizer:
             "default": "lightgray",
         }
 
-    def _default_edge_colors(self) -> Dict[str, str]:
+    def _default_edge_colors(self) -> dict[str, str]:
         """Get default edge color map."""
         return {"default": "gray", "error": "red", "conditional": "orange"}
 
@@ -66,7 +66,7 @@ class WorkflowVisualizer:
             return self.node_colors["ai"]
         return self.node_colors["default"]
 
-    def _get_node_colors(self) -> List[str]:
+    def _get_node_colors(self) -> list[str]:
         """Get colors for all nodes in workflow."""
         colors = []
         for node_id in self.workflow.graph.nodes():
@@ -75,7 +75,7 @@ class WorkflowVisualizer:
             colors.append(self._get_node_color(node_type))
         return colors
 
-    def _get_node_labels(self) -> Dict[str, str]:
+    def _get_node_labels(self) -> dict[str, str]:
         """Get labels for nodes in workflow."""
         labels = {}
         for node_id in self.workflow.graph.nodes():
@@ -93,7 +93,7 @@ class WorkflowVisualizer:
                     labels[node_id] = node_id
         return labels
 
-    def _get_edge_labels(self) -> Dict[Tuple[str, str], str]:
+    def _get_edge_labels(self) -> dict[tuple[str, str], str]:
         """Get labels for edges in workflow."""
         edge_labels = {}
 
@@ -119,7 +119,7 @@ class WorkflowVisualizer:
 
         return edge_labels
 
-    def _calculate_layout(self) -> Dict[str, Tuple[float, float]]:
+    def _calculate_layout(self) -> dict[str, tuple[float, float]]:
         """Calculate node positions for visualization."""
         # Try to use stored positions first
         pos = {}
@@ -148,7 +148,7 @@ class WorkflowVisualizer:
 
         return pos
 
-    def _create_layers(self) -> Dict[int, list]:
+    def _create_layers(self) -> dict[int, list]:
         """Create layers of nodes for hierarchical layout."""
         layers = {}
         remaining = set(self.workflow.graph.nodes())
@@ -173,8 +173,8 @@ class WorkflowVisualizer:
         return layers
 
     def _hierarchical_layout(
-        self, layers: Dict[int, list]
-    ) -> Dict[str, Tuple[float, float]]:
+        self, layers: dict[int, list]
+    ) -> dict[str, tuple[float, float]]:
         """Create hierarchical layout from layers."""
         pos = {}
         layer_height = 2.0
@@ -196,8 +196,8 @@ class WorkflowVisualizer:
 
     def _draw_graph(
         self,
-        pos: Dict[str, Tuple[float, float]],
-        node_colors: List[str],
+        pos: dict[str, tuple[float, float]],
+        node_colors: list[str],
         show_labels: bool,
         show_connections: bool,
     ) -> None:
@@ -235,9 +235,9 @@ class WorkflowVisualizer:
 
     def visualize(
         self,
-        output_path: Optional[str] = None,
-        figsize: Tuple[int, int] = (12, 8),
-        title: Optional[str] = None,
+        output_path: str | None = None,
+        figsize: tuple[int, int] = (12, 8),
+        title: str | None = None,
         show_labels: bool = True,
         show_connections: bool = True,
         dpi: int = 300,
@@ -303,7 +303,7 @@ class WorkflowVisualizer:
         self.visualize(output_path=output_path, **kwargs)
 
     def create_execution_graph(
-        self, run_id: str, task_manager: Any, output_path: Optional[str] = None
+        self, run_id: str, task_manager: Any, output_path: str | None = None
     ) -> str:
         """Create a Mermaid visualization showing execution status.
 
@@ -423,8 +423,8 @@ class WorkflowVisualizer:
         return str(output_path)
 
     def create_performance_dashboard(
-        self, run_id: str, task_manager: TaskManager, output_dir: Optional[Path] = None
-    ) -> Dict[str, Path]:
+        self, run_id: str, task_manager: TaskManager, output_dir: Path | None = None
+    ) -> dict[str, Path]:
         """Create integrated performance dashboard with workflow visualization.
 
         Args:
@@ -461,7 +461,7 @@ class WorkflowVisualizer:
         return outputs
 
     def _create_dashboard_html(
-        self, run_id: str, outputs: Dict[str, Path], dashboard_path: Path
+        self, run_id: str, outputs: dict[str, Path], dashboard_path: Path
     ) -> None:
         """Create HTML dashboard integrating all visualizations."""
         html_content = f"""
@@ -578,7 +578,7 @@ class WorkflowVisualizer:
 def add_visualization_to_workflow():
     """Add visualization method to Workflow class."""
 
-    def visualize(self, output_path: Optional[str] = None, **kwargs) -> None:
+    def visualize(self, output_path: str | None = None, **kwargs) -> None:
         """Visualize the workflow.
 
         Args:

{kailash-0.3.0.dist-info → kailash-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.3.0
+Version: 0.3.1
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum
@@ -109,7 +109,8 @@ Dynamic: requires-python
 - 🔁 **Cyclic Workflows (v0.2.0)**: Universal Hybrid Cyclic Graph Architecture with 30,000+ iterations/second performance
 - 🛠️ **Developer Tools**: CycleAnalyzer, CycleDebugger, CycleProfiler for production-ready cyclic workflows
 - 📈 **High Performance**: Optimized execution engine supporting 100,000+ iteration workflows
-- 📁 **Enhanced Documentation (v0.2.2)**: Reorganized structure with production-ready workflow library
+- 📁 **Complete Finance Workflow Library (v0.3.1)**: Production-ready financial workflows with AI analysis
+- 💼 **Enterprise Workflow Patterns**: Credit risk, portfolio optimization, trading signals, fraud detection
 
 ## 🎯 Who Is This For?
 
@@ -197,6 +198,52 @@ from kailash.utils.export import export_workflow
 export_workflow(workflow, "customer_analysis.yaml")
 ```
 
+## 💼 Finance Workflow Library (New in v0.3.1)
+
+Complete production-ready financial workflows using AI and modern quantitative methods:
+
+### Credit Risk Assessment
+
+```python
+from kailash.workflow import Workflow
+from kailash.nodes.data import CSVReaderNode
+from kailash.nodes.code import PythonCodeNode
+from kailash.nodes.ai import LLMAgentNode
+
+def calculate_risk_metrics(customers, transactions):
+    """Calculate comprehensive risk metrics."""
+    # Modern risk scoring with AI analysis
+    # 100+ lines of production risk calculation
+    return {"result": risk_scores}
+
+workflow = Workflow("credit-risk", "Credit Risk Assessment")
+workflow.add_node("customer_reader", CSVReaderNode())
+workflow.add_node("risk_calculator", PythonCodeNode.from_function(func=calculate_risk_metrics))
+workflow.add_node("ai_analyzer", LLMAgentNode(model="gpt-4",
+    system_prompt="You are a financial risk expert..."))
+```
+
+### Portfolio Optimization
+
+```python
+def optimize_portfolio(holdings, market_data, risk_profile="moderate"):
+    """Modern Portfolio Theory optimization with rebalancing."""
+    # Sharpe ratio optimization, correlation analysis
+    # Risk-adjusted returns with AI market insights
+    return {"result": optimization_plan}
+
+workflow = Workflow("portfolio-opt", "Portfolio Optimization")
+workflow.add_node("optimizer", PythonCodeNode.from_function(func=optimize_portfolio))
+# Generates rebalancing trades, risk metrics, AI market analysis
+```
+
+### Trading Signals & Fraud Detection
+
+- **Trading Signals**: Technical indicators (RSI, MACD, Bollinger Bands) + AI sentiment
+- **Fraud Detection**: Real-time transaction monitoring with velocity analysis
+
+**See complete examples**: `sdk-users/workflows/by-industry/finance/`
+
 ## 📚 Documentation
 
 ### For SDK Users
@@ -204,9 +251,10 @@ export_workflow(workflow, "customer_analysis.yaml")
 **Build solutions with the SDK:**
 - `sdk-users/` - Everything you need to build with Kailash
   - `developer/` - Node creation patterns and troubleshooting
-  - `workflows/` - Production-ready workflow library (NEW in v0.2.2)
+  - `workflows/` - Complete production workflow library (v0.3.1)
+    - Finance workflows: Credit risk, portfolio optimization, trading signals, fraud detection
     - Quick-start patterns (30-second workflows)
-    - Industry-specific solutions (healthcare, finance)
+    - Industry-specific solutions by vertical
     - Enterprise integration patterns
   - `essentials/` - Quick reference and cheatsheets
   - `nodes/` - Comprehensive node catalog (66+ nodes)
@@ -347,7 +395,7 @@ assert results["analyze"]["result"]["total_customers"] == len(test_data)
 3. **Monitor in real-time**:
    ```python
    from kailash.visualization import DashboardServer
-   
+
    server = DashboardServer(port=8080)
    server.start()
    # Open http://localhost:8080 for live monitoring