kailash 0.3.0__tar.gz → 0.3.1__tar.gz

{kailash-0.3.0/src/kailash.egg-info → kailash-0.3.1}/PKG-INFO

@@ -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

{kailash-0.3.0 → kailash-0.3.1}/README.md

@@ -39,7 +39,8 @@
 - 🔁 **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?
 
@@ -127,6 +128,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
@@ -134,9 +181,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)
@@ -277,7 +325,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

{kailash-0.3.0 → kailash-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kailash"
-version = "0.3.0"
+version = "0.3.1"
 description = "Python SDK for the Kailash container-node architecture"
 authors = [
     {name = "Integrum", email = "info@integrum.com"}
@@ -89,3 +89,19 @@ dev = [
 
 [project.scripts]
 kailash = "kailash.cli:main"
+
+[tool.ruff]
+line-length = 88
+target-version = "py311"
+
+[tool.ruff.lint]
+select = ["E", "F"]
+ignore = ["E203", "E501", "E722", "F401", "F841"]
+
+[tool.ruff.lint.per-file-ignores]
+"sdk-users/workflows/by-industry/finance/scripts/*.py" = ["E402"]
+"examples/feature_examples/workflows/cyclic/*.py" = ["E722"]
+"examples/feature_examples/workflows/parallel/*.py" = ["E722"]
+"scripts/refactor-pythoncode-to-functions.py" = ["E722"]
+"src/kailash/nodes/api/security.py" = ["E722"]
+"tests/conftest_sdk_dev.py" = ["E722"]

{kailash-0.3.0 → kailash-0.3.1}/setup.py

@@ -13,7 +13,7 @@ with open(os.path.join(this_directory, "README.md"), encoding="utf-8") as f:
 # Package configuration
 setup(
     name="kailash",
-    version="0.3.0",
+    version="0.3.1",
     author="Integrum",
     author_email="info@integrum.com",
     description="Python SDK for the Kailash container-node architecture",

{kailash-0.3.0 → kailash-0.3.1}/src/kailash/__init__.py

@@ -15,7 +15,7 @@ from kailash.workflow.visualization import WorkflowVisualizer
 # For backward compatibility
 WorkflowGraph = Workflow
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 __all__ = [
     "Workflow",

{kailash-0.3.0 → kailash-0.3.1}/src/kailash/access_control.py

@@ -51,10 +51,11 @@ Future Enhancements:
 
 import logging
 import threading
+from collections.abc import Callable
 from dataclasses import dataclass, field
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from enum import Enum
-from typing import Any, Callable, Dict, List, Optional, Set, Union
+from typing import Any
 
 logger = logging.getLogger(__name__)
 
@@ -136,11 +137,11 @@ class UserContext:
     user_id: str
     tenant_id: str
     email: str
-    roles: List[str] = field(default_factory=list)
-    permissions: List[str] = field(default_factory=list)
-    attributes: Dict[str, Any] = field(default_factory=dict)  # Custom attributes
-    session_id: Optional[str] = None
-    ip_address: Optional[str] = None
+    roles: list[str] = field(default_factory=list)
+    permissions: list[str] = field(default_factory=list)
+    attributes: dict[str, Any] = field(default_factory=dict)  # Custom attributes
+    session_id: str | None = None
+    ip_address: str | None = None
 
 
 @dataclass
@@ -195,21 +196,21 @@ class PermissionRule:
     id: str
     resource_type: str  # "workflow" or "node"
     resource_id: str  # workflow_id or node_id
-    permission: Union[WorkflowPermission, NodePermission]
+    permission: WorkflowPermission | NodePermission
     effect: PermissionEffect
 
     # Who does this apply to?
-    user_id: Optional[str] = None  # Specific user
-    role: Optional[str] = None  # Any user with role
-    tenant_id: Optional[str] = None  # All users in tenant
+    user_id: str | None = None  # Specific user
+    role: str | None = None  # Any user with role
+    tenant_id: str | None = None  # All users in tenant
 
     # Conditions
-    conditions: Dict[str, Any] = field(default_factory=dict)
+    conditions: dict[str, Any] = field(default_factory=dict)
 
     # Metadata
-    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
-    created_by: Optional[str] = None
-    expires_at: Optional[datetime] = None
+    created_at: datetime = field(default_factory=lambda: datetime.now(UTC))
+    created_by: str | None = None
+    expires_at: datetime | None = None
     priority: int = 0  # Higher priority rules evaluated first
 
 
@@ -262,17 +263,17 @@ class AccessDecision:
 
     allowed: bool
     reason: str
-    applied_rules: List[PermissionRule] = field(default_factory=list)
-    conditions_met: Dict[str, bool] = field(default_factory=dict)
-    masked_fields: List[str] = field(default_factory=list)  # Fields to mask in output
-    redirect_node: Optional[str] = None  # Alternative node to execute
+    applied_rules: list[PermissionRule] = field(default_factory=list)
+    conditions_met: dict[str, bool] = field(default_factory=dict)
+    masked_fields: list[str] = field(default_factory=list)  # Fields to mask in output
+    redirect_node: str | None = None  # Alternative node to execute
 
 
 class ConditionEvaluator:
     """Evaluates conditions for conditional permissions"""
 
     def __init__(self):
-        self.evaluators: Dict[str, Callable] = {
+        self.evaluators: dict[str, Callable] = {
             "time_range": self._eval_time_range,
             "data_contains": self._eval_data_contains,
             "user_attribute": self._eval_user_attribute,
@@ -281,7 +282,7 @@ class ConditionEvaluator:
         }
 
     def evaluate(
-        self, condition_type: str, condition_value: Any, context: Dict[str, Any]
+        self, condition_type: str, condition_value: Any, context: dict[str, Any]
     ) -> bool:
         """Evaluate a condition"""
         evaluator = self.evaluators.get(condition_type)
@@ -295,7 +296,7 @@ class ConditionEvaluator:
             logger.error(f"Error evaluating condition {condition_type}: {e}")
             return False
 
-    def _eval_time_range(self, value: Dict[str, str], context: Dict[str, Any]) -> bool:
+    def _eval_time_range(self, value: dict[str, str], context: dict[str, Any]) -> bool:
         """Check if current time is within range"""
         from datetime import datetime, time
 
@@ -305,7 +306,7 @@ class ConditionEvaluator:
         return start <= now <= end
 
     def _eval_data_contains(
-        self, value: Dict[str, Any], context: Dict[str, Any]
+        self, value: dict[str, Any], context: dict[str, Any]
     ) -> bool:
         """Check if data contains specific values"""
         data = context.get("data", {})
@@ -317,7 +318,7 @@ class ConditionEvaluator:
         return False
 
     def _eval_user_attribute(
-        self, value: Dict[str, Any], context: Dict[str, Any]
+        self, value: dict[str, Any], context: dict[str, Any]
     ) -> bool:
         """Check user attributes"""
         user = context.get("user")
@@ -329,7 +330,7 @@ class ConditionEvaluator:
 
         return user.attributes.get(attr_name) == expected
 
-    def _eval_ip_range(self, value: Dict[str, Any], context: Dict[str, Any]) -> bool:
+    def _eval_ip_range(self, value: dict[str, Any], context: dict[str, Any]) -> bool:
         """Check if IP is in allowed range"""
         # Simplified IP check - in production use ipaddress module
         allowed_ips = value.get("allowed", [])
@@ -337,7 +338,7 @@ class ConditionEvaluator:
 
         return user_ip in allowed_ips
 
-    def _eval_custom(self, value: Dict[str, Any], context: Dict[str, Any]) -> bool:
+    def _eval_custom(self, value: dict[str, Any], context: dict[str, Any]) -> bool:
         """Evaluate custom condition"""
         # This would call a custom function registered by the user
         return True
@@ -409,7 +410,7 @@ class AccessControlManager:
 
     def __init__(self, enabled: bool = False):
         self.enabled = enabled  # Disabled by default
-        self.rules: List[PermissionRule] = []
+        self.rules: list[PermissionRule] = []
         self.condition_evaluator = ConditionEvaluator()
         self._cache = {}  # Cache access decisions
         self._cache_lock = threading.Lock()
@@ -458,7 +459,7 @@ class AccessControlManager:
         user: UserContext,
         node_id: str,
         permission: NodePermission,
-        runtime_context: Dict[str, Any] = None,
+        runtime_context: dict[str, Any] = None,
     ) -> AccessDecision:
         """Check if user has permission on node"""
         cache_key = f"node:{node_id}:{user.user_id}:{permission.value}"
@@ -494,7 +495,7 @@ class AccessControlManager:
 
     def get_accessible_nodes(
         self, user: UserContext, workflow_id: str, permission: NodePermission
-    ) -> Set[str]:
+    ) -> set[str]:
         """Get all nodes user can access in a workflow"""
         accessible = set()
 
@@ -518,9 +519,9 @@ class AccessControlManager:
         self,
         user: UserContext,
         conditional_node_id: str,
-        true_path_nodes: List[str],
-        false_path_nodes: List[str],
-    ) -> List[str]:
+        true_path_nodes: list[str],
+        false_path_nodes: list[str],
+    ) -> list[str]:
         """Determine which path user should take based on permissions"""
         # Check if user has access to nodes in true path
         true_path_accessible = all(
@@ -544,8 +545,8 @@ class AccessControlManager:
                 return []
 
     def mask_node_output(
-        self, user: UserContext, node_id: str, output: Dict[str, Any]
-    ) -> Dict[str, Any]:
+        self, user: UserContext, node_id: str, output: dict[str, Any]
+    ) -> dict[str, Any]:
         """Mask sensitive fields in node output"""
         decision = self.check_node_access(user, node_id, NodePermission.READ_OUTPUT)
 
@@ -568,8 +569,8 @@ class AccessControlManager:
         user: UserContext,
         resource_type: str,
         resource_id: str,
-        permission: Union[WorkflowPermission, NodePermission],
-        runtime_context: Dict[str, Any],
+        permission: WorkflowPermission | NodePermission,
+        runtime_context: dict[str, Any],
     ) -> AccessDecision:
         """Evaluate all applicable rules"""
         applicable_rules = []
@@ -584,7 +585,7 @@ class AccessControlManager:
             ):
 
                 # Check expiration
