h-message-bus 0.0.31__tar.gz → 0.0.32__tar.gz

{h_message_bus-0.0.31 → h_message_bus-0.0.32}/PKG-INFO

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

{h_message_bus-0.0.31 → h_message_bus-0.0.32}/pyproject.toml

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

{h_message_bus-0.0.31 → h_message_bus-0.0.32}/src/h_message_bus/domain/models/request_message_topic.py

@@ -63,6 +63,9 @@ class RequestMessageTopic(str, Enum):
     GRAPH_GET_ALL = "hai.graph.get.all"
     GRAPH_NODES_BY_PROPERTY = "hai.graph.nodes.by.property"
 
+    GRAPH_QUERY_OPERATION = "hai.graph.query.operation"
+    GRAPH_QUERY_OPERATION_RESPONSE = "hai.graph.query.operation.response"
+
     GRAPH_NODE_ADD_RESPONSE = "hai.graph.node.add.response"
     GRAPH_NODE_UPDATE_RESPONSE = "hai.graph.node.update.response"
     GRAPH_NODE_GET_RESPONSE = "hai.graph.node.get.response"

h_message_bus-0.0.32/src/h_message_bus/domain/request_messages/graph_query_operation_request_message.py

@@ -0,0 +1,114 @@
+import json
+from typing import Dict, Any, Type, TypeVar
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+
+class GraphQueryOperationRequestMessage(HaiMessage):
+    """Message to request graph query operations"""
+
+    @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,
+                       operation_type: str,
+                       anchor_node: str,
+                       relationship_direction: str = None,
+                       relationship_type: str = None,
+                       limit: int = None,
+                       traversal_depth: int = None) -> 'GraphQueryOperationRequestMessage':
+        """
+        Create a message requesting a graph query operation
+
+        Args:
+            operation_type: One of 'FIND_RELATED_NODES', 'GET_NODE_INFO', 'COUNT_RELATIONSHIPS'
+            anchor_node: The central node in the query (e.g., 'hyperliquid')
+            relationship_direction: One of 'INCOMING', 'OUTGOING', or 'BOTH'
+            relationship_type: The relationship type to traverse (e.g., 'BUILDS_ON')
+            limit: Numerical limit of results to return
+            traversal_depth: How many relationship hops to traverse
+
+        Returns:
+            A new GraphQueryOperationRequestMessage
+        """
+        # Build the query operation dictionary
+        query_operation = {
+            "operation_type": operation_type,
+            "anchor_node": anchor_node,
+        }
+
+        # Add optional parameters if provided
+        if relationship_direction is not None:
+            query_operation["relationship_direction"] = relationship_direction
+
+        if relationship_type is not None:
+            query_operation["relationship_type"] = relationship_type
+
+        if limit is not None:
+            query_operation["limit"] = f"{limit}"
+
+        if traversal_depth is not None:
+            query_operation["traversal_depth"] = f"{traversal_depth}"
+
+        return cls.create(
+            topic=RequestMessageTopic.GRAPH_QUERY_OPERATION,
+            payload={"query_operation": json.dumps(query_operation)},
+        )
+
+    @property
+    def query_operation(self) -> Dict[str, Any]:
+        """Get the query operation from the payload"""
+        query_operation_payload = self.payload.get("query_operation", "{}")
+        return json.loads(query_operation_payload)
+
+    @property
+    def operation_type(self) -> str:
+        """Get the operation type from the query operation"""
+        return self.query_operation.get("operation_type", "")
+
+    @property
+    def anchor_node(self) -> str:
+        """Get the anchor node from the query operation"""
+        return self.query_operation.get("anchor_node", "")
+
+    @property
+    def relationship_direction(self) -> str:
+        """Get the relationship direction from the query operation"""
+        return self.query_operation.get("relationship_direction", "BOTH")
+
+    @property
+    def relationship_type(self) -> str:
+        """Get the relationship type from the query operation"""
+        return self.query_operation.get("relationship_type", "")
+
+    @property
+    def limit(self) -> int:
+        """Get the limit from the query operation"""
+        return int(self.query_operation.get("limit", 10))
+
+    @property
+    def traversal_depth(self) -> int:
+        """Get the traversal depth from the query operation"""
+        return int(self.query_operation.get("traversal_depth", 1))
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'GraphQueryOperationRequestMessage':
+        """Create a GraphQueryOperationRequestMessage from a HaiMessage"""
+        payload = message.payload
+        query_operation_str = payload.get("query_operation", "{}")
+        query_operation = json.loads(query_operation_str)
+
+        return cls.create_message(
+            operation_type=query_operation.get("operation_type", ""),
+            anchor_node=query_operation.get("anchor_node", ""),
+            relationship_direction=query_operation.get("relationship_direction"),
+            relationship_type=query_operation.get("relationship_type"),
+            limit=query_operation.get("limit"),
+            traversal_depth=query_operation.get("traversal_depth")
+        )

