h-message-bus 0.0.28__tar.gz → 0.0.30__tar.gz

{h_message_bus-0.0.28 → h_message_bus-0.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_message_bus
-Version: 0.0.28
+Version: 0.0.30
 Summary: Message bus integration for HAI
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3

{h_message_bus-0.0.28 → h_message_bus-0.0.30}/pyproject.toml

@@ -4,7 +4,7 @@
 
    [project]
    name = "h_message_bus"
-   version = "0.0.28"
+   version = "0.0.30"
    authors = [
        {name = "shoebill", email = "shoebill.hai@gmail.com"},
    ]

h_message_bus-0.0.30/src/h_message_bus/domain/request_messages/graph_node_update_response_message.py

@@ -0,0 +1,69 @@
+from typing import TypeVar, Dict, Any, Type, Optional
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class GraphNodeUpdateResponseMessage(HaiMessage):
+    """Message indicating a node was successfully added to the graph"""
+
+    @classmethod
+    def create(cls: Type[T], topic: str, payload: Dict[Any, Any]) -> T:
+        """Create a message - inherited from HaiMessage"""
+        return super().create(topic=topic, payload=payload)
+
+    @classmethod
+    def create_message(cls, node_id: str, label: str, properties: dict = None,
+                       description: str = None, success: bool = True) -> 'GraphNodeUpdateResponseMessage':
+        """Create a message confirming a node was added"""
+        if properties is None:
+            properties = {}
+
+        return cls.create(
+            topic=RequestMessageTopic.GRAPH_NODE_UPDATE_RESPONSE,
+            payload={
+                "node_id": node_id,
+                "label": label,
+                "properties": properties,
+                "description": description,
+                "success": success
+            },
+        )
+
+    @property
+    def node_id(self) -> str:
+        """Get the node ID from the payload"""
+        return self.payload.get("node_id")
+
+    @property
+    def label(self) -> str:
+        """Get the label from the payload"""
+        return self.payload.get("label")
+
+    @property
+    def properties(self) -> dict:
+        """Get the properties from the payload"""
+        return self.payload.get("properties", {})
+
+    @property
+    def description(self) -> Optional[str]:
+        """Get the description from the payload"""
+        return self.payload.get("description")
+
+    @property
+    def success(self) -> bool:
+        """Check if the operation was successful"""
+        return self.payload.get("success", False)
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'GraphNodeUpdateResponseMessage':
+        # Extract the necessary fields from the message payload
+        payload = message.payload
+
+        return cls.create_message(
+            node_id=payload.get("node_id", ''),
+            label=payload.get("label", ''),
+            properties=payload.get("properties", {}),
+            description=payload.get("description")
+        )

{h_message_bus-0.0.28 → h_message_bus-0.0.30}/src/h_message_bus.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_message_bus
-Version: 0.0.28
+Version: 0.0.30
 Summary: Message bus integration for HAI
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3

{h_message_bus-0.0.28 → h_message_bus-0.0.30}/src/h_message_bus.egg-info/SOURCES.txt

@@ -34,6 +34,7 @@ src/h_message_bus/domain/request_messages/graph_node_added_response_message.py
 src/h_message_bus/domain/request_messages/graph_node_get_request_message.py
 src/h_message_bus/domain/request_messages/graph_node_get_result_response_message.py
 src/h_message_bus/domain/request_messages/graph_node_update_request_message.py
+src/h_message_bus/domain/request_messages/graph_node_update_response_message.py
 src/h_message_bus/domain/request_messages/graph_nodes_by_property_request_message.py
 src/h_message_bus/domain/request_messages/graph_nodes_by_property_response_message.py
 src/h_message_bus/domain/request_messages/graph_query_request_message.py