h-message-bus 0.0.30__py3-none-any.whl → 0.0.32__py3-none-any.whl

h_message_bus/domain/models/request_message_topic.py

@@ -48,6 +48,10 @@ class RequestMessageTopic(str, Enum):
     WEB_FIND_RELATED = "hai.tools.web.find.related"
     WEB_FIND_RELATED_RESPONSE = "hai.tools.web.find.related.response"
 
+    INIT_KNOWLEDGE_BASE = "hai.tools.init.knowledge.base"
+    INIT_KNOWLEDGE_BASE_RESPONSE = "hai.tools.init.knowledge.base.response"
+
+    # graph database
     GRAPH_NODE_ADD = "hai.graph.node.add"
     GRAPH_NODE_UPDATE = "hai.graph.node.update"
     GRAPH_NODE_GET = "hai.graph.node.get"
@@ -59,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/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/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/domain/request_messages/init_knowledgebase_request.py

@@ -0,0 +1,26 @@
+from typing import TypeVar, Dict, Any, Type
+
+from ..models.request_message_topic import RequestMessageTopic
+from ...domain.models.hai_message import HaiMessage
+
+T = TypeVar('T', bound='HaiMessage')
+
+class InitKnowledgeBaseRequestMessage(HaiMessage):
+
+
+    @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) -> 'InitKnowledgeBaseRequestMessage':
+        """Create a message requesting to clear the graph"""
+        return cls.create(
+            topic=RequestMessageTopic.INIT_KNOWLEDGE_BASE,
+            payload={}
+        )
+
+    @classmethod
+    def from_hai_message(cls, message: HaiMessage) -> 'InitKnowledgeBaseRequestMessage':
+        return cls.create_message()

{h_message_bus-0.0.30.dist-info → h_message_bus-0.0.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_message_bus
-Version: 0.0.30
+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.30.dist-info → h_message_bus-0.0.32.dist-info}/RECORD

@@ -13,7 +13,7 @@ h_message_bus/domain/event_messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 h_message_bus/domain/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/domain/models/event_message_topic.py,sha256=fSjRMBwMD2RE9GBUud38XMrgLZcngURERrbuoAhQ0Hk,717
 h_message_bus/domain/models/hai_message.py,sha256=b5CfX7hi5uNq77IVnZzEi9iotc4b_U2MNYwV6JY7JcU,2146
-h_message_bus/domain/models/request_message_topic.py,sha256=JA8BXe4-cjyrfhLyu_y22AUx2QPLUNi3k-8QYGff_Jo,2851
+h_message_bus/domain/models/request_message_topic.py,sha256=Ec9ilhsR0ylcSqz0Uv11Eim-q9sBVm4JH8yg4IVh9lE,3145
 h_message_bus/domain/models/twitter_message_metadata.py,sha256=Cj0dSVQi0Com9nPlJeEM0c-ZAQKGqIq_txvdzYg0myw,1203
 h_message_bus/domain/models/twitter_user_metadata.py,sha256=N7GVyn2txmlMQ19aatNh7lvzO6FNSj4dKVujestha9o,2827
 h_message_bus/domain/models/vector_collection_metadata.py,sha256=fSy3ZQ-zkYISTpc-_M6XWf1JkufVAlzlS2ZT60TzA2s,891
@@ -30,8 +30,11 @@ h_message_bus/domain/request_messages/graph_node_update_request_message.py,sha25
 h_message_bus/domain/request_messages/graph_node_update_response_message.py,sha256=VMCOF4FnKNvTItex8daeR2BSwN9juubj0ksHPHh0BT4,2424
 h_message_bus/domain/request_messages/graph_nodes_by_property_request_message.py,sha256=L-oZfZqsI9DG-CFL0G7r-2A-6wTqFuMNkDww2JUR73s,1712
 h_message_bus/domain/request_messages/graph_nodes_by_property_response_message.py,sha256=UDCA769I14BNP2Qmfwg5c6m98nBNVSZ10Kpi8Z8AGAQ,3051
+h_message_bus/domain/request_messages/graph_query_operation_request_message.py,sha256=vc0NZQpjFDJT4ms0zsoSluBOZxePVuXuZZSDHlCKHPU,4568
+h_message_bus/domain/request_messages/graph_query_operation_response_message.py,sha256=-rML2iTebOmkN-_UczA4VZd6kMHbgeqjMtuPypRVuuY,3004
 h_message_bus/domain/request_messages/graph_query_request_message.py,sha256=UXs_m6pgPkYRSRHrhI-c3WAQ31Yj1idZBgavufm32s0,1635
 h_message_bus/domain/request_messages/graph_relationship_added_request_message.py,sha256=a2hJyF78ITCjkvdR_v_NbnUvxFkolnCyN1-71rF-J2I,2388
+h_message_bus/domain/request_messages/init_knowledgebase_request.py,sha256=ydX5J784wW0aoXosffSJBAQ6xFmKY5Y8bsWITaIJqbY,891
 h_message_bus/domain/request_messages/tg_message_request_message.py,sha256=bta24fXwwxTMCwdDHlbB20T-0SNkNWdPwNsQbo_OGN0,1447
 h_message_bus/domain/request_messages/tg_user_message_reply_request_message.py,sha256=desDxVddPTp7iOYq7126zDEyaJF1bu9LWEamj-e87IA,1767
 h_message_bus/domain/request_messages/tg_user_message_request_message.py,sha256=pHmmGJpMEnwzbTVVvoD6UOMflQOuQKgErZA9MyRwWt8,3121
@@ -51,7 +54,7 @@ h_message_bus/domain/request_messages/web_search_request_message.py,sha256=ZoB4i
 h_message_bus/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h_message_bus/infrastructure/nats_client_repository.py,sha256=bhaWRK94h2EmFIef2DALnoBJKnuTnx6-iU7IXO2S_EU,3642
 h_message_bus/infrastructure/nats_config.py,sha256=Yzqqd1bCfmUv_4FOnA1dvqIpakzV0BUL2_nXQcndWvo,1304
-h_message_bus-0.0.30.dist-info/METADATA,sha256=rg9Z6lf9ukHebwDXUxmSq52eFtwdATGhXn8afJkmU80,8834
-h_message_bus-0.0.30.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-h_message_bus-0.0.30.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
-h_message_bus-0.0.30.dist-info/RECORD,,
+h_message_bus-0.0.32.dist-info/METADATA,sha256=EDqM6sULqdLtkUqAgf_4S2SDXeD78CPorlnl9GbK72o,8834
+h_message_bus-0.0.32.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+h_message_bus-0.0.32.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
+h_message_bus-0.0.32.dist-info/RECORD,,