kailash 0.6.2__py3-none-any.whl → 0.6.4__py3-none-any.whl

kailash/nodes/rag/similarity.py

@@ -53,7 +53,7 @@ class DenseRetrievalNode(Node):
         )
 
         # Finds semantically similar content even without exact keywords
-        results = await dense_retriever.run(
+        results = await dense_retriever.execute(
             query="How to make AI systems more intelligent",
             documents=documents
         )
@@ -189,7 +189,7 @@ class SparseRetrievalNode(Node):
         )
 
         # Excellent for technical queries with specific terms
-        results = await sparse_retriever.run(
+        results = await sparse_retriever.execute(
             query="sklearn RandomForestClassifier hyperparameters",
             documents=technical_docs
         )
@@ -478,7 +478,7 @@ class ColBERTRetrievalNode(Node):
         )
 
         # Excellent for queries with multiple specific requirements
-        results = await colbert.run(
+        results = await colbert.execute(
             query="transformer architecture with attention mechanism for NLP tasks",
             documents=research_papers
         )
@@ -739,7 +739,7 @@ class MultiVectorRetrievalNode(Node):
         multi_vector = MultiVectorRetrievalNode()
 
         # Retrieves based on full content + summary + keywords
-        results = await multi_vector.run(
+        results = await multi_vector.execute(
             query="machine learning optimization techniques",
             documents=long_documents
         )
@@ -1037,7 +1037,7 @@ class CrossEncoderRerankNode(Node):
         )
 
         # Reranks initial results for maximum precision
-        reranked = await reranker.run(
+        reranked = await reranker.execute(
             initial_results=fast_retrieval_results,
             query="specific implementation details of BERT fine-tuning"
         )
@@ -1277,7 +1277,7 @@ class HybridFusionNode(Node):
         )
 
         # Combines dense and sparse retrieval results optimally
-        fused = await fusion.run(
+        fused = await fusion.execute(
             retrieval_results=[dense_results, sparse_results]
         )
 
@@ -1568,7 +1568,7 @@ class PropositionBasedRetrievalNode(Node):
         proposition_rag = PropositionBasedRetrievalNode()
 
         # Retrieves specific facts without surrounding noise
-        facts = await proposition_rag.run(
+        facts = await proposition_rag.execute(
             documents=knowledge_base,
             query="What is the speed of light in vacuum?"
         )

kailash/nodes/rag/strategies.py

@@ -412,7 +412,7 @@ class SemanticRAGNode(Node):
             self.workflow_node = create_semantic_rag_workflow(self.config)
 
         # Delegate to WorkflowNode
-        return self.workflow_node.run(**kwargs)
+        return self.workflow_node.execute(**kwargs)
 
 
 @register_node()
@@ -458,7 +458,7 @@ class StatisticalRAGNode(Node):
         if not self.workflow_node:
             self.workflow_node = create_statistical_rag_workflow(self.config)
 
-        return self.workflow_node.run(**kwargs)
+        return self.workflow_node.execute(**kwargs)
 
 
 @register_node()
@@ -517,7 +517,7 @@ class HybridRAGNode(Node):
             self.fusion_method = fusion_method
             self.workflow_node = create_hybrid_rag_workflow(self.config, fusion_method)
 
-        return self.workflow_node.run(**kwargs)
+        return self.workflow_node.execute(**kwargs)
 
 
 @register_node()
@@ -563,4 +563,4 @@ class HierarchicalRAGNode(Node):
         if not self.workflow_node:
             self.workflow_node = create_hierarchical_rag_workflow(self.config)
 
-        return self.workflow_node.run(**kwargs)
+        return self.workflow_node.execute(**kwargs)

kailash/nodes/security/abac_evaluator.py

@@ -194,7 +194,7 @@ class ABACPermissionEvaluatorNode(SecurityMixin, PerformanceMixin, LoggingMixin,
         ...     "network": "corporate"
         ... }
         >>>
-        >>> result = evaluator.run(
+        >>> result = evaluator.execute(
         ...     user_context=user_context,
         ...     resource_context=resource_context,
         ...     environment_context=env_context,
@@ -770,7 +770,7 @@ class ABACPermissionEvaluatorNode(SecurityMixin, PerformanceMixin, LoggingMixin,
 
     async def execute_async(self, **kwargs) -> Dict[str, Any]:
         """Async execution method for test compatibility."""
-        return self.run(**kwargs)
+        return self.execute(**kwargs)
 
     def _evaluate_permission(
         self, context: ABACContext, permission: str
@@ -1026,7 +1026,7 @@ class ABACPermissionEvaluatorNode(SecurityMixin, PerformanceMixin, LoggingMixin,
             )
 
             # Run AI analysis
-            ai_response = self.ai_agent.run(
+            ai_response = self.ai_agent.execute(
                 provider="ollama",
                 model=self.ai_model.replace("ollama:", ""),
                 messages=[{"role": "user", "content": prompt}],
@@ -1275,7 +1275,7 @@ Return a JSON object with this structure:
         }
 
         try:
-            self.audit_log_node.run(**audit_entry)
+            self.audit_log_node.execute(**audit_entry)
         except Exception as e:
             self.log_with_context(
                 "WARNING", f"Failed to audit permission decision: {e}"
@@ -1355,7 +1355,7 @@ RESPONSE FORMAT:
 }}
 """
 
-            ai_response = self.ai_agent.run(
+            ai_response = self.ai_agent.execute(
                 provider="ollama",
                 model=self.ai_model.replace("ollama:", ""),
                 messages=[{"role": "user", "content": prompt}],
@@ -1408,4 +1408,4 @@ RESPONSE FORMAT:
 
     async def async_run(self, **kwargs) -> Dict[str, Any]:
         """Async execution method for enterprise integration."""
-        return self.run(**kwargs)
+        return self.execute(**kwargs)

kailash/nodes/security/behavior_analysis.py

@@ -16,7 +16,6 @@ from datetime import UTC, datetime, timedelta
 from typing import Any, Dict, List, Optional, Set, Tuple
 
 import numpy as np
-
 from kailash.nodes.ai.llm_agent import LLMAgentNode
 from kailash.nodes.base import Node, NodeParameter, register_node
 from kailash.nodes.mixins import LoggingMixin, PerformanceMixin, SecurityMixin
@@ -107,7 +106,7 @@ class BehaviorAnalysisNode(SecurityMixin, PerformanceMixin, LoggingMixin, Node):
         ...     "data_volume_mb": 15.5
         ... }
         >>>
-        >>> result = behavior_analyzer.run(
+        >>> result = behavior_analyzer.execute(
         ...     action="analyze",
         ...     user_id="user123",
         ...     recent_activity=[activity]
@@ -417,7 +416,7 @@ class BehaviorAnalysisNode(SecurityMixin, PerformanceMixin, LoggingMixin, Node):
             # Create audit log entry
             if result.get("success", False):
                 try:
-                    self.audit_log_node.run(
+                    self.audit_log_node.execute(
                         action="behavior_analysis",
                         user_id=user_id or "unknown",
                         result="success",
@@ -1185,7 +1184,7 @@ RESPONSE FORMAT:
 """
 
             # Run AI analysis
-            ai_response = self.ai_agent.run(
+            ai_response = self.ai_agent.execute(
                 provider="ollama",
                 model=self.ai_model.replace("ollama:", ""),
                 messages=[{"role": "user", "content": prompt}],
@@ -1370,7 +1369,7 @@ RESPONSE FORMAT:
         }
 
         try:
-            self.security_event_node.run(**security_event)
+            self.security_event_node.execute(**security_event)
         except Exception as e:
             self.log_with_context("WARNING", f"Failed to log anomaly event: {e}")
 
@@ -1890,4 +1889,4 @@ RESPONSE FORMAT:
 
     async def async_run(self, **kwargs) -> Dict[str, Any]:
         """Async execution method for enterprise integration."""
-        return self.run(**kwargs)
+        return self.execute(**kwargs)

kailash/nodes/security/credential_manager.py

@@ -398,4 +398,4 @@ class CredentialManagerNode(Node):
 
     async def async_run(self, **kwargs) -> Dict[str, Any]:
         """Async execution method for enterprise integration."""
-        return self.run(**kwargs)
+        return self.execute(**kwargs)

kailash/nodes/security/rotating_credentials.py

@@ -255,7 +255,7 @@ class RotatingCredentialNode(Node):
         """Check if a credential is approaching expiration."""
         try:
             # Get current credential
-            credential_result = self._credential_manager.run(
+            credential_result = self._credential_manager.execute(
                 operation="get_credential", credential_name=credential_name
             )
 
@@ -314,7 +314,7 @@ class RotatingCredentialNode(Node):
             # Try each refresh source in order
             for source in refresh_sources:
                 try:
-                    refresh_result = self._credential_manager.run(
+                    refresh_result = self._credential_manager.execute(
                         operation="get_credential",
                         credential_name=credential_name,
                         credential_sources=[source],
@@ -363,7 +363,7 @@ class RotatingCredentialNode(Node):
             # Step 1: Get current credential (for rollback if needed)
             current_credential = None
             if rollback_on_failure:
-                current_result = self._credential_manager.run(
+                current_result = self._credential_manager.execute(
                     operation="get_credential", credential_name=credential_name
                 )
                 if current_result.get("success"):
@@ -390,7 +390,7 @@ class RotatingCredentialNode(Node):
             new_credential = refresh_result["credential"]
 
             # Step 3: Validate new credential
-            validation_result = self._credential_manager.run(
+            validation_result = self._credential_manager.execute(
                 operation="validate_credential",
                 credential_name=credential_name,
                 credential_data=new_credential,
@@ -422,7 +422,7 @@ class RotatingCredentialNode(Node):
 
                 temp_credential_name = f"{credential_name}_rotating_{int(time.time())}"
 
-                store_result = self._credential_manager.run(
+                store_result = self._credential_manager.execute(
                     operation="store_credential",
                     credential_name=temp_credential_name,
                     credential_data=new_credential,
@@ -445,7 +445,7 @@ class RotatingCredentialNode(Node):
                 # For this example, we'll assume it passes
 
                 # Atomic switch
-                final_store_result = self._credential_manager.run(
+                final_store_result = self._credential_manager.execute(
                     operation="store_credential",
                     credential_name=credential_name,
                     credential_data=new_credential,
@@ -454,7 +454,7 @@ class RotatingCredentialNode(Node):
                 if not final_store_result.get("success"):
                     # Rollback if requested
                     if rollback_on_failure and current_credential:
-                        self._credential_manager.run(
+                        self._credential_manager.execute(
                             operation="store_credential",
                             credential_name=credential_name,
                             credential_data=current_credential,
@@ -476,13 +476,13 @@ class RotatingCredentialNode(Node):
                     }
 
                 # Clean up temporary credential
-                self._credential_manager.run(
+                self._credential_manager.execute(
                     operation="delete_credential", credential_name=temp_credential_name
                 )
 
             else:
                 # Direct replacement
-                store_result = self._credential_manager.run(
+                store_result = self._credential_manager.execute(
                     operation="store_credential",
                     credential_name=credential_name,
                     credential_data=new_credential,
@@ -491,7 +491,7 @@ class RotatingCredentialNode(Node):
                 if not store_result.get("success"):
                     # Rollback if requested
                     if rollback_on_failure and current_credential:
-                        self._credential_manager.run(
+                        self._credential_manager.execute(
                             operation="store_credential",
                             credential_name=credential_name,
                             credential_data=current_credential,
@@ -757,4 +757,4 @@ class RotatingCredentialNode(Node):
 
     async def async_run(self, **kwargs) -> Dict[str, Any]:
         """Async execution method for enterprise integration."""
-        return self.run(**kwargs)
+        return self.execute(**kwargs)

kailash/nodes/security/threat_detection.py

@@ -45,7 +45,7 @@ class ThreatDetectionNode(SecurityMixin, PerformanceMixin, LoggingMixin, Node):
         ...     {"type": "login", "user": "admin", "ip": "192.168.1.100", "failed": True}
         ... ]
         >>>
-        >>> threats = threat_detector.run(events=events)
+        >>> threats = threat_detector.execute(events=events)
         >>> print(f"Detected {len(threats['threats'])} threats")
     """
 
@@ -661,7 +661,7 @@ class ThreatDetectionNode(SecurityMixin, PerformanceMixin, LoggingMixin, Node):
             prompt = self._create_ai_analysis_prompt(event_summary, context)
 
             # Run AI analysis
-            ai_response = self.ai_agent.run(
+            ai_response = self.ai_agent.execute(
                 provider="ollama",
                 model=self.ai_model.replace("ollama:", ""),
                 messages=[{"role": "user", "content": prompt}],
@@ -923,7 +923,7 @@ If no threats are detected, return an empty array: []
             "source_ip": threat.get("source_ip", "unknown"),
         }
 
-        self.security_event_node.run(**alert_event)
+        self.security_event_node.execute(**alert_event)
 
     def _block_ip(self, threat: Dict[str, Any]) -> None:
         """Block IP address associated with threat.
@@ -949,7 +949,7 @@ If no threats are detected, return an empty array: []
                 "source_ip": ip,
             }
 
-            self.security_event_node.run(**block_event)
+            self.security_event_node.execute(**block_event)
 
     def _lock_account(self, threat: Dict[str, Any]) -> None:
         """Lock user account associated with threat.
@@ -974,7 +974,7 @@ If no threats are detected, return an empty array: []
                 "source_ip": threat.get("source_ip", "unknown"),
             }
 
-            self.security_event_node.run(**lock_event)
+            self.security_event_node.execute(**lock_event)
 
     def _quarantine_resource(self, threat: Dict[str, Any]) -> None:
         """Quarantine resource associated with threat.
@@ -996,7 +996,7 @@ If no threats are detected, return an empty array: []
             "source_ip": threat.get("source_ip", "unknown"),
         }
 
-        self.security_event_node.run(**quarantine_event)
+        self.security_event_node.execute(**quarantine_event)
 
     def _log_threat(self, threat: Dict[str, Any]) -> None:
         """Log threat to audit trail.
@@ -1014,7 +1014,7 @@ If no threats are detected, return an empty array: []
             "ip_address": threat.get("source_ip", "unknown"),
         }
 
-        self.audit_log_node.run(**log_entry)
+        self.audit_log_node.execute(**log_entry)
 
     def _update_stats(
         self, events_processed: int, threats_detected: int, processing_time_ms: float
@@ -1100,4 +1100,4 @@ If no threats are detected, return an empty array: []
 
     async def async_run(self, **kwargs) -> Dict[str, Any]:
         """Async execution method for enterprise integration."""
-        return self.run(**kwargs)
+        return self.execute(**kwargs)

kailash/nodes/testing/credential_testing.py

@@ -39,7 +39,7 @@ class CredentialTestingNode(Node):
     Example:
         >>> # Test OAuth2 token expiration
         >>> tester = CredentialTestingNode()
-        >>> result = tester.run(
+        >>> result = tester.execute(
         ...     credential_type='oauth2',
         ...     scenario='expired',
         ...     mock_data={'client_id': 'test_client'}
@@ -48,7 +48,7 @@ class CredentialTestingNode(Node):
         >>> assert 'expired_token' in result['error_details']
         >>>
         >>> # Test successful API key validation
-        >>> result = tester.run(
+        >>> result = tester.execute(
         ...     credential_type='api_key',
         ...     scenario='success',
         ...     validation_rules={'key_length': 32}

kailash/nodes/transform/processors.py

@@ -68,7 +68,7 @@ class FilterNode(Node):
     Examples:
         >>> # Filter list of numbers
         >>> filter_node = FilterNode()
-        >>> result = filter_node.run(
+        >>> result = filter_node.execute(
         ...     data=[1, 2, 3, 4, 5],
         ...     operator=">",
         ...     value=3
@@ -81,7 +81,7 @@ class FilterNode(Node):
         ...     {"name": "Bob", "age": 25},
         ...     {"name": "Charlie", "age": 35}
         ... ]
-        >>> result = filter_node.run(
+        >>> result = filter_node.execute(
         ...     data=users,
         ...     field="age",
         ...     operator=">=",
@@ -96,7 +96,7 @@ class FilterNode(Node):
         ...     {"title": "Java Development"},
         ...     {"title": "Python for Data Science"}
         ... ]
-        >>> result = filter_node.run(
+        >>> result = filter_node.execute(
         ...     data=items,
         ...     field="title",
         ...     operator="contains",
@@ -110,7 +110,7 @@ class FilterNode(Node):
         ...     {"value": None},
         ...     {"value": 20}
         ... ]
-        >>> result = filter_node.run(
+        >>> result = filter_node.execute(
         ...     data=data_with_nulls,
         ...     field="value",
         ...     operator="!=",
@@ -570,7 +570,7 @@ class ContextualCompressorNode(Node):
         ...     compression_ratio=0.6,
         ...     relevance_threshold=0.7
         ... )
-        >>> result = compressor.run(
+        >>> result = compressor.execute(
         ...     query="machine learning algorithms",
         ...     retrieved_docs=[{"content": "...", "metadata": {}}],
         ...     compression_target=1500