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

kailash/nodes/api/auth.py

@@ -12,7 +12,7 @@ Key Components:
 
 import base64
 import time
-from typing import Any, Dict
+from typing import Any
 
 import requests
 
@@ -42,7 +42,7 @@ class BasicAuthNode(Node):
     - RESTClientNode: Uses auth headers for API calls
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -63,7 +63,7 @@ class BasicAuthNode(Node):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -84,7 +84,7 @@ class BasicAuthNode(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Generate Basic Authentication headers.
 
         Args:
@@ -162,7 +162,7 @@ class OAuth2Node(Node):
         self.token_data = None  # Will store token information
         self.token_expires_at = 0  # Timestamp when token expires
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -236,7 +236,7 @@ class OAuth2Node(Node):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -269,7 +269,7 @@ class OAuth2Node(Node):
             ),
         }
 
-    def _get_token(self, **kwargs) -> Dict[str, Any]:
+    def _get_token(self, **kwargs) -> dict[str, Any]:
         """Get an OAuth token using the configured grant type.
 
         This method handles different grant types with appropriate parameters.
@@ -359,7 +359,7 @@ class OAuth2Node(Node):
                 f"Failed to parse OAuth token response: {str(e)}"
             ) from e
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Get OAuth authentication headers.
 
         This method handles token acquisition, caching, and renewal based on
@@ -444,7 +444,7 @@ class APIKeyNode(Node):
     - RESTClientNode: Uses auth data for API calls
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -477,7 +477,7 @@ class APIKeyNode(Node):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -510,7 +510,7 @@ class APIKeyNode(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Generate API key authentication data.
 
         Args:

kailash/nodes/api/graphql.py

@@ -10,7 +10,7 @@ Key Components:
 - GraphQL query building and response handling utilities
 """
 
-from typing import Any, Dict, Optional
+from typing import Any
 
 from kailash.nodes.api.http import AsyncHTTPRequestNode, HTTPRequestNode
 from kailash.nodes.base import Node, NodeParameter, register_node
@@ -60,7 +60,7 @@ class GraphQLClientNode(Node):
         super().__init__(**kwargs)
         self.http_node = HTTPRequestNode(**kwargs)
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -137,7 +137,7 @@ class GraphQLClientNode(Node):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -173,9 +173,9 @@ class GraphQLClientNode(Node):
     def _build_graphql_payload(
         self,
         query: str,
-        variables: Dict[str, Any] = None,
-        operation_name: Optional[str] = None,
-    ) -> Dict[str, Any]:
+        variables: dict[str, Any] = None,
+        operation_name: str | None = None,
+    ) -> dict[str, Any]:
         """Build a GraphQL request payload.
 
         Args:
@@ -197,8 +197,8 @@ class GraphQLClientNode(Node):
         return payload
 
     def _process_graphql_response(
-        self, response: Dict[str, Any], flatten_response: bool = False
-    ) -> Dict[str, Any]:
+        self, response: dict[str, Any], flatten_response: bool = False
+    ) -> dict[str, Any]:
         """Process a GraphQL response.
 
         Args:
@@ -241,7 +241,7 @@ class GraphQLClientNode(Node):
 
         return {"data": data, "errors": errors, "success": success}
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Execute a GraphQL query or mutation.
 
         Args:
@@ -362,7 +362,7 @@ class AsyncGraphQLClientNode(AsyncNode):
         self.http_node = AsyncHTTPRequestNode(**kwargs)
         self.graphql_node = GraphQLClientNode(**kwargs)
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -371,7 +371,7 @@ class AsyncGraphQLClientNode(AsyncNode):
         # Same parameters as the synchronous version
         return self.graphql_node.get_parameters()
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -380,7 +380,7 @@ class AsyncGraphQLClientNode(AsyncNode):
         # Same output schema as the synchronous version
         return self.graphql_node.get_output_schema()
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Synchronous version of the GraphQL request, for compatibility.
 
         This is implemented for compatibility but users should use the
@@ -399,7 +399,7 @@ class AsyncGraphQLClientNode(AsyncNode):
         # Forward to the synchronous GraphQL node
         return self.graphql_node.run(**kwargs)
 
