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

kailash/nodes/data/streaming.py

@@ -138,16 +138,31 @@ class KafkaConsumerNode(Node):
         tags={"streaming", "kafka", "consumer"},
     )
 
-    def __init__(self):
+    def __init__(self, name: str = None, id: str = None, **kwargs):
         """Initialize the Kafka consumer node.
 
         Sets up the node and prepares for consumer initialization.
         The actual consumer is created during configuration.
         """
-        super().__init__()
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self._consumer = None
         self._topic = None
 
+        # Call parent constructor with all kwargs
+        super().__init__(name=self.name, **kwargs)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Define parameters for the Kafka consumer node."""
         return {
@@ -411,15 +426,30 @@ class StreamPublisherNode(Node):
         tags={"streaming", "publisher", "messaging"},
     )
 
-    def __init__(self):
+    def __init__(self, name: str = None, id: str = None, **kwargs):
         """Initialize the stream publisher node.
 
         Sets up the node and prepares for publisher initialization.
         """
-        super().__init__()
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self._publisher = None
         self._protocol = None
 
+        # Call parent constructor with all kwargs
+        super().__init__(name=self.name, **kwargs)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Define parameters for the stream publisher node."""
         return {
@@ -673,16 +703,31 @@ class WebSocketNode(Node):
         tags={"streaming", "websocket", "data"},
     )
 
-    def __init__(self):
+    def __init__(self, name: str = None, id: str = None, **kwargs):
         """Initialize the WebSocket node.
 
         Sets up the node and prepares for WebSocket connection.
         """
-        super().__init__()
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self._ws = None
         self._connected = False
         self._message_queue = []
 
+        # Call parent constructor with all kwargs
+        super().__init__(name=self.name, **kwargs)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Get the parameters for this node.
 
@@ -984,16 +1029,31 @@ class EventStreamNode(Node):
         tags={"streaming", "sse", "data"},
     )
 
-    def __init__(self):
+    def __init__(self, name: str = None, id: str = None, **kwargs):
         """Initialize the EventStream node.
 
         Sets up the node and prepares for SSE connection.
         """
-        super().__init__()
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self._stream = None
         self._connected = False
         self._last_event_id = None
 
+        # Call parent constructor with all kwargs
+        super().__init__(name=self.name, **kwargs)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Get the parameters for this node.
 

kailash/nodes/data/vector_db.py

@@ -132,16 +132,31 @@ class EmbeddingNode(Node):
         tags={"embedding", "nlp", "vector"},
     )
 
-    def __init__(self):
+    def __init__(self, name: str = None, id: str = None, **kwargs):
         """Initialize the embedding node.
 
         Sets up the node with default configuration and prepares for
         model initialization. The actual model is loaded during configuration.
         """
-        super().__init__()
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self._model = None
         self._model_info = {}
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Define parameters for the embedding node."""
         return {
@@ -432,16 +447,31 @@ class VectorDatabaseNode(Node):
         tags={"vector", "database", "storage"},
     )
 
-    def __init__(self):
+    def __init__(self, name: str = None, id: str = None, **kwargs):
         """Initialize the vector database node.
 
         Sets up the node and prepares for database connection.
         The actual connection is established during configuration.
         """
-        super().__init__()
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self._client = None
         self._index = None
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Define parameters for the vector database node."""
         return {
@@ -748,6 +778,26 @@ class TextSplitterNode(Node):
         tags={"text", "processing", "nlp"},
     )
 
+    def __init__(self, name: str = None, id: str = None, **kwargs):
+        """Initialize the text splitter node.
+
+        Sets up the node with default configuration.
+        """
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         """Define parameters for the text splitter node."""
         return {

kailash/nodes/enterprise/__init__.py

@@ -0,0 +1,13 @@
+"""Enterprise integration nodes for the Kailash SDK.
+
+This module provides enterprise-grade nodes for complex business integration
+patterns, data lineage tracking, and batch processing optimizations.
+"""
+
+from kailash.nodes.enterprise.batch_processor import BatchProcessorNode
+from kailash.nodes.enterprise.data_lineage import DataLineageNode
+
+__all__ = [
+    "DataLineageNode",
+    "BatchProcessorNode",
+]