-                if rule.expires_at and rule.expires_at < datetime.now(timezone.utc):
+                if rule.expires_at and rule.expires_at < datetime.now(UTC):
                     continue
 
                 applicable_rules.append(rule)
@@ -593,7 +594,7 @@ class AccessControlManager:
         context = {
             "user": user,
             "runtime": runtime_context,
-            "timestamp": datetime.now(timezone.utc),
+            "timestamp": datetime.now(UTC),
         }
 
         final_effect = PermissionEffect.DENY  # Default deny
@@ -660,7 +661,7 @@ class AccessControlManager:
         user: UserContext,
         resource_type: str,
         resource_id: str,
-        permission: Union[WorkflowPermission, NodePermission],
+        permission: WorkflowPermission | NodePermission,
         decision: AccessDecision,
     ):
         """Log access attempt for audit"""

{kailash-0.3.0 → kailash-0.3.1}/src/kailash/api/auth.py

@@ -50,8 +50,8 @@ Future Enhancements:
 import os
 import secrets
 import threading
-from datetime import datetime, timedelta, timezone
-from typing import Any, Dict, List, Optional
+from datetime import UTC, datetime, timedelta
+from typing import Any
 
 from fastapi import Depends, HTTPException, Request, status
 from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
@@ -101,14 +101,14 @@ class User(Base):
 
     # User roles and permissions
     roles = Column(JSON, default=lambda: ["user"])
