h-message-bus 0.0.29__py3-none-any.whl → 0.0.31__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"

h_message_bus/domain/request_messages/graph_node_update_response_message.py

@@ -15,7 +15,7 @@ class GraphNodeUpdateResponseMessage(HaiMessage):
 
     @classmethod
     def create_message(cls, node_id: str, label: str, properties: dict = None,
-                       description: str = None) -> 'GraphNodeUpdateResponseMessage':
+                       description: str = None, success: bool = True) -> 'GraphNodeUpdateResponseMessage':
         """Create a message confirming a node was added"""
         if properties is None:
             properties = {}
@@ -27,7 +27,7 @@ class GraphNodeUpdateResponseMessage(HaiMessage):
                 "label": label,
                 "properties": properties,
                 "description": description,
-                "success": True
+                "success": success
             },
         )
 

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.29.dist-info → h_message_bus-0.0.31.dist-info}/METADATA

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

{h_message_bus-0.0.29.dist-info → h_message_bus-0.0.31.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=mfELwwQ6Kxdt_Bxrt0OpEZTNoYQPJbtsCYSEhxVNOWU,3011
 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
@@ -27,11 +27,12 @@ h_message_bus/domain/request_messages/graph_node_added_response_message.py,sha25
 h_message_bus/domain/request_messages/graph_node_get_request_message.py,sha256=G_JlCnwU6a9HEU0gFt8ilOwx1KF-FJC0EqpWw1C4km8,1302
 h_message_bus/domain/request_messages/graph_node_get_result_response_message.py,sha256=kzde_K65UezKm2NLSFehiLrDAVTHXZM9Gd-GDYUZ1dg,2457
 h_message_bus/domain/request_messages/graph_node_update_request_message.py,sha256=gNXcQmpPF6Ut6M8wtRx7Xi7XdpCT9ZJvqPzNoSdwUZQ,1977
-h_message_bus/domain/request_messages/graph_node_update_response_message.py,sha256=Adm_4H9n1NliW4VqSjT8P_Bmi7l-Ktk1cTgqwLH74BM,2399
+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_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 +52,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.29.dist-info/METADATA,sha256=TuE20kbc6p0TNq_-xeMtoTQ5t95eSbO2Z-QjVGqSu18,8834
-h_message_bus-0.0.29.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-h_message_bus-0.0.29.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
-h_message_bus-0.0.29.dist-info/RECORD,,
+h_message_bus-0.0.31.dist-info/METADATA,sha256=LLckLQauJsj94-bkox63f-OgJ4CQ4HFn8hK4br7FgLE,8834
+h_message_bus-0.0.31.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+h_message_bus-0.0.31.dist-info/top_level.txt,sha256=BArjhm_lwFR9yJJEIf-LT_X64psuLkXFdbpQRJUreFE,23
+h_message_bus-0.0.31.dist-info/RECORD,,