-    async def async_run(self, **kwargs) -> Dict[str, Any]:
+    async def async_run(self, **kwargs) -> dict[str, Any]:
         """Execute a GraphQL query or mutation asynchronously.
 
         Args:

kailash/nodes/api/http.py

@@ -8,7 +8,7 @@ import asyncio
 import base64
 import time
 from enum import Enum
-from typing import Any, Dict, Optional
+from typing import Any
 
 import aiohttp
 import requests
@@ -48,8 +48,8 @@ class HTTPResponse(BaseModel):
     """
 
     status_code: int
-    headers: Dict[str, str]
-    content_type: Optional[str] = None
+    headers: dict[str, str]
+    content_type: str | None = None
     content: Any  # Can be dict, str, bytes depending on response format
     response_time_ms: float
     url: str
@@ -194,7 +194,7 @@ class HTTPRequestNode(Node):
         super().__init__(**kwargs)
         self.session = requests.Session()
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -328,7 +328,7 @@ class HTTPRequestNode(Node):
             ),
         }
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -358,10 +358,10 @@ class HTTPRequestNode(Node):
     def _apply_authentication(
         self,
         headers: dict,
-        auth_type: Optional[str],
-        auth_token: Optional[str],
-        auth_username: Optional[str],
-        auth_password: Optional[str],
+        auth_type: str | None,
+        auth_token: str | None,
+        auth_username: str | None,
+        auth_password: str | None,
         api_key_header: str,
     ) -> dict:
         """Apply authentication to request headers.
@@ -398,7 +398,7 @@ class HTTPRequestNode(Node):
 
         return auth_headers
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Execute an HTTP request.
 
         Args:
@@ -683,7 +683,7 @@ class AsyncHTTPRequestNode(AsyncNode):
         super().__init__(**kwargs)
         self._session = None  # Will be created when needed
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -692,7 +692,7 @@ class AsyncHTTPRequestNode(AsyncNode):
         # Same parameters as the synchronous version
         return HTTPRequestNode().get_parameters()
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -704,10 +704,10 @@ class AsyncHTTPRequestNode(AsyncNode):
     def _apply_authentication(
         self,
         headers: dict,
-        auth_type: Optional[str],
-        auth_token: Optional[str],
-        auth_username: Optional[str],
-        auth_password: Optional[str],
+        auth_type: str | None,
+        auth_token: str | None,
+        auth_username: str | None,
+        auth_password: str | None,
         api_key_header: str,
     ) -> dict:
         """Apply authentication to request headers.
@@ -744,7 +744,7 @@ class AsyncHTTPRequestNode(AsyncNode):
 
         return auth_headers
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Synchronous version of the request, for compatibility.
 
         This is implemented for compatibility but users should use the
@@ -763,7 +763,7 @@ class AsyncHTTPRequestNode(AsyncNode):
         http_node = HTTPRequestNode(**self.config)
         return http_node.run(**kwargs)
 
-    async def async_run(self, **kwargs) -> Dict[str, Any]:
+    async def async_run(self, **kwargs) -> dict[str, Any]:
         """Execute an HTTP request asynchronously.
 
         Args:
@@ -937,7 +937,7 @@ class AsyncHTTPRequestNode(AsyncNode):
 
                     return result
 
-            except (aiohttp.ClientError, asyncio.TimeoutError) as e:
+            except (TimeoutError, aiohttp.ClientError) as e:
                 self.logger.warning(f"Async request failed: {str(e)}")
 
                 # Last attempt, no more retries

kailash/nodes/api/monitoring.py

@@ -4,8 +4,8 @@ import asyncio
 import socket
 import subprocess
 import time
-from datetime import datetime, timezone
-from typing import Any, Dict, List
+from datetime import UTC, datetime
+from typing import Any
 
 import requests
 
@@ -92,7 +92,7 @@ class HealthCheckNode(Node):
         >>> assert 'health_results' in result
     """
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "targets": NodeParameter(
                 name="targets",
@@ -130,7 +130,7 @@ class HealthCheckNode(Node):
             ),
         }
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         targets = kwargs["targets"]
         timeout = kwargs.get("timeout", 30)
         retries = kwargs.get("retries", 2)
@@ -162,12 +162,12 @@ class HealthCheckNode(Node):
             "healthy_count": len([r for r in results if r["status"] == "healthy"]),
             "unhealthy_count": len([r for r in results if r["status"] == "unhealthy"]),
             "execution_time": execution_time,