h_message_bus-0.0.32/src/h_message_bus/domain/request_messages/graph_query_operation_response_message.py

@@ -0,0 +1,85 @@
+import json
+from typing import Dict, Any, TypeVar, Type
+
+from ...domain.models.hai_message import HaiMessage
+from ..models.request_message_topic import RequestMessageTopic
+
+T = TypeVar('T', bound='HaiMessage')
+
+class GraphQueryOperationResponseMessage(HaiMessage):
+    """Message with results from graph query operations"""
+
+    @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,
+                       operation_type: str,
+                       result: Dict[str, Any],
+                       success: bool = True,
+                       error_message: str = None) -> 'GraphQueryOperationResponseMessage':
+        """
+        Create a response message with graph query operation results
+
+        Args:
+            operation_type: The type of operation that was performed
+            result: Dictionary containing the operation results
+            success: Whether the operation was successful
+            error_message: Error message if the operation failed
+
+        Returns:
+            A new GraphQueryOperationResponseMessage
+        """
+        payload = {
+            "operation_type": operation_type,
+            "result": json.dumps(result),
+            "success": success
+        }
+
+        if error_message:
+            payload["error_message"] = error_message
+
+        return cls.create(
+            topic=RequestMessageTopic.GRAPH_QUERY_OPERATION_RESPONSE,
+            payload=payload,
+        )
+
+    @property
+    def operation_type(self) -> str:
+        """Get the operation type from the payload"""
+        return self.payload.get("operation_type", "")
+
+    @property
+    def result(self) -> Dict[str, Any]:
+        """Get the result from the payload"""
+        result_str = self.payload.get("result", "{}")
+        return json.loads(result_str)
+
+    @property
+    def success(self) -> bool:
+        """Get the success status from the payload"""
+        return self.payload.get("success", False)
+
+    @property
+    def error_message(self) -> str:
+        """Get the error message from the payload"""
+        return self.payload.get("error_message", "")
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'GraphQueryOperationResponseMessage':
+        """Create a GraphQueryOperationResponseMessage from a HaiMessage"""
+        payload = message.payload
+        operation_type = payload.get("operation_type", "")
+        result_str = payload.get("result", "{}")
+        result = json.loads(result_str)
+        success = payload.get("success", False)
+        error_message = payload.get("error_message")
+
+        return cls.create_message(
+            operation_type=operation_type,
+            result=result,
+            success=success,
+            error_message=error_message
+        )

{h_message_bus-0.0.31 → h_message_bus-0.0.32}/src/h_message_bus.egg-info/PKG-INFO

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

{h_message_bus-0.0.31 → h_message_bus-0.0.32}/src/h_message_bus.egg-info/SOURCES.txt

@@ -37,6 +37,8 @@ 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_operation_request_message.py
+src/h_message_bus/domain/request_messages/graph_query_operation_response_message.py
 src/h_message_bus/domain/request_messages/graph_query_request_message.py
 src/h_message_bus/domain/request_messages/graph_relationship_added_request_message.py
 src/h_message_bus/domain/request_messages/init_knowledgebase_request.py