-    permissions = Column(JSON, default=lambda: [])
+    permissions = Column(JSON, default=list)
 
     # Timestamps
-    created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
+    created_at = Column(DateTime, default=lambda: datetime.now(UTC))
     updated_at = Column(
         DateTime,
-        default=lambda: datetime.now(timezone.utc),
-        onupdate=lambda: datetime.now(timezone.utc),
+        default=lambda: datetime.now(UTC),
+        onupdate=lambda: datetime.now(UTC),
     )
     last_login = Column(DateTime)
 
@@ -150,11 +150,11 @@ class Tenant(Base):
     subscription_tier = Column(String(50), default="free")
 
     # Timestamps
-    created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
+    created_at = Column(DateTime, default=lambda: datetime.now(UTC))
     updated_at = Column(
         DateTime,
-        default=lambda: datetime.now(timezone.utc),
-        onupdate=lambda: datetime.now(timezone.utc),
+        default=lambda: datetime.now(UTC),
+        onupdate=lambda: datetime.now(UTC),
     )
 
     # Relationships
@@ -187,7 +187,7 @@ class APIKey(Base):
     usage_count = Column(JSON, default=lambda: {"total": 0, "monthly": 0})
 
     # Timestamps
-    created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
+    created_at = Column(DateTime, default=lambda: datetime.now(UTC))
 
     # Relationships
     user = relationship("User", back_populates="api_keys")
@@ -203,7 +203,7 @@ class UserCreate(BaseModel):
     email: EmailStr
     username: str = Field(..., min_length=3, max_length=100)
     password: str = Field(..., min_length=8)
-    tenant_id: Optional[str] = None  # If None, create new tenant
+    tenant_id: str | None = None  # If None, create new tenant
 
 
 class UserLogin(BaseModel):
@@ -227,9 +227,9 @@ class TokenData(BaseModel):
 
     sub: str
     tenant_id: str
-    roles: List[str] = ["user"]
-    permissions: List[str] = []
-    exp: Optional[datetime] = None
+    roles: list[str] = ["user"]
+    permissions: list[str] = []
+    exp: datetime | None = None
 
 
 class JWTAuth:
@@ -240,33 +240,27 @@ class JWTAuth:
         self.algorithm = ALGORITHM
 
     def create_access_token(
-        self, data: Dict[str, Any], expires_delta: Optional[timedelta] = None
+        self, data: dict[str, Any], expires_delta: timedelta | None = None
     ) -> str:
         """Create a JWT access token"""
         to_encode = data.copy()
 
         if expires_delta:
-            expire = datetime.now(timezone.utc) + expires_delta
+            expire = datetime.now(UTC) + expires_delta
         else:
-            expire = datetime.now(timezone.utc) + timedelta(
-                hours=ACCESS_TOKEN_EXPIRE_HOURS
-            )
+            expire = datetime.now(UTC) + timedelta(hours=ACCESS_TOKEN_EXPIRE_HOURS)
 
-        to_encode.update(
-            {"exp": expire, "iat": datetime.now(timezone.utc), "type": "access"}
-        )
+        to_encode.update({"exp": expire, "iat": datetime.now(UTC), "type": "access"})
 
         encoded_jwt = jwt.encode(to_encode, self.secret_key, algorithm=self.algorithm)
         return encoded_jwt
 
-    def create_refresh_token(self, data: Dict[str, Any]) -> str:
+    def create_refresh_token(self, data: dict[str, Any]) -> str:
         """Create a JWT refresh token"""
         to_encode = data.copy()
-        expire = datetime.now(timezone.utc) + timedelta(days=REFRESH_TOKEN_EXPIRE_DAYS)
+        expire = datetime.now(UTC) + timedelta(days=REFRESH_TOKEN_EXPIRE_DAYS)
 
-        to_encode.update(
-            {"exp": expire, "iat": datetime.now(timezone.utc), "type": "refresh"}
-        )
+        to_encode.update({"exp": expire, "iat": datetime.now(UTC), "type": "refresh"})
 
         encoded_jwt = jwt.encode(to_encode, self.secret_key, algorithm=self.algorithm)
         return encoded_jwt
@@ -421,13 +415,13 @@ async def verify_api_key(
         )
 
     # Check expiration
-    if valid_key.expires_at and valid_key.expires_at < datetime.now(timezone.utc):
+    if valid_key.expires_at and valid_key.expires_at < datetime.now(UTC):
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED, detail="API key expired"
         )
 
     # Update usage
-    valid_key.last_used_at = datetime.now(timezone.utc)
+    valid_key.last_used_at = datetime.now(UTC)
     valid_key.usage_count["total"] += 1
     valid_key.usage_count["monthly"] += 1
     session.commit()
@@ -514,7 +508,7 @@ class TenantContext:
 _tenant_context = threading.local()
 
 
-def get_current_tenant_id() -> Optional[str]:
+def get_current_tenant_id() -> str | None:
     """Get current tenant ID from context"""
     return getattr(_tenant_context, "tenant_id", None)
 
@@ -615,7 +609,7 @@ class AuthService:
             )
 
         # Update last login
-        user.last_login = datetime.now(timezone.utc)
+        user.last_login = datetime.now(UTC)
         self.session.commit()
 
         # Generate tokens
@@ -646,7 +640,7 @@ class AuthService:
         return self.auth.create_tokens(user)
 
     def create_api_key(
-        self, name: str, user: User, scopes: List[str] = None
+        self, name: str, user: User, scopes: list[str] = None
     ) -> tuple[str, APIKey]:
         """Create an API key for a user"""
         key, key_hash = create_api_key()