-            "timestamp": datetime.now(timezone.utc).isoformat() + "Z",
+            "timestamp": datetime.now(UTC).isoformat() + "Z",
         }
 
     async def _run_checks_parallel(
-        self, targets: List[Dict], timeout: int, retries: int, include_metrics: bool
-    ) -> List[Dict[str, Any]]:
+        self, targets: list[dict], timeout: int, retries: int, include_metrics: bool
+    ) -> list[dict[str, Any]]:
         """Run health checks in parallel using asyncio."""
 
         async def run_single_check(target):
@@ -184,8 +184,8 @@ class HealthCheckNode(Node):
         return await asyncio.gather(*tasks, return_exceptions=True)
 
     def _run_checks_sequential(
-        self, targets: List[Dict], timeout: int, retries: int, include_metrics: bool
-    ) -> List[Dict[str, Any]]:
+        self, targets: list[dict], timeout: int, retries: int, include_metrics: bool
+    ) -> list[dict[str, Any]]:
         """Run health checks sequentially."""
         return [
             self._perform_health_check(target, timeout, retries, include_metrics)
@@ -193,8 +193,8 @@ class HealthCheckNode(Node):
         ]
 
     def _perform_health_check(
-        self, target: Dict, timeout: int, retries: int, include_metrics: bool
-    ) -> Dict[str, Any]:
+        self, target: dict, timeout: int, retries: int, include_metrics: bool
+    ) -> dict[str, Any]:
         """Perform a single health check with retry logic."""
 
         check_type = target.get("type", "unknown")
@@ -228,7 +228,7 @@ class HealthCheckNode(Node):
                 result["check_id"] = check_id
                 result["check_type"] = check_type
                 result["target"] = target
-                result["timestamp"] = datetime.now(timezone.utc).isoformat() + "Z"
+                result["timestamp"] = datetime.now(UTC).isoformat() + "Z"
 
                 # If successful, return immediately
                 if result["status"] == "healthy":
@@ -245,7 +245,7 @@ class HealthCheckNode(Node):
                         "details": {"error": str(e), "error_type": type(e).__name__},
                         "response_time": time.time() - start_time,
                         "attempt": attempt + 1,
-                        "timestamp": datetime.now(timezone.utc).isoformat() + "Z",
+                        "timestamp": datetime.now(UTC).isoformat() + "Z",
                     }
 
                 # Wait before retry (exponential backoff)
@@ -253,7 +253,7 @@ class HealthCheckNode(Node):
 
         return result
 
-    def _check_http(self, target: Dict, timeout: int) -> Dict[str, Any]:
+    def _check_http(self, target: dict, timeout: int) -> dict[str, Any]:
         """Perform HTTP health check."""
         url = target["url"]
         expected_status = target.get("expected_status", 200)
@@ -296,7 +296,7 @@ class HealthCheckNode(Node):
             },
         }
 
-    def _check_tcp(self, target: Dict, timeout: int) -> Dict[str, Any]:
+    def _check_tcp(self, target: dict, timeout: int) -> dict[str, Any]:
         """Perform TCP port connectivity check."""
         host = target["host"]
         port = target["port"]
@@ -321,7 +321,7 @@ class HealthCheckNode(Node):
         finally:
             sock.close()
 
-    def _check_disk(self, target: Dict) -> Dict[str, Any]:
+    def _check_disk(self, target: dict) -> dict[str, Any]:
         """Perform disk space check."""
         import shutil
 
@@ -365,7 +365,7 @@ class HealthCheckNode(Node):
                 "details": {"path": path, "error": str(e)},
             }
 
-    def _check_command(self, target: Dict, timeout: int) -> Dict[str, Any]:
+    def _check_command(self, target: dict, timeout: int) -> dict[str, Any]:
         """Perform custom command health check."""
         command = target["command"]
         expected_exit_code = target.get("expected_exit_code", 0)
@@ -409,7 +409,7 @@ class HealthCheckNode(Node):
                 "details": {"command": command, "timeout": timeout},
             }
 
-    def _check_database(self, target: Dict, timeout: int) -> Dict[str, Any]:
+    def _check_database(self, target: dict, timeout: int) -> dict[str, Any]:
         """Perform database connectivity check."""
         # This is a simplified example - in production, you'd use actual database drivers
         db_type = target.get("db_type", "postgresql")
@@ -421,8 +421,8 @@ class HealthCheckNode(Node):
         return self._check_tcp({"host": host, "port": port}, timeout)
 
     def _generate_summary(
-        self, results: List[Dict], execution_time: float
-    ) -> Dict[str, Any]:
+        self, results: list[dict], execution_time: float
+    ) -> dict[str, Any]:
         """Generate summary statistics from health check results."""
         total_checks = len(results)
         healthy_checks = len([r for r in results if r.get("status") == "healthy"])

kailash/nodes/api/rate_limiting.py

@@ -17,7 +17,7 @@ import time
 from abc import ABC, abstractmethod
 from collections import deque
 from dataclasses import dataclass
-from typing import Any, Dict, Optional
+from typing import Any
 
 from kailash.nodes.base import Node, NodeParameter, register_node
 from kailash.nodes.base_async import AsyncNode
@@ -34,7 +34,7 @@ class RateLimitConfig:
     max_requests: int = 100  # Maximum requests allowed
     time_window: float = 60.0  # Time window in seconds
     strategy: str = "token_bucket"  # Rate limiting strategy
-    burst_limit: Optional[int] = None  # Maximum burst requests (for token bucket)
+    burst_limit: int | None = None  # Maximum burst requests (for token bucket)
     backoff_factor: float = 1.0  # Backoff factor when rate limited
     max_backoff: float = 300.0  # Maximum backoff time in seconds
 
@@ -61,7 +61,6 @@ class RateLimiter(ABC):
         Returns:
             True if request can proceed, False if rate limited
         """
-        pass
 
     @abstractmethod
     def wait_time(self) -> float:
@@ -70,7 +69,6 @@ class RateLimiter(ABC):
         Returns:
             Wait time in seconds (0 if can proceed immediately)
         """
-        pass
 
     @abstractmethod
     def consume(self) -> bool:
@@ -79,12 +77,10 @@ class RateLimiter(ABC):
         Returns:
             True if token was consumed, False if rate limited
         """
-        pass
 
     @abstractmethod
     def reset(self) -> None:
         """Reset the rate limiter state."""
-        pass
 
 
 class TokenBucketRateLimiter(RateLimiter):
@@ -295,7 +291,7 @@ class RateLimitedAPINode(Node):
         self.rate_limiter = create_rate_limiter(rate_limit_config)
         self.config = rate_limit_config
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -326,7 +322,7 @@ class RateLimitedAPINode(Node):
 
         return params
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -345,7 +341,7 @@ class RateLimitedAPINode(Node):
 
         return schema
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Execute the wrapped node with rate limiting.
 
         Args:
@@ -463,7 +459,7 @@ class AsyncRateLimitedAPINode(AsyncNode):
         self.config = rate_limit_config
         self.sync_node = RateLimitedAPINode(wrapped_node, rate_limit_config, **kwargs)
 
-    def get_parameters(self) -> Dict[str, NodeParameter]:
+    def get_parameters(self) -> dict[str, NodeParameter]:
         """Define the parameters this node accepts.
 
         Returns:
@@ -471,7 +467,7 @@ class AsyncRateLimitedAPINode(AsyncNode):
         """
         return self.sync_node.get_parameters()
 
-    def get_output_schema(self) -> Dict[str, NodeParameter]:
+    def get_output_schema(self) -> dict[str, NodeParameter]:
         """Define the output schema for this node.
 
         Returns:
@@ -479,7 +475,7 @@ class AsyncRateLimitedAPINode(AsyncNode):
         """
         return self.sync_node.get_output_schema()
 
-    def run(self, **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> dict[str, Any]:
         """Synchronous version for compatibility.
 
         Args:
@@ -490,7 +486,7 @@ class AsyncRateLimitedAPINode(AsyncNode):
         """
         return self.sync_node.run(**kwargs)
 
-    async def async_run(self, **kwargs) -> Dict[str, Any]:
+    async def async_run(self, **kwargs) -> dict[str, Any]:
         """Execute the wrapped async node with rate limiting.
 
